postgres

type ArtifactModel struct {
	ID        string    `gorm:"primaryKey;type:uuid;default:gen_random_uuid()"`
	SessionID string    `gorm:"type:uuid;not null;index:idx_artifacts"`
	Name      string    `gorm:"type:varchar(255);not null;index:idx_artifacts"`
	Version   int       `gorm:"not null;index:idx_artifacts"`
	Content   []byte    `gorm:"type:bytea"`
	MimeType  string    `gorm:"type:varchar(100)"`
	Size      int64     `gorm:"default:0"`
	CreatedAt time.Time `gorm:"not null;default:now()"`

	// 唯一约束: 同一 session 下，同一工件的同一版本只能有一条记录
	// 通过 GORM 索引实现
	Session SessionModel `gorm:"foreignKey:SessionID;references:ID"`
}

type Config struct {
	// DSN 数据库连接字符串
	// 例如: "host=localhost user=postgres password=postgres dbname=agentsdk port=5432 sslmode=disable"
	DSN string

	// MaxIdleConns 最大空闲连接数
	MaxIdleConns int

	// MaxOpenConns 最大打开连接数
	MaxOpenConns int

	// ConnMaxLifetime 连接最大生命周期
	ConnMaxLifetime time.Duration

	// LogLevel GORM 日志级别
	LogLevel logger.LogLevel

	// AutoMigrate 是否自动迁移表结构
	AutoMigrate bool
}

type EventModel struct {
	ID           string    `gorm:"primaryKey;type:uuid;default:gen_random_uuid()"`
	SessionID    string    `gorm:"type:uuid;not null;index:idx_session_events"`
	InvocationID string    `gorm:"type:varchar(255);not null;index:idx_invocation_events"`
	Branch       string    `gorm:"type:varchar(500);not null;index:idx_branch_events"`
	Author       string    `gorm:"type:varchar(255);not null"`
	AgentID      string    `gorm:"type:varchar(255);not null"`
	Timestamp    time.Time `gorm:"not null;default:now();index:idx_session_events"`

	// 内容 - JSONB 存储
	Content []byte `gorm:"type:jsonb;not null"`

	// 动作 - JSONB 存储
	Actions []byte `gorm:"type:jsonb"`

	// 长时运行工具 ID 列表
	LongRunningToolIDs pq.StringArray `gorm:"type:text[]"`

	// 元数据 - JSONB 存储
	Metadata []byte `gorm:"type:jsonb"`

	// 关联关系
	Session SessionModel `gorm:"foreignKey:SessionID;references:ID"`
}

type Service struct {
	// contains filtered or unexported fields
}

type SessionModel struct {
	ID        string    `gorm:"primaryKey;type:uuid;default:gen_random_uuid()"`
	AppName   string    `gorm:"type:varchar(255);not null;index:idx_app_sessions"`
	UserID    string    `gorm:"type:varchar(255);not null;index:idx_user_sessions"`
	AgentID   string    `gorm:"type:varchar(255);not null;index:idx_agent_sessions"`
	CreatedAt time.Time `gorm:"not null;default:now()"`
	UpdatedAt time.Time `gorm:"not null;default:now();index:idx_user_sessions,idx_app_sessions"`

	// 关联关系
	States    []StateModel    `gorm:"foreignKey:SessionID;constraint:OnDelete:CASCADE"`
	Events    []EventModel    `gorm:"foreignKey:SessionID;constraint:OnDelete:CASCADE"`
	Artifacts []ArtifactModel `gorm:"foreignKey:SessionID;constraint:OnDelete:CASCADE"`
}

type StateModel struct {
	SessionID string    `gorm:"primaryKey;type:uuid"`
	Scope     string    `gorm:"primaryKey;type:varchar(50)"` // app, user, session, temp
	Key       string    `gorm:"primaryKey;type:varchar(255)"`
	Value     []byte    `gorm:"type:jsonb;not null"` // JSONB 存储
	CreatedAt time.Time `gorm:"not null;default:now()"`
	UpdatedAt time.Time `gorm:"not null;default:now()"`

	// 关联关系
	Session SessionModel `gorm:"foreignKey:SessionID;references:ID"`
}