AllowedGroups string `json:"allowed_groups" gorm:"type:text;default:''"`

AllowedGroups string `json:"allowed_groups" gorm:"type:text;default:''"`

// isGroupAllowed checks if a group is allowed by the allowedGroups setting.
// Empty allowedGroups means all groups are allowed.
func isGroupAllowed(allowedGroups string, group string) bool {
    if strings.TrimSpace(allowedGroups) == "" {
        return true
    }
    groups := strings.Split(allowedGroups, ",")
    for _, g := range groups {
        if strings.TrimSpace(g) == group {
            return true
        }
    }
    return false
}

sub := &UserSubscription{
    // ... 现有字段 ...
    AllowedGroups: plan.AllowedGroups,  // 新增：拷贝套餐的分组限制
}

for _, candidate := range subs {
    sub := candidate
    // 新增：检查分组是否匹配
    if !isGroupAllowed(sub.AllowedGroups, usingGroup) {
        continue
    }
    // ... 现有逻辑 ...
}

// HasActiveUserSubscriptionForGroup checks if user has active subscription for specific group.
func HasActiveUserSubscriptionForGroup(userId int, usingGroup string) (bool, error) {
    if userId <= 0 {
        return false, errors.New("invalid userId")
    }
    now := common.GetTimestamp()
    var subs []UserSubscription
    if err := DB.Model(&UserSubscription{}).
        Select("id, allowed_groups").
        Where("user_id = ? AND status = ? AND end_time > ?", userId, "active", now).
        Find(&subs).Error; err != nil {
        return false, err
    }
    for _, sub := range subs {
        if isGroupAllowed(sub.AllowedGroups, usingGroup) {
            return true, nil
        }
    }
    return false, nil
}

type SubscriptionFunding struct {
    // ... 现有字段 ...
    usingGroup string  // 新增
}

res, err := model.PreConsumeUserSubscription(
    s.requestId,
    s.userId,
    s.modelName,
    0,
    s.amount,
    s.usingGroup,  // 新增参数
)

session := &BillingSession{
    relayInfo: relayInfo,
    funding: &SubscriptionFunding{
        requestId: relayInfo.RequestId,
        userId:    relayInfo.UserId,
        modelName: relayInfo.OriginModelName,
        amount:    subConsume,
        usingGroup: relayInfo.UsingGroup,  // 新增
    },
}

hasSub, subCheckErr := model.HasActiveUserSubscriptionForGroup(relayInfo.UserId, relayInfo.UsingGroup)

// 验证 AllowedGroups
req.Plan.AllowedGroups = strings.TrimSpace(req.Plan.AllowedGroups)
if req.Plan.AllowedGroups != "" {
    groups := strings.Split(req.Plan.AllowedGroups, ",")
    groupRatios := ratio_setting.GetGroupRatioCopy()
    for _, g := range groups {
        g = strings.TrimSpace(g)
        if g != "" {
            if _, ok := groupRatios[g]; !ok {
                common.ApiErrorMsg(c, fmt.Sprintf("分组 %s 不存在", g))
                return
            }
        }
    }
}

updateMap := map[string]interface{}{
    // ... 现有字段 ...
    "allowed_groups": req.Plan.AllowedGroups,  // 新增
}