--no-header flag implementation for generate systemd

Signed-off-by: Jakub Guzik <jakubmguzik@gmail.com>

5 files changed, 106 insertions, 20 deletions

diff --git a/pkg/systemd/generate/common.go b/pkg/systemd/generate/common.go
index e9902319c..bbd1a5f92 100644
--- a/pkg/systemd/generate/common.go
+++ b/pkg/systemd/generate/common.go
@@ -31,10 +31,12 @@ func validateRestartPolicy(restart string) error {
 }
 
 const headerTemplate = `# {{{{.ServiceName}}}}.service
+{{{{- if (eq .GenerateNoHeader false) }}}}
 # autogenerated by Podman {{{{.PodmanVersion}}}}
 {{{{- if .TimeStamp}}}}
 # {{{{.TimeStamp}}}}
 {{{{- end}}}}
+{{{{- end}}}}
 
 [Unit]
 Description=Podman {{{{.ServiceName}}}}.service
diff --git a/pkg/systemd/generate/containers.go b/pkg/systemd/generate/containers.go
index acee7be65..8fbbc727b 100644
--- a/pkg/systemd/generate/containers.go
+++ b/pkg/systemd/generate/containers.go
@@ -65,7 +65,8 @@ type containerInfo struct {
 	ExecStop string
 	// ExecStopPost of the unit.
 	ExecStopPost string
-
+	// Removes autogenerated by Podman and timestamp if set to true
+	GenerateNoHeader bool
 	// If not nil, the container is part of the pod.  We can use the
 	// podInfo to extract the relevant data.
 	Pod *podInfo
@@ -292,10 +293,15 @@ func executeContainerTemplate(info *containerInfo, options entities.GenerateSyst
 	if info.PodmanVersion == "" {
 		info.PodmanVersion = version.Version.String()
 	}
+
+	if options.NoHeader {
+		info.GenerateNoHeader = true
+		info.GenerateTimestamp = false
+	}
+
 	if info.GenerateTimestamp {
 		info.TimeStamp = fmt.Sprintf("%v", time.Now().Format(time.UnixDate))
 	}
-
 	// Sort the slices to assure a deterministic output.
 	sort.Strings(info.BoundToServices)
 
diff --git a/pkg/systemd/generate/containers_test.go b/pkg/systemd/generate/containers_test.go
index dad14ece9..2ba73ef8d 100644
--- a/pkg/systemd/generate/containers_test.go
+++ b/pkg/systemd/generate/containers_test.go
@@ -37,9 +37,11 @@ func TestValidateRestartPolicyContainer(t *testing.T) {
 }
 
 func TestCreateContainerSystemdUnit(t *testing.T) {
-	goodID := `# container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
-# autogenerated by Podman CI
-
+	serviceInfo := `# container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
+`
+	headerInfo := `# autogenerated by Podman CI
+`
+	goodIDContent := `
 [Unit]
 Description=Podman container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401.service
 Documentation=man:podman-generate-systemd(1)
@@ -59,6 +61,8 @@ Type=forking
 [Install]
 WantedBy=multi-user.target default.target
 `
+	goodID := serviceInfo + headerInfo + goodIDContent
+	goodIDNoHeaderInfo := serviceInfo + goodIDContent
 
 	goodName := `# container-foobar.service
 # autogenerated by Podman CI
@@ -377,11 +381,12 @@ Type=forking
 WantedBy=multi-user.target default.target
 `
 	tests := []struct {
-		name    string
-		info    containerInfo
-		want    string
-		new     bool
-		wantErr bool
+		name     string
+		info     containerInfo
+		want     string
+		new      bool
+		noHeader bool
+		wantErr  bool
 	}{
 
 		{"good with id",
@@ -398,6 +403,23 @@ WantedBy=multi-user.target default.target
 			goodID,
 			false,
 			false,
+			false,
+		},
+		{"good with noHeader",
+			containerInfo{
+				Executable:        "/usr/bin/podman",
+				ServiceName:       "container-639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401",
+				ContainerNameOrID: "639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401",
+				RestartPolicy:     "always",
+				PIDFile:           "/run/containers/storage/overlay-containers/639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401/userdata/conmon.pid",
+				StopTimeout:       22,
+				PodmanVersion:     "CI",
+				EnvVariable:       EnvVariable,
+			},
+			goodIDNoHeaderInfo,
+			false,
+			true,
+			false,
 		},
 		{"good with name",
 			containerInfo{
@@ -413,6 +435,7 @@ WantedBy=multi-user.target default.target
 			goodName,
 			false,
 			false,
+			false,
 		},
 		{"good with name and bound to",
 			containerInfo{
@@ -429,6 +452,7 @@ WantedBy=multi-user.target default.target
 			goodNameBoundTo,
 			false,
 			false,
+			false,
 		},
 		{"bad restart policy",
 			containerInfo{
@@ -442,6 +466,7 @@ WantedBy=multi-user.target default.target
 			},
 			"",
 			false,
+			false,
 			true,
 		},
 		{"good with name and generic",
@@ -459,6 +484,7 @@ WantedBy=multi-user.target default.target
 			goodWithNameAndGeneric,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit short detach param",
 			containerInfo{
@@ -475,6 +501,7 @@ WantedBy=multi-user.target default.target
 			goodWithExplicitShortDetachParam,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit short detach param and podInfo",
 			containerInfo{
@@ -494,6 +521,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewWithPodFile,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit full detach param",
 			containerInfo{
@@ -510,6 +538,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetach,
 			true,
 			false,
+			false,
 		},
 		{"good with id and no param",
 			containerInfo{
@@ -526,6 +555,7 @@ WantedBy=multi-user.target default.target
 			goodIDNew,
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=true param",
 			containerInfo{
@@ -542,6 +572,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("--detach=true"),
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=false param",
 			containerInfo{
@@ -558,6 +589,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-d"),
 			true,
 			false,
+			false,
 		},
 		{"good with explicit detach=false param",
 			containerInfo{
@@ -574,6 +606,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetachFalseWithCmd,
 			true,
 			false,
+			false,
 		},
 		{"good with multiple detach=false params",
 			containerInfo{
@@ -590,6 +623,7 @@ WantedBy=multi-user.target default.target
 			goodNameNewDetachFalseWithCmd,
 			true,
 			false,
+			false,
 		},
 		{"good with multiple shorthand params detach first",
 			containerInfo{
@@ -606,6 +640,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-dti"),
 			true,
 			false,
+			false,
 		},
 		{"good with multiple shorthand params detach last",
 			containerInfo{
@@ -622,6 +657,7 @@ WantedBy=multi-user.target default.target
 			genGoodNewDetach("-tid"),
 			true,
 			false,
+			false,
 		},
 		{"good with root flags",
 			containerInfo{
@@ -638,6 +674,7 @@ WantedBy=multi-user.target default.target
 			goodNewRootFlags,
 			true,
 			false,
+			false,
 		},
 		{"good with container create",
 			containerInfo{
@@ -654,6 +691,7 @@ WantedBy=multi-user.target default.target
 			goodContainerCreate,
 			true,
 			false,
+			false,
 		},
 		{"good with journald log tag (see #9034)",
 			containerInfo{
@@ -670,6 +708,7 @@ WantedBy=multi-user.target default.target
 			goodNewWithJournaldTag,
 			true,
 			false,
+			false,
 		},
 		{"good with special chars",
 			containerInfo{
@@ -686,13 +725,15 @@ WantedBy=multi-user.target default.target
 			goodNewWithSpecialChars,
 			true,
 			false,
+			false,
 		},
 	}
 	for _, tt := range tests {
 		test := tt
 		t.Run(tt.name, func(t *testing.T) {
 			opts := entities.GenerateSystemdOptions{
-				New: test.new,
+				New:      test.new,
+				NoHeader: test.noHeader,
 			}
 			got, err := executeContainerTemplate(&test.info, opts)
 			if (err != nil) != test.wantErr {
diff --git a/pkg/systemd/generate/pods.go b/pkg/systemd/generate/pods.go
index ff8ce3a03..14b1b7a89 100644
--- a/pkg/systemd/generate/pods.go
+++ b/pkg/systemd/generate/pods.go
@@ -70,6 +70,8 @@ type podInfo struct {
 	ExecStop string
 	// ExecStopPost of the unit.
 	ExecStopPost string
+	// Removes autogenerated by Podman and timestamp if set to true
+	GenerateNoHeader bool
 }
 
 const podTemplate = headerTemplate + `Requires={{{{- range $index, $value := .RequiredServices -}}}}{{{{if $index}}}} {{{{end}}}}{{{{ $value }}}}.service{{{{end}}}}
@@ -319,6 +321,12 @@ func executePodTemplate(info *podInfo, options entities.GenerateSystemdOptions)
 	if info.PodmanVersion == "" {
 		info.PodmanVersion = version.Version.String()
 	}
+
+	if options.NoHeader {
+		info.GenerateNoHeader = true
+		info.GenerateTimestamp = false
+	}
+
 	if info.GenerateTimestamp {
 		info.TimeStamp = fmt.Sprintf("%v", time.Now().Format(time.UnixDate))
 	}
diff --git a/pkg/systemd/generate/pods_test.go b/pkg/systemd/generate/pods_test.go
index 2b430226b..f6ca9bced 100644
--- a/pkg/systemd/generate/pods_test.go
+++ b/pkg/systemd/generate/pods_test.go
@@ -37,9 +37,11 @@ func TestValidateRestartPolicyPod(t *testing.T) {
 }
 
 func TestCreatePodSystemdUnit(t *testing.T) {
-	podGood := `# pod-123abc.service
-# autogenerated by Podman CI
-
+	serviceInfo := `# pod-123abc.service
+`
+	headerInfo := `# autogenerated by Podman CI
+`
+	podContent := `
 [Unit]
 Description=Podman pod-123abc.service
 Documentation=man:podman-generate-systemd(1)
@@ -61,6 +63,8 @@ Type=forking
 [Install]
 WantedBy=multi-user.target default.target
 `
+	podGood := serviceInfo + headerInfo + podContent
+	podGoodNoHeaderInfo := serviceInfo + podContent
 
 	podGoodNamedNew := `# pod-123abc.service
 # autogenerated by Podman CI
@@ -171,11 +175,12 @@ WantedBy=multi-user.target default.target
 `
 
 	tests := []struct {
-		name    string
-		info    podInfo
-		want    string
-		new     bool
-		wantErr bool
+		name     string
+		info     podInfo
+		want     string
+		new      bool
+		noHeader bool
+		wantErr  bool
 	}{
 		{"pod",
 			podInfo{
@@ -192,6 +197,24 @@ WantedBy=multi-user.target default.target
 			podGood,
 			false,
 			false,
+			false,
+		},
+		{"pod noHeader",
+			podInfo{
+				Executable:       "/usr/bin/podman",
+				ServiceName:      "pod-123abc",
+				InfraNameOrID:    "jadda-jadda-infra",
+				RestartPolicy:    "always",
+				PIDFile:          "/run/containers/storage/overlay-containers/639c53578af4d84b8800b4635fa4e680ee80fd67e0e6a2d4eea48d1e3230f401/userdata/conmon.pid",
+				StopTimeout:      42,
+				PodmanVersion:    "CI",
+				RequiredServices: []string{"container-1", "container-2"},
+				CreateCommand:    []string{"podman", "pod", "create", "--name", "foo", "bar=arg with space"},
+			},
+			podGoodNoHeaderInfo,
+			false,
+			true,
+			false,
 		},
 		{"pod with root args",
 			podInfo{
@@ -208,6 +231,7 @@ WantedBy=multi-user.target default.target
 			podGood,
 			false,
 			false,
+			false,
 		},
 		{"pod --new",
 			podInfo{
@@ -224,6 +248,7 @@ WantedBy=multi-user.target default.target
 			podGoodNamedNew,
 			true,
 			false,
+			false,
 		},
 		{"pod --new with root args",
 			podInfo{
@@ -240,6 +265,7 @@ WantedBy=multi-user.target default.target
 			podGoodNamedNewWithRootArgs,
 			true,
 			false,
+			false,
 		},
 		{"pod --new with --replace=false",
 			podInfo{
@@ -256,6 +282,7 @@ WantedBy=multi-user.target default.target
 			podGoodNamedNewWithReplaceFalse,
 			true,
 			false,
+			false,
 		},
 		{"pod --new with double curly braces",
 			podInfo{
@@ -272,6 +299,7 @@ WantedBy=multi-user.target default.target
 			podNewLabelWithCurlyBraces,
 			true,
 			false,
+			false,
 		},
 	}
 
@@ -279,7 +307,8 @@ WantedBy=multi-user.target default.target
 		test := tt
 		t.Run(tt.name, func(t *testing.T) {
 			opts := entities.GenerateSystemdOptions{
-				New: test.new,
+				New:      test.new,
+				NoHeader: test.noHeader,
 			}
 			got, err := executePodTemplate(&test.info, opts)
 			if (err != nil) != test.wantErr {