cli/cmd/pastee.go

package cmd

import (
	"fmt"
	"github.com/mitchellh/go-homedir"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"os"
	"path"
)

var rootCmd = &cobra.Command{
	Use:   "pastee",
	Short: "A quick and easy to use Command-Line Paste Interface",
	Long:  `Fast, ad-free, secure, and feature filled! Submit pastes to Paste.ee from any operating system, anywhere!`,
	Args:  cobra.ArbitraryArgs,
	Run:   uploadCommandHandler,
}

var (
	cfgFile string
	Key     = ""
)

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
}

func init() {
	cobra.OnInitialize(initConfig)

	rootCmd.PersistentFlags().BoolP("encrypt", "e", false, "Encrypt uploaded contents")
	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.cobra.yaml)")
	rootCmd.PersistentFlags().Bool("viper", true, "use Viper for configuration")
	viper.BindPFlag("encrypt", rootCmd.PersistentFlags().Lookup("encrypt"))
	viper.SetDefault("apiKey", Key)

	rootCmd.AddCommand(loginCmd)
}

func initConfig() {
	if cfgFile != "" {
		// Use config file from the flag.
		viper.SetConfigFile(cfgFile)
	} else {
		home, err := homedir.Dir()
		cobra.CheckErr(err)

		viper.SetConfigName("pastee")
		viper.SetConfigType("yaml")
		viper.AddConfigPath(path.Join(home, ".config", "pastee"))
		viper.AddConfigPath("/etc")

		if err := viper.ReadInConfig(); err != nil {
			if _, ok := err.(viper.ConfigFileNotFoundError); ok {

			} else {
				fmt.Println("Error: Configuration file was unable to be read -", err)
				os.Exit(1)
			}
		}
	}

	viper.AutomaticEnv()
}