feat: add special user usable group setting

service/group.go

@@ -0,0 +1,54 @@
+package service
+
+import (
+	"strings"
+
+	"github.com/QuantumNous/new-api/setting"
+	"github.com/QuantumNous/new-api/setting/ratio_setting"
+)
+
+func GetUserUsableGroups(userGroup string) map[string]string {
+	groupsCopy := setting.GetUserUsableGroupsCopy()
+	if userGroup != "" {
+		specialSettings, b := ratio_setting.GetGroupRatioSetting().GroupSpecialUsableGroup.Get(userGroup)
+		if b {
+			// 处理特殊可用分组
+			for specialGroup, desc := range specialSettings {
+				if strings.HasPrefix(specialGroup, "-:") {
+					// 移除分组
+					groupToRemove := strings.TrimPrefix(specialGroup, "-:")
+					delete(groupsCopy, groupToRemove)
+				} else if strings.HasPrefix(specialGroup, "+:") {
+					// 添加分组
+					groupToAdd := strings.TrimPrefix(specialGroup, "+:")
+					groupsCopy[groupToAdd] = desc
+				} else {
+					// 直接添加分组
+					groupsCopy[specialGroup] = desc
+				}
+			}
+		}
+		// 如果userGroup不在UserUsableGroups中，返回UserUsableGroups + userGroup
+		if _, ok := groupsCopy[userGroup]; !ok {
+			groupsCopy[userGroup] = "用户分组"
+		}
+	}
+	return groupsCopy
+}
+
+func GroupInUserUsableGroups(userGroup, groupName string) bool {
+	_, ok := GetUserUsableGroups(userGroup)[groupName]
+	return ok
+}
+
+// GetUserAutoGroup 根据用户分组获取自动分组设置
+func GetUserAutoGroup(userGroup string) []string {
+	groups := GetUserUsableGroups(userGroup)
+	autoGroups := make([]string, 0)
+	for _, group := range setting.GetAutoGroups() {
+		if _, ok := groups[group]; ok {
+			autoGroups = append(autoGroups, group)
+		}
+	}
+	return autoGroups
+}