Merge pull request #5998 from vrothberg/generate-systemd

generate systemd

3 files changed, 85 insertions, 0 deletions

diff --git a/cmd/podman/generate/generate.go b/cmd/podman/generate/generate.go
new file mode 100644
index 000000000..f04ef58a5
--- /dev/null
+++ b/cmd/podman/generate/generate.go
@@ -0,0 +1,27 @@
+package pods
+
+import (
+	"github.com/containers/libpod/cmd/podman/registry"
+	"github.com/containers/libpod/pkg/domain/entities"
+	"github.com/containers/libpod/pkg/util"
+	"github.com/spf13/cobra"
+)
+
+var (
+	// Command: podman _generate_
+	generateCmd = &cobra.Command{
+		Use:              "generate",
+		Short:            "Generate structured data based on containers and pods.",
+		Long:             "Generate structured data (e.g., Kubernetes yaml or systemd units) based on containers and pods.",
+		TraverseChildren: true,
+		RunE:             registry.SubCommandExists,
+	}
+	containerConfig = util.DefaultContainerConfig()
+)
+
+func init() {
+	registry.Commands = append(registry.Commands, registry.CliCommand{
+		Mode:    []entities.EngineMode{entities.ABIMode},
+		Command: generateCmd,
+	})
+}
diff --git a/cmd/podman/generate/systemd.go b/cmd/podman/generate/systemd.go
new file mode 100644
index 000000000..55d770249
--- /dev/null
+++ b/cmd/podman/generate/systemd.go
@@ -0,0 +1,57 @@
+package pods
+
+import (
+	"fmt"
+
+	"github.com/containers/libpod/cmd/podman/registry"
+	"github.com/containers/libpod/cmd/podman/utils"
+	"github.com/containers/libpod/pkg/domain/entities"
+	"github.com/spf13/cobra"
+)
+
+var (
+	systemdTimeout     uint
+	systemdOptions     = entities.GenerateSystemdOptions{}
+	systemdDescription = `Generate systemd units for a pod or container.
+  The generated units can later be controlled via systemctl(1).`
+
+	systemdCmd = &cobra.Command{
+		Use:   "systemd [flags] CTR|POD",
+		Short: "Generate systemd units.",
+		Long:  systemdDescription,
+		RunE:  systemd,
+		Args:  cobra.MinimumNArgs(1),
+		Example: `podman generate systemd CTR
+  podman generate systemd --new --time 10 CTR
+  podman generate systemd --files --name POD`,
+	}
+)
+
+func init() {
+	registry.Commands = append(registry.Commands, registry.CliCommand{
+		Mode:    []entities.EngineMode{entities.ABIMode, entities.TunnelMode},
+		Command: systemdCmd,
+		Parent:  generateCmd,
+	})
+	flags := systemdCmd.Flags()
+	flags.BoolVarP(&systemdOptions.Name, "name", "n", false, "Use container/pod names instead of IDs")
+	flags.BoolVarP(&systemdOptions.Files, "files", "f", false, "Generate .service files instead of printing to stdout")
+	flags.UintVarP(&systemdTimeout, "time", "t", containerConfig.Engine.StopTimeout, "Stop timeout override")
+	flags.StringVar(&systemdOptions.RestartPolicy, "restart-policy", "on-failure", "Systemd restart-policy")
+	flags.BoolVarP(&systemdOptions.New, "new", "", false, "Create a new container instead of starting an existing one")
+	flags.SetNormalizeFunc(utils.AliasFlags)
+}
+
+func systemd(cmd *cobra.Command, args []string) error {
+	if cmd.Flags().Changed("time") {
+		systemdOptions.StopTimeout = &systemdTimeout
+	}
+
+	report, err := registry.ContainerEngine().GenerateSystemd(registry.GetContext(), args[0], systemdOptions)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println(report.Output)
+	return nil
+}
diff --git a/cmd/podman/main.go b/cmd/podman/main.go
index 4cd0beee8..481214a38 100644
--- a/cmd/podman/main.go
+++ b/cmd/podman/main.go
@@ -4,6 +4,7 @@ import (
 	"os"
 
 	_ "github.com/containers/libpod/cmd/podman/containers"
+	_ "github.com/containers/libpod/cmd/podman/generate"
 	_ "github.com/containers/libpod/cmd/podman/healthcheck"
 	_ "github.com/containers/libpod/cmd/podman/images"
 	_ "github.com/containers/libpod/cmd/podman/manifest"