cli: templatecreate: add CLI flags --max-ttl --min-autostart-interval

johnstcn · johnstcn · commit 245795ec4401 · 2022-06-03T16:13:55.000Z
diff --git a/cli/templatecreate.go b/cli/templatecreate.go
@@ -14,16 +14,19 @@ import (
 
 	"github.com/coder/coder/cli/cliui"
 	"github.com/coder/coder/coderd/database"
+	"github.com/coder/coder/coderd/util/ptr"
 	"github.com/coder/coder/codersdk"
 	"github.com/coder/coder/provisionerd"
 	"github.com/coder/coder/provisionersdk"
 )
 
 func templateCreate() *cobra.Command {
 	var (
-		directory     string
-		provisioner   string
-		parameterFile string
+		directory            string
+		provisioner          string
+		parameterFile        string
+		maxTTL               time.Duration
+		minAutostartInterval time.Duration
 	)
 	cmd := &cobra.Command{
 		Use:   "create [name]",
@@ -92,11 +95,15 @@ func templateCreate() *cobra.Command {
 				return err
 			}
 
-			_, err = client.CreateTemplate(cmd.Context(), organization.ID, codersdk.CreateTemplateRequest{
-				Name:            templateName,
-				VersionID:       job.ID,
-				ParameterValues: parameters,
-			})
+			createReq := codersdk.CreateTemplateRequest{
+				Name:                       templateName,
+				VersionID:                  job.ID,
+				ParameterValues:            parameters,
+				MaxTTLMillis:               ptr.Ref(maxTTL.Milliseconds()),
+				MinAutostartIntervalMillis: ptr.Ref(minAutostartInterval.Milliseconds()),
+			}
+
+			_, err = client.CreateTemplate(cmd.Context(), organization.ID, createReq)
 			if err != nil {
 				return err
 			}
@@ -115,6 +122,8 @@ func templateCreate() *cobra.Command {
 	cmd.Flags().StringVarP(&directory, "directory", "d", currentDirectory, "Specify the directory to create from")
 	cmd.Flags().StringVarP(&provisioner, "test.provisioner", "", "terraform", "Customize the provisioner backend")
 	cmd.Flags().StringVarP(&parameterFile, "parameter-file", "", "", "Specify a file path with parameter values.")
+	cmd.Flags().DurationVarP(&maxTTL, "max-ttl", "", 168*time.Hour, "Specify a maximum TTL for worksapces created from this template.")
+	cmd.Flags().DurationVarP(&minAutostartInterval, "min-autostart-interval", "", time.Hour, "Specify a minimum autostart interval for workspaces created from this template.")
 	// This is for testing!
 	err := cmd.Flags().MarkHidden("test.provisioner")
 	if err != nil {
diff --git a/cli/templatecreate_test.go b/cli/templatecreate_test.go
@@ -24,7 +24,16 @@ func TestTemplateCreate(t *testing.T) {
 			Parse:     echo.ParseComplete,
 			Provision: echo.ProvisionComplete,
 		})
-		cmd, root := clitest.New(t, "templates", "create", "my-template", "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho))
+		args := []string{
+			"templates",
+			"create",
+			"my-template",
+			"--directory", source,
+			"--test.provisioner", string(database.ProvisionerTypeEcho),
+			"--max-ttl", "24h",
+			"--min-autostart-interval", "2h",
+		}
+		cmd, root := clitest.New(t, args...)
 		clitest.SetupConfig(t, client, root)
 		pty := ptytest.New(t)
 		cmd.SetIn(pty.Input())
