initial commit

2025-02-03 18:16:07 -05:00
commit 8ef3c029f2
19 changed files with 1354 additions and 0 deletions
--- a/main.go
+++ b/main.go
@ -0,0 +1,129 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+	"github.com/uptrace/bun/extra/bundebug"
+
+	"git.savin.nyc/alex/go-receipt-tracker-api/config"
+	"git.savin.nyc/alex/go-receipt-tracker-api/database"
+	"git.savin.nyc/alex/go-receipt-tracker-api/handlers"
+)
+
+var cfg = &config.Config{}
+
+const (
+	LoggingOutputJson = "JSON"
+	LoggingOutputText = "TEXT"
+)
+
+func configureLogging(config *config.Logging) *slog.Logger { //nolint:unparam
+	if config == nil {
+		return nil
+	}
+
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(config.Level)); err != nil {
+		slog.Warn(err.Error())
+		slog.Warn(fmt.Sprintf("logging level not recognized, defaulting to level %s", slog.LevelInfo.String()))
+		level = slog.LevelInfo
+	}
+
+	var handler slog.Handler
+	switch config.Output {
+	case LoggingOutputJson:
+		handler = slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{AddSource: cfg.Logging.Source, Level: level})
+	case LoggingOutputText:
+		handler = slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{AddSource: cfg.Logging.Source, Level: level})
+	default:
+		handler = slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{AddSource: cfg.Logging.Source, Level: level})
+	}
+
+	return slog.New(handler)
+}
+
+func main() {
+	sigs := make(chan os.Signal, 1)
+	done := make(chan bool, 1)
+	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		<-sigs
+		done <- true
+	}()
+
+	var err error
+	cfg, err = config.New()
+	if err != nil { // Handle errors reading the config file
+		slog.Error("Fatal error config file", "error", err.Error())
+		os.Exit(1)
+	}
+
+	// LOGGER
+	logger := configureLogging(cfg.Logging)
+	logger.Debug("CONFIG", "config", cfg)
+	// Create the "tasks" table in the database if it doesn't exist
+	db, err := database.NewDB(cfg.DataBase, logger)
+	if err != nil {
+		logger.Error(err.Error())
+		os.Exit(1)
+	}
+
+	// Add a query hook for logging
+	db.AddQueryHook(bundebug.NewQueryHook(
+		bundebug.WithVerbose(true),
+		bundebug.FromEnv("BUNDEBUG"),
+	))
+
+	// Ping the database to test the connection
+	err = db.Ping()
+	if err != nil {
+		logger.Debug("Failed to connect to the database")
+		return
+	}
+	// Connection successful
+	logger.Debug("Connected to the database")
+
+	handlers.DB = db
+	router := gin.Default()
+	router.Use(cors.Default())
+
+	v1 := router.Group("/api/v1")
+	{
+		v1.GET("/", handlers.HomePage)
+		v1.GET("/receipts", handlers.GetReceipts)
+		v1.GET("/receipts/:id", handlers.GetReceipt)
+		v1.DELETE("/receipts/:id", handlers.RemoveReceipt)
+		v1.POST("/receipts", handlers.AddReceipt)
+		v1.PUT("/receipts/:id", handlers.UpdateReceipt)
+		v1.GET("/receipts/:id/items", handlers.GetItems)
+	}
+	srv := &http.Server{
+		Addr:    ":8080",
+		Handler: router.Handler(),
+	}
+
+	go func() {
+		// service connections
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			logger.Error("cannot start ListenAndServe", "error", err)
+		}
+	}()
+
+	<-done
+	logger.Warn("caught signal, stopping...")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		logger.Error("cannot shutdown server gracefully", "error", err)
+	}
+	logger.Info("main.go finished")
+}