chore: initial commit

main.go

@@ -0,0 +1,70 @@
+package main
+
+import (
+	"ai-router/internal/config"
+	"ai-router/internal/server"
+	"context"
+	"embed"
+	"flag"
+	"io/fs"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+)
+
+//go:embed all:web/dist
+var frontend embed.FS
+
+func main() {
+	path := flag.String("config", "config.yaml", "YAML configuration path")
+	flag.Parse()
+	cfg, err := config.Load(*path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	assets, err := fs.Sub(frontend, "web/dist")
+	if err != nil {
+		log.Fatal(err)
+	}
+	app, err := server.New(cfg.Storage.Path, cfg.Admin.Password, assets)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer func() {
+		if err := app.Close(); err != nil {
+			log.Print("AI Router shutdown: failed to persist all request logs or close database")
+		}
+	}()
+	if cfg.Server.PublicOrigin != "" {
+		if err := app.SetPublicOrigin(cfg.Server.PublicOrigin); err != nil {
+			log.Fatal(err)
+		}
+	}
+	if err := app.SetTrustedProxies(cfg.Server.TrustedProxies); err != nil {
+		log.Fatal(err)
+	}
+	if err := app.SetLimits(cfg.Limits.RequestsPerMinute, cfg.Limits.PerKeyConcurrency, cfg.Limits.GlobalConcurrency); err != nil {
+		log.Fatal(err)
+	}
+	app.SetLogRetention(cfg.Storage.LogRetentionDays)
+	srv := &http.Server{Addr: cfg.Server.Listen, Handler: app, ReadTimeout: 30 * time.Second, ReadHeaderTimeout: 10 * time.Second, IdleTimeout: 90 * time.Second, MaxHeaderBytes: 1 << 20}
+	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
+	defer stop()
+	done := make(chan error, 1)
+	go func() { log.Printf("AI Router listening on %s", cfg.Server.Listen); done <- srv.ListenAndServe() }()
+	select {
+	case err := <-done:
+		if err != http.ErrServerClosed {
+			log.Printf("server: %v", err)
+		}
+	case <-ctx.Done():
+		shutdown, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+		defer cancel()
+		if err := srv.Shutdown(shutdown); err != nil {
+			_ = srv.Close()
+		}
+	}
+}