package handler

import (
	"strconv"

	"kra/internal/server/httpx"
	"kra/internal/service"
	"kra/internal/service/dto"

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

type Menu struct{ service *service.MenuService }

func NewMenu(service *service.MenuService) *Menu { return &Menu{service: service} }

func (h *Menu) List(c *gin.Context) {
	items, err := h.service.Tree(c.Request.Context())
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, items, "获取成功")
}

func (h *Menu) Tree(c *gin.Context) {
	items, err := h.service.Tree(c.Request.Context())
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{"menus": items}, "获取成功")
}

func (h *Menu) Create(c *gin.Context) {
	var req dto.MenuRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	if err := h.service.Create(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "添加失败："+err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "添加成功")
}

func (h *Menu) Update(c *gin.Context) {
	var req dto.MenuRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	if err := h.service.Update(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "更新失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "更新成功")
}

func (h *Menu) Delete(c *gin.Context) {
	var req dto.DeleteMenuRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	if err := h.service.Delete(c.Request.Context(), req.ID); err != nil {
		httpx.Fail(c, "删除失败:"+err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "删除成功")
}

func (h *Menu) Find(c *gin.Context) {
	var req dto.GetMenuRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	item, err := h.service.Find(c.Request.Context(), req.ID)
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{"menu": item}, "获取成功")
}

func (h *Menu) SetAuthorityMenus(c *gin.Context) {
	var req dto.SetAuthorityMenusRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	if err := h.service.SetAuthorityMenus(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "添加失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "添加成功")
}

func (h *Menu) AuthorityMenus(c *gin.Context) {
	var req dto.GetAuthorityMenusRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	menus, err := h.service.AuthorityMenus(c.Request.Context(), req.AuthorityID)
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{"menus": menus}, "获取成功")
}

func (h *Menu) RoleIDs(c *gin.Context) {
	id, _ := strconv.ParseUint(c.Query("menuId"), 10, 64)
	ids, err := h.service.RoleIDs(c.Request.Context(), uint(id))
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	if ids == nil {
		ids = []uint{}
	}
	defaultIDs, err := h.service.DefaultRouterRoleIDs(c.Request.Context(), uint(id))
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	if defaultIDs == nil {
		defaultIDs = []uint{}
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{"authorityIds": ids, "defaultRouterAuthorityIds": defaultIDs}, "获取成功")
}

func (h *Menu) SetRoles(c *gin.Context) {
	var req dto.SetMenuRolesRequest
	if c.ShouldBindJSON(&req) != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	if err := h.service.SetRoles(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "设置失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "设置成功")
}