Added shell completions

cmd/root.go

@@ -54,6 +54,7 @@ func New() *cobra.Command {
 		createDeleteCmd(),
 		createNewCmd(),
 		createMergeCmd(),
+		createBashCompletionCmd(rootCmd),
 	)
 	return rootCmd
 }

cmd/shell_completion.go

@@ -0,0 +1,46 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+var shellVariant = "bash"
+
+func createBashCompletionCmd(rootCmd *cobra.Command) *cobra.Command {
+	var completionCmd = &cobra.Command{
+		Use:   "shell-completion",
+		Short: "Generates shell completion scripts",
+		Long: `Example usage (for bash): to load completion run
+	
+	. <(yq bash-completion)
+	
+	To configure your bash shell to load completions for each session add to your bashrc
+	
+	# ~/.bashrc or ~/.profile
+	. <(yq bash-completion)
+	`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			switch shellVariant {
+			case "bash", "":
+				rootCmd.GenBashCompletion(os.Stdout)
+				return nil
+			case "zsh":
+				rootCmd.GenZshCompletion(os.Stdout)
+				return nil
+			case "fish":
+				rootCmd.GenFishCompletion(os.Stdout, true)
+				return nil
+			case "powershell":
+				rootCmd.GenPowerShellCompletion(os.Stdout)
+				return nil
+			default:
+				return fmt.Errorf("Unknown variant %v", shellVariant)
+			}
+		},
+	}
+	completionCmd.PersistentFlags().StringVarP(&shellVariant, "variation", "V", "", "shell variation: bash (default), zsh, fish, powershell")
+	return completionCmd
+}