fathom/fathom.go

package main

import (
	"math/rand"
	"os"

	"github.com/joho/godotenv"
	"github.com/usefathom/fathom/pkg/commands"
	"github.com/usefathom/fathom/pkg/datastore"

	"github.com/kelseyhightower/envconfig"
	log "github.com/sirupsen/logrus"
	kingpin "gopkg.in/alecthomas/kingpin.v2"
)

type Config struct {
	Database *datastore.Config

	Secret string
}

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()
)

func main() {
	cfg := parseConfig()

	// setup database connection
	db := datastore.Init(cfg.Database)
	defer db.Close()

	// 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)
	}
}

func parseConfig() *Config {
	var cfg Config
	godotenv.Load()
	err := envconfig.Process("Fathom", &cfg)
	if err != nil {
		log.Fatalf("Error parsing Fathom config from environment: %s", err)
	}

	// alias sqlite to sqlite3
	if cfg.Database.Driver == "sqlite" {
		cfg.Database.Driver = "sqlite3"
	}

	// if secret key is empty, use a randomly generated one to ease first-time installation
	if cfg.Secret == "" {
		cfg.Secret = randomString(40)
		os.Setenv("FATHOM_SECRET", cfg.Secret)
	}

	return &cfg
}

func randomString(len int) string {
	bytes := make([]byte, len)
	for i := 0; i < len; i++ {
		bytes[i] = byte(65 + rand.Intn(25)) //A=65 and Z = 65+25
	}

	return string(bytes)
}