[#82] Refactor command in smaller functions

cmd/root.go

@@ -20,43 +20,36 @@ var rootCmd = &cobra.Command{
 }
 
 func Execute(setupScripts *scripts.SetupScripts) {
-	var selectedCIProvider *string
-	var setupScript string
-
 	selectedCIProvider, err := getCIProvider(os.Args)
 	if err != nil {
 		fmt.Printf(err.Error())
 		return
 	}
 
-	switch *selectedCIProvider {
-	case scripts.Bitbucket:
-		setupScript = setupScripts.BitBucket
-	case scripts.CircleCI:
-		setupScript = setupScripts.CircleCI
-	case scripts.GithubActions:
-		setupScript = setupScripts.GitHubActions
-	case scripts.GitLabCI:
-		setupScript = setupScripts.GitLabCI
-	case scripts.TravisCI:
-		setupScript = setupScripts.TravisCI
-	default:
-		fmt.Println("Unknown CI provider")
+	setupScript, err := scripts.MapCIProviderToScript(selectedCIProvider, setupScripts)
+	if err != nil {
+		fmt.Printf(err.Error())
 		return
 	}
 
-	stringReader := strings.NewReader(setupScript)
-	stringReadCloser := io.NopCloser(stringReader)
-	execScriptCmd := exec.Command("bash")
+	fmt.Println("This might take a few seconds...")
 
-	execScriptCmd.Stdin = stringReadCloser
-	res, err := execScriptCmd.CombinedOutput()
+	res, err := executeCIInstallerScript(setupScript)
 	if err != nil {
 		fmt.Println("error executing script: ", err.Error())
 	}
 	fmt.Println("output: ", string(res))
 }
 
+func executeCIInstallerScript(setupScript *string) ([]byte, error) {
+	stringReader := strings.NewReader(*setupScript)
+	stringReadCloser := io.NopCloser(stringReader)
+	execScriptCmd := exec.Command("bash")
+
+	execScriptCmd.Stdin = stringReadCloser
+	return execScriptCmd.CombinedOutput()
+}
+
 func getCIProvider(args []string) (*string, error) {
 	if len(args) > 1 {
 		return &args[1], nil

scripts/scripts.go

@@ -1,6 +1,9 @@
 package scripts
 
-import _ "embed"
+import (
+	_ "embed"
+	"fmt"
+)
 
 const (
 	Bitbucket     = "Bitbucket"
@@ -42,3 +45,25 @@ func LoadSetupScripts() *SetupScripts {
 		TravisCI:      setupTravisCI,
 	}
 }
+
+func MapCIProviderToScript(ciProvider *string, setupScripts *SetupScripts) (*string, error) {
+	var setupScript *string
+	var err error
+
+	switch *ciProvider {
+	case Bitbucket:
+		setupScript = &setupScripts.BitBucket
+	case CircleCI:
+		setupScript = &setupScripts.CircleCI
+	case GithubActions:
+		setupScript = &setupScripts.GitHubActions
+	case GitLabCI:
+		setupScript = &setupScripts.GitLabCI
+	case TravisCI:
+		setupScript = &setupScripts.TravisCI
+	default:
+		err = fmt.Errorf("Unknown CI provider")
+	}
+
+	return setupScript, err
+}