DEV-2465: add atmos pro stack locking (#678)

cmd/pro.go

@@ -0,0 +1,17 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+)
+
+// proCmd executes 'atmos pro' CLI commands
+var proCmd = &cobra.Command{
+	Use:                "pro",
+	Short:              "Execute 'pro' commands",
+	Long:               `This command executes 'atmos pro' CLI commands`,
+	FParseErrWhitelist: struct{ UnknownFlags bool }{UnknownFlags: false},
+}
+
+func init() {
+	RootCmd.AddCommand(proCmd)
+}

cmd/pro_lock.go

@@ -0,0 +1,33 @@
+package cmd
+
+import (
+	e "github.com/cloudposse/atmos/internal/exec"
+	u "github.com/cloudposse/atmos/pkg/utils"
+	"github.com/spf13/cobra"
+)
+
+// proLockCmd executes 'pro lock' CLI command
+var proLockCmd = &cobra.Command{
+	Use:                "lock",
+	Short:              "Lock a stack",
+	Long:               `This command calls the atmos pro API and locks a stack`,
+	FParseErrWhitelist: struct{ UnknownFlags bool }{UnknownFlags: false},
+	Run: func(cmd *cobra.Command, args []string) {
+		// Check Atmos configuration
+		checkAtmosConfig()
+
+		err := e.ExecuteProLockCommand(cmd, args)
+		if err != nil {
+			u.LogErrorAndExit(err)
+		}
+	},
+}
+
+func init() {
+	proLockCmd.PersistentFlags().StringP("component", "c", "", "Specify the Atmos component to lock")
+	proLockCmd.PersistentFlags().StringP("stack", "s", "", "Specify the Atmos stack to lock")
+	proLockCmd.PersistentFlags().StringP("message", "m", "", "The lock message to display if someone else tries to lock the stack. Defaults to 'Locked by Atmos'")
+	proLockCmd.PersistentFlags().Int32P("ttl", "t", 0, "The amount of time in seconds to lock the stack for. Defaults to 30")
+
+	proCmd.AddCommand(proLockCmd)
+}

cmd/pro_unlock.go

@@ -0,0 +1,31 @@
+package cmd
+
+import (
+	e "github.com/cloudposse/atmos/internal/exec"
+	u "github.com/cloudposse/atmos/pkg/utils"
+	"github.com/spf13/cobra"
+)
+
+// proUnlockCmd executes 'pro unlock' CLI command
+var proUnlockCmd = &cobra.Command{
+	Use:                "unlock",
+	Short:              "Unlock a stack",
+	Long:               `This command calls the atmos pro API and unlocks a stack`,
+	FParseErrWhitelist: struct{ UnknownFlags bool }{UnknownFlags: false},
+	Run: func(cmd *cobra.Command, args []string) {
+		// Check Atmos configuration
+		checkAtmosConfig()
+
+		err := e.ExecuteProUnlockCommand(cmd, args)
+		if err != nil {
+			u.LogErrorAndExit(err)
+		}
+	},
+}
+
+func init() {
+	proUnlockCmd.PersistentFlags().StringP("component", "c", "", "Specify the Atmos component to lock")
+	proUnlockCmd.PersistentFlags().StringP("stack", "s", "", "Specify the Atmos stack to lock")
+
+	proCmd.AddCommand(proUnlockCmd)
+}

demo/screengrabs/demo-stacks.txt

@@ -24,6 +24,8 @@ atmos describe dependents --help
 atmos describe stacks --help
 atmos describe workflows --help
 atmos docs --help
+atmos pro lock --help
+atmos pro unlock --help
 atmos terraform --help
 atmos terraform clean --help
 atmos terraform deploy --help

go.mod

@@ -206,6 +206,7 @@ require (
 	github.com/sourcegraph/conc v0.3.0 // indirect
 	github.com/spf13/afero v1.11.0 // indirect
 	github.com/spf13/cast v1.6.0 // indirect
+	github.com/stretchr/objx v0.5.2 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
 	github.com/tchap/go-patricia/v2 v2.3.1 // indirect
 	github.com/ugorji/go/codec v1.2.7 // indirect