mirror of https://github.com/status-im/fathom.git
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/joho/godotenv"
|
|
"github.com/robfig/cron"
|
|
"github.com/usefathom/fathom/pkg/commands"
|
|
"github.com/usefathom/fathom/pkg/count"
|
|
"github.com/usefathom/fathom/pkg/datastore"
|
|
"gopkg.in/alecthomas/kingpin.v2"
|
|
"os"
|
|
|
|
"github.com/kelseyhightower/envconfig"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Config struct {
|
|
Database struct {
|
|
Driver string `default:"mysql"`
|
|
Host string `default:"localhost"`
|
|
User string `required:"true"`
|
|
Password string `required:"true"`
|
|
Name string `default:"fathom"`
|
|
}
|
|
|
|
Secret string `required:"true"`
|
|
}
|
|
|
|
var (
|
|
app = kingpin.New("fathom", "Simple website analytics.")
|
|
register = app.Command("register", "Register a new user.")
|
|
registerEmail = register.Arg("email", "Email for user.").Required().String()
|
|
registerPassword = register.Arg("password", "Password for user.").Required().String()
|
|
server = app.Command("server", "Start webserver.").Default()
|
|
serverPort = server.Flag("port", "Port to listen on.").Default("8080").Int()
|
|
serverWebRoot = server.Flag("webroot", "Root directory of static assets").Default("./").String()
|
|
archive = app.Command("archive", "Process unarchived data.")
|
|
seed = app.Command("seed", "Seed the database.")
|
|
seedN = seed.Flag("n", "Number of records to seed.").Int()
|
|
)
|
|
|
|
func main() {
|
|
// load .env file
|
|
var cfg Config
|
|
godotenv.Load()
|
|
err := envconfig.Process("Fathom", &cfg)
|
|
if err != nil {
|
|
log.Fatalf("Error parsing Fathom config from environment: %s", err)
|
|
}
|
|
|
|
// setup database connection
|
|
dbcfg := cfg.Database
|
|
db := datastore.Init(dbcfg.Driver, dbcfg.Host, dbcfg.Name, dbcfg.User, dbcfg.Password)
|
|
defer db.Close()
|
|
|
|
// setup cron to run count.Archive every hour
|
|
c := cron.New()
|
|
c.AddFunc("@hourly", count.Archive)
|
|
c.Start()
|
|
|
|
// parse & run cli commands
|
|
app.Version("1.0")
|
|
app.UsageTemplate(kingpin.CompactUsageTemplate)
|
|
switch kingpin.MustParse(app.Parse(os.Args[1:])) {
|
|
case "register":
|
|
commands.Register(*registerEmail, *registerPassword)
|
|
|
|
case "server":
|
|
commands.Server(*serverPort, *serverWebRoot)
|
|
|
|
case "archive":
|
|
commands.Archive()
|
|
|
|
case "seed":
|
|
commands.Seed(*seedN)
|
|
}
|
|
|
|
}
|