main.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"

	"github.com/sachaos/toggl/cache"
	"github.com/sachaos/toggl/command"
	"github.com/sachaos/toggl/lib"
	"github.com/spf13/viper"
	"github.com/urfave/cli"
)

var (
	ConfigPath = os.Getenv("HOME")
)

const (
	ConfigName = ".toggl"
	ConfigType = "json"
)


const Name string = "toggl"

var version string

func main() {
	cache.New(os.Getenv("HOME") + "/.toggl.cache.json")
	cache.Init()

	initialize()

	cmdApp := command.NewApp(viper.GetString("token"))

	app := cli.NewApp()
	app.Name = Name
	app.Version = version
	app.Author = "sachaos"
	app.Email = "sakataku7@gmail.com"
	app.Usage = "Toggl API CLI Client"

	app.Flags = GlobalFlags
	app.Commands = []cli.Command{
		{
			Name:   "start",
			Usage:  "Start time entry",
			Action: cmdApp.CmdStart,
			Flags: []cli.Flag{
				projectIDFlag,
			},
		},
		{
			Name:   "stop",
			Usage:  "End time entry",
			Action: cmdApp.CmdStop,
			Flags:  []cli.Flag{},
		},
		{
			Name:   "current",
			Usage:  "Show current time entry",
			Action: cmdApp.CmdCurrent,
			Flags:  []cli.Flag{},
		},
		{
			Name:   "workspaces",
			Usage:  "Show workspaces",
			Action: cmdApp.CmdWorkspaces,
		},
		{
			Name:   "projects",
			Usage:  "Show projects on current workspaces",
			Action: cmdApp.CmdProjects,
		},
		{
			Name:   "local",
			Usage:  "Set current dir workspace",
			Action: CmdLocal,
			Flags: []cli.Flag{
				projectIDFlag,
			},
		},
		{
			Name:   "global",
			Usage:  "Set global workspace",
			Action: CmdGlobal,
		},
	}
	app.CommandNotFound = CommandNotFound

	if err := app.Run(os.Args); err != nil {
		log.Fatalf("Error: %s", err.Error())
		os.Exit(1)
	}
}

func requireToken() error {
	var token string
	var workspaces []toggl.Workspace
	var err error
	count := 0
	for count < 3 {
		fmt.Printf("Input API Token: ")
		fmt.Scan(&token)
		client := toggl.NewDefaultClient(token)
		workspaces, err = client.FetchWorkspaces()
		if err == nil {
			viper.Set("token", token)
			viper.Set("wid", workspaces[0].ID)
			return nil
		}
		count++
	}
	panic(fmt.Errorf("Invalid token"))
}

func RootConfigFilePath() string {
	return filepath.Join(ConfigPath, ConfigName+"."+ConfigType)
}

func LocalConfigFilePath() string {
	return filepath.Join(".", ConfigName+"."+ConfigType)
}

func CreateConfig(path string, hash interface{}) {
	buf, _ := json.MarshalIndent(hash, "", "  ")
	err := ioutil.WriteFile(path, buf, os.ModePerm)
	if err != nil {
		panic(fmt.Errorf("Fatal error config file: %s \n", err))
	}
}

func LoadLocalConfig() error {
	localFilePath := LocalConfigFilePath()
	file, err := os.Open(localFilePath)
	if err != nil {
		return err
	}
	viper.MergeConfig(file)
	return nil
}

func initialize() {
	viper.SetConfigType(ConfigType)
	viper.SetConfigName(ConfigName)
	viper.AddConfigPath(ConfigPath)
	viper.AddConfigPath(".")
	viper.ReadInConfig()

	LoadLocalConfig()

	if !viper.IsSet("token") {
		requireToken()
		CreateConfig(RootConfigFilePath(), viper.AllSettings())
	}
}