mirror of https://github.com/status-im/fathom.git
51 lines
1.4 KiB
Go
51 lines
1.4 KiB
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)
|
|
}
|
|
}
|