root.go 3.43 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
// Copyright © 2019 NAME HERE <EMAIL ADDRESS>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package cmd

import (
	"fmt"
	"os"
20
	"strings"
21

Christopher Harm's avatar
Christopher Harm committed
22
	"github.com/fatih/color"
23 24 25 26 27
	homedir "github.com/mitchellh/go-homedir"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

Christopher Harm's avatar
Christopher Harm committed
28 29 30 31
const (
	configFile string = ".devtool.config"
)

32 33 34 35 36 37 38 39 40
var cfgFile string

// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
	Use:   "devtool",
	Short: "A developer's tool",
	Long:  `A tool used by software engineering to build applications.`,
	// Uncomment the following line if your bare application
	// has an action associated with it:
Christopher Harm's avatar
Christopher Harm committed
41 42 43 44 45 46
	Run: func(cmd *cobra.Command, args []string) {
		cmd.HelpFunc()(cmd, args)
	},
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		PrintLogo()
	},
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
}

// Execute adds all child commands to the root command and sets flags appropriately.
// This is called by main.main(). It only needs to happen once to the rootCmd.
func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func init() {
	cobra.OnInitialize(initConfig)

	// Here you will define your flags and configuration settings.
	// Cobra supports persistent flags, which, if defined here,
	// will be global for your application.
	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.devtool.yaml)")

	// Cobra also supports local flags, which will only run
	// when this action is called directly.
	// rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

// initConfig reads in config file and ENV variables if set.
func initConfig() {
	if cfgFile != "" {
		// Use config file from the flag.
		viper.SetConfigFile(cfgFile)
	} else {
		// Find home directory.
		home, err := homedir.Dir()
		if err != nil {
			fmt.Println(err)
			os.Exit(1)
		}

		// Search config in home directory with name ".devtool" (without extension).
		viper.AddConfigPath(home)
		viper.SetConfigName(".devtool")
	}

	viper.AutomaticEnv() // read in environment variables that match
90 91
	replacer := strings.NewReplacer(".", "_")
	viper.SetEnvKeyReplacer(replacer)
92 93

	// If a config file is found, read it in.
94 95 96
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error reading config file: %s\n", err)
97
	} else {
98
		fmt.Fprintf(os.Stderr, "Using config file: %s\n", viper.ConfigFileUsed())
99
	}
100
}
Christopher Harm's avatar
Christopher Harm committed
101 102 103 104 105 106 107 108 109

func PrintLogo() {
	color.Blue("__________  _____________ ___      .___             __                .__   ")
	color.Blue("\\______   \\/   _____/    |   \\   __| _/_______  ___/  |_  ____   ____ |  |  ")
	color.Blue(" |     ___/\\_____  \\|    |   /  / __ |/ __ \\  \\/ /\\   __\\/  _ \\ /  _ \\|  |  ")
	color.Blue(" |    |    /        \\    |  /  / /_/ \\  ___/\\   /  |  | (  <_> |  <_> )  |__")
	color.Blue(" |____|   /_______  /______/   \\____ |\\___  >\\_/   |__|  \\____/ \\____/|____/")
	color.Blue("                  \\/                \\/    \\/                                ")
}