imagini/cmd/main.go

package main

import (
    "os"
    "os/signal"
    "github.com/urfave/cli/v2"
    log "github.com/sirupsen/logrus"

    "reichard.io/imagini/cmd/server"

    "reichard.io/imagini/plugin"
    "github.com/99designs/gqlgen/api"
	"github.com/99designs/gqlgen/codegen/config"
)

type UTCFormatter struct {
    log.Formatter
}

func (u UTCFormatter) Format(e *log.Entry) ([]byte, error) {
    e.Time = e.Time.UTC()
    return u.Formatter.Format(e)
}

func main() {
    log.SetFormatter(UTCFormatter{&log.TextFormatter{FullTimestamp: true}})

    app := &cli.App{
        Name: "Imagini",
        Usage: "A self hosted photo library.",
        Commands: []*cli.Command{
            {
                Name:    "serve",
                Aliases: []string{"s"},
                Usage:   "Start Imagini web server.",
                Action:  cmdServer,
            },
            {
				Name:  "generate",
				Usage: "generate graphql schema",
				Action: cmdGenerate,
			},
        },
    }
    err := app.Run(os.Args)
    if err != nil {
        log.Fatal(err)
    }
}

func cmdServer(ctx *cli.Context) error {
    log.Info("Starting Imagini Server")
    server := server.NewServer()
    server.StartServer()

    c := make(chan os.Signal, 1)
    signal.Notify(c, os.Interrupt)
    <-c

    server.StopServer()
    os.Exit(0)

    return nil
}

func cmdGenerate(ctx *cli.Context) error {
    log.Info("Generating Imagini Models")
    gqlgenConf, err := config.LoadConfigFromDefaultLocations()
    if err != nil {
        log.Panic("Failed to load config", err.Error())
        os.Exit(2)
    }

    log.Info("Generating Schema...")
    err = api.Generate(gqlgenConf,
        api.AddPlugin(plugin.New()),
    )
    log.Info("Schema Generation Done")
    if err != nil {
        log.Panic(err.Error())
        os.Exit(3)
    }
    os.Exit(0)
    return nil
}