feat:manager check

Author: whxxxxxxxxxx

internal/app/fastgpt/handler/v1/app.go

@@ -7,6 +7,7 @@ import (
 	"HelpStudent/internal/app/fastgpt/dao"
 	"HelpStudent/internal/app/fastgpt/dto"
 	"HelpStudent/internal/app/fastgpt/model"
+	dao2 "HelpStudent/internal/app/managers/dao"
 	"errors"
 
 	"github.com/flamego/binding"
@@ -21,6 +22,12 @@ func HandleCreateApp(c flamego.Context, r flamego.Render, req dto.CreateAppReque
 		return
 	}
 
+	// 检查是否是管理员
+	if !dao2.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
+
 	// 检查 AppID 是否已存在
 	exists, err := dao.FastgptApp.CheckAppIDExists(req.AppID, 0)
 	if err != nil {
@@ -61,6 +68,12 @@ func HandleGetAppList(c flamego.Context, r flamego.Render, req dto.GetAppListReq
 		req.Limit = 100
 	}
 
+	// 检查是否是管理员
+	if !dao2.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
+
 	apps, total, err := dao.FastgptApp.GetAllApps(req.Offset, req.Limit)
 	if err != nil {
 		logx.SystemLogger.CtxError(c.Request().Context(), err)
@@ -97,6 +110,12 @@ func HandleUpdateApp(c flamego.Context, r flamego.Render, req dto.UpdateAppReque
 		return
 	}
 
+	// 检查是否是管理员
+	if !dao2.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
+
 	// 检查应用是否存在
 	_, err := dao.FastgptApp.GetAppByPrimaryID(req.ID)
 	if err != nil {
@@ -146,6 +165,12 @@ func HandleDeleteApp(c flamego.Context, r flamego.Render, req dto.DeleteAppReque
 		return
 	}
 
+	// 检查是否是管理员
+	if !dao2.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
+
 	// 检查应用是否存在
 	_, err := dao.FastgptApp.GetAppByPrimaryID(req.ID)
 	if err != nil {

internal/app/fastgpt/model/fastgpt.go

@@ -14,7 +14,3 @@ type FastgptApp struct {
 	Status      int    `gorm:"default:1;comment:状态(1:启用,0:禁用)"`
 	CreatedBy   string `gorm:"type:varchar(50);comment:创建者"`
 }
-
-func (FastgptApp) TableName() string {
-	return "fastgpt_apps"
-}

internal/app/fastgpt/router/fastgpt.go

@@ -62,5 +62,3 @@ func AppFastgptInit(e *flamego.Flame) {
 		})
 	}, web.Authorization)
 }
-
-func FastgptGroup(e *flamego.Flame) {}

internal/app/managers/handler/v1/managers.go

@@ -23,7 +23,12 @@ import (
 )
 
 // HandleImportStudentSubjectsExcel 处理Excel导入学生科目
-func HandleImportStudentSubjectsExcel(c flamego.Context, r flamego.Render) {
+func HandleImportStudentSubjectsExcel(c flamego.Context, r flamego.Render, authInfo auth.Info) {
+	// 检查是否是管理员
+	if !dao.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
 	// 从 FormFile 获取文件
 	file, header, err := c.Request().FormFile("file")
 	if err != nil {
@@ -186,6 +191,11 @@ func HandleAddManager(r flamego.Render, c flamego.Context, req dto.AddManagerReq
 		response.InValidParam(r, errs)
 		return
 	}
+	// 检查是否是管理员
+	if !dao.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
 	if authInfo.Uid == req.StaffId {
 		response.HTTPFail(r, 403001, "不能添加自己")
 		return
@@ -226,6 +236,11 @@ func HandleDeleteManager(r flamego.Render, c flamego.Context, req dto.DeleteMana
 		response.InValidParam(r, errs)
 		return
 	}
+	// 检查是否是管理员
+	if !dao.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
 
 	// 防止删除自己
 	if req.StaffId == authInfo.Uid {
@@ -266,6 +281,11 @@ func HandleGetManagerList(r flamego.Render, c flamego.Context, authInfo auth.Inf
 		response.HTTPFail(r, 403002, "permission denied")
 		return
 	}
+	// 检查是否是管理员
+	if !dao.Managers.IsManager(authInfo.Uid) {
+		response.HTTPFail(r, 400013, "非管理员用户无法创建应用")
+		return
+	}
 	managers, total, err := dao.Managers.GetAllManagers()
 	if err != nil {
 		logx.SystemLogger.CtxError(c.Request().Context(), err)

internal/app/managers/router/managers.go

@@ -36,5 +36,3 @@ func AppManagersInit(e *flamego.Flame) {
 	}, web.Authorization)
 
 }
-
-func ManagersGroup(e *flamego.Flame) {}