Esempio n. 1
0
func loadEnvCommandsReturnNewRoot(osArgs []string, rootCmd *cobra.Command) *cobra.Command {
	logs.WithField("path", ggn.Home.Config.WorkPath).Debug("Loading envs")
	work := work.NewWork(ggn.Home.Config.WorkPath)

	newRootCmd := &cobra.Command{
		Use: "ggn",
	}

	for _, envNames := range work.ListEnvs() {
		env := work.LoadEnv(envNames)

		envCmd := &cobra.Command{
			Use:   env.GetName(),
			Short: "Run command for " + env.GetName(),
			Run: func(cmd *cobra.Command, args []string) {

				newRootCmd.AddCommand(prepareEnvCommands(env))
				newRootCmd.SetArgs(osArgs[1:])
				newRootCmd.Execute()
			},
		}
		rootCmd.AddCommand(envCmd)
	}
	return newRootCmd
}
Esempio n. 2
0
or just source them in directly:
	$ . /etc/bash_completion`,

	Run: func(cmd *cobra.Command, args []string) {
		if autocompleteTarget == "" {
			autocompleteTarget = ggn.Home.Path + "/ggn_completion.sh"
		}
		if autocompleteType != "bash" {
			logs.WithField("type", autocompleteType).Fatal("Only Bash is supported for now")
		}

		work := work.NewWork(ggn.Home.Config.WorkPath)

		for _, command := range cmd.Root().Commands() {
			if command.Use != "genautocomplete" && command.Use != "version" && command.Use != "help [command]" { // TODO sux
				command.AddCommand(prepareEnvCommands(work.LoadEnv(command.Use)))
			}
		}
		err := cmd.Root().GenBashCompletionFile(autocompleteTarget)
		if err != nil {
			logs.WithE(err).Fatal("Failed to generate shell completion file")
		} else {
			logs.WithField("path", autocompleteTarget).Info("Bash completion saved")
		}
		os.Exit(0)
	},
}

func init() {
	genautocompleteCmd.PersistentFlags().StringVarP(&autocompleteTarget, "completionfile", "", "", "Autocompletion file")
	genautocompleteCmd.PersistentFlags().StringVarP(&autocompleteType, "type", "", "bash", "Autocompletion type (currently only bash supported)")