package main

import (
	"log"
	"os"
	"path/filepath"
	"xboard-go/internal/config"
	"xboard-go/internal/database"
	"xboard-go/internal/handler"
	"xboard-go/internal/middleware"
	"xboard-go/internal/service"

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

func main() {
	config.LoadConfig()
	database.InitDB()
	database.InitCache()

	router := gin.New()
	router.Use(gin.Logger(), gin.Recovery())

	api := router.Group("/api")
	registerV1(api.Group("/v1"))
	registerV2(api.Group("/v2"))
	registerWebRoutes(router)

	log.Printf("server starting on port %s", config.AppConfig.AppPort)
	if err := router.Run(":" + config.AppConfig.AppPort); err != nil {
		log.Fatalf("failed to start server: %v", err)
	}
}

func registerV1(v1 *gin.RouterGroup) {
	registerPassportRoutes(v1)
	registerGuestRoutes(v1)
	registerUserRoutes(v1)
	registerClientRoutes(v1)
	registerServerRoutesV1(v1)
}

func registerV2(v2 *gin.RouterGroup) {
	registerPassportRoutes(v2)
	registerUserRoutesV2(v2)
	registerClientRoutesV2(v2)
	registerServerRoutesV2(v2)
	registerAdminRoutesV2(v2)
}

func registerPassportRoutes(group *gin.RouterGroup) {
	passport := group.Group("/passport")
	passport.POST("/auth/register", handler.Register)
	passport.POST("/auth/login", handler.Login)
	passport.GET("/auth/token2Login", handler.Token2Login)
	passport.POST("/auth/forget", handler.ForgetPassword)
	passport.POST("/auth/getQuickLoginUrl", handler.PassportGetQuickLoginURL)
	passport.POST("/auth/loginWithMailLink", handler.PassportLoginWithMailLink)
	passport.POST("/comm/sendEmailVerify", handler.SendEmailVerify)
	passport.POST("/comm/pv", handler.PassportPV)
}

func registerGuestRoutes(v1 *gin.RouterGroup) {
	guest := v1.Group("/guest")
	guest.GET("/plan/fetch", handler.GuestPlanFetch)
	guest.POST("/telegram/webhook", handler.NotImplemented("guest.telegram.webhook"))
	guest.Any("/payment/notify/:method/:uuid", handler.NotImplemented("guest.payment.notify"))
	guest.GET("/comm/config", handler.GuestConfig)
}

func registerUserRoutes(v1 *gin.RouterGroup) {
	user := v1.Group("/user")
	user.Use(middleware.Auth())
	user.GET("/resetSecurity", handler.UserResetSecurity)
	user.GET("/info", handler.UserInfo)
	user.POST("/changePassword", handler.UserChangePassword)
	user.POST("/update", handler.UserUpdate)
	user.GET("/getSubscribe", handler.UserGetSubscribe)
	user.GET("/getStat", handler.UserGetStat)
	user.GET("/checkLogin", handler.UserCheckLogin)
	user.GET("/plan/fetch", handler.GuestPlanFetch)
	user.GET("/server/fetch", handler.UserServerFetch)
	user.GET("/comm/config", handler.UserCommConfig)

	user.POST("/transfer", handler.UserTransfer)
	user.POST("/getQuickLoginUrl", handler.UserGetQuickLoginURL)
	user.GET("/notice/fetch", handler.UserNoticeFetch)
	user.POST("/coupon/check", handler.UserCouponCheck)
	user.POST("/gift-card/check", handler.UserGiftCardCheck)
	user.POST("/gift-card/redeem", handler.UserGiftCardRedeem)
	user.GET("/gift-card/history", handler.UserGiftCardHistory)
	user.GET("/gift-card/detail", handler.UserGiftCardDetail)
	user.GET("/gift-card/types", handler.UserGiftCardTypes)
	user.GET("/telegram/getBotInfo", handler.UserTelegramBotInfo)
	user.POST("/comm/getStripePublicKey", handler.UserGetStripePublicKey)
	user.GET("/stat/getTrafficLog", handler.UserTrafficLog)
	user.POST("/order/save", handler.UserOrderSave)
	user.POST("/order/checkout", handler.UserOrderCheckout)
	user.GET("/order/check", handler.UserOrderCheck)
	user.GET("/order/detail", handler.UserOrderDetail)
	user.GET("/order/fetch", handler.UserOrderFetch)
	user.GET("/order/getPaymentMethod", handler.UserOrderGetPaymentMethod)
	user.POST("/order/cancel", handler.UserOrderCancel)
	user.GET("/invite/save", handler.UserInviteSave)
	user.GET("/invite/fetch", handler.UserInviteFetch)
	user.GET("/invite/details", handler.UserInviteDetails)
	user.GET("/getActiveSession", handler.UserGetActiveSession)
	user.POST("/removeActiveSession", handler.UserRemoveActiveSession)
	user.GET("/knowledge/fetch", handler.UserKnowledgeFetch)
	user.GET("/knowledge/getCategory", handler.UserKnowledgeCategories)
	user.POST("/ticket/reply", handler.UserTicketReply)
	user.POST("/ticket/close", handler.UserTicketClose)
	user.POST("/ticket/save", handler.UserTicketSave)
	user.GET("/ticket/fetch", handler.UserTicketFetch)
	user.POST("/ticket/withdraw", handler.UserTicketWithdraw)

	// Integrated User Features
	user.GET("/real-name-verification/status", handler.PluginRealNameStatus)
	user.POST("/real-name-verification/submit", handler.PluginRealNameSubmit)
	user.GET("/user-online-devices/get-ip", handler.PluginUserOnlineDevicesGetIP)
	user.POST("/user-add-ipv6-subscription/enable", handler.PluginUserAddIPv6Enable)
	user.POST("/user-add-ipv6-subscription/sync-password", handler.PluginUserAddIPv6SyncPassword)
	user.GET("/user-add-ipv6-subscription/check", handler.PluginUserAddIPv6Check)
}

func registerUserRoutesV2(v2 *gin.RouterGroup) {
	user := v2.Group("/user")
	user.Use(middleware.Auth())
	user.GET("/resetSecurity", handler.UserResetSecurity)
	user.GET("/info", handler.UserInfo)
}

func registerClientRoutes(v1 *gin.RouterGroup) {
	client := v1.Group("/client")
	client.Use(middleware.ClientAuth())
	client.GET("/subscribe", handler.ClientSubscribe)
	client.GET("/app/getConfig", handler.ClientAppConfigV1)
	client.GET("/app/getVersion", handler.ClientAppVersion)
}

func registerClientRoutesV2(v2 *gin.RouterGroup) {
	client := v2.Group("/client")
	client.Use(middleware.ClientAuth())
	client.GET("/app/getConfig", handler.ClientAppConfigV2)
	client.GET("/app/getVersion", handler.ClientAppVersion)
}

func registerServerRoutesV1(v1 *gin.RouterGroup) {
	server := v1.Group("/server")

	uniProxy := server.Group("/UniProxy")
	uniProxy.Use(middleware.NodeAuth())
	uniProxy.GET("/config", handler.NodeConfig)
	uniProxy.GET("/user", handler.NodeUser)
	uniProxy.POST("/push", handler.NodePush)
	uniProxy.POST("/alive", handler.NodeAlive)
	uniProxy.GET("/alivelist", handler.NodeAliveList)
	uniProxy.POST("/status", handler.NodeStatus)

	shadowsocks := server.Group("/ShadowsocksTidalab")
	shadowsocks.Use(middleware.NodeAuth())
	shadowsocks.GET("/user", handler.NodeShadowsocksTidalabUser)
	shadowsocks.POST("/submit", handler.NodeTidalabSubmit)

	trojan := server.Group("/TrojanTidalab")
	trojan.Use(middleware.NodeAuth())
	trojan.GET("/config", handler.NodeTrojanTidalabConfig)
	trojan.GET("/user", handler.NodeTrojanTidalabUser)
	trojan.POST("/submit", handler.NodeTidalabSubmit)
}

func registerServerRoutesV2(v2 *gin.RouterGroup) {
	server := v2.Group("/server")
	server.Use(middleware.NodeAuth())
	server.POST("/handshake", handler.NodeHandshake)
	server.POST("/report", handler.NodeReport)
	server.GET("/config", handler.NodeConfig)
	server.GET("/user", handler.NodeUser)
	server.POST("/push", handler.NodePush)
	server.POST("/alive", handler.NodeAlive)
	server.GET("/alivelist", handler.NodeAliveList)
	server.POST("/status", handler.NodeStatus)
}

func registerAdminRoutesV2(v2 *gin.RouterGroup) {
	admin := v2.Group("/" + service.GetAdminSecurePath())
	admin.Use(middleware.Auth(), middleware.AdminAuth())

	admin.GET("/config/fetch", handler.AdminConfigFetch)
	admin.POST("/config/save", handler.AdminConfigSave)
	admin.GET("/dashboard/summary", handler.AdminDashboardSummary)

	admin.GET("/stat/getStats", handler.AdminDashboardSummary)
	admin.GET("/stat/getOverride", handler.AdminDashboardSummary)
	admin.GET("/stat/getTrafficRank", handler.AdminGetTrafficRank)
	admin.GET("/stat/getOrder", handler.AdminGetOrderStats)

	admin.GET("/system/getSystemStatus", handler.AdminSystemStatus)
	admin.GET("/system/getQueueStats", handler.AdminSystemQueueStats)
	admin.GET("/system/getQueueWorkload", handler.AdminSystemQueueStats)
	admin.GET("/system/getQueueMasters", handler.AdminSystemQueueStats)
	admin.GET("/system/getHorizonFailedJobs", handler.AdminSystemQueueStats)

	admin.GET("/server/group/fetch", handler.AdminServerGroupsFetch)
	admin.POST("/server/group/save", handler.AdminServerGroupSave)
	admin.POST("/server/group/drop", handler.AdminServerGroupDrop)
	admin.GET("/server/route/fetch", handler.AdminServerRoutesFetch)
	admin.POST("/server/route/save", handler.AdminServerRouteSave)
	admin.POST("/server/route/drop", handler.AdminServerRouteDrop)
	admin.GET("/server/manage/getNodes", handler.AdminServerManageGetNodes)
	admin.POST("/server/manage/sort", handler.AdminServerManageSort)
	admin.POST("/server/manage/update", handler.AdminServerManageUpdate)
	admin.POST("/server/manage/save", handler.AdminServerManageSave)
	admin.POST("/server/manage/drop", handler.AdminServerManageDrop)
	admin.POST("/server/manage/copy", handler.AdminServerManageCopy)
	admin.POST("/server/manage/batchDelete", handler.AdminServerManageBatchDelete)
	admin.POST("/server/manage/resetTraffic", handler.AdminServerManageResetTraffic)
	admin.POST("/server/manage/batchResetTraffic", handler.AdminServerManageBatchResetTraffic)
	admin.GET("/plan/fetch", handler.AdminPlansFetch)
	admin.POST("/plan/save", handler.AdminPlanSave)
	admin.POST("/plan/drop", handler.AdminPlanDrop)
	admin.POST("/plan/sort", handler.AdminPlanSort)

	admin.GET("/order/fetch", handler.AdminOrdersFetch)
	admin.POST("/order/paid", handler.AdminOrderPaid)
	admin.POST("/order/cancel", handler.AdminOrderCancel)

	admin.GET("/coupon/fetch", handler.AdminCouponsFetch)
	admin.POST("/coupon/save", handler.AdminCouponSave)
	admin.POST("/coupon/drop", handler.AdminCouponDrop)

	admin.GET("/user/fetch", handler.AdminUsersFetch)
	admin.POST("/user/update", handler.AdminUserUpdate)
	admin.POST("/user/ban", handler.AdminUserBan)
	admin.POST("/user/resetTraffic", handler.AdminUserResetTraffic)
	admin.POST("/user/drop", handler.AdminUserDelete)
	admin.GET("/ticket/fetch", handler.AdminTicketsFetch)
	admin.GET("/gift-card/status", handler.AdminGiftCardStatus)
	admin.GET("/traffic-reset/fetch", handler.AdminTrafficResetFetch)
	admin.GET("/traffic-reset/logs", handler.AdminTrafficResetFetch)
	admin.POST("/traffic-reset/reset-user", handler.AdminTrafficResetUser)
	admin.GET("/traffic-reset/user/:id/history", handler.AdminTrafficResetUserHistory)

	// Integrated Admin Features
	admin.GET("/realname/records", handler.PluginRealNameRecords)
	admin.POST("/realname/clear-cache", handler.PluginRealNameClearCache)
	admin.POST("/realname/review/:userId", handler.PluginRealNameReview)
	admin.POST("/realname/reset/:userId", handler.PluginRealNameReset)
	admin.POST("/realname/sync-all", handler.PluginRealNameSyncAll)
	admin.POST("/realname/approve-all", handler.PluginRealNameApproveAll)
	admin.GET("/user-online-devices/users", handler.PluginUserOnlineDevicesUsers)
}

func registerWebRoutes(router *gin.Engine) {
	themeRoot := filepath.Join(".", "frontend", "theme")
	adminRoot := filepath.Join(".", "frontend", "admin")

	if _, err := os.Stat(themeRoot); err == nil {
		router.Static("/theme", themeRoot)
	}
	if _, err := os.Stat(adminRoot); err == nil {
		router.Static("/admin-assets", adminRoot)
	}

	securePath := "/" + service.GetAdminSecurePath()
	router.GET("/", handler.UserThemePage)
	router.GET("/dashboard", handler.UserThemePage)
	router.GET(securePath, handler.AdminAppPage)
	router.GET(securePath+"/", handler.AdminAppPage)
	router.GET(securePath+"/plugins/:plugin", handler.AdminAppPage)
}