go(repositories): add sub menu for ee repository

go/commands/yarn.go

@@ -0,0 +1,15 @@
+package commands
+
+import (
+	"ocl/portainer-devtool/utils"
+)
+
+// RunClient starts the portainer client
+func RunPortainerClient(workdir string) error {
+	err := utils.RunCommandWithStdoutPipe(workdir, "yarn")
+	if err != nil {
+		return err
+	}
+
+	return utils.RunCommandWithStdoutPipe(workdir, "yarn", "start:client")
+}

go/main.go

@@ -1,10 +1,10 @@
 package main
 
 import (
-	"fmt"
 	"log"
 	"ocl/portainer-devtool/repositories"
 	"ocl/portainer-devtool/utils"
+	"os"
 )
 
 const (
@@ -12,29 +12,42 @@ const (
 	MENU_OPTION_CE_REPO
 	MENU_OPTION_AGENT_REPO
 	MENU_OPTION_OTHERS
+	MENU_OPTION_QUIT
 )
 
 func main() {
 
-	utils.MenuPrint()
+	for {
 
-	var option int
+		printMainMenu := func() {
-	_, err := fmt.Scanf("%d", &option)
+			utils.MenuPrint("Which repository or action do you want to operate:", `
-	if err != nil {
+ 1. Portainer EE Repository
-		log.Fatal(err)
+ 2. Portainer CE Repository
+ 3. Portainer Agent Repository
+ 4. Others
+ 5. Quit`)
+		}
+
+		option := utils.PromptMenu(printMainMenu)
+
+		var action repositories.Actioner
+		switch option {
+		case MENU_OPTION_EE_REPO:
+			action = repositories.NewPortainerEERepository()
+		case MENU_OPTION_CE_REPO:
+
+		case MENU_OPTION_AGENT_REPO:
+
+		case MENU_OPTION_OTHERS:
+
+		case MENU_OPTION_QUIT:
+			os.Exit(0)
+		}
+
+		err := action.Execute()
+		if err != nil {
+			log.Fatalln(err)
+		}
 	}
 
-	var action repositories.Actioner
-	switch option {
-	case MENU_OPTION_EE_REPO:
-		action = repositories.NewPortainerEERepository()
-	case MENU_OPTION_CE_REPO:
-
-	case MENU_OPTION_AGENT_REPO:
-
-	case MENU_OPTION_OTHERS:
-
-	}
-
-	log.Fatal(action.Execute())
 }

go/repositories/ee.go

@@ -6,6 +6,18 @@ import (
 	"ocl/portainer-devtool/utils"
 )
 
+const (
+	ACTION_EE_BUILD_ALL int = iota + 1
+	ACTION_EE_RUN_FRONTEND
+	ACTION_EE_RUN_BACKEND
+	ACTION_EE_VALIDATE_ALL
+	ACTION_EE_VALIDATE_FRONTEND
+	ACTION_EE_VALIDATE_BACKEND
+	ACTION_EE_RUN_UNIT_TEST_ALL
+	ACTION_EE_RUN_UNIT_TEST_FRONTEND
+	ACTION_EE_RUN_UNIT_TEST_BACKEND
+)
+
 type PortainerEE struct {
 	WorkDir     string
 	FrontendDir string
@@ -33,5 +45,27 @@ func (repo *PortainerEE) Execute() error {
 		return nil
 	}
 
+	option := utils.PromptMenu(repo.listSubMenu)
+	switch option {
+	case ACTION_EE_BUILD_ALL:
+
+	case ACTION_EE_RUN_FRONTEND:
+		commands.RunPortainerClient(repo.WorkDir)
+
+	}
+
 	return nil
 }
+
+func (repo *PortainerEE) listSubMenu() {
+	utils.MenuPrint("Do you want?", `
+ 1. Build both front-end and backend
+ 2. Run front-end only
+ 3. Run backend only
+ 4. Validate both fornt-end and backend before commit 
+ 5. Validate front-end only before commit
+ 6. Validate backend only before commit
+ 7. Run unit tests for both front-end and backend
+ 8. Run unit tests for front-end only
+ 9. Run unit tests for backend only`)
+}

go/utils/command.go

@@ -0,0 +1,45 @@
+package utils
+
+import (
+	"bufio"
+	"fmt"
+	"os/exec"
+)
+
+func RunCommandWithStdoutPipe(workdir, progName string, args ...string) error {
+	cmd := exec.Command(progName, args...)
+	cmd.Dir = workdir
+	out, err := cmd.StdoutPipe()
+	if err != nil {
+		return err
+	}
+
+	scanner := bufio.NewScanner(out)
+	go func() {
+		counter := 0
+		for scanner.Scan() {
+			if counter > 10 {
+				// Swallow the output
+				if counter%50 == 0 {
+					fmt.Printf("output %d lines in total.\n", counter)
+				}
+				counter++
+				continue
+			}
+			PrintOutput("", scanner.Bytes())
+			counter++
+		}
+	}()
+
+	err = cmd.Start()
+	if err != nil {
+		return err
+	}
+
+	err = cmd.Wait()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

go/utils/print.go

@@ -41,8 +41,10 @@ func InputPrint(message string) {
 	fmt.Println(colorYellow, message, colorReset)
 }
 
-func MenuPrint() {
+func MenuPrint(question, menu string) {
-	InputPrint("Which repository or action do you want to operate:")
+	if question != "" {
+		InputPrint(fmt.Sprintf("[%s]", question))
+	}
 
 	// 	menu := `
 	// 1. Build Portainer EE/CE All
@@ -56,10 +58,5 @@ func MenuPrint() {
 	// 9. Cleanup Temporary Volume
 	// 	`
 
-	menu := `1. Portainer EE Repository
- 2. Portainer CE Repository
- 3. Portainer Agent Repository
- 4. Others`
-
 	fmt.Println(colorCyan, menu, colorReset)
 }

go/utils/prompt.go

@@ -14,6 +14,14 @@ func PromptContinue() bool {
 	return false
 }
 
+func PromptMenu(listMenu func()) int {
+	listMenu()
+
+	var option int
+	fmt.Scanf("%d", &option)
+	return option
+}
+
 func prompt(question string) string {
 	fmt.Printf("%s %s :%s", colorYellow, question, colorReset)
 	var ret string