gime/cmd/root.go

/*
Copyright © 2022 Brian Buller <brian@bullercodeworks.com>

*/
package cmd

import (
	"fmt"
	"os"

	"git.bullercodeworks.com/brian/gime/util"
	gap "github.com/muesli/go-app-paths"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var (
	Version string
	Build   string

	configFile string

	// rootCmd represents the base command when called without any subcommands
	rootCmd = &cobra.Command{
		Use:   "gime",
		Short: "Timer.txt Client",
		Long:  fmt.Sprintf("Gime is a timer.txt client to make all aspects of timekeeping simpler.\nVersion: %s\nBuild Date: %s\n", Version, Build),
		RunE:  opStatus,
	}
)

// 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() {
	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}

func init() {
	rootCmd.Version = Version
	initConfig()
}

func initConfig() {
	viper.SetDefault("roundto", "15m")
	viper.SetDefault("timerfile", "timer.txt")
	viper.SetDefault("donefile", "done.txt")
	viper.SetDefault("reportfile", "report.txt")
	viper.SetDefault("copytags", false)
	var firstDir string // In case we need to make directories
	if configFile != "" {
		viper.SetConfigFile(configFile)
	} else {
		scope := gap.NewScope(gap.User, "gime")
		dirs, err := scope.ConfigDirs()
		if err != nil {
			fmt.Println("Can't retrieve default config. Please manually pass a config file with '--config'")
			os.Exit(1)
		}
		firstDir = dirs[0]
		for _, v := range dirs {
			viper.AddConfigPath(v)
		}
		viper.SetConfigName("gime")
		viper.SetConfigType("yaml")
	}
	var createConfig bool
	v2Path := fmt.Sprintf("%s%s%s", firstDir, string(os.PathSeparator), "gime.yaml")
	if err := viper.ReadInConfig(); err != nil {
		createConfig = true
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			// Config file not found; Check if we have a v1 config file
			fmt.Println("Config file not found.")
		} else {
			fmt.Println("Found config file, but another error occurred.")
			fmt.Println(err)
		}
	}
	if createConfig {
		_, err := os.Stat(firstDir)
		if os.IsNotExist(err) {
			err := os.Mkdir(firstDir, 0755)
			if err != nil {
				fmt.Println(err)
				os.Exit(1)
			}
		}
		if err = viper.WriteConfigAs(v2Path); err != nil {
			fmt.Println(err)
			os.Exit(1)
		}
	}
	dir := viper.GetString("directory")
	if dir == "" {
		dir = util.PromptUser("Path to timer.txt")
		if dir == "" {
			fmt.Println("No path to timer.txt given")
			os.Exit(1)
		}
		viper.Set("directory", dir)
		viper.WriteConfig()
	}
}