package router

import (
	"bookhoard/internal/handlers"

	"github.com/golang-jwt/jwt/v5"
	"github.com/labstack/echo-jwt/v4"
	"github.com/labstack/echo/v4"
)

func registerSyncRoutes(cfg *Config) {
	e := cfg.Echo

	// JWT middleware for protected routes
	jwtMiddleware := echojwt.WithConfig(echojwt.Config{
		SigningKey: []byte(cfg.Cfg.JWTSecret),
		ContextKey: "user",
		SuccessHandler: func(c echo.Context) {
			token := c.Get("user").(*jwt.Token)
			claims := token.Claims.(jwt.MapClaims)
			c.Set("user_id", claims["user_id"])
			c.Set("user_role", claims["user_role"])
			c.Set("user_email", claims["user_email"])
			c.Set("user_username", claims["user_username"])
		},
	})

	protected := e.Group("/api", jwtMiddleware)

	// Setup ebook handler routes first
	h := handlers.SetupRoutes(protected, cfg.Queries, cfg.ConnManager)

	// Book matching and unlinked book resolution routes
	sync := protected.Group("/sync")
	sync.POST("/bulk-link-books", h.BulkLinkBooks)
	sync.POST("/auto-link-books", h.AutoLinkBooks)
	sync.GET("/unlinked-books/:id/suggestions", h.GetUnlinkedBookSuggestions)

	// KOReader sync routes (device authentication required)
	koreaderSync := e.Group("/api/sync/koreader")
	koreaderSync.POST("/progress", cfg.DeviceAuthMiddleware.Authenticate(cfg.KOReaderHandler.SyncProgress))
	koreaderSync.GET("/metadata/:uuid", cfg.DeviceAuthMiddleware.Authenticate(cfg.KOReaderHandler.GetMetadata))
	koreaderSync.GET("/library", cfg.DeviceAuthMiddleware.Authenticate(cfg.KOReaderHandler.GetLibrary))
	koreaderSync.POST("/bookmarks", cfg.DeviceAuthMiddleware.Authenticate(cfg.KOReaderHandler.SyncBookmarks))

	// Kobo sync routes (device authentication required)
	koboHandler := handlers.NewKoboHandler(cfg.Queries, cfg.ConnManager)
	koboSync := e.Group("/api/sync/kobo")
	koboSync.POST("/markup", cfg.DeviceAuthMiddleware.Authenticate(koboHandler.Markup))
	koboSync.POST("/bookmark", cfg.DeviceAuthMiddleware.Authenticate(koboHandler.Bookmark))
	koboSync.POST("/v1/analytics/gettests", cfg.DeviceAuthMiddleware.Authenticate(koboHandler.AnalyticsGettests))
	koboSync.GET("/v1/initialization", cfg.DeviceAuthMiddleware.Authenticate(koboHandler.Initialization))
	koboSync.POST("/sync-from-server", cfg.DeviceAuthMiddleware.Authenticate(koboHandler.SyncFromServer))
}

func registerWebSocketRoutes(cfg *Config) {
	e := cfg.Echo

	// WebSocket endpoint for real-time sync
	e.GET("/ws/sync", cfg.WSHandler.HandleWebSocket)
}