main.go

package main

import (
	"context"
	"embed"
	"io/fs"
	"math/rand"
	"os"
	"os/signal"
	"time"

	"bfp/api"
	"bfp/conf"
	"bfp/echologrus"
	"bfp/model"
	"github.com/oschwald/geoip2-golang"
	log "github.com/sirupsen/logrus"
)

// @title BFP API
// @version 1.0
// @description BFP API

// @license.name Apache 2.0
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html

// @BasePath /api/

// go get -u github.com/swaggo/swag/cmd/swag
//go:generate swag init --parseDependency
//go:generate rm -f docs/swagger.json docs/docs.go

//go:embed static
var frontendStatic embed.FS

func FrontendStatic() fs.FS {
	fsys, err := fs.Sub(frontendStatic, "static")
	if err != nil {
		panic(err)
	}
	return fsys
}

func main() {
	rand.Seed(time.Now().UnixNano())

	cfg, err := conf.ReadConfig()
	if err != nil {
		log.Fatalf("error loading config file: %s", err)
	}
	initLogger(cfg.LogLevel)

	geoDb, err := geoip2.Open("GeoLite2.mmdb")
	if err != nil {
		log.Panicf("error opening GeoLite2.mmdb: %v", err)
	}
	defer geoDb.Close()

	db, err := model.NewDB(cfg.DBConnectionString)
	if err != nil {
		log.Panicf("init db: %v", err)
	}
	defer db.Close()

	fpRep := model.NewFingerprint(db)
	wordsRep, err := model.NewWordsRepository(db)
	if err != nil {
		log.Panicf("init words: %v", err)
	}

	handler := api.NewHandler(fpRep, wordsRep, geoDb)
	handler.SetupAPI(cfg)
	handler.SetupFrontend(FrontendStatic())

	quit := make(chan os.Signal, 2)
	signal.Notify(quit, os.Interrupt)
	<-quit
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := handler.Shutdown(ctx); err != nil {
		log.Errorf("main: shutdown http server: %s", err)
	}
}

func initLogger(level string) {
	log.SetFormatter(&log.TextFormatter{})
	log.SetOutput(os.Stdout)
	log.SetLevel(log.DebugLevel)
	lvl, err := log.ParseLevel(level)
	if err != nil {
		log.Warnf("invalid log level %s", level)
	}
	log.SetLevel(lvl)

	echologrus.Logger = log.StandardLogger()
}