templates.go

package cmd

import (
	"github.com/spf13/cobra"
	"github.com/spf13/pflag"
	"golang.org/x/term"
)

func getUsageTemplate() string {
	return `Usage:{{if .Runnable}}
  {{.UseLine}}{{end}}{{if .HasAvailableSubCommands}}
  {{.CommandPath}} [command]{{end}}
{{if not .HasParent}}
Commands:
  {{rpad "setup" 29}} Create your first feature flag using a step-by-step guide
  {{rpad "config" 29}} View and modify specific configuration values
  {{rpad "completion" 29}} Enable command autocompletion within supported shells
  {{rpad "login" 29}} Log in to your LaunchDarkly account
  {{rpad "signup" 29}} Create a new LaunchDarkly account
  {{rpad "dev-server" 29}} Run a development server to serve flags locally

Common resource commands:
  {{rpad "flags" 29}} List, create, and modify feature flags and their targeting
  {{rpad "environments" 29}} List, create, and manage environments
  {{rpad "projects" 29}} List, create, and manage projects
  {{rpad "members" 29}} Invite new members to an account
  {{rpad "segments" 29}} List, create, modify, and delete segments
  {{rpad "sourcemaps" 29}} Manage sourcemaps for error monitoring
  {{rpad "symbols" 29}} Manage symbol files for error monitoring
  {{rpad "..." 29}} To see more resource commands, run 'ldcli resources'

Flags:
{{.LocalFlags.FlagUsages | trimTrailingWhitespaces}}
{{else}}{{if .HasAvailableSubCommands}}{{$cmds := .Commands}}

Available Commands:{{range $cmds}}{{if (or .IsAvailableCommand (eq .Name "help"))}}
  {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableLocalFlags}}

Flags:
{{.LocalFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasInheritedFlags}}

Global flags:
{{.InheritedFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasAvailableSubCommands}}

Use "{{.CommandPath}} [command] --help" for more information about a command.{{end}}
{{end}}`
}

func HasRequiredFlags(cmd *cobra.Command) bool {
	var numFlags int
	cmd.LocalFlags().VisitAll(func(flag *pflag.Flag) {
		_, ok := flag.Annotations["required"]
		if ok {
			numFlags += 1
		}
	})

	return numFlags > 0
}

func HasOptionalFlags(cmd *cobra.Command) bool {
	var numFlags int
	cmd.LocalFlags().VisitAll(func(flag *pflag.Flag) {
		_, ok := flag.Annotations["required"]
		if !ok && flag.Name != "help" {
			numFlags += 1
		}
	})

	return numFlags > 0
}

func WrappedRequiredFlagUsages(cmd *cobra.Command) string {
	flagSet := pflag.NewFlagSet("request", pflag.ExitOnError)

	cmd.LocalFlags().VisitAll(func(flag *pflag.Flag) {
		if _, ok := flag.Annotations["required"]; ok {
			flagSet.AddFlag(flag)
		}
	})

	return flagSet.FlagUsagesWrapped(getTerminalWidth())
}

func WrappedOptionalFlagUsages(cmd *cobra.Command) string {
	flagSet := pflag.NewFlagSet("request", pflag.ExitOnError)

	cmd.LocalFlags().VisitAll(func(flag *pflag.Flag) {
		_, ok := flag.Annotations["required"]
		if !ok && flag.Name != "help" {
			flagSet.AddFlag(flag)
		}
	})

	return flagSet.FlagUsagesWrapped(getTerminalWidth())
}

func getTerminalWidth() int {
	var width int

	width, _, err := term.GetSize(0)
	if err != nil {
		width = 80
	}

	return width
}