package store

import (
	"context"

	"gorm.io/gorm"
)

// LLMModel 是一个 LLM 后端配置（控制面：管理员在此登记可用模型）。
// 表名 sundynix_model（遵守前缀约定）。同一时刻仅一条 Active=true。
type LLMModel struct {
	ID       uint   `gorm:"primaryKey"`
	Provider string `gorm:"size:32"`  // openai-compatible / vllm
	BaseURL  string `gorm:"size:255"` // 如 https://api.deepseek.com/v1
	APIKey   string `gorm:"size:255"`
	Model    string `gorm:"size:64"` // 如 deepseek-chat
	Active   bool
}

func (LLMModel) TableName() string { return "sundynix_model" }

// ListModels 列出全部模型配置。
func (p *Postgres) ListModels(ctx context.Context) ([]LLMModel, error) {
	if p.db == nil {
		return nil, nil
	}
	var rows []LLMModel
	err := p.db.WithContext(ctx).Order("id").Find(&rows).Error
	return rows, err
}

// SaveModel 新增或更新一条模型配置（ID==0 新增）。
func (p *Postgres) SaveModel(ctx context.Context, m *LLMModel) error {
	if p.db == nil {
		return errStoreDisabled
	}
	return p.db.WithContext(ctx).Save(m).Error
}

// SetActiveModel 把指定模型设为激活（其余取消），事务保证唯一激活。
func (p *Postgres) SetActiveModel(ctx context.Context, id uint) error {
	if p.db == nil {
		return errStoreDisabled
	}
	return p.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(&LLMModel{}).Where("active = ?", true).Update("active", false).Error; err != nil {
			return err
		}
		return tx.Model(&LLMModel{}).Where("id = ?", id).Update("active", true).Error
	})
}

// GetActiveModel 返回当前激活模型（无则 nil）。
func (p *Postgres) GetActiveModel(ctx context.Context) (*LLMModel, error) {
	if p.db == nil {
		return nil, nil
	}
	var m LLMModel
	err := p.db.WithContext(ctx).Where("active = ?", true).First(&m).Error
	if err != nil {
		return nil, nil // 未配置激活模型
	}
	return &m, nil
}

// DeleteModel 删除一条模型配置。
func (p *Postgres) DeleteModel(ctx context.Context, id uint) error {
	if p.db == nil {
		return errStoreDisabled
	}
	return p.db.WithContext(ctx).Delete(&LLMModel{}, id).Error
}