Respect SUMMARY: or USAGE: (#5)

zph · Aug 8, 2024 · a44fe7d · a44fe7d
1 parent 14fc1b8
commit a44fe7d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cmd/help.go b/cmd/help.go
@@ -13,6 +13,7 @@ import (
 )
 
 var UsageKey = "USAGE: "
+var LegacyUsageKey = "SUMMARY: "
 
 // helpCmd represents the help command
 var helpCmd = &cobra.Command{

diff --git a/cmd/struct.go b/cmd/struct.go
@@ -54,11 +54,11 @@ func (s *Script) parse() error {
 		return err
 	}
 
-	if strings.Contains(string(b), UsageKey) {
+	if strings.Contains(string(b), UsageKey) || strings.Contains(string(b), LegacyUsageKey) {
 		lines := strings.Split(string(b), "\n")
 		var linesStart int
 		for idx, line := range lines {
-			if strings.Contains(line, UsageKey) {
+			if strings.Contains(line, UsageKey) || strings.Contains(line, LegacyUsageKey) {
 				linesStart = idx
 				break
 			}
@@ -74,7 +74,7 @@ func (s *Script) parse() error {
 		helpTextLines := lines[linesStart:helpEnds]
 		helpText := strings.Join(helpTextLines, "\n")
 
-		s.usage = strings.TrimSpace(strings.Split(lines[linesStart], UsageKey)[1])
+		s.usage = strings.TrimSpace(strings.SplitN(lines[linesStart], ":", 2)[1])
 		s.help = helpText
 	}
 	return nil