wip: oci image management

main.go

@@ -0,0 +1,94 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"gitea.loveuer.com/loveuer/cluster/api"
+	"gitea.loveuer.com/loveuer/cluster/internal/config"
+	"gitea.loveuer.com/loveuer/cluster/internal/database"
+	"gitea.loveuer.com/loveuer/cluster/internal/middleware"
+	"gitea.loveuer.com/loveuer/cluster/internal/registry/storage"
+
+	"github.com/gofiber/fiber/v3"
+)
+
+func main() {
+	// 解析命令行参数
+	var (
+		debug   = flag.Bool("debug", false, "Enable debug mode")
+		address = flag.String("address", ":8080", "API server listen address")
+		dataDir = flag.String("data-dir", "./storage", "Data directory for storing all data")
+	)
+	flag.Parse()
+
+	// 加载配置
+	cfg, err := config.LoadFromFlags(*debug, *address, *dataDir)
+	if err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
+
+	log.Printf("Starting server with config: debug=%v, address=%s, data-dir=%s",
+		cfg.Server.Debug, cfg.Server.Address, cfg.Storage.RootPath)
+
+	// 初始化数据库
+	if err := database.Init(cfg.Storage.RootPath); err != nil {
+		log.Fatalf("Failed to initialize database: %v", err)
+	}
+
+	// 初始化存储（使用 GORM 元数据 + 文件系统）
+	store, err := storage.NewGormStorage(cfg.Storage.RootPath)
+	if err != nil {
+		log.Fatalf("Failed to initialize storage: %v", err)
+	}
+
+	// 创建 Fiber 应用
+	config := fiber.Config{}
+	if !cfg.Server.Debug {
+		// Fiber v3 默认不显示启动消息
+	}
+	app := fiber.New(config)
+
+	// 全局中间件
+	app.Use(middleware.Logger())
+	app.Use(middleware.Recovery())
+	app.Use(middleware.CORS())
+
+	// 健康检查
+	app.Get("/health", func(c fiber.Ctx) error {
+		return c.JSON(fiber.Map{"status": "ok"})
+	})
+
+	// 设置路由
+	api.SetupRoutes(app, store)
+
+	// 启动服务器（在 goroutine 中）
+	go func() {
+		log.Printf("Server listening on %s", cfg.Server.Address)
+		if err := app.Listen(cfg.Server.Address); err != nil {
+			log.Fatalf("Failed to start server: %v", err)
+		}
+	}()
+
+	// 等待中断信号以优雅关闭服务器
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	log.Println("Shutting down server...")
+
+	// 优雅关闭，等待 5 秒
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := app.ShutdownWithContext(ctx); err != nil {
+		log.Fatalf("Server forced to shutdown: %v", err)
+	}
+
+	log.Println("Server exited")
+}