83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/ent/group"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
func ensureSimpleModeDefaultGroups(ctx context.Context, client *dbent.Client) error {
|
|
if client == nil {
|
|
return fmt.Errorf("nil ent client")
|
|
}
|
|
|
|
requiredByPlatform := map[string]int{
|
|
service.PlatformAnthropic: 1,
|
|
service.PlatformOpenAI: 1,
|
|
service.PlatformGemini: 1,
|
|
service.PlatformAntigravity: 2,
|
|
}
|
|
|
|
for platform, minCount := range requiredByPlatform {
|
|
count, err := client.Group.Query().
|
|
Where(group.PlatformEQ(platform), group.DeletedAtIsNil()).
|
|
Count(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("count groups for platform %s: %w", platform, err)
|
|
}
|
|
|
|
if platform == service.PlatformAntigravity {
|
|
if count < minCount {
|
|
for i := count; i < minCount; i++ {
|
|
name := fmt.Sprintf("%s-default-%d", platform, i+1)
|
|
if err := createGroupIfNotExists(ctx, client, name, platform); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
continue
|
|
}
|
|
|
|
// Non-antigravity platforms: ensure <platform>-default exists.
|
|
name := platform + "-default"
|
|
if err := createGroupIfNotExists(ctx, client, name, platform); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func createGroupIfNotExists(ctx context.Context, client *dbent.Client, name, platform string) error {
|
|
exists, err := client.Group.Query().
|
|
Where(group.NameEQ(name), group.DeletedAtIsNil()).
|
|
Exist(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("check group exists %s: %w", name, err)
|
|
}
|
|
if exists {
|
|
return nil
|
|
}
|
|
|
|
_, err = client.Group.Create().
|
|
SetName(name).
|
|
SetDescription("Auto-created default group").
|
|
SetPlatform(platform).
|
|
SetStatus(service.StatusActive).
|
|
SetSubscriptionType(service.SubscriptionTypeStandard).
|
|
SetRateMultiplier(1.0).
|
|
SetIsExclusive(false).
|
|
Save(ctx)
|
|
if err != nil {
|
|
if dbent.IsConstraintError(err) {
|
|
// Concurrent server startups may race on creation; treat as success.
|
|
return nil
|
|
}
|
|
return fmt.Errorf("create default group %s: %w", name, err)
|
|
}
|
|
return nil
|
|
}
|