package router

import (
	"eslogad-be/config"
	"eslogad-be/internal/middleware"

	"github.com/gin-gonic/gin"
)

type Router struct {
	config           *config.Config
	authHandler      AuthHandler
	healthHandler    HealthHandler
	authMiddleware   AuthMiddleware
	userHandler      UserHandler
	fileHandler      FileHandler
	rbacHandler      RBACHandler
	masterHandler    MasterHandler
	letterHandler    LetterHandler
	dispRouteHandler DispositionRouteHandler
	voteEventHandler VoteEventHandler
}

func NewRouter(
	cfg *config.Config,
	authHandler AuthHandler,
	authMiddleware AuthMiddleware,
	healthHandler HealthHandler,
	userHandler UserHandler,
	fileHandler FileHandler,
	rbacHandler RBACHandler,
	masterHandler MasterHandler,
	letterHandler LetterHandler,
	dispRouteHandler DispositionRouteHandler,
	voteEventHandler VoteEventHandler,
) *Router {
	return &Router{
		config:           cfg,
		authHandler:      authHandler,
		authMiddleware:   authMiddleware,
		healthHandler:    healthHandler,
		userHandler:      userHandler,
		fileHandler:      fileHandler,
		rbacHandler:      rbacHandler,
		masterHandler:    masterHandler,
		letterHandler:    letterHandler,
		dispRouteHandler: dispRouteHandler,
		voteEventHandler: voteEventHandler,
	}
}

func (r *Router) Init() *gin.Engine {
	gin.SetMode(gin.ReleaseMode)
	engine := gin.New()
	engine.Use(
		middleware.CORS(),
		middleware.JsonAPI(),
		middleware.CorrelationID(),
		middleware.Recover(),
		middleware.HTTPStatLogger(),
		middleware.PopulateContext(),
	)

	r.addAppRoutes(engine)
	return engine
}

func (r *Router) addAppRoutes(rg *gin.Engine) {
	rg.GET("/health", r.healthHandler.HealthCheck)

	v1 := rg.Group("/api/v1")
	{
		auth := v1.Group("/auth")
		{
			auth.POST("/login", r.authHandler.Login)
			auth.POST("/refresh", r.authHandler.RefreshToken)
			auth.GET("/profile", r.authHandler.GetProfile)
		}

		users := v1.Group("/users")
		users.Use(r.authMiddleware.RequireAuth())
		{
			users.GET("", r.userHandler.ListUsers)
			users.POST("/bulk", r.userHandler.BulkCreateUsersAsync)
			users.POST("/bulk/async", r.userHandler.BulkCreateUsersAsync)
			users.GET("/bulk/job/:jobId", r.userHandler.GetBulkJobStatus)
			users.GET("/profile", r.userHandler.GetProfile)
			users.PUT("/profile", r.userHandler.UpdateProfile)
			users.PUT(":id/password", r.userHandler.ChangePassword)
			users.GET("/titles", r.userHandler.ListTitles)
			users.GET("/mention", r.userHandler.GetActiveUsersForMention)
			users.POST("/profile/avatar", r.fileHandler.UploadProfileAvatar)
		}

		files := v1.Group("/files")
		files.Use(r.authMiddleware.RequireAuth())
		{
			files.POST("/documents", r.fileHandler.UploadDocument)
		}

		rbac := v1.Group("/rbac")
		rbac.Use(r.authMiddleware.RequireAuth())
		{
			rbac.GET("/permissions", r.rbacHandler.ListPermissions)
			rbac.POST("/permissions", r.rbacHandler.CreatePermission)
			rbac.PUT("/permissions/:id", r.rbacHandler.UpdatePermission)
			rbac.DELETE("/permissions/:id", r.rbacHandler.DeletePermission)

			rbac.GET("/roles", r.rbacHandler.ListRoles)
			rbac.POST("/roles", r.rbacHandler.CreateRole)
			rbac.PUT("/roles/:id", r.rbacHandler.UpdateRole)
			rbac.DELETE("/roles/:id", r.rbacHandler.DeleteRole)
		}

		master := v1.Group("/master")
		master.Use(r.authMiddleware.RequireAuth())
		{
			master.GET("/labels", r.masterHandler.ListLabels)
			master.POST("/labels", r.masterHandler.CreateLabel)
			master.PUT("/labels/:id", r.masterHandler.UpdateLabel)
			master.DELETE("/labels/:id", r.masterHandler.DeleteLabel)

			master.GET("/priorities", r.masterHandler.ListPriorities)
			master.POST("/priorities", r.masterHandler.CreatePriority)
			master.PUT("/priorities/:id", r.masterHandler.UpdatePriority)
			master.DELETE("/priorities/:id", r.masterHandler.DeletePriority)

			master.GET("/institutions", r.masterHandler.ListInstitutions)
			master.POST("/institutions", r.masterHandler.CreateInstitution)
			master.PUT("/institutions/:id", r.masterHandler.UpdateInstitution)
			master.DELETE("/institutions/:id", r.masterHandler.DeleteInstitution)

			master.GET("/disposition-actions", r.masterHandler.ListDispositionActions)
			master.POST("/disposition-actions", r.masterHandler.CreateDispositionAction)
			master.PUT("/disposition-actions/:id", r.masterHandler.UpdateDispositionAction)
			master.DELETE("/disposition-actions/:id", r.masterHandler.DeleteDispositionAction)
		}

		lettersch := v1.Group("/letters")
		lettersch.Use(r.authMiddleware.RequireAuth())
		{
			lettersch.POST("/incoming", r.letterHandler.CreateIncomingLetter)
			lettersch.GET("/incoming/:id", r.letterHandler.GetIncomingLetter)
			lettersch.GET("/incoming", r.letterHandler.ListIncomingLetters)
			lettersch.PUT("/incoming/:id", r.letterHandler.UpdateIncomingLetter)
			lettersch.DELETE("/incoming/:id", r.letterHandler.DeleteIncomingLetter)

			lettersch.POST("/dispositions/:letter_id", r.letterHandler.CreateDispositions)
			lettersch.GET("/dispositions/:letter_id", r.letterHandler.GetEnhancedDispositionsByLetter)

			lettersch.POST("/discussions/:letter_id", r.letterHandler.CreateDiscussion)
			lettersch.PUT("/discussions/:letter_id/:discussion_id", r.letterHandler.UpdateDiscussion)
		}

		droutes := v1.Group("/disposition-routes")
		droutes.Use(r.authMiddleware.RequireAuth())
		{
			droutes.POST("", r.dispRouteHandler.Create)
			droutes.GET(":id", r.dispRouteHandler.Get)
			droutes.PUT(":id", r.dispRouteHandler.Update)
			droutes.GET("department", r.dispRouteHandler.ListByFromDept)
			droutes.PUT(":id/active", r.dispRouteHandler.SetActive)
		}

		voteEvents := v1.Group("/vote-events")
		voteEvents.Use(r.authMiddleware.RequireAuth())
		{
			voteEvents.POST("", r.voteEventHandler.CreateVoteEvent)
			voteEvents.GET("", r.voteEventHandler.ListVoteEvents)
			voteEvents.GET("/active", r.voteEventHandler.GetActiveEvents)
			voteEvents.GET("/:id", r.voteEventHandler.GetVoteEvent)
			voteEvents.PUT("/:id", r.voteEventHandler.UpdateVoteEvent)
			voteEvents.DELETE("/:id", r.voteEventHandler.DeleteVoteEvent)
			voteEvents.GET("/:id/candidates", r.voteEventHandler.GetCandidates)
			voteEvents.GET("/:id/results", r.voteEventHandler.GetVoteResults)
			voteEvents.GET("/:id/vote-status", r.voteEventHandler.CheckVoteStatus)
		}

		candidates := v1.Group("/candidates")
		candidates.Use(r.authMiddleware.RequireAuth())
		{
			candidates.POST("", r.voteEventHandler.CreateCandidate)
		}

		votes := v1.Group("/votes")
		votes.Use(r.authMiddleware.RequireAuth())
		{
			votes.POST("", r.voteEventHandler.SubmitVote)
		}
	}
}