mirror of https://github.com/status-im/fathom.git
74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli"
|
|
"github.com/usefathom/fathom/pkg/config"
|
|
"github.com/usefathom/fathom/pkg/datastore"
|
|
)
|
|
|
|
type App struct {
|
|
*cli.App
|
|
database datastore.Datastore
|
|
config *config.Config
|
|
}
|
|
|
|
// CLI application
|
|
var app *App
|
|
|
|
// Run parses the CLI arguments & run application command
|
|
func Run(version string, commit string, buildDate string) error {
|
|
// force all times in UTC, regardless of server timezone
|
|
time.Local = time.UTC
|
|
|
|
// setup CLI app
|
|
app = &App{cli.NewApp(), nil, nil}
|
|
app.Name = "Fathom"
|
|
app.Usage = "simple & transparent website analytics"
|
|
app.Version = fmt.Sprintf("%v, commit %v, built at %v", strings.TrimPrefix(version, "v"), commit, buildDate)
|
|
app.HelpName = "fathom"
|
|
app.Flags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "config, c",
|
|
Value: ".env",
|
|
Usage: "Load configuration from `FILE`",
|
|
},
|
|
}
|
|
app.Before = before
|
|
app.After = after
|
|
app.Commands = []cli.Command{
|
|
serverCmd,
|
|
userCmd,
|
|
statsCmd,
|
|
}
|
|
|
|
if len(os.Args) < 2 || os.Args[1] != "--version" {
|
|
log.Printf("%s version %s", app.Name, app.Version)
|
|
}
|
|
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func before(c *cli.Context) error {
|
|
configFile := c.String("config")
|
|
config.LoadEnv(configFile)
|
|
app.config = config.Parse()
|
|
app.database = datastore.New(app.config.Database)
|
|
return nil
|
|
}
|
|
|
|
func after(c *cli.Context) error {
|
|
err := app.database.Close()
|
|
return err
|
|
}
|