迁移

2026-08-13 21:51:20 +08:00
parent be5362913b
commit 21c264fcd7
60 changed files with 10174 additions and 69 deletions
@@ -0,0 +1,97 @@
+package server
+
+import (
+	"strconv"
+
+	"kra/internal/biz"
+	"kra/internal/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+func registerPolicyRoutes(group *gin.RouterGroup, svc *service.AccessService) {
+	casbin := group.Group("/casbin")
+	casbin.POST("/updateCasbin", func(c *gin.Context) {
+		var req struct {
+			AuthorityID uint `json:"authorityId"`
+			Infos       []struct {
+				Path   string `json:"path"`
+				Method string `json:"method"`
+			} `json:"casbinInfos"`
+		}
+		if c.ShouldBindJSON(&req) != nil {
+			fail(c, "参数错误")
+			return
+		}
+		paths := make([]*biz.API, 0, len(req.Infos))
+		for _, v := range req.Infos {
+			paths = append(paths, &biz.API{Path: v.Path, Method: v.Method})
+		}
+		if err := svc.SetPolicyPaths(c.Request.Context(), req.AuthorityID, paths); err != nil {
+			fail(c, "更新失败")
+			return
+		}
+		ok(c)
+	})
+	casbin.POST("/getPolicyPathByAuthorityId", func(c *gin.Context) {
+		var req struct {
+			AuthorityID uint `json:"authorityId"`
+		}
+		if c.ShouldBindJSON(&req) != nil {
+			fail(c, "参数错误")
+			return
+		}
+		paths, err := svc.PolicyPaths(c.Request.Context(), req.AuthorityID)
+		if err != nil {
+			fail(c, "获取失败")
+			return
+		}
+		writeResult(c, codeSuccess, gin.H{"paths": paths}, "获取成功")
+	})
+	buttons := group.Group("/authorityBtn")
+	buttons.POST("/getAuthorityBtn", func(c *gin.Context) {
+		var req struct {
+			MenuID      uint `json:"menuID"`
+			AuthorityID uint `json:"authorityId"`
+		}
+		if c.ShouldBindJSON(&req) != nil {
+			fail(c, "参数错误")
+			return
+		}
+		selected, err := svc.SelectedButtons(c.Request.Context(), req.AuthorityID, req.MenuID)
+		if err != nil {
+			fail(c, "查询失败")
+			return
+		}
+		writeResult(c, codeSuccess, gin.H{"selected": selected}, "查询成功")
+	})
+	buttons.POST("/setAuthorityBtn", func(c *gin.Context) {
+		var req struct {
+			MenuID      uint   `json:"menuID"`
+			AuthorityID uint   `json:"authorityId"`
+			Selected    []uint `json:"selected"`
+		}
+		if c.ShouldBindJSON(&req) != nil {
+			fail(c, "参数错误")
+			return
+		}
+		if err := svc.SetSelectedButtons(c.Request.Context(), req.AuthorityID, req.MenuID, req.Selected); err != nil {
+			fail(c, "分配失败")
+			return
+		}
+		ok(c)
+	})
+	buttons.POST("/canRemoveAuthorityBtn", func(c *gin.Context) {
+		id, _ := strconv.ParseUint(c.Query("id"), 10, 64)
+		allowed, err := svc.CanRemoveButton(c.Request.Context(), uint(id))
+		if err != nil {
+			fail(c, "检查失败")
+			return
+		}
+		if !allowed {
+			fail(c, "此按钮正在被使用无法删除")
+			return
+		}
+		ok(c)
+	})
+}