fathom/commands/commands.go

package commands

import (
	"flag"

	"github.com/dannyvankooten/ana/count"
	"github.com/dannyvankooten/ana/db"
)

var runCreateUserCommand bool
var runDeleteUserCommand bool
var runStartServerCommand bool
var runSeedDataCommand bool
var runArchiveDataCommand bool
var idArg int
var emailArg string
var passwordArg string
var nArg int
var portArg int

// Parse CLI arguments
func Parse() {
	// parse commands
	flag.BoolVar(&runCreateUserCommand, "create_user", false, "Create a new user")
	flag.BoolVar(&runDeleteUserCommand, "delete_user", false, "Deletes a user")
	flag.BoolVar(&runStartServerCommand, "start_server", true, "Start the API web server, listen on -port")
	flag.BoolVar(&runSeedDataCommand, "seed_data", false, "Seed the database -n times")
	flag.BoolVar(&runArchiveDataCommand, "archive_data", false, "Aggregates data into daily totals")
	flag.StringVar(&emailArg, "email", "", "Email address")
	flag.StringVar(&passwordArg, "password", "", "Password")
	flag.IntVar(&idArg, "id", 0, "Object ID")
	flag.IntVar(&nArg, "n", 0, "Number")
	flag.IntVar(&portArg, "port", 8080, "Port")
	flag.Parse()
}

// Run parsed CLI command. Defaults to starting the HTTP server.
func Run() {
	if runCreateUserCommand {
		createUser()
	} else if runDeleteUserCommand {
		deleteUser()
	} else if runSeedDataCommand {
		db.Seed(nArg)
	} else if runArchiveDataCommand {
		count.Archive()
	} else if runStartServerCommand {
		startServer(portArg)
	}
}