store

type AISessionRecord struct {
	ID        string `json:"id"`        // 会话 ID
	ConnID    string `json:"connId"`    // 所属连接
	TabID     string `json:"tabId"`     // 所属 query tab（按 tab 隔离对话；空 = 不隔离）
	DB        string `json:"db"`        // 目标库名
	Dialect   string `json:"dialect"`   // 方言标签
	Lang      string `json:"lang"`      // 会话语言（恢复时沿用，历史会话不回溯）
	Messages  []any  `json:"messages"`  // 对话消息（schema.Message 序列化，含 system/user/assistant/tool）
	Usage     any    `json:"usage"`     // 累计 token（llm.Usage）
	CreatedAt int64  `json:"createdAt"` // Unix 毫秒
	UpdatedAt int64  `json:"updatedAt"` // Unix 毫秒（每次对话更新，用于过期判定）
}

type CompareOptions = engine.CompareOptions

type CompareResult = engine.CompareResult

type ConnInfo struct {
	ID        string     `json:"id"`
	Name      string     `json:"name"`
	ShortName string     `json:"shortName,omitempty"`
	Env       string     `json:"env,omitempty"`
	Conn      DBConnInfo `json:"conn"`
	SubTypes  []string   `json:"subTypes"` // 该类型可用的子类型（兼容数据库产品）
}

type ConnRecord struct {
	ID        string     `json:"id"`
	Name      string     `json:"name"`
	ShortName string     `json:"shortName,omitempty"` // 命令行简写（如 prod/my-dev），可选；不填时用 Name 匹配
	Env       string     `json:"env,omitempty"`       // dev/test/staging/prod，留空视为 prod
	Conn      DBConnInfo `json:"conn"`
}

type Contributor = engine.Contributor

type ContributorImportRequest = engine.ContributorImportRequest

type ContributorRequest = engine.ContributorRequest

type ContributorResult = engine.ContributorResult

type CreateSnapshotOptions = engine.CreateSnapshotOptions

type DBConnInfo = engine.DBConnInfo

type DataEntry = engine.DataEntry

type DataPackage = engine.DataPackage

type DataPrepareRequest = engine.DataPrepareRequest

type DataPreparer = engine.DataPreparer

type DictionaryOptions = engine.DictionaryOptions

type ExecutionRecord struct {
	ID           string   `json:"id"`                     // 同 taskID
	TaskType     string   `json:"taskType"`               // export/import/migrate/compare/dictionary
	TaskConfigID string   `json:"taskConfigId,omitempty"` // 关联的任务配置 ID（可选）
	Status       string   `json:"status"`                 // running/done/error/cancelled
	StartedAt    int64    `json:"startedAt"`
	FinishedAt   int64    `json:"finishedAt"`
	Duration     int64    `json:"duration"`   // 毫秒
	TotalUnits   int      `json:"totalUnits"` // 工作单元数（表 + 对象）
	DoneUnits    int      `json:"doneUnits"`  // 终态时已完成的单元数（失败/取消时用于还原真实进度）
	TotalRows    int64    `json:"totalRows"`
	OutputPath   string   `json:"outputPath,omitempty"`
	FileSize     int64    `json:"fileSize,omitempty"`
	ErrorMsg     string   `json:"errorMsg,omitempty"`
	Logs         []string `json:"logs,omitempty"`   // 执行日志快照（截断保留最近若干条，供终态回放展示）
	Target       string   `json:"target,omitempty"` // 操作目标描述（连接名 + 库表对象），如 "生产库 · db1(93表)"
	Summary      string   `json:"summary"`          // 如 "3表, 40000行, 15.3MB"
}

type ExportFormat = engine.ExportFormat

type ExportOptions = engine.ExportOptions

type ImportOptions = engine.ImportOptions

type MigrateOptions = engine.MigrateOptions

type ProgressFunc = engine.ProgressFunc

type ProgressInfo = engine.Progress

type QueryHooks = engine.QueryHooks

type ResetMode = engine.ResetMode

type SQLAuditEntry struct {
	ID        string `json:"id"`
	ConnID    string `json:"connId"`
	DB        string `json:"db,omitempty"`   // 执行时目标库名
	Mode      string `json:"mode,omitempty"` // 执行模式 transform/raw
	Source    string `json:"source"`         // manual=用户手写 / tree=对象树自动查询 / cell=单元格内联编辑
	SQL       string `json:"sql"`            // 实际执行的 SQL（cell 类型下为摘要）
	IsWrite   bool   `json:"isWrite"`
	RowCount  int    `json:"rowCount"`
	Elapsed   int64  `json:"elapsedMs"`
	Status    string `json:"status"` // ok / error
	ErrorMsg  string `json:"error,omitempty"`
	CreatedAt int64  `json:"createdAt"` // Unix 毫秒

	// 单元格内联编辑的结构化参数（source=cell 时才有值）
	Table     string   `json:"table,omitempty"`
	Column    string   `json:"column,omitempty"`
	NewValue  any      `json:"newValue,omitempty"`
	PKColumns []string `json:"pkColumns,omitempty"`
	PKValues  []any    `json:"pkValues,omitempty"`
}

type SQLFavorite struct {
	ID        string `json:"id"`
	ConnID    string `json:"connId"`
	Title     string `json:"title"` // 用户可重命名，默认取 SQL 去注释后首行前 40 字符
	DB        string `json:"db"`    // 执行上下文：目标库（仅 replace_all 回填时还原）
	Mode      string `json:"mode"`  // 执行模式 transform/raw（仅 replace_all 回填时还原）
	SQL       string `json:"sql"`
	CreatedAt int64  `json:"createdAt"` // Unix 毫秒
}

type SQLHistoryItem struct {
	ID        string `json:"id"`
	ConnID    string `json:"connId"`
	DB        string `json:"db,omitempty"`   // 执行时目标库名（空=连接默认库）
	Mode      string `json:"mode,omitempty"` // 执行模式 transform/raw
	SQL       string `json:"sql"`
	IsWrite   bool   `json:"isWrite"`
	RowCount  int    `json:"rowCount"`
	Elapsed   int64  `json:"elapsedMs"`
	Status    string `json:"status"` // ok / error
	ErrorMsg  string `json:"error,omitempty"`
	CreatedAt int64  `json:"createdAt"` // Unix 毫秒
}

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

type Snapshot = engine.Snapshot

type SnapshotCompareOptions = engine.SnapshotCompareOptions

type SnapshotInfo = engine.SnapshotInfo

type SnapshotTable = engine.SnapshotTable

type Store interface {
	// Close 关闭底层数据库连接。
	Close() error

	// Migrate 执行自动迁移（建表/补列），基于 allModels 注册表。
	Migrate() error

	// SaveConn 保存连接配置：rec.ID 非空则按主键更新，否则生成新 ID。
	SaveConn(rec ConnRecord) (ConnRecord, error)
	// LoadConns 加载全部连接配置（按 ID 索引）。
	LoadConns() map[string]ConnRecord
	// GetConn 按主键 ID 查找；兼容按名称或短名查找。
	GetConn(key string) (ConnRecord, bool)
	// DeleteConn 删除连接（按主键 ID，兼容名称或短名）。
	DeleteConn(key string) error

	// SaveTask 保存任务配置（按 ID 更新或新增）。
	SaveTask(task TaskConfig) error
	// LoadTasks 加载全部任务配置。
	LoadTasks() []TaskConfig
	// GetTask 获取指定任务配置。
	GetTask(id string) (TaskConfig, bool)
	// DeleteTask 删除任务配置。
	DeleteTask(taskID string) error
	// MarkLastUsed 标记指定类型为最近使用（同类型其他任务取消标记）。
	MarkLastUsed(taskID, taskType string) error
	// GetLastUsed 获取指定类型最近使用的任务配置。
	GetLastUsed(taskType string) *TaskConfig

	// SaveHistory 保存执行历史（按 ID 更新或新增，超出上限裁剪最旧记录）。
	SaveHistory(record ExecutionRecord) error
	// LoadHistory 加载执行历史（taskType 为空=全部，taskConfigID 为空=不过滤）。
	LoadHistory(taskType, taskConfigID string) []ExecutionRecord
	// GetHistory 获取指定执行记录。
	GetHistory(id string) (ExecutionRecord, error)
	// DeleteHistory 删除指定执行记录。
	DeleteHistory(id string) error

	// SaveWebAccess 保存 Web 访问凭证。
	SaveWebAccess(info WebAccessInfo) error
	// LoadWebAccess 读取 Web 访问凭证；无有效内容时 ok=false。
	LoadWebAccess() (WebAccessInfo, bool)

	// AddSQLHistory 追加一条 SQL 执行历史（每连接环形保留最近 N 条）。
	// user 为所属用户域；item.ConnID 为裸连接 key（三列同写：user/conn_key/conn_id 作用域键）。
	AddSQLHistory(user string, item SQLHistoryItem) error
	// ListSQLHistory 返回某用户域下某连接的历史（新→旧）。
	// connID 为裸连接 key（user 列 + conn_key 双条件过滤）；为空时仅按 user 列过滤
	// （该用户域全部连接）。
	ListSQLHistory(user, connID string) ([]SQLHistoryItem, error)
	// ClearSQLHistory 清空某用户域下某连接的历史（connID 为空=该用户域全部连接）。
	ClearSQLHistory(user, connID string) error

	// AddFavorite 新增一条收藏（user 为所属用户域；f.ConnID 为裸连接 key，三列同写）。
	AddFavorite(user string, f *SQLFavorite) error
	// ListFavorites 返回该用户域的全部收藏（新→旧）。
	ListFavorites(user string) ([]*SQLFavorite, error)
	// DeleteFavorite 删除收藏（按 id 定位，仅限本用户域）。
	DeleteFavorite(user, id string) error
	// RenameFavorite 重命名收藏（按 id 定位，仅限本用户域）。
	RenameFavorite(user, id, title string) error

	// AppendSQLAudit 追加一条 SQL 审计日志（只追加，不提供删除）。
	// user 为所属用户域；entry.ConnID 为裸连接 key（三列同写）。
	AppendSQLAudit(user string, entry SQLAuditEntry) error
	// ListSQLAudit 读取审计日志（倒序，分页）。connID 为裸连接 key（user 列 + conn_key
	// 双条件过滤）；为空时仅按 user 列过滤（该用户域全部连接，跨用户隔离由 user 列保证）。
	ListSQLAudit(user, connID string, limit, offset int) ([]SQLAuditEntry, error)

	// SaveWorkspace 保存某用户域下某连接的工作区（整体覆盖）。
	// connID 为裸连接 key（conn_id 物理列存作用域键作主键，conn_key 存裸 key）；
	// user 为所属用户域。
	SaveWorkspace(user, connID string, state WorkspaceState) error
	// LoadWorkspace 读取某用户域下某连接的工作区（user 列 + conn_key 双条件）；无记录时 ok=false。
	LoadWorkspace(user, connID string) (WorkspaceState, bool)
	// DeleteWorkspace 删除某用户域下某连接的工作区（user 列 + conn_key 双条件）。
	DeleteWorkspace(user, connID string) error
	// DeleteWorkspacesByConnAllUsers 跨用户域删除某连接的全部工作区（连接删除级联清理，
	// connID 为裸连接 key，按 conn_key 列等值删除所有用户域的行）。
	DeleteWorkspacesByConnAllUsers(connID string) error

	// SaveAISession 保存/更新一个 AI 会话（整组消息覆盖写）。
	// user 为所属用户域；rec.ConnID 为裸连接 key（三列同写：user/conn_key/conn_id 作用域键）。
	SaveAISession(user string, rec AISessionRecord) error
	// LoadAISession 读取指定会话（user 为所属用户域，按 user 列归属校验，不属于该用户域
	// 视为不存在）；无记录时 ok=false。rec.ConnID 还原为裸连接 key。
	LoadAISession(user, sessionID string) (AISessionRecord, bool)
	// AISessionExists 判断会话 ID 是否已落盘（跨用户域存在性判定，不含归属校验），
	// 供透明重建复用原 ID 前区分「彻底不存在」与「存在但归属其他用户域」。
	AISessionExists(sessionID string) bool
	// ListAISessions 列出某用户域下某连接（可选指定 tab）的会话（新→旧，仅元信息不含
	// 消息，供前端恢复选择）。connID 为裸连接 key，按 user 列 + conn_key 双条件过滤。
	ListAISessions(user, connID, tabID string) ([]AISessionRecord, error)
	// DeleteAISession 删除指定会话（user 为所属用户域，按 user 列 + 主键双条件删除）。
	DeleteAISession(user, sessionID string) error
	// DeleteAISessionByTab 删除某用户域下某连接指定 tab 的会话（tab 关闭时调用）。
	DeleteAISessionByTab(user, connID, tabID string) error
	// DeleteAISessionsByConn 删除某用户域下某连接的全部会话。
	DeleteAISessionsByConn(user, connID string) error
	// DeleteAISessionsByConnAllUsers 跨用户域删除某连接的全部会话（连接删除级联清理，
	// connID 为裸连接 key，按 conn_key 列等值删除所有用户域的会话）。
	DeleteAISessionsByConnAllUsers(connID string) error
	// PurgeExcessAISessions 清理超额会话：当某连接会话数 > maxPerConn 时，
	// 删除其中「超过 keepDays 天未活动」的会话（从最旧开始），返回删除条数。
	PurgeExcessAISessions(maxPerConn int, keepDays int) (int64, error)

	// UpsertSnapshot 写入/更新快照索引行（按 info.ID 主键幂等；存量 JSON 索引迁移与
	// 新建快照共用）。conn_id/created_at 同步写入真实列供 SQL 过滤。
	UpsertSnapshot(info SnapshotInfo) error
	// ListSnapshots 列出快照索引（created_at 倒序）。connID 非空时按 conn_id 列等值过滤
	// （连接 ID 即创建时连接，库模式下虚拟连接为 env:<id>，用于按环境隔离）。
	ListSnapshots(connID string) ([]SnapshotInfo, error)
	// DeleteSnapshot 删除快照索引行（按主键 ID）。
	DeleteSnapshot(id string) error

	// GetMeta 读取 KV 元信息；无记录时 ok=false。
	GetMeta(key string) (string, bool, error)
	// SetMeta 写入/更新 KV 元信息（按 key 幂等）。
	SetMeta(key, value string) error
}

type TabSettings struct {
	MaxTabs     int      `json:"maxTabs,omitempty"`     // 最大标签页数（默认 20，范围 5~100）
	EvictOrder  []string `json:"evictOrder,omitempty"`  // 淘汰优先级顺序（5 类分类的排列）
	MaxTabWidth int      `json:"maxTabWidth,omitempty"` // 标签页最大宽度（像素，默认 160，范围 80~300）
}

type TableAlias = engine.TableAlias

type TableCondition = engine.TableCondition

type TaskConfig struct {
	ID             string             `json:"id" yaml:"id"`
	Name           string             `json:"name" yaml:"name"`
	Type           string             `json:"type" yaml:"type"` // export/import/migrate/compare/dictionary
	ExportOpts     *ExportOptions     `json:"exportOpts,omitempty" yaml:"exportOpts,omitempty"`
	ImportOpts     *ImportOptions     `json:"importOpts,omitempty" yaml:"importOpts,omitempty"`
	MigrateOpts    *MigrateOptions    `json:"migrateOpts,omitempty" yaml:"migrateOpts,omitempty"`
	CompareOpts    *CompareOptions    `json:"compareOpts,omitempty" yaml:"compareOpts,omitempty"`
	DictionaryOpts *DictionaryOptions `json:"dictionaryOpts,omitempty" yaml:"dictionaryOpts,omitempty"`
	CreatedAt      int64              `json:"createdAt" yaml:"createdAt"`
	UpdatedAt      int64              `json:"updatedAt" yaml:"updatedAt"`
	IsLastUsed     bool               `json:"isLastUsed" yaml:"isLastUsed"`
}

type WebAccessInfo struct {
	Addr     string `json:"addr"`               // 监听地址 host:port
	Token    string `json:"token"`              // 访问令牌（空=启动时禁用了认证）
	IssuedAt int64  `json:"issuedAt,omitempty"` // 令牌签发时间（Unix 毫秒），用于过期判断
}

type WorkspaceState struct {
	Tabs        []WorkspaceTab `json:"tabs"`
	ActiveID    string         `json:"activeId"`
	TabSettings *TabSettings   `json:"tabSettings,omitempty"` // 标签页设置（可选）
}

type WorkspaceTab struct {
	ID    string `json:"id"`
	Kind  string `json:"kind"`            // "query" | "object"
	Seq   int    `json:"seq,omitempty"`   // 查询序号（query tab 专用）
	Title string `json:"title,omitempty"` // 展示标题（query tab 重命名后）
	DB    string `json:"db,omitempty"`    // 目标库名（空 = 连接默认库）
	SQL   string `json:"sql,omitempty"`   // 查询 SQL（query tab 专用）
	Mode  string `json:"mode,omitempty"`  // 执行模式 transform/raw（query tab 专用）
	// object tab 专用
	Name    string `json:"name,omitempty"`
	ObjType string `json:"objType,omitempty"` // table / view / function / procedure
	SubTab  string `json:"subTab,omitempty"`  // data / struct / ddl
	// 表浏览视图布局（过滤/排序/列显隐/页大小），原样 JSON 存取，后端不解析语义
	ViewLayout json.RawMessage `json:"viewLayout,omitempty"`
	// 固定标签页（不参与自动淘汰）
	Pinned bool `json:"pinned,omitempty"`
}