main.go

/*
This is a simple program for the recording of Chaturbate live streams.
*/

package main

import (
	"os"

	"github.com/mitchellh/cli"
	"github.com/sirupsen/logrus"
	"github.com/spf13/viper"
)

var log = logrus.New()

func main() {
	os.Exit(realMain())
}

var ui cli.Ui

func realMain() int {
	//Define version and print on startup
	version := "0.0.2"
	log.Info("EVEDBTool ", version)

	//Initialize config, logging and closehandler
	initConfig()
	setupLogging()

	ui = &cli.BasicUi{Writer: os.Stdout}

	cli := &cli.CLI{
		Args: os.Args[1:],
		Commands: map[string]cli.CommandFactory{
			"install": func() (cli.Command, error) {
				return &InstallCommand{}, nil
			},
			"up": func() (cli.Command, error) {
				return &UpCommand{}, nil
			},
			"down": func() (cli.Command, error) {
				return &DownCommand{}, nil
			},
			"redo": func() (cli.Command, error) {
				return &RedoCommand{}, nil
			},
			"status": func() (cli.Command, error) {
				return &StatusCommand{}, nil
			},
			"new": func() (cli.Command, error) {
				return &NewCommand{}, nil
			},
			"skip": func() (cli.Command, error) {
				return &SkipCommand{}, nil
			},
		},
		HelpFunc: cli.BasicHelpFunc("evedbtool"),
		Version:  version,
	}

	exitCode, err := cli.Run()
	if err != nil {
		log.Error(os.Stderr, "Error executing CLI: %s\n", err.Error())
		return 1
	}

	return exitCode
}

//Sets up logging levels based upon configuration file
func setupLogging() {
	log_level := viper.Get("log-level")

	if log_level == "Debug" {
		log.Level = logrus.DebugLevel
	} else if log_level == "Trace" {
		log.Level = logrus.TraceLevel
	} else if log_level == "Warn" {
		log.Level = logrus.WarnLevel
	} else if log_level == "Error" {
		log.Level = logrus.ErrorLevel
	} else {
		log.Level = logrus.InfoLevel
	}
}