package model

import (
	"time"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

// MCP MCP工具配置
type MCP struct {
	ID          string    `json:"id" gorm:"primaryKey"`
	Name        string    `json:"name" gorm:"uniqueIndex;size:100;not null"`
	Description string    `json:"description" gorm:"type:text"`       // 英文描述
	DescriptionCN string `json:"description_cn" gorm:"type:text"`    // 中文描述
	Category    string    `json:"category" gorm:"size:50;not null"`  // 分类
	Transport   string    `json:"transport" gorm:"size:20;default:'stdio'"` // stdio, http, sse
	Command     string    `json:"command" gorm:"size:500"`            // 启动命令
	Args        string    `json:"args" gorm:"type:text"`             // 参数，JSON数组格式
	Env         string    `json:"env" gorm:"type:text"`               // 环境变量，JSON对象格式
	Status      string    `json:"status" gorm:"size:20;default:'active'"`
	CreatedAt   time.Time `json:"created_at"`
	UpdatedAt   time.Time `json:"updated_at"`
}

// BeforeCreate 创建前自动生成ID
func (m *MCP) BeforeCreate(tx *gorm.DB) error {
	if m.ID == "" {
		m.ID = uuid.New().String()
	}
	return nil
}