[Live Now] Join the 1-day Testing & QA Summit featuring 15+ Expert Speakers.Join for FREE! Join TestMu Conference

How to use setGlobalFlags method of cmd Package

Best Gauge code snippet using cmd.setGlobalFlags

kctrl.go

Source:kctrl.go

1// Copyright 2020 VMware, Inc.2// SPDX-License-Identifier: Apache-2.03package cmd4import (5	"io"6	"github.com/cppforlife/cobrautil"7	"github.com/cppforlife/go-cli-ui/ui"8	cmdcore "github.com/rohitagg2020/kctrl/pkg/kctrl/cmd/core"9	cmdpkg "github.com/rohitagg2020/kctrl/pkg/kctrl/cmd/package"10	pkgcreate "github.com/rohitagg2020/kctrl/pkg/kctrl/cmd/package/create"11	"github.com/rohitagg2020/kctrl/pkg/kctrl/logger"12	"github.com/rohitagg2020/kctrl/pkg/kctrl/version"13	"github.com/spf13/cobra"14)15type KctrlOptions struct {16	ui            *ui.ConfUI17	logger        *logger.UILogger18	configFactory cmdcore.ConfigFactory19	depsFactory   cmdcore.DepsFactory20	UIFlags         UIFlags21	LoggerFlags     LoggerFlags22	KubeAPIFlags    cmdcore.KubeAPIFlags23	KubeconfigFlags cmdcore.KubeconfigFlags24}25func NewKctrlOptions(ui *ui.ConfUI, configFactory cmdcore.ConfigFactory,26	depsFactory cmdcore.DepsFactory) *KctrlOptions {27	return &KctrlOptions{ui: ui, logger: logger.NewUILogger(ui),28		configFactory: configFactory, depsFactory: depsFactory}29}30func NewDefaultKctrlCmd(ui *ui.ConfUI) *cobra.Command {31	configFactory := cmdcore.NewConfigFactoryImpl()32	depsFactory := cmdcore.NewDepsFactoryImpl(configFactory, ui)33	options := NewKctrlOptions(ui, configFactory, depsFactory)34	flagsFactory := cmdcore.NewFlagsFactory(configFactory, depsFactory)35	return NewKctrlCmd(options, flagsFactory)36}37func NewKctrlCmd(o *KctrlOptions, flagsFactory cmdcore.FlagsFactory) *cobra.Command {38	cmd := &cobra.Command{39		Use:   "kctrl",40		Short: "kctrl helps to manage packages and repositories on your Kubernetes cluster",41		RunE: cobrautil.ShowHelp,42		// Affects children as well43		SilenceErrors: true,44		SilenceUsage:  true,45		// Disable docs header46		DisableAutoGenTag: true,47		Version:           version.Version,48		// TODO bash completion49	}50	cmd.SetOutput(uiBlockWriter{o.ui}) // setting output for cmd.Help()51	cmd.SetUsageTemplate(cobrautil.HelpSectionsUsageTemplate([]cobrautil.HelpSection{52		cmdcore.PackageHelpGroup,53		cmdcore.RestOfCommandsHelpGroup,54	}))55	pkgOpts := cmdcore.PackageCommandTreeOpts{BinaryName: "kctrl", PositionalArgs: false, Color: true, JSON: true}56	SetGlobalFlags(o, cmd, flagsFactory, pkgOpts)57	ConfigurePathResolvers(o, cmd, flagsFactory)58	cmd.AddCommand(NewVersionCmd(NewVersionOptions(o.ui, o.depsFactory), flagsFactory))59	pkgCmd := cmdpkg.NewCmd()60	AddPackageCommands(o, pkgCmd, flagsFactory, pkgOpts)61	cmd.AddCommand(pkgCmd)62	ConfigureGlobalFlags(o, cmd, flagsFactory, pkgOpts.PositionalArgs)63	return cmd64}65func SetGlobalFlags(o *KctrlOptions, cmd *cobra.Command, flagsFactory cmdcore.FlagsFactory, opts cmdcore.PackageCommandTreeOpts) {66	o.UIFlags.Set(cmd, flagsFactory, opts)67	o.LoggerFlags.Set(cmd, flagsFactory)68	o.KubeAPIFlags.Set(cmd, flagsFactory)69	o.KubeconfigFlags.Set(cmd, flagsFactory)70}71func ConfigurePathResolvers(o *KctrlOptions, cmd *cobra.Command, flagsFactory cmdcore.FlagsFactory) {72	o.configFactory.ConfigurePathResolver(o.KubeconfigFlags.Path.Value)73	o.configFactory.ConfigureContextResolver(o.KubeconfigFlags.Context.Value)74	o.configFactory.ConfigureYAMLResolver(o.KubeconfigFlags.YAML.Value)75}76func ConfigureGlobalFlags(o *KctrlOptions, cmd *cobra.Command, flagsFactory cmdcore.FlagsFactory, positionalNameArg bool) {77	finishDebugLog := func(cmd *cobra.Command) {78		origRunE := cmd.RunE79		if origRunE != nil {80			cmd.RunE = func(cmd2 *cobra.Command, args []string) error {81				defer o.logger.DebugFunc("CommandRun").Finish()82				return origRunE(cmd2, args)83			}84		}85	}86	configureGlobal := cobrautil.WrapRunEForCmd(func(*cobra.Command, []string) error {87		o.UIFlags.ConfigureUI(o.ui)88		o.LoggerFlags.Configure(o.logger)89		o.KubeAPIFlags.Configure(o.configFactory)90		return nil91	})92	// Last one runs first93	// TODO: Add validation for number of arguments when positionalNameArg is true94	if positionalNameArg {95		cobrautil.VisitCommands(cmd, finishDebugLog, cobrautil.ReconfigureCmdWithSubcmd,96			configureGlobal, cobrautil.WrapRunEForCmd(cobrautil.ResolveFlagsForCmd))97	} else {98		cobrautil.VisitCommands(cmd, finishDebugLog, cobrautil.ReconfigureCmdWithSubcmd, configureGlobal, cobrautil.WrapRunEForCmd(cobrautil.ResolveFlagsForCmd))99	}100}101func AddPackageCommands(o *KctrlOptions, cmd *cobra.Command, flagsFactory cmdcore.FlagsFactory, opts cmdcore.PackageCommandTreeOpts) {102	cmd.AddCommand(pkgcreate.NewCreateCmd(pkgcreate.NewCreateOptions(o.logger, "kapp-a", opts), flagsFactory))103}104func AttachGlobalFlags(o *KctrlOptions, cmd *cobra.Command, flagsFactory cmdcore.FlagsFactory, opts cmdcore.PackageCommandTreeOpts) {105	SetGlobalFlags(o, cmd, flagsFactory, opts)106	ConfigurePathResolvers(o, cmd, flagsFactory)107	ConfigureGlobalFlags(o, cmd, flagsFactory, opts.PositionalArgs)108}109func AttachKctrlPackageCommandTree(cmd *cobra.Command, confUI *ui.ConfUI, opts cmdcore.PackageCommandTreeOpts) {110	configFactory := cmdcore.NewConfigFactoryImpl()111	depsFactory := cmdcore.NewDepsFactoryImpl(configFactory, confUI)112	options := NewKctrlOptions(confUI, configFactory, depsFactory)113	flagsFactory := cmdcore.NewFlagsFactory(configFactory, depsFactory)114	AddPackageCommands(options, cmd, flagsFactory, opts)115	AttachGlobalFlags(options, cmd, flagsFactory, opts)116}117type uiBlockWriter struct {118	ui ui.UI119}120var _ io.Writer = uiBlockWriter{}121func (w uiBlockWriter) Write(p []byte) (n int, err error) {122	w.ui.PrintBlock(p)123	return len(p), nil124}...

console_test.go

Source:console_test.go

...70			out := &bytes.Buffer{}71			cmd.SetOut(out)72			cmd.SetErr(out)73			viper.Reset()74			setGlobalFlags(test.flags)75			cmd.SetArgs(test.args)76			err := cmd.Execute()77			if test.expectedErr != "" {78				require.EqualError(t, err, test.expectedErr)79			}80		})81	}82}83func setGlobalFlags(flags map[string]string) {84	for k, v := range flags {85		viper.Set(k, v)86	}87}...

root.go

Source:root.go

...18	RunE:              createCmd.RunE,19	ValidArgsFunction: createCmd.ValidArgsFunction,20}21func init() {22	if err := setGlobalFlags(rootCmd.PersistentFlags()); err != nil {23		panic(fmt.Sprintf("unable to set global flags: %+v", err))24	}25	setCreateFlags(rootCmd.Flags())26}27func setGlobalFlags(flags *pflag.FlagSet) error {28	flags.StringVarP(&persistentOpts.ConfigPath, "config", "c", "", "application config file")29	flag := "quiet"30	flags.BoolP(31		flag, "q", false,32		"suppress all logging output",33	)34	if err := viper.BindPFlag(flag, flags.Lookup(flag)); err != nil {35		return err36	}37	flags.CountVarP(&persistentOpts.Verbosity, "verbose", "v", "increase verbosity (-v = info, -vv = debug)")38	return nil39}...

setGlobalFlags

Using AI Code Generation

1import (2type cmd struct {3}4func printUsage() {5	fmt.Printf("Usage: %s [-options] class [args...]", os.Args[0])6}7func (cmd *cmd) parseCmd() {8	flag.BoolVar(&cmd.helpFlag, "help", false, "print help message")9	flag.BoolVar(&cmd.helpFlag, "?", false, "print help message")10	flag.BoolVar(&cmd.versionFlag, "version", false, "print version and exit")11	flag.BoolVar(&cmd.versionFlag, "v", false, "print version and exit")12	flag.StringVar(&cmd.cpOption, "classpath", "", "classpath")13	flag.StringVar(&cmd.cpOption, "cp", "", "classpath")14	flag.StringVar(&cmd.XjreOption, "Xjre", "", "path to jre")15	flag.Parse()16	args := flag.Args()17	if len(args) > 0 {18	}19}20func main() {21	cmd := cmd{}22	cmd.parseCmd()23	if cmd.versionFlag {24		fmt.Println("version 0.0.1")25	} else if cmd.helpFlag || cmd.class == "" {26		printUsage()27	} else {28		startJVM(cmd)29	}30}31func startJVM(cmd cmd) {32	fmt.Printf("classpath:%s class:%s args:%v\n", cmd.cpOption, cmd.class, cmd.args)33}34import (35type cmd struct {36}37func printUsage() {38	fmt.Printf("Usage: %s [-options] class [args...]", os.Args[0])39}40func (cmd *cmd) parseCmd() {41	flag.BoolVar(&cmd.helpFlag, "help", false, "print help message")42	flag.BoolVar(&cmd.helpFlag, "?", false, "print help message")43	flag.BoolVar(&cmd.versionFlag, "

setGlobalFlags

Using AI Code Generation

1import (2type Cmd struct {3	Run func(cmd *Cmd, args []string)4}5func (c *Cmd) Usage() {6	fmt.Fprintf(os.Stderr, "usage: %s7	os.Exit(2)8}9func Main() {10	flag.Parse()11	args := flag.Args()12	if len(args) < 1 {13		usage()14	}15	for _, cmd := range commands {16		if cmd.Name() == args[0] {17			cmd.Flag.Usage = func() { cmd.Usage() }18			cmd.Flag.Parse(args[1:])19			args = cmd.Flag.Args()20			cmd.Run(cmd, args)21		}22	}23	fmt.Fprintf(os.Stderr, "go: unknown subcommand %q24	os.Exit(2)25}26func usage() {27	fmt.Fprintf(os.Stderr, "usage: go command [arguments]28	for _, cmd := range commands {29		fmt.Fprintf(os.Stderr, "\t%s\t%s30", cmd.Name(), cmd.Short)31	}32	fmt.Fprintf(os.Stderr, `33	for _, topic := range helpTopics {34		fmt.Fprintf(os.Stderr, "\t%s35	}36	fmt.Fprintf(os.Stderr, `37	os.Exit(2)38}

setGlobalFlags

Using AI Code Generation

1import (2func main() {3	cmd := &cobra.Command{4		Run: func(cmd *cobra.Command, args []string) {5			fmt.Println("Hello World")6		},7	}8	setGlobalFlags(cmd)9	cmd.Execute()10}11func setGlobalFlags(cmd *cobra.Command) {12	cmd.PersistentFlags().StringP("config", "c", "", "config file (default is $HOME/.cobra.yaml)")13}14  -c, --config string   config file (default is $HOME/.cobra.yaml)15      --config string   config file (default is $HOME/.cobra.yaml)16import (17func main() {18	cmd := &cobra.Command{19		Run: func(cmd *cobra.Command, args []string) {20			fmt.Println("Hello World")21		},22	}23	setGlobalFlags(cmd)24	cmd.Execute()25}26func setGlobalFlags(cmd *cobra.Command) {27	cmd.PersistentFlags().StringP("config", "c", "", "config file (default is $HOME/.cobra.yaml)")28}29func init() {30	cobra.OnInitialize(initConfig)31}32func initConfig() {33	viper.SetConfigFile(viper.GetString("config"))34	viper.SetConfigType("yaml")35	viper.AutomaticEnv()36	if err := viper.ReadInConfig(); err != nil {37		fmt.Println("Error reading config file, %s", err)38		os.Exit(1)39	}40}

setGlobalFlags

Using AI Code Generation

1import (2var (3	globalFlags = struct {4	}{}5func main() {6	var cmd = &cobra.Command{7		Run: func(cmd *cobra.Command, args []string) {8			fmt.Println("Hello World")9		},10	}11	cmd.PersistentFlags().BoolVarP(&globalFlags.verbose, "verbose", "v", false, "verbose output")12	if err := cmd.Execute(); err != nil {13		fmt.Println(err)14	}15}

setGlobalFlags

Using AI Code Generation

1import (2func main() {3    file := flag.String("file", "", "file name")4    path := flag.String("path", "", "path of the file")5    flag.Parse()6    if *file == "" || *path == "" {7        fmt.Println("Please provide file name and path")8        os.Exit(1)9    }10    if _, err := os.Stat(*path + "/" + *file); os.IsNotExist(err) {11        fmt.Println("Please provide a valid file name and path")12        os.Exit(1)13    }14    err := os.Chdir(*path)15    if err != nil {16        fmt.Println("Please provide a valid path")17        os.Exit(1)18    }19    wd, err := os.Getwd()20    if err != nil {21        fmt.Println("Error in getting path")22        os.Exit(1)23    }24    absPath, err := filepath.Abs(*file)25    if err != nil {26        fmt.Println("Error in getting file path")27        os.Exit(1)28    }29    relPath, err := filepath.Rel(wd, absPath)30    if err != nil {31        fmt.Println("Error in getting relative path")32        os.Exit(1)33    }34    cmd := exec.Command("./" + relPath)35    err = cmd.Run()36    if err != nil {37        fmt.Println("Error in executing file")38        os.Exit(1)39    }40}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.