add web service

cmd/remembos/main.go

@@ -0,0 +1,105 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"git.vakhrushev.me/av/remembos/internal/config"
+	"git.vakhrushev.me/av/remembos/internal/memory"
+	"git.vakhrushev.me/av/remembos/internal/memos"
+	"git.vakhrushev.me/av/remembos/internal/search"
+	"git.vakhrushev.me/av/remembos/internal/storage"
+	"git.vakhrushev.me/av/remembos/internal/web"
+)
+
+func main() {
+	configPath := flag.String("config", "config.toml", "path to config file")
+	flag.Parse()
+
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		slog.Error("failed to load config", "error", err)
+		os.Exit(1)
+	}
+
+	// Logger
+	var logLevel slog.Level
+	switch cfg.General.LogLevel {
+	case "debug":
+		logLevel = slog.LevelDebug
+	case "warn":
+		logLevel = slog.LevelWarn
+	case "error":
+		logLevel = slog.LevelError
+	default:
+		logLevel = slog.LevelInfo
+	}
+	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel}))
+	slog.SetDefault(logger)
+
+	// Timezone
+	loc, err := time.LoadLocation(cfg.General.Timezone)
+	if err != nil {
+		logger.Error("invalid timezone", "timezone", cfg.General.Timezone, "error", err)
+		os.Exit(1)
+	}
+
+	// Storage
+	store, err := storage.Open(cfg.Database.Path)
+	if err != nil {
+		logger.Error("failed to open storage", "error", err)
+		os.Exit(1)
+	}
+	defer store.Close()
+
+	// Memos client
+	client := memos.NewClient(cfg.Memos.URL, cfg.Memos.Token)
+
+	// Search selector
+	selector := search.NewSelector(client, store, cfg.Search, loc, logger)
+
+	// Memory service
+	memorySvc := memory.NewService(selector, store, loc, logger)
+
+	// Web handler
+	handler := web.NewHandler(memorySvc, logger)
+
+	// HTTP server
+	srv := &http.Server{
+		Addr:         cfg.Web.Listen,
+		Handler:      handler,
+		ReadTimeout:  10 * time.Second,
+		WriteTimeout: 30 * time.Second,
+		IdleTimeout:  60 * time.Second,
+	}
+
+	// Graceful shutdown
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	go func() {
+		logger.Info("starting server", "addr", cfg.Web.Listen)
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			logger.Error("server error", "error", err)
+			os.Exit(1)
+		}
+	}()
+
+	<-ctx.Done()
+	logger.Info("shutting down")
+
+	shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := srv.Shutdown(shutdownCtx); err != nil {
+		logger.Error("shutdown error", "error", err)
+	}
+
+	logger.Info("stopped")
+}