package main

import (
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

var (
	aFlag bool
	bFlag bool
	cFlag string
)

func main() {
	// Define commands
	var rootCmd = &cobra.Command{
		Use: "mycli",
	}
	var firstCmd = &cobra.Command{
		Use: "first",
	}
	var secondCmd = &cobra.Command{
		Use: "second",
		Run: func(cmd *cobra.Command, args []string) {
			if aFlag {
				fmt.Println("flag A is set")
			}
			if bFlag {
				fmt.Println("flag B is set")
			}
			if cFlag != "" {
				fmt.Printf("flag C is set with value: %s\n", cFlag)
			}
		},
	}

	// Define flags
	rootCmd.PersistentFlags().BoolVarP(&aFlag, "set-A", "A", false, "Set the A flag")
	rootCmd.PersistentFlags().BoolVarP(&bFlag, "set-B", "B", false, "Set the B flag")
	rootCmd.PersistentFlags().StringVarP(&cFlag, "set-C", "C", "", "Set the C flag with a value")

	firstCmd.AddCommand(secondCmd)
	rootCmd.AddCommand(firstCmd)

	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}