🎉 初始化项目

server/plugin/plugin-tool/utils/check.go

@@ -0,0 +1,137 @@
+package utils
+
+import (
+	"github.com/pkg/errors"
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+	"path/filepath"
+	"runtime"
+	"strings"
+	"sync"
+
+	"git.echol.cn/loser/st/server/global"
+	"git.echol.cn/loser/st/server/model/system"
+)
+
+var (
+	ApiMap  = make(map[string][]system.SysApi)
+	MenuMap = make(map[string][]system.SysBaseMenu)
+	DictMap = make(map[string][]system.SysDictionary)
+	rw      sync.Mutex
+)
+
+func getPluginName() string {
+	_, file, _, ok := runtime.Caller(2)
+	pluginName := ""
+	if ok {
+		file = filepath.ToSlash(file)
+		const key = "server/plugin/"
+		if idx := strings.Index(file, key); idx != -1 {
+			remain := file[idx+len(key):]
+			parts := strings.Split(remain, "/")
+			if len(parts) > 0 {
+				pluginName = parts[0]
+			}
+		}
+	}
+	return pluginName
+}
+
+func RegisterApis(apis ...system.SysApi) {
+	name := getPluginName()
+	if name != "" {
+		rw.Lock()
+		ApiMap[name] = apis
+		rw.Unlock()
+	}
+
+	err := global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		for _, api := range apis {
+			err := tx.Model(system.SysApi{}).Where("path = ? AND method = ? AND api_group = ? ", api.Path, api.Method, api.ApiGroup).FirstOrCreate(&api).Error
+			if err != nil {
+				zap.L().Error("注册API失败", zap.Error(err), zap.String("api", api.Path), zap.String("method", api.Method), zap.String("apiGroup", api.ApiGroup))
+				return err
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		zap.L().Error("注册API失败", zap.Error(err))
+	}
+}
+
+func RegisterMenus(menus ...system.SysBaseMenu) {
+	name := getPluginName()
+	if name != "" {
+		rw.Lock()
+		MenuMap[name] = menus
+		rw.Unlock()
+	}
+
+	parentMenu := menus[0]
+	otherMenus := menus[1:]
+	err := global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		err := tx.Model(system.SysBaseMenu{}).Where("name = ? ", parentMenu.Name).FirstOrCreate(&parentMenu).Error
+		if err != nil {
+			zap.L().Error("注册菜单失败", zap.Error(err))
+			return errors.Wrap(err, "注册菜单失败")
+		}
+		pid := parentMenu.ID
+		for i := range otherMenus {
+			otherMenus[i].ParentId = pid
+			err = tx.Model(system.SysBaseMenu{}).Where("name = ? ", otherMenus[i].Name).FirstOrCreate(&otherMenus[i]).Error
+			if err != nil {
+				zap.L().Error("注册菜单失败", zap.Error(err))
+				return errors.Wrap(err, "注册菜单失败")
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		zap.L().Error("注册菜单失败", zap.Error(err))
+	}
+
+}
+
+func RegisterDictionaries(dictionaries ...system.SysDictionary) {
+	name := getPluginName()
+	if name != "" {
+		rw.Lock()
+		DictMap[name] = dictionaries
+		rw.Unlock()
+	}
+
+	err := global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		for _, dict := range dictionaries {
+			details := dict.SysDictionaryDetails
+			dict.SysDictionaryDetails = nil
+			err := tx.Model(system.SysDictionary{}).Where("type = ?", dict.Type).FirstOrCreate(&dict).Error
+			if err != nil {
+				zap.L().Error("注册字典失败", zap.Error(err), zap.String("type", dict.Type))
+				return err
+			}
+			for _, detail := range details {
+				detail.SysDictionaryID = int(dict.ID)
+				err = tx.Model(system.SysDictionaryDetail{}).Where("sys_dictionary_id = ? AND value = ?", dict.ID, detail.Value).FirstOrCreate(&detail).Error
+				if err != nil {
+					zap.L().Error("注册字典详情失败", zap.Error(err), zap.String("value", detail.Value))
+					return err
+				}
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		zap.L().Error("注册字典失败", zap.Error(err))
+	}
+}
+
+func Pointer[T any](in T) *T {
+	return &in
+}
+
+func GetPluginData(pluginName string) ([]system.SysApi, []system.SysBaseMenu, []system.SysDictionary) {
+	rw.Lock()
+	defer rw.Unlock()
+	return ApiMap[pluginName], MenuMap[pluginName], DictMap[pluginName]
+}