117 lines
2.2 KiB
Go
117 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/lmas/ss13_se/src"
|
|
)
|
|
|
|
var (
|
|
// General flags
|
|
f_debug = flag.Bool("debug", false, "Run in debug mode")
|
|
f_verbose = flag.Bool("verbose", false, "Show verbose messages")
|
|
|
|
// Flags for the run (webserver) command
|
|
f_addr = flag.String("addr", "127.0.0.1:8000", "Server's listening address")
|
|
|
|
// Flags for the update command
|
|
f_daemon = flag.Bool("daemon", false, "Continuously update")
|
|
f_timeout = flag.Int64("timeout", 15, "Time (in minutes) between each update in daemon mode")
|
|
)
|
|
|
|
type Command struct {
|
|
Name string
|
|
Desc string
|
|
DoFunc func()
|
|
}
|
|
|
|
var commands = []Command{
|
|
// Main commands
|
|
Command{"serve", "Run a local web server.", doserve},
|
|
Command{"update", "Update the population stats.", doupdate},
|
|
|
|
// Misc. commands
|
|
Command{"version", "Show current version and exit.", doversion},
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
ss13.SetDebug(*f_debug)
|
|
|
|
command := flag.Arg(0)
|
|
for _, a := range commands {
|
|
if command == a.Name {
|
|
a.DoFunc()
|
|
return
|
|
}
|
|
}
|
|
|
|
dohelp()
|
|
}
|
|
|
|
func dohelp() {
|
|
fmt.Println(`SS13 population stats server.
|
|
|
|
Usage:
|
|
TODO_NAME [flags] command [arguments]
|
|
|
|
Available commands:`)
|
|
for _, a := range commands {
|
|
fmt.Printf("\t%s\t%s\n", a.Name, a.Desc)
|
|
}
|
|
fmt.Println(`
|
|
Run TODO_NAME -h to get a list of command line flags.
|
|
`)
|
|
}
|
|
|
|
func doversion() {
|
|
fmt.Println("TODO: Use a fancy name, add a banner and set a version.")
|
|
}
|
|
|
|
func doserve() {
|
|
if *f_verbose {
|
|
fmt.Printf("Listening on %s.\n\n", *f_addr)
|
|
}
|
|
|
|
instance := &ss13.Instance{
|
|
Debug: *f_debug,
|
|
DB: ss13.OpenSqliteDB("new.db"), // TODO
|
|
}
|
|
instance.Init()
|
|
instance.Serve(*f_addr)
|
|
}
|
|
|
|
func doupdate() {
|
|
td := time.Duration(*f_timeout) * time.Minute
|
|
if *f_verbose {
|
|
if *f_daemon {
|
|
fmt.Printf("Running updates every %v minutes\n", *f_timeout)
|
|
} else {
|
|
fmt.Println("Updating...")
|
|
}
|
|
}
|
|
|
|
instance := &ss13.Instance{
|
|
Debug: *f_debug,
|
|
DB: ss13.OpenSqliteDB("new.db"), // TODO
|
|
}
|
|
instance.Init()
|
|
|
|
for {
|
|
start := time.Now()
|
|
instance.UpdateServers()
|
|
stop := time.Now()
|
|
if *f_verbose {
|
|
fmt.Printf("Update completed in %v\n", stop.Sub(start))
|
|
}
|
|
|
|
if !*f_daemon {
|
|
return
|
|
}
|
|
fmt.Println("Next at ", time.Now().Add(td))
|
|
time.Sleep(td)
|
|
}
|
|
}
|