1 files changed, 55 insertions, 3 deletions
diff --git a/main.go b/main.go
index 36463ff..2f08ba6 100644
--- a/main.go
+++ b/main.go
@@ -1,10 +1,13 @@
 package main
 
 import (
+	"database/sql"
+	"errors"
 	"flag"
-	"log"
 	"log/slog"
 	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/SayaAndy/saya-today-web/config"
 	"github.com/SayaAndy/saya-today-web/internal/b2"
@@ -14,9 +17,14 @@ import (
 	"github.com/SayaAndy/saya-today-web/locale"
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/middleware/redirect"
+	"github.com/golang-migrate/migrate/v4"
+	"github.com/golang-migrate/migrate/v4/database/sqlite3"
 	"github.com/yuin/goldmark"
 	"github.com/yuin/goldmark/parser"
 	gmhtml "github.com/yuin/goldmark/renderer/html"
+
+	_ "github.com/golang-migrate/migrate/v4/source/file"
+	_ "github.com/mattn/go-sqlite3"
 )
 
 var (
@@ -52,6 +60,32 @@ func main() {
 	slog.SetLogLoggerLevel(cfg.LogLevel)
 	slog.Info("starting sayana-web server...")
 
+	db, err := sql.Open(cfg.Auth.Db.Type, cfg.Auth.Db.Cfg.DSN)
+	if err != nil {
+		slog.Error("fail to initialize db", slog.String("error", err.Error()))
+		os.Exit(1)
+	}
+
+	driver, err := sqlite3.WithInstance(db, &sqlite3.Config{})
+	if err != nil {
+		slog.Error("fail to initialize driver for migrating db", slog.String("error", err.Error()))
+		os.Exit(1)
+	}
+
+	m, err := migrate.NewWithDatabaseInstance(
+		"file://migrations",
+		cfg.Auth.Db.Type, driver)
+	if err != nil {
+		slog.Error("fail to initialize migration client", slog.String("error", err.Error()))
+		os.Exit(1)
+	}
+
+	if err = m.Up(); err != nil && err == errors.New("no change") {
+		slog.Error("fail to apply migrations", slog.String("error", err.Error()))
+		os.Exit(1)
+	}
+	slog.Info("successfully applied migrations")
+
 	b2Client, err = b2.NewB2Client(&cfg.BlogPages.Storage.Config)
 	if err != nil {
 		slog.Error("fail to initialize b2 client", slog.String("error", err.Error()))
@@ -80,7 +114,11 @@ func main() {
 		StatusCode: 301,
 	}))
 
-	router.CCache = router.NewClientCache([]byte(cfg.Auth.Salt))
+	router.CCache, err = router.NewClientCache(db, []byte(cfg.Auth.Salt))
+	if err != nil {
+		slog.Error("fail to initialize cache", slog.String("error", err.Error()))
+		os.Exit(1)
+	}
 
 	app.Get("/", router.Root(cfg.AvailableLanguages))
 	app.Get("/:lang/map", router.Lang_Map(localization, availableLanguages, b2Client))
@@ -93,5 +131,19 @@ func main() {
 
 	app.Static("/", "./static")
 
-	log.Fatal(app.Listen(":3000"))
+	go func() {
+		if err := app.Listen(":3000"); err != nil {
+			slog.Error("error while running fiber server", slog.String("error", err.Error()))
+			panic(err)
+		}
+	}()
+
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+
+	<-sigChan
+	slog.Info("gracefully shutting down...")
+	app.Shutdown()
+	router.CCache.Close()
+	db.Close()
 }