[cmd] - Move the CLI root command to its own package

v2/cmd/stratus/cleanup_cmd.go → v2/cmd/stratus/cmd/cleanup_cmd.go

@@ -1,12 +1,13 @@
-package main
+package cmd
 
 import (
 	"errors"
+	"log"
+	"os"
+
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"github.com/spf13/cobra"
-	"log"
-	"os"
 )
 
 var flagForceCleanup bool

v2/cmd/stratus/detonate_cmd.go → v2/cmd/stratus/cmd/detonate_cmd.go

@@ -1,13 +1,13 @@
-package main
+package cmd
 
 import (
 	"errors"
-	"github.com/datadog/stratus-red-team/v2/internal/utils"
-	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
-	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"os"
 	"strings"
 
+	"github.com/datadog/stratus-red-team/v2/internal/utils"
+	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
+	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"github.com/spf13/cobra"
 )
 

v2/cmd/stratus/list_cmd.go → v2/cmd/stratus/cmd/list_cmd.go

@@ -1,14 +1,15 @@
-package main
+package cmd
 
 import (
 	"fmt"
+	"log"
+	"strings"
+
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus/mitreattack"
 	"github.com/fatih/color"
 	"github.com/jedib0t/go-pretty/v6/table"
 	"github.com/spf13/cobra"
-	"log"
-	"strings"
 )
 
 var listPlatform string

v2/cmd/stratus/revert_cmd.go → v2/cmd/stratus/cmd/revert_cmd.go

@@ -1,12 +1,12 @@
-package main
+package cmd
 
 import (
 	"errors"
-	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
-	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"log"
 	"os"
 
+	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
+	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"github.com/spf13/cobra"
 )
 

v2/cmd/stratus/cmd/root.go

@@ -0,0 +1,39 @@
+package cmd
+
+import (
+	"log"
+	"os"
+
+	_ "github.com/datadog/stratus-red-team/v2/internal/attacktechniques"
+	"github.com/spf13/cobra"
+)
+
+var RootCmd = &cobra.Command{
+	Use: "stratus",
+}
+
+func init() {
+	setupLogging()
+
+	listCmd := buildListCmd()
+	showCmd := buildShowCmd()
+	warmupCmd := buildWarmupCmd()
+	detonateCmd := buildDetonateCmd()
+	revertCmd := buildRevertCmd()
+	statusCmd := buildStatusCmd()
+	cleanupCmd := buildCleanupCmd()
+	versionCmd := buildVersionCmd()
+
+	RootCmd.AddCommand(listCmd)
+	RootCmd.AddCommand(showCmd)
+	RootCmd.AddCommand(warmupCmd)
+	RootCmd.AddCommand(detonateCmd)
+	RootCmd.AddCommand(revertCmd)
+	RootCmd.AddCommand(statusCmd)
+	RootCmd.AddCommand(cleanupCmd)
+	RootCmd.AddCommand(versionCmd)
+}
+
+func setupLogging() {
+	log.SetOutput(os.Stdout)
+}

v2/cmd/stratus/show_cmd.go → v2/cmd/stratus/cmd/show_cmd.go

@@ -1,8 +1,9 @@
-package main
+package cmd
 
 import (
 	"errors"
 	"fmt"
+
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
 	"github.com/spf13/cobra"
 )

v2/cmd/stratus/status_cmd.go → v2/cmd/stratus/cmd/status_cmd.go

@@ -1,4 +1,4 @@
-package main
+package cmd
 
 import (
 	"github.com/datadog/stratus-red-team/v2/internal/state"

v2/cmd/stratus/util.go → v2/cmd/stratus/cmd/util.go

@@ -1,12 +1,13 @@
-package main
+package cmd
 
 import (
 	"errors"
-	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
-	"github.com/jedib0t/go-pretty/v6/table"
 	"log"
 	"os"
 	"strings"
+
+	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
+	"github.com/jedib0t/go-pretty/v6/table"
 )
 
 func GetDisplayTable() table.Writer {

v2/cmd/stratus/version_cmd.go → v2/cmd/stratus/cmd/version_cmd.go

@@ -1,4 +1,4 @@
-package main
+package cmd
 
 import (
 	"github.com/spf13/cobra"

v2/cmd/stratus/warmup_cmd.go → v2/cmd/stratus/cmd/warmup_cmd.go

@@ -1,11 +1,12 @@
-package main
+package cmd
 
 import (
 	"errors"
+	"os"
+
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus"
 	"github.com/datadog/stratus-red-team/v2/pkg/stratus/runner"
 	"github.com/spf13/cobra"
-	"os"
 )
 
 var forceWarmup bool

v2/cmd/stratus/main.go

@@ -1,42 +1,7 @@
 package main
 
-import (
-	_ "github.com/datadog/stratus-red-team/v2/internal/attacktechniques"
-	"github.com/spf13/cobra"
-	"log"
-	"os"
-)
-
-var rootCmd = &cobra.Command{
-	Use: "stratus",
-}
-
-func init() {
-	setupLogging()
-
-	listCmd := buildListCmd()
-	showCmd := buildShowCmd()
-	warmupCmd := buildWarmupCmd()
-	detonateCmd := buildDetonateCmd()
-	revertCmd := buildRevertCmd()
-	statusCmd := buildStatusCmd()
-	cleanupCmd := buildCleanupCmd()
-	versionCmd := buildVersionCmd()
-
-	rootCmd.AddCommand(listCmd)
-	rootCmd.AddCommand(showCmd)
-	rootCmd.AddCommand(warmupCmd)
-	rootCmd.AddCommand(detonateCmd)
-	rootCmd.AddCommand(revertCmd)
-	rootCmd.AddCommand(statusCmd)
-	rootCmd.AddCommand(cleanupCmd)
-	rootCmd.AddCommand(versionCmd)
-}
-
-func setupLogging() {
-	log.SetOutput(os.Stdout)
-}
+import "github.com/datadog/stratus-red-team/v2/cmd/stratus/cmd"
 
 func main() {
-	rootCmd.Execute()
+	cmd.RootCmd.Execute()
 }