internal/config/config.go

package config

import (
	"crypto/sha1"
	"encoding/hex"
	"fmt"
	"strings"
)

type Config struct {
	Keys              []string                `json:"keys,omitempty"`
	APIKeys           []APIKey                `json:"api_keys,omitempty"`
	Accounts          []Account               `json:"accounts,omitempty"`
	Proxies           []Proxy                 `json:"proxies,omitempty"`
	ModelAliases      map[string]string       `json:"model_aliases,omitempty"`
	Admin             AdminConfig             `json:"admin,omitempty"`
	Runtime           RuntimeConfig           `json:"runtime,omitempty"`
	Responses         ResponsesConfig         `json:"responses,omitempty"`
	Embeddings        EmbeddingsConfig        `json:"embeddings,omitempty"`
	AutoDelete        AutoDeleteConfig        `json:"auto_delete"`
	CurrentInputFile  CurrentInputFileConfig  `json:"current_input_file,omitempty"`
	ThinkingInjection ThinkingInjectionConfig `json:"thinking_injection,omitempty"`
	Vercel            VercelConfig            `json:"vercel,omitempty"`
	Platform          PlatformConfig          `json:"platform,omitempty"`
	VercelSyncHash    string                  `json:"_vercel_sync_hash,omitempty"`
	VercelSyncTime    int64                   `json:"_vercel_sync_time,omitempty"`
	AdditionalFields  map[string]any          `json:"-"`
}

type Account struct {
	Name     string `json:"name,omitempty"`
	Remark   string `json:"remark,omitempty"`
	Email    string `json:"email,omitempty"`
	Mobile   string `json:"mobile,omitempty"`
	Password string `json:"password,omitempty"`
	Token    string `json:"token,omitempty"`
	ProxyID  string `json:"proxy_id,omitempty"`
	Role     string `json:"role,omitempty"`   // "normal" (default) or "standby"
	Banned   bool   `json:"banned,omitempty"` // true when USER_IS_BANNED detected
}

type APIKey struct {
	Key    string `json:"key"`
	Name   string `json:"name,omitempty"`
	Remark string `json:"remark,omitempty"`
}

type Proxy struct {
	ID       string `json:"id,omitempty"`
	Name     string `json:"name,omitempty"`
	Type     string `json:"type,omitempty"`
	Host     string `json:"host,omitempty"`
	Port     int    `json:"port,omitempty"`
	Username string `json:"username,omitempty"`
	Password string `json:"password,omitempty"`
	Disabled bool   `json:"disabled,omitempty"`
}

func NormalizeProxy(p Proxy) Proxy {
	p.ID = strings.TrimSpace(p.ID)
	p.Name = strings.TrimSpace(p.Name)
	p.Type = strings.ToLower(strings.TrimSpace(p.Type))
	p.Host = strings.TrimSpace(p.Host)
	p.Username = strings.TrimSpace(p.Username)
	p.Password = strings.TrimSpace(p.Password)
	if p.ID == "" {
		p.ID = StableProxyID(p)
	}
	if p.Name == "" && p.Host != "" && p.Port > 0 {
		p.Name = fmt.Sprintf("%s:%d", p.Host, p.Port)
	}
	return p
}

func StableProxyID(p Proxy) string {
	sum := sha1.Sum([]byte(strings.ToLower(strings.TrimSpace(p.Type)) + "|" + strings.ToLower(strings.TrimSpace(p.Host)) + "|" + fmt.Sprintf("%d", p.Port) + "|" + strings.TrimSpace(p.Username)))
	return "proxy_" + hex.EncodeToString(sum[:6])
}

func (c *Config) ClearAccountTokens() {
	if c == nil {
		return
	}
	for i := range c.Accounts {
		c.Accounts[i].Token = ""
	}
}

func (c *Config) NormalizeCredentials() {
	if c == nil {
		return
	}
	normalizedAPIKeys := normalizeAPIKeys(c.APIKeys)
	if len(normalizedAPIKeys) > 0 {
		c.APIKeys = normalizedAPIKeys
		c.Keys = apiKeysToStrings(c.APIKeys)
	} else {
		c.Keys = normalizeKeys(c.Keys)
		c.APIKeys = apiKeysFromStrings(c.Keys, nil)
	}

	for i := range c.Accounts {
		c.Accounts[i].Name = strings.TrimSpace(c.Accounts[i].Name)
		c.Accounts[i].Remark = strings.TrimSpace(c.Accounts[i].Remark)
	}

	c.Vercel = NormalizeVercelConfig(c.Vercel)
	c.normalizeModelAliases()
}

// DropInvalidAccounts removes accounts that cannot be addressed by admin APIs
// (no email and no normalizable mobile). This prevents legacy token-only
// records from becoming orphaned empty entries after token stripping.
func (c *Config) DropInvalidAccounts() {
	if c == nil || len(c.Accounts) == 0 {
		return
	}
	kept := make([]Account, 0, len(c.Accounts))
	for _, acc := range c.Accounts {
		if acc.Identifier() == "" {
			continue
		}
		kept = append(kept, acc)
	}
	c.Accounts = kept
}

func (c *Config) normalizeModelAliases() {
	if c == nil {
		return
	}

	aliases := map[string]string{}
	for k, v := range c.ModelAliases {
		key := strings.TrimSpace(lower(k))
		val := strings.TrimSpace(lower(v))
		if key == "" || val == "" {
			continue
		}
		aliases[key] = val
	}
	if len(aliases) == 0 {
		c.ModelAliases = nil
	} else {
		c.ModelAliases = aliases
	}
}

type AdminConfig struct {
	PasswordHash      string `json:"password_hash,omitempty"`
	JWTExpireHours    int    `json:"jwt_expire_hours,omitempty"`
	JWTValidAfterUnix int64  `json:"jwt_valid_after_unix,omitempty"`
}

type RuntimeConfig struct {
	AccountMaxInflight        int `json:"account_max_inflight,omitempty"`
	AccountMaxQueue           int `json:"account_max_queue,omitempty"`
	GlobalMaxInflight         int `json:"global_max_inflight,omitempty"`
	TokenRefreshIntervalHours int `json:"token_refresh_interval_hours,omitempty"`
	BackupReleaseCount        int `json:"backup_release_count,omitempty"`
}

type ResponsesConfig struct {
	StoreTTLSeconds int `json:"store_ttl_seconds,omitempty"`
}

type EmbeddingsConfig struct {
	Provider string `json:"provider,omitempty"`
}

type AutoDeleteConfig struct {
	Mode     string `json:"mode,omitempty"`
	Sessions bool   `json:"sessions,omitempty"`
}

type CurrentInputFileConfig struct {
	Enabled  *bool `json:"enabled,omitempty"`
	MinChars int   `json:"min_chars,omitempty"`
}

type ThinkingInjectionConfig struct {
	Enabled *bool  `json:"enabled,omitempty"`
	Prompt  string `json:"prompt,omitempty"`
}

type VercelConfig struct {
	Token     string `json:"token,omitempty"`
	ProjectID string `json:"project_id,omitempty"`
	TeamID    string `json:"team_id,omitempty"`
}

type PlatformConfig struct {
	Mode            string `json:"mode,omitempty"`              // "android" (default) or "web"
	AndroidVersion  string `json:"android_version,omitempty"`   // override Android client version (e.g. "2.1.5")
	WebVersion      string `json:"web_version,omitempty"`       // override Web client version (e.g. "2.0.0")
	AndroidAPILevel string `json:"android_api_level,omitempty"` // override Android API level (e.g. "32")
}

func NormalizeVercelConfig(v VercelConfig) VercelConfig {
	return VercelConfig{
		Token:     strings.TrimSpace(v.Token),
		ProjectID: strings.TrimSpace(v.ProjectID),
		TeamID:    strings.TrimSpace(v.TeamID),
	}
}

func (c *Config) ClearVercelCredentials() {
	if c == nil {
		return
	}
	c.Vercel = VercelConfig{}
}

func (c *Config) PlatformMode() string {
	if c == nil {
		return "android"
	}
	mode := strings.TrimSpace(strings.ToLower(c.Platform.Mode))
	if mode == "web" {
		return "web"
	}
	return "android"
}

func (c *Config) IsWebPlatform() bool {
	return c.PlatformMode() == "web"
}

func (c *Config) IsAndroidPlatform() bool {
	return c.PlatformMode() == "android"
}