package cmd

import (
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
)

var (
	target string
	center string
	mode   string
	debugGoc bool
	buildFlags string
	packages string
	appArgs string
)

// addBasicFlags adds a
func addBasicFlags(cmdset *pflag.FlagSet) *pflag.FlagSet {
	cmdset.StringVar(&center, "center", "http://127.0.0.1:7777", "cover profile host center")
	// bind to viper
	viper.BindPFlags(cmdset)
	return cmdset
}

func addCommonFlags(cmdset *pflag.FlagSet) *pflag.FlagSet {
	addBasicFlags(cmdset)
	cmdset.StringVar(&mode, "mode", "count", "coverage mode: set, count, atomic")
	cmdset.StringVar(&buildFlags, "buildflags", "",  "specify the build flags")
	// bind to viper
	viper.BindPFlags(cmdset)
	return cmdset
}

func addBuildFlags(cmdset *pflag.FlagSet) *pflag.FlagSet {
	addCommonFlags(cmdset)
	cmdset.StringVar(&packages, "packages", ".", "specify the package name, only . and ./... are supported")
	// bind to viper
	viper.BindPFlags(cmdset)
	return cmdset
}

func addRunFlags(cmdset *pflag.FlagSet) {
	addBuildFlags(cmdset)
	cmdset.Lookup("packages").Usage = "specify the package name, only ., ./... and *.go are supported"
	cmdset.StringVar(&appArgs, "appargs", "", "specify the application's arguments")
	// bind to viper
	viper.BindPFlags(cmdset)
}