cli.go

package habit

import (
	"flag"
	"fmt"

	"github.com/mitchellh/go-homedir"
	"io"
)

//RunCLI parses arguments and passes them to habit.Controller
func RunCLI(args []string, output io.Writer) {
	flagSet := flag.NewFlagSet("habit", flag.ContinueOnError)
	flagSet.SetOutput(output)
	flagSet.Usage = func() {
		fmt.Fprintln(output,
			`habit is an application to assist you in building habits
Usage: habit <Option Flags> <HABIT_NAME> -- to create/update a new habit
       habit all   --   to list all habits
Option Flags:`)
		flagSet.PrintDefaults()
	}

	frequency := flagSet.String("f", "daily", "Set the frequency of the habit: daily, weekly.")
	storeType := flagSet.String("s", "db", "Set the store backend for habit tracker: db, file.")
	homeDir, err := homedir.Dir()
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	storeDir := flagSet.String("d", homeDir, "Set the store directory.")

	err = flagSet.Parse(args)
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}

	if len(flagSet.Args()) == 0 {
		flagSet.Usage()
		return
	}

	if len(flagSet.Args()) > 1 {
		fmt.Fprintln(output, "too many args")
		flagSet.Usage()
		return
	}

	store, err := storeFactory(*storeType, *storeDir)
	if err != nil {
		fmt.Fprintln(output, err)
		flagSet.Usage()
		return
	}
	controller, err := NewController(store)
	if err != nil {
		fmt.Fprintln(output, err)
		flagSet.Usage()
		return
	}

	if flagSet.Args()[0] == "all" {
		fmt.Fprintln(output, controller.GetAllHabits())
		return
	}

	h, err := parseHabit(flagSet.Args()[0], *frequency)
	if err != nil {
		fmt.Fprintln(output, err)
		flagSet.Usage()
		return
	}

	h, err = controller.Handle(h)
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	fmt.Fprintln(output, h)
}

//RunServer parses args and starts HTTP habit server on provided address
func RunServer(args []string, output io.Writer) {
	if len(args) == 0 {
		fmt.Fprintln(output, "no address provided")
		return
	}
	if len(args) > 1 {
		fmt.Fprintln(output, "too many args provided")
		return
	}
	homeDir, err := homedir.Dir()
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	store, err := OpenDBStore(homeDir + "/.habitTracker.db")
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	controller, err := NewController(store)
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	server, err := NewServer(&controller, args[0])
	if err != nil {
		fmt.Fprintln(output, err)
		return
	}
	fmt.Fprintln(output, "Starting HTTP server")
	server.Run()
}

func storeFactory(storeType string, dir string) (store Store, err error) {
	switch storeType {
	case "db":
		store, err = OpenDBStore(dir + "/.habitTracker.db")
	case "file":
		store, err = OpenFileStore(dir + "/.habitTracker")
	default:
		return nil, fmt.Errorf("unknown store type %s", storeType)
	}
	if err != nil {
		return nil, err
	}
	return store, nil
}