main.go

package main

import (
	"github.com/jinzhu/gorm"

	flag "github.com/ogier/pflag"
	"github.com/sirupsen/logrus"

	"github.com/jclebreton/opensirene/api/router"
	"github.com/jclebreton/opensirene/conf"
	"github.com/jclebreton/opensirene/database"
	"github.com/jclebreton/opensirene/logic"
)

// This variable is overridden at compile time when using script/build.sh
var version = "dev"
var buildDate = "no build date"

func main() {
	var err error
	var config string
	var fullImport bool

	// Configuration
	flag.StringVarP(&config, "config", "c", "conf.yml", "Path to the configuration file")
	flag.BoolVarP(&fullImport, "drop", "", false, "Truncate database and run a full import")
	flag.Parse()
	if err = conf.Load(config); err != nil {
		logrus.WithError(err).Fatal("Couldn't parse configuration")
	}

	// Init PGX database client
	var pgxClient *database.PgxClient
	if pgxClient, err = database.NewImportClient(); err != nil {
		logrus.WithError(err).Fatal("Couldn't initialize PGX client")
	}
	defer pgxClient.Conn.Close()

	// Full import
	if fullImport {
		if err = logic.ResetDatabase(pgxClient); err != nil {
			logrus.WithError(err).Fatal("Couldn't reset database")
		}
		logrus.Info("Database has been reset to trigger automatic update")
	}

	// Start automatic updates
	crontab := &logic.Crontab{PgxClient: pgxClient, Config: conf.C.Crontab}
	go crontab.Start()

	// Start API
	var gormClient *gorm.DB
	if gormClient, err = database.NewGORMClient(); err != nil {
		logrus.WithError(err).Fatal("Couldn't initialize GORM")
	}
	defer func() {
		err = gormClient.Close()
		logrus.WithError(err).Fatal("Couldn't close GORM")
	}()
	if err = router.SetupAndRun(gormClient, version, buildDate); err != nil {
		logrus.WithError(err).Fatal("Could not setup and run API")
	}
}