github.com/sohaha/zlsgo@v1.7.13-0.20240501141223-10dd1a906f76/zcli/cli.go

// Package zcli quickly build cli applications
package zcli

import (
	"bufio"
	"flag"
	"fmt"
	"os"
	"strings"

	"github.com/sohaha/zlsgo/zlog"
	"github.com/sohaha/zlsgo/zshell"
	"github.com/sohaha/zlsgo/zstring"
	"github.com/sohaha/zlsgo/ztype"
)

func init() {
	Log = zlog.New()
	Log.ResetFlags(zlog.BitLevel)
	// flag.CommandLine.SetOutput(ioutil.Discard)
	flag.CommandLine.SetOutput(&errWrite{})
	flag.Usage = func() {
		usage()
	}
}

// Add registers a cmd for the provided subCommand name
func Add(name, description string, command Cmd) *cmdCont {
	if name == "" {
		Log.Error(GetLangText("command_empty"))
		return &cmdCont{}
	}
	cmd := &cmdCont{
		name:          name,
		desc:          description,
		command:       command,
		requiredFlags: []string{},
	}
	cmds[name] = cmd
	cmdsKey = append(cmdsKey, name)
	return cmd
}

// SetUnknownCommand set unknown command handle
func SetUnknownCommand(fn func(_ string)) {
	unknownCommandFn = fn
}

func usage() {
	showHeadr()
	showFlagsAndRequired := func() {
		if numOfGlobalFlags() > 0 {
			showFlags(flag.CommandLine)
			showRequired(flag.CommandLine, requiredFlags)
		}
	}
	if len(cmds) == 0 {
		Log.Printf("usage of %s\n\n", showText(FirstParameter))
		showFlagsAndRequired()
		return
	}
	Log.Printf("usage: %s <command>\n\n\n", FirstParameter)
	Log.Println("where <command> is one of:")
	for _, name := range cmdsKey {
		if cont, ok := cmds[name]; ok {
			// for name, cont := range cmds {
			Log.Printf("    "+tipText("%-19s")+" %s\n", name, cont.desc)
		}
	}

	showFlagsAndRequired()
	if !HidePrompt {
		Log.Printf(showText("\nMore Command information, please use: %s <command> --help\n"), FirstParameter)
	}
}

func showFlags(fg *flag.FlagSet) {
	Log.Printf("\noptional flags:\n")
	max := 40
	showFlagsHelp()
	flagsItems := zstring.Buffer()
	fg.VisitAll(func(f *flag.Flag) {
		s := zstring.Buffer()
		flagsTitle := strings.Replace(f.Name, cliPrefix, "", 1)
		for _, key := range varShortsKey {
			if flagsTitle == key {
				return
			}
		}
		output := false
		if flagsTitle == "version" {
			output = true
		}
		name, usage := flag.UnquoteUsage(f)
		for key, v := range varsKey {
			shorts := v.shorts
			if key == flagsTitle && len(shorts) > 0 {
				for key := range shorts {
					shorts[key] = "-" + shorts[key]
				}
				flagsTitle += ", " + strings.Join(shorts, ", ")
			}
		}
		// if name == "" {
		// 	name = "bool"
		// }
		sf := "    -%-12s"
		if len(name) > 0 {
			newName := showText("<" + name + ">")
			namePadLen := 12 + len(newName) - len(name)
			flagsTitle += " " + newName
			sf = "    -%-" + ztype.ToString(namePadLen) + "s"
		}
		s.WriteString(warnText(fmt.Sprintf(sf, flagsTitle)))
		if zstring.Len(s.String()) <= max {
			s.WriteString("\t")
		} else {
			s.WriteString("\n    \t")
		}
		s.WriteString(strings.Replace(usage, "\n", "\n    \t", -1))
		defValue := ztype.ToString(f.DefValue)
		if defValue != "" && defValue != "0" && defValue != "false" {
			s.WriteString(fmt.Sprintf(" (default %v)", defValue))
		}
		if output {
			Log.Println(s.String())
		} else {
			s.WriteString("\n")
			flagsItems.WriteString(s.String())
		}
	})

	Log.Println(flagsItems.String())
}

// Start app
func Start(runFunc ...runFunc) {
	if *flagDetach {
		err := zshell.BgRun(strings.Join(runCmd, " "))
		if err != nil {
			Error(err.Error())
		}
		return
	}
	if matchingCmd != nil {
		if *flagHelp {
			showSubcommandUsage(flag.CommandLine, matchingCmd)
		} else {
			matchingCmd.command.Run(args)
		}
		return
	}
	requiredErr := parseRequiredFlags(flag.CommandLine, requiredFlags)
	if requiredErr != nil {
		Error(requiredErr.Error())
	}

	isRunFunc := len(runFunc) > 0
	if isRunFunc {
		runFunc[0]()
	} else {
		Help()
	}
}

// Run runnable
func Run(runFunc ...runFunc) (ok bool) {
	isRunFunc := len(runFunc) > 0
	parse(!isRunFunc)
	Start(runFunc...)
	return
}

func Input(problem string, required bool) (text string) {
	if problem != "" {
		fmt.Print(problem)
	}
	reader := bufio.NewReader(os.Stdin)
	text, _ = reader.ReadString('\n')
	if required && zstring.TrimSpace(text) == "" {
		return Input(problem, required)
	}
	return
}

func Inputln(problem string, required bool) (text string) {
	if problem != "" {
		problem = problem + "\n"
	}
	return Input(problem, required)
}

func Current() (interface{}, bool) {
	if matchingCmd == nil {
		return nil, false
	}
	return matchingCmd.command, true
}