feat: spin on plan and destroy

cli/parse.go

@@ -0,0 +1,17 @@
+package cli
+
+import (
+	"os"
+)
+
+func ParseArgs() (string, []string) {
+	command := ""
+	args := []string{}
+
+	if len(os.Args) > 1 {
+		command = os.Args[1]
+		args = os.Args[2:]
+	}
+
+	return command, args
+}

main.go

@@ -2,8 +2,11 @@ package main
 
 import (
 	"terraform-roulette/cli"
+	"terraform-roulette/roulette"
 )
 
 func main() {
-	cli.RunCli()
+	command, args := cli.ParseArgs()
+
+	roulette.Play(command, args)
 }

roulette/game.go

@@ -7,11 +7,13 @@ import (
 	"strings"
 )
 
-var options = map[string]int{
+var overrides = map[string]int{
 	"destroy":              25,
 	"apply --auto-approve": 50,
 }
 
+var overrideables = []string{"apply", "plan", "destroy"}
+
 func assertTerraformCommand(command string) {
 	cmd := exec.Command("terraform", command, "--help")
 	cmd.Stderr = os.Stderr
@@ -36,22 +38,37 @@ func execTerraform(commands []string, args []string) {
 }
 
 func spinTheWheel(commands []string) []string {
-	for option, weight := range options {
-		if rand.Intn(100) < weight {
-			commands = strings.Split(option, " ")
+	target := rand.Intn(100)
+
+	for override, weight := range overrides {
+		if target < weight {
+			commands = strings.Split(override, " ")
 			break
 		}
 	}
 
 	return commands
 }
 
+func isOverridable(command string) bool {
+	result := false
+
+	for _, overrideable := range overrideables {
+		if command == overrideable {
+			result = true
+			break
+		}
+	}
+
+	return result
+}
+
 func Play(command string, args []string) {
 	assertTerraformCommand(command)
 
 	commands := strings.Split(command, " ")
 
-	if !strings.HasPrefix(command, "apply") {
+	if !isOverridable(command) {
 		execTerraform(commands, args)
 		os.Exit(0)
 	}