Merge pull request #1782 from aluzzardi/do-help-improvements

dagger do: Improve help message

cmd/dagger/cmd/do.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"context"
 	"fmt"
+	"io"
 	"os"
 	"path/filepath"
 	"strings"
@@ -110,34 +111,44 @@ func getTargetPath(args []string) cue.Path {
 func doHelpCmd(cmd *cobra.Command, _ []string) {
 	lg := logger.New()
 
-	fmt.Printf("%s\n\n%s", cmd.Short, cmd.UsageString())
+	fmt.Println(cmd.Short)
+
+	err := printActions(os.Stdout, getTargetPath(cmd.Flags().Args()))
+
+	fmt.Printf("\n%s", cmd.UsageString())
 
-	p, err := loadPlan()
 	if err != nil {
 		lg.Fatal().Err(err).Msg("failed to load plan")
 	}
+}
+
+func printActions(w io.Writer, target cue.Path) error {
+	p, err := loadPlan()
+	if err != nil {
+		return err
+	}
 
-	target := getTargetPath(cmd.Flags().Args())
 	action := p.Action().FindByPath(target)
 	if action == nil {
-		lg.Fatal().Msg(fmt.Sprintf("action %s not found", target.String()))
-		return
+		return fmt.Errorf("action %s not found", target.String())
 	}
 
 	if len(action.Name) < 1 {
-		return
+		return nil
 	}
 
-	fmt.Println("")
-	w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', tabwriter.StripEscape)
-	defer w.Flush()
+	fmt.Printf("\nAvailable Actions:\n")
+	tw := tabwriter.NewWriter(w, 0, 0, 1, ' ', tabwriter.StripEscape)
+	defer tw.Flush()
 
 	for _, a := range action.Children {
 		if !a.Hidden {
-			lineParts := []string{"", a.Name, strings.TrimSpace(a.Comment)}
-			fmt.Fprintln(w, strings.Join(lineParts, "\t"))
+			lineParts := []string{"", a.Name, a.Documentation}
+			fmt.Fprintln(tw, strings.Join(lineParts, "\t"))
 		}
 	}
+
+	return nil
 }
 
 func init() {

compiler/value.go

@@ -6,6 +6,7 @@ import (
 	"path/filepath"
 	"sort"
 	"strconv"
+	"strings"
 
 	"cuelang.org/go/cue"
 	"cuelang.org/go/cue/ast"
@@ -346,3 +347,16 @@ func (v *Value) Default() (*Value, bool) {
 func (v *Value) Doc() []*ast.CommentGroup {
 	return v.Cue().Doc()
 }
+
+// DocSummary returns the first line of documentation
+func (v *Value) DocSummary() string {
+	doc := ""
+	for _, c := range v.Doc() {
+		doc += c.Text()
+	}
+
+	doc = strings.TrimSpace(doc)
+	doc = strings.SplitN(doc, "\n", 2)[0]
+
+	return doc
+}

plan/action.go

@@ -5,12 +5,11 @@ import (
 )
 
 type Action struct {
-	Name     string
-	Hidden   bool
-	Path     cue.Path
-	Comment  string
-	Children []*Action
-	// pkg      string
+	Name          string
+	Hidden        bool
+	Path          cue.Path
+	Documentation string
+	Children      []*Action
 }
 
 func (a *Action) AddChild(c *Action) {

plan/plan.go

@@ -199,9 +199,7 @@ func (p *Plan) fillAction() {
 	if !actions.Exists() {
 		return
 	}
-	for _, cg := range actions.Doc() {
-		p.action.Comment += cg.Text()
-	}
+	p.action.Documentation = actions.DocSummary()
 
 	tasks := flow.Tasks()
 
@@ -214,17 +212,12 @@ func (p *Plan) fillAction() {
 			a := prevAction.FindByPath(path)
 			if a == nil {
 				v := p.Source().LookupPath(path)
-				childComment := ""
-				for _, cg := range v.Doc() {
-					childComment += cg.Text()
-				}
-
 				a = &Action{
-					Name:     s.String(),
-					Hidden:   s.PkgPath() != "",
-					Path:     path,
-					Comment:  childComment,
-					Children: []*Action{},
+					Name:          s.String(),
+					Hidden:        s.PkgPath() != "",
+					Path:          path,
+					Documentation: v.DocSummary(),
+					Children:      []*Action{},
 				}
 				prevAction.AddChild(a)
 			}