UPDATE: Auth module, User module

2026-03-30 00:27:57 +07:00
parent 92d44bb00c
commit f04441bf2a
59 changed files with 4246 additions and 521 deletions
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@@ -0,0 +1,119 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	_ "history-api/docs"
+	"history-api/pkg/cache"
+	"history-api/pkg/config"
+	"history-api/pkg/database"
+	_ "history-api/pkg/log"
+	"history-api/pkg/mbtiles"
+	"os/signal"
+	"syscall"
+	"time"
+	"github.com/rs/zerolog/log"
+)
+
+func gracefulShutdown(fiberServer *FiberServer, done chan bool) {
+	// Create context that listens for the interrupt signal from the OS.
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	// Listen for the interrupt signal.
+	<-ctx.Done()
+
+	log.Info().Msg("shutting down gracefully, press Ctrl+C again to force")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := fiberServer.App.ShutdownWithContext(ctx); err != nil {
+		log.Info().Msgf("Server forced to shutdown with error: %v", err)
+	}
+
+	log.Info().Msg("Server exiting")
+
+	done <- true
+}
+
+//export StartServer
+func StartServer() {
+	err := config.LoadEnv()
+	if err != nil {
+		log.Error().Msg(err.Error())
+		panic(err)
+	}
+	poolPg, err := database.NewPostgresqlDB()
+	if err != nil {
+		log.Error().Msg(err.Error())
+		panic(err)
+	}
+	defer poolPg.Close()
+
+	sqlTile, err := mbtiles.NewMBTilesDB("data/map.mbtiles")
+	if err != nil {
+		log.Error().Msg(err.Error())
+		panic(err)
+	}
+	defer sqlTile.Close()
+
+	redisClient, err := cache.NewRedisClient()
+	if err != nil {
+		log.Error().Msg(err.Error())
+		panic(err)
+	}
+
+	serverIp, _ := config.GetConfig("SERVER_IP")
+	if serverIp == "" {
+		serverIp = "127.0.0.1"
+	}
+
+	httpPort, _ := config.GetConfig("SERVER_PORT")
+	if httpPort == "" {
+		httpPort = "3434"
+	}
+
+	serverHttp := NewHttpServer()
+	serverHttp.SetupServer(poolPg, sqlTile, redisClient)
+	Singleton = serverHttp
+
+	done := make(chan bool, 1)
+
+	err = serverHttp.App.Listen(fmt.Sprintf("%s:%s", serverIp, httpPort))
+	if err != nil {
+		log.Error().Msgf("Error: app failed to start on port %s, %v", httpPort, err)
+		panic(err)
+	}
+
+	// Run graceful shutdown in a separate goroutine
+	go gracefulShutdown(serverHttp, done)
+
+	// Wait for the graceful shutdown to complete
+	<-done
+	log.Info().Msg("Graceful shutdown complete.")
+}
+
+// @title           History API
+// @version         1.0
+// @description     This is a sample server for History API.
+// @termsOfService  http://swagger.io/terms/
+
+// @contact.name   API Support
+// @contact.url    http://www.swagger.io/support
+// @contact.email  support@swagger.io
+
+// @license.name  Apache 2.0
+// @license.url   http://www.apache.org/licenses/LICENSE-2.0.html
+
+// @host      history-api.kain.id.vn
+// @BasePath  /
+// @schemes   https http
+
+// @securityDefinitions.apikey BearerAuth
+// @in header
+// @name Authorization
+// @description Type "Bearer " followed by a space and JWT token.
+func main() {
+	StartServer()
+}
--- a/cmd/api/server.go
+++ b/cmd/api/server.go
@@ -0,0 +1,84 @@
+package main
+
+import (
+	"database/sql"
+	_ "embed"
+	"history-api/docs"
+	"history-api/internal/controllers"
+	"history-api/internal/gen/sqlc"
+	"history-api/internal/repositories"
+	"history-api/internal/routes"
+	"history-api/internal/services"
+	"history-api/pkg/cache"
+	"os"
+
+	swagger "github.com/gofiber/contrib/v3/swaggerui"
+	middleware "github.com/gofiber/contrib/v3/zerolog"
+	"github.com/gofiber/fiber/v3"
+	"github.com/gofiber/fiber/v3/middleware/cors"
+	"github.com/rs/zerolog"
+)
+
+var (
+	Singleton *FiberServer
+)
+
+type FiberServer struct {
+	App *fiber.App
+}
+
+func NewHttpServer() *FiberServer {
+	server := &FiberServer{
+		App: fiber.New(fiber.Config{
+			ServerHeader: "http-server",
+			AppName:      "http-server",
+		}),
+	}
+	cfg := swagger.Config{
+		BasePath:    "/",
+		FileContent: docs.SwaggerJSON,
+		Path:        "swagger",
+		Title:       "Swagger API Docs",
+	}
+
+	server.App.Use(swagger.New(cfg))
+
+	logger := zerolog.New(os.Stderr).With().Timestamp().Logger()
+	server.App.Use(middleware.New(middleware.Config{
+		Logger: &logger,
+	}))
+	return server
+}
+
+func (s *FiberServer) SetupServer(sqlPg sqlc.DBTX, sqlTile *sql.DB, redis cache.Cache) {
+	// Apply CORS middleware
+	s.App.Use(cors.New(cors.Config{
+		AllowOrigins:     []string{"*"},
+		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS", "PATCH"},
+		AllowHeaders:     []string{"Accept", "Authorization", "Content-Type", "Origin"},
+		AllowCredentials: false,
+		MaxAge:           300,
+	}))
+
+	// repo setup
+	userRepo := repositories.NewUserRepository(sqlPg, redis)
+	roleRepo := repositories.NewRoleRepository(sqlPg, redis)
+	tileRepo := repositories.NewTileRepository(sqlTile, redis)
+	tokenRepo := repositories.NewTokenRepository(redis)
+
+	// service setup
+	authService := services.NewAuthService(userRepo, roleRepo, tokenRepo, redis)
+	userService := services.NewUserService(userRepo, roleRepo)
+	tileService := services.NewTileService(tileRepo)
+
+	// controller setup
+	authController := controllers.NewAuthController(authService)
+	userController := controllers.NewUserController(userService)
+	tileController := controllers.NewTileController(tileService)
+
+	// route setup
+	routes.AuthRoutes(s.App, authController, userRepo)
+	routes.UserRoutes(s.App, userController, userRepo)
+	routes.TileRoutes(s.App, tileController)
+	routes.NotFoundRoute(s.App)
+}