Add retention flag, call to update retention

app/flags.go

@@ -10,6 +10,7 @@ import (
 const (
 	ServerFlagName          = "server"
 	ConfigDirFlagName       = "config-dir"
+	RetentionFlagName       = "retention-days"
 	NamespaceFlagName       = "namespace"
 	RequestIDFlagName       = "request-id"
 	ResourceVersionFlagName = "resource-version"
@@ -32,6 +33,13 @@ var (
 		Hidden:   true,
 		Required: false,
 	}
+	RetentionFlag = &cli.StringFlag{
+		Name:     RetentionFlagName,
+		Usage:    "The retention of the namespace",
+		Aliases:  []string{"ret"},
+		EnvVars:  []string{"NAMESPACE_RETENTION"},
+		Required: true,
+	}
 	NamespaceFlag = &cli.StringFlag{
 		Name:     NamespaceFlagName,
 		Usage:    "The namespace hosted on temporal cloud",

app/namespace.go

@@ -574,6 +574,33 @@ func NewNamespaceCommand(getNamespaceClientFn GetNamespaceClientFn) (CommandOut,
 						},
 					},
 				},
+				{
+					Name:    "retention",
+					Usage:   "The length of time a closed workflow will be preserved before deletion",
+					Aliases: []string{"r"},
+					Subcommands: []*cli.Command{{
+						Name:    "set",
+						Aliases: []string{"s"},
+						Usage:   "Set the length of time a closed workflow will be preserved before deletion per namespace",
+						Flags: []cli.Flag{
+							NamespaceFlag,
+							RetentionFlag,
+							RequestIDFlag,
+						},
+						Action: func(ctx *cli.Context) error {
+							retention := ctx.Int(RetentionFlagName)
+							n, err := c.getNamespace(ctx.String(NamespaceFlagName))
+							if err != nil {
+								return err
+							}
+							if int32(retention) == n.Spec.RetentionDays {
+								return fmt.Errorf("retention for namespace is already set at %d days", ctx.Int(RetentionFlagName))
+							}
+							n.Spec.RetentionDays = int32(retention)
+							return c.updateNamespace(ctx, n)
+						},
+					}},
+				},
 				{
 					Name:    "search-attributes",
 					Usage:   "Manage search attributes used by namespace",