improve(*): more beautiful prompt.

Author: Wine93

cli/command/clean.go

@@ -24,7 +24,6 @@ package command
 
 import (
 	"fmt"
-	"strings"
 
 	"github.com/opencurve/curveadm/cli/cli"
 	"github.com/opencurve/curveadm/internal/configure/topology"
@@ -41,7 +40,11 @@ var (
   $ curveadm clean --only='log,data'             # Clean log and data for all services
   $ curveadm clean --role=etcd --only=container  # Clean container for etcd services`
 
-	supportOnlyFlag = map[string]bool{"log": true, "data": true, "container": true}
+	supportOnlyFlag = map[string]bool{
+		task.ITEM_LOG:       true,
+		task.ITEM_DATA:      true,
+		task.ITEM_CONTAINER: true,
+	}
 )
 
 type cleanOptions struct {
@@ -105,10 +108,7 @@ func runClean(curveadm *cli.CurveAdm, options cleanOptions) error {
 		return nil
 	}
 
-	// clean service
-	curveadm.WriteOut("Warning: clean role=%s host=%s id=%s only=%s now, %s in it will be cleaned.\n",
-		role, host, id, strings.Join(only, ","), strings.Join(only, ","))
-	if pass := tui.ConfirmYes("Do you want to continue? [YES/No]: "); !pass {
+	if pass := tui.ConfirmYes(tui.PromptCleanService(role, host, only)); !pass {
 		curveadm.WriteOut("Clean canceled\n")
 		return nil
 	}

cli/command/cluster/remove.go

@@ -26,7 +26,7 @@ import (
 	"fmt"
 
 	"github.com/opencurve/curveadm/cli/cli"
-	"github.com/opencurve/curveadm/internal/tui/common"
+	tui "github.com/opencurve/curveadm/internal/tui/common"
 	cliutil "github.com/opencurve/curveadm/internal/utils"
 	"github.com/opencurve/curveadm/pkg/log"
 	"github.com/spf13/cobra"
@@ -66,8 +66,7 @@ func runRemove(curveadm *cli.CurveAdm, options removeOptions) error {
 	}
 
 	// TODO(@Wine93): check all service has removed before delete cluster
-	curveadm.WriteOut("Warning: This operation will remove cluster %s, and all data in it will be cleaned.\n", clusterName)
-	if pass := common.ConfirmYes("Do you want to continue? [YES/No]: "); !pass {
+	if pass := tui.ConfirmYes(tui.PromptRemoveCluster(clusterName)); !pass {
 		curveadm.WriteOut("Remove cluster canceled\n")
 		return nil
 	} else if err := curveadm.Storage().DeleteCluster(clusterName); err != nil {

cli/command/config/commit.go

@@ -103,7 +103,7 @@ func runCommit(curveadm *cli.CurveAdm, options commitOptions) error {
 
 	if err := validateTopology(oldData, newData); err != nil {
 		return err
-	} else if pass := common.ConfirmYes("Do you want to continue? [YES/No]: "); !pass {
+	} else if pass := common.ConfirmYes("Do you want to continue?"); !pass {
 		return nil
 	} else if err := curveadm.Storage().SetClusterTopology(curveadm.ClusterId(), newData); err != nil {
 		return err

cli/command/stop.go

@@ -76,9 +76,7 @@ func runStop(curveadm *cli.CurveAdm, options stopOptions) error {
 		return fmt.Errorf("service not found")
 	}
 
-	// stop service
-	curveadm.WriteOut("Warning: Stop all service now, client IO will be hang!\n")
-	if pass := tui.ConfirmYes("Do you want to continue? [YES/No]: "); !pass {
+	if pass := tui.ConfirmYes(tui.PromptStopService()); !pass {
 		curveadm.WriteOut("Stop canceled\n")
 		return nil
 	}

cli/command/support.go

@@ -33,22 +33,12 @@ import (
 	task "github.com/opencurve/curveadm/internal/task/task/common"
 	"github.com/opencurve/curveadm/internal/task/tasks"
 	"github.com/opencurve/curveadm/internal/tools"
-	"github.com/opencurve/curveadm/internal/tui/common"
+	tui "github.com/opencurve/curveadm/internal/tui/common"
 	"github.com/opencurve/curveadm/internal/utils"
 	cliutil "github.com/opencurve/curveadm/internal/utils"
 	"github.com/spf13/cobra"
 )
 
-const (
-	PROMPT = `FYI:
-  > We have collected logs for troubleshooting,
-  > and now we will send these logs to the curve center.
-  > Please don't worry about the data security,
-  > we guarantee that all logs are encrypted
-  > and only you have the secret key.
-`
-)
-
 type supportOptions struct {
 }
 
@@ -142,8 +132,7 @@ func runSupport(curveadm *cli.CurveAdm, options supportOptions) error {
 		return err
 	}
 
-	curveadm.WriteOut(color.YellowString(PROMPT))
-	if pass := common.ConfirmYes("Do you want to continue? [YES/No]: "); !pass {
+	if pass := tui.ConfirmYes(tui.PromptCollectService()); !pass {
 		return nil
 	}
 

internal/configure/format/fc.go

@@ -27,7 +27,7 @@ import (
 )
 
 const (
-	DEFAULT_SSH_TIMEOUT = 10
+	DEFAULT_SSH_TIMEOUT = 10 // seconds
 )
 
 type (

internal/tui/common/prompt.go

@@ -16,8 +16,90 @@
 
 /*
  * Project: CurveAdm
- * Created Date: 2021-12-22
+ * Created Date: 2022-01-14
  * Author: Jingli Chen (Wine93)
  */
 
 package common
+
+import (
+	"bytes"
+	"fmt"
+	"strings"
+	"text/template"
+
+	"github.com/fatih/color"
+)
+
+const (
+	PROMPT_REMOVE_CLUSTER = `{{.warning}}
+Do you want to continue?`
+
+	PROMPT_STOP_SERVICE = `{{.warning}}
+Do you want to continue?`
+
+	PROMPT_CLEAN_SERVICE = `{{.warning}}
+  - Service role: {{.role}} ("*" means all roles)
+  - Service host: {{.host}} ("*" means all hosts)
+  - Clean items : [{{.items}}]
+Do you want to continue?`
+
+	PROMPT_COLLECT_SERVICE = `
+FYI:
+  > We have collected logs for troubleshooting,
+  > and now we will send these logs to the curve center.
+  > Please don't worry about the data security,
+  > we guarantee that all logs are encrypted
+  > and only you have the secret key.
+`
+
+	DEFAULT_CONFIRM_PROMPT = "Do you want to continue?"
+)
+
+type Prompt struct {
+	tmpl *template.Template
+	data map[string]interface{}
+}
+
+func NewPrompt(text string) *Prompt {
+	return &Prompt{
+		tmpl: template.Must(template.New("prompt").Parse(text)),
+		data: map[string]interface{}{},
+	}
+}
+
+func (p *Prompt) Build() string {
+	buffer := bytes.NewBufferString("")
+	err := p.tmpl.Execute(buffer, p.data)
+	if err != nil {
+		return ""
+	}
+	return buffer.String()
+}
+
+func PromptRemoveCluster(clusterName string) string {
+	prompt := NewPrompt(PROMPT_REMOVE_CLUSTER)
+	prompt.data["warning"] = fmt.Sprintf("WARNING: cluster '%s' will be removed,\n"+
+		"and all data in it will be cleaned up", clusterName)
+	return prompt.Build()
+}
+
+func PromptStopService() string {
+	prompt := NewPrompt(PROMPT_STOP_SERVICE)
+	prompt.data["warning"] = "WARNING: stop service may cause client IO be hang"
+	return prompt.Build()
+}
+
+func PromptCleanService(role, host string, items []string) string {
+	prompt := NewPrompt(PROMPT_CLEAN_SERVICE)
+	prompt.data["warning"] = "WARNING: service items which matched will be cleaned up"
+	prompt.data["role"] = role
+	prompt.data["host"] = host
+	prompt.data["items"] = strings.Join(items, ",")
+	return prompt.Build()
+}
+
+func PromptCollectService() string {
+	prompt := NewPrompt(color.YellowString(PROMPT_COLLECT_SERVICE) + DEFAULT_CONFIRM_PROMPT)
+	return prompt.Build()
+}

internal/tui/common/tui.go

@@ -141,16 +141,6 @@ func prompt(prompt string) string {
 	return strings.TrimSuffix(input, "\n")
 }
 
-func ConfirmNo(format string, a ...interface{}) bool {
-	ans := prompt(fmt.Sprintf(format, a...) + "(default=Y)")
-	switch strings.TrimSpace(strings.ToLower(ans)) {
-	case "n", "no":
-		return true
-	default:
-		return false
-	}
-}
-
 func ConfirmYes(format string, a ...interface{}) bool {
 	ans := prompt(fmt.Sprintf(format, a...) + " [yes/no]: (default=no)")
 	switch strings.TrimSpace(ans) {