package model

import "encoding/json"

type SettingKey string

const (
	SettingKeyPublic  SettingKey = "public"
	SettingKeyPrivate SettingKey = "private"
)

// ModelChannel 模型渠道配置。
type ModelChannel struct {
	Protocol string   `json:"protocol"`
	Name     string   `json:"name"`
	BaseURL  string   `json:"baseUrl"`
	APIKey   string   `json:"apiKey"`
	Models   []string `json:"models"`
	Weight   int      `json:"weight"`
	Enabled  bool     `json:"enabled"`
	Remark   string   `json:"remark"`
}

// PublicSetting 公开配置。
type PublicSetting struct {
	AvailableModels    []string `json:"availableModels"`
	DefaultModel       string   `json:"defaultModel"`
	DefaultImageModel  string   `json:"defaultImageModel"`
	DefaultTextModel   string   `json:"defaultTextModel"`
	SystemPrompt       string   `json:"systemPrompt"`
	AllowCustomChannel bool     `json:"allowCustomChannel"`
}

// PrivateSetting 私有配置。
type PrivateSetting struct {
	Channels []ModelChannel `json:"channels"`
}

// Setting 系统配置。
type Setting struct {
	Key       SettingKey      `json:"key" gorm:"primaryKey"`
	Value     json.RawMessage `json:"value" gorm:"serializer:json"`
	CreatedAt string          `json:"createdAt"`
	UpdatedAt string          `json:"updatedAt"`
}

// Settings 系统公开和私有配置。
type Settings struct {
	Public  PublicSetting  `json:"public"`
	Private PrivateSetting `json:"private"`
}

func (setting *PublicSetting) UnmarshalJSON(data []byte) error {
	type alias struct {
		AvailableModels    []string `json:"availableModels"`
		DefaultModel       string   `json:"defaultModel"`
		DefaultImageModel  string   `json:"defaultImageModel"`
		DefaultTextModel   string   `json:"defaultTextModel"`
		SystemPrompt       string   `json:"systemPrompt"`
		AllowCustomChannel bool     `json:"allowCustomChannel"`
		AllowCustomModel   bool     `json:"allowCustomModel"`
	}
	var value alias
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	setting.AvailableModels = value.AvailableModels
	setting.DefaultModel = value.DefaultModel
	setting.DefaultImageModel = value.DefaultImageModel
	setting.DefaultTextModel = value.DefaultTextModel
	setting.SystemPrompt = value.SystemPrompt
	setting.AllowCustomChannel = value.AllowCustomChannel || value.AllowCustomModel
	return nil
}