parser

type AlterAction struct {
	Type    string      `json:"type"` // ADD, DROP, MODIFY, CHANGE, etc.
	Column  *ColumnInfo `json:"column,omitempty"`
	OldName string      `json:"old_name,omitempty"`
	NewName string      `json:"new_name,omitempty"`
}

type AlterStatement struct {
	Type    string        `json:"type"` // TABLE, etc.
	Name    string        `json:"name"`
	Actions []AlterAction `json:"actions,omitempty"`
}

type BlockStmt struct {
	Declarations []Declaration
	Statements   []Statement
}

type BoundType int

type CTEContext struct {
	// CTE缓存
	CTEResults map[string][]map[string]interface{}

	// CTE是否已物化
	CTEMaterialized map[string]bool
}

type CTEInfo struct {
	Name      string           // CTE名称
	Alias     string           // CTE别名
	Subquery  *SelectStatement // CTE子查询
	Columns   []string         // 列别名(可选)
	Recursive bool             // 是否为递归CTE
}

type CTEOptimizer struct {
	// 优化配置
	InlineThreshold int  // 内联阈值(行数)
	CacheEnabled    bool // 是否启用缓存
}

type CallStmt struct {
	ProcedureName string
	Args          []Expression // 参数
}

type CaseStmt struct {
	Expression Expression // 可选的表达式
	Cases      []CaseWhen
	Else       *BlockStmt
}

type CaseWhen struct {
	Condition Expression
	Then      *BlockStmt
}

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

type ColumnInfo struct {
	Name       string          `json:"name"`
	Type       string          `json:"type"`
	Nullable   bool            `json:"nullable"`
	Primary    bool            `json:"primary"`
	Default    interface{}     `json:"default,omitempty"`
	Unique     bool            `json:"unique,omitempty"`
	AutoInc    bool            `json:"auto_increment,omitempty"`
	ForeignKey *ForeignKeyInfo `json:"foreign_key,omitempty"`
	Comment    string          `json:"comment,omitempty"`

	// Generated Columns 支持
	IsGenerated      bool     `json:"is_generated,omitempty"`      // 是否为生成列
	GeneratedType    string   `json:"generated_type,omitempty"`    // "STORED" (第一阶段) 或 "VIRTUAL" (第二阶段)
	GeneratedExpr    string   `json:"generated_expr,omitempty"`    // 表达式字符串
	GeneratedDepends []string `json:"generated_depends,omitempty"` // 依赖的列名

	// Vector Columns 支持
	VectorDim  int    `json:"vector_dim,omitempty"`  // 向量维度
	VectorType string `json:"vector_type,omitempty"` // 向量类型（如 "float32"）
}

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

type CreateIndexStatement struct {
	IndexName string   `json:"index_name"`
	TableName string   `json:"table_name"`
	Columns   []string `json:"columns"`    // Support composite index (multi-column)
	IndexType string   `json:"index_type"` // BTREE, HASH, FULLTEXT, VECTOR
	Unique    bool     `json:"unique"`
	IfExists  bool     `json:"if_exists"`

	// Vector Index 配置
	IsVectorIndex   bool                   `json:"is_vector_index,omitempty"`
	VectorIndexType string                 `json:"vector_index_type,omitempty"` // hnsw, ivf_flat, flat
	VectorMetric    string                 `json:"vector_metric,omitempty"`     // cosine, l2, inner_product
	VectorDim       int                    `json:"vector_dim,omitempty"`
	VectorParams    map[string]interface{} `json:"vector_params,omitempty"`
}

type CreateStatement struct {
	Type       string                 `json:"type"` // TABLE, DATABASE, INDEX, etc.
	Name       string                 `json:"name"`
	Columns    []ColumnInfo           `json:"columns,omitempty"`
	Options    map[string]interface{} `json:"options,omitempty"`
	Persistent bool                   `json:"persistent,omitempty"` // PERSISTENT=1 for hybrid storage
}

type CreateUserStatement struct {
	Username    string `json:"username"`
	Host        string `json:"host,omitempty"`     // Default is '%'
	Password    string `json:"password,omitempty"` // IDENTIFIED BY
	IfNotExists bool   `json:"if_not_exists"`
}

type CreateViewStatement struct {
	OrReplace   bool             `json:"or_replace"`
	Algorithm   string           `json:"algorithm,omitempty"` // UNDEFINED, MERGE, TEMPTABLE
	Definer     string           `json:"definer,omitempty"`   // 'user'@'host'
	Security    string           `json:"security,omitempty"`  // DEFINER, INVOKER
	Name        string           `json:"name"`
	ColumnList  []string         `json:"column_list,omitempty"`
	Select      *SelectStatement `json:"select"`
	CheckOption string           `json:"check_option,omitempty"` // NONE, CASCADED, LOCAL
}

type DDLHandler struct{}

type DDLResult struct {
	Type      string
	Tables    []string
	Databases []string
}

type DMLHandler struct{}

type DMLResult struct {
	Type     string
	Tables   []string
	Columns  []string
	Affected int64
}

type Declaration struct {
	Name     string
	DataType string
	Initial  interface{} // 初始值(可选)
}

type DeclareStmt struct {
	Variables []Declaration
}

type DefaultHandler struct{}

type DefaultResult struct {
	Type string
	Stmt ast.StmtNode
}

type DeleteStatement struct {
	Table   string        `json:"table"`
	Where   *Expression   `json:"where,omitempty"`
	OrderBy []OrderByItem `json:"order_by,omitempty"`
	Limit   *int64        `json:"limit,omitempty"`
}

type DescribeStatement struct {
	Table  string `json:"table"`
	Column string `json:"column,omitempty"`
}

type DropIndexStatement struct {
	IndexName string `json:"index_name"`
	TableName string `json:"table_name"`
	IfExists  bool   `json:"if_exists"`
}

type DropStatement struct {
	Type     string `json:"type"` // TABLE, DATABASE, INDEX, etc.
	Name     string `json:"name"`
	IfExists bool   `json:"if_exists"`
}

type DropUserStatement struct {
	Username string `json:"username"`
	Host     string `json:"host,omitempty"` // Default is '%'
	IfExists bool   `json:"if_exists"`
}

type DropViewStatement struct {
	Views    []string `json:"views"`     // 视图名称列表
	IfExists bool     `json:"if_exists"` // IF EXISTS
	Restrict bool     `json:"restrict"`  // RESTRICT (TiDB 不支持，保留用于兼容性)
	Cascade  bool     `json:"cascade"`   // CASCADE (TiDB 不支持，保留用于兼容性)
}

type ElseIfStmt struct {
	Condition Expression
	Then      *BlockStmt
}

type ExplainStatement struct {
	Query     *SelectStatement `json:"query,omitempty"`      // The query to explain
	TargetSQL string           `json:"target_sql,omitempty"` // Raw SQL string
	Format    string           `json:"format,omitempty"`     // Format type (e.g., "TREE", "JSON")
	Analyze   bool             `json:"analyze,omitempty"`    // EXPLAIN ANALYZE
}

type ExprConverter struct{}

type ExprType string

type Expression struct {
	Type     ExprType     `json:"type"`
	Column   string       `json:"column,omitempty"`
	Value    interface{}  `json:"value,omitempty"`
	Operator string       `json:"operator,omitempty"`
	Left     *Expression  `json:"left,omitempty"`
	Right    *Expression  `json:"right,omitempty"`
	Args     []Expression `json:"args,omitempty"`
	Function string       `json:"function,omitempty"`
}

type ForeignKeyInfo struct {
	RefTable  string `json:"ref_table"`
	RefColumn string `json:"ref_column"`
	OnDelete  string `json:"on_delete,omitempty"`
	OnUpdate  string `json:"on_update,omitempty"`
}

type FrameBound struct {
	Type  BoundType
	Value Expression // 帧偏移值(可为空)
}

type FrameMode int

type FunctionInfo struct {
	Name       string
	Params     []ProcedureParam
	ReturnType string
	Body       *BlockStmt
}

type FunctionStmt struct {
	Name       string
	Params     []ProcedureParam
	ReturnType string
	Body       *BlockStmt
}

type GrantStatement struct {
	Privileges      []string `json:"privileges"`
	On              string   `json:"on"` // e.g., "db.*", "db.table"
	To              string   `json:"to"` // e.g., "'user'@'host'"
	WithGrantOption bool     `json:"with_grant_option"`
}

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

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

type IfStmt struct {
	Condition Expression
	Then      *BlockStmt
	ElseIfs   []*ElseIfStmt
	Else      *BlockStmt
}

type InsertStatement struct {
	Table       string           `json:"table"`
	Columns     []string         `json:"columns,omitempty"`
	Values      [][]interface{}  `json:"values"`
	OnDuplicate *UpdateStatement `json:"on_duplicate,omitempty"`
}

type JoinInfo struct {
	Type      JoinType    `json:"type"`
	Table     string      `json:"table"`
	Alias     string      `json:"alias,omitempty"`
	Condition *Expression `json:"condition,omitempty"`
}

type JoinType string

type OrderByItem struct {
	Column    string `json:"column"`
	Direction string `json:"direction"`           // ASC, DESC
	Collation string `json:"collation,omitempty"` // COLLATE clause (optional)
}

type OrderItem struct {
	Expr      Expression
	Direction string
	Collation string // COLLATE clause (optional)
}

type ParamType int

type ParseResult struct {
	Statement *SQLStatement `json:"statement"`
	Success   bool          `json:"success"`
	Error     string        `json:"error,omitempty"`
}

type ParsedHints struct {
	// JOIN hints
	HashJoinTables     []string
	MergeJoinTables    []string
	INLJoinTables      []string
	INLHashJoinTables  []string
	INLMergeJoinTables []string
	NoHashJoinTables   []string
	NoMergeJoinTables  []string
	NoIndexJoinTables  []string
	LeadingOrder       []string
	StraightJoin       bool

	// INDEX hints
	UseIndex     map[string][]string
	ForceIndex   map[string][]string
	IgnoreIndex  map[string][]string
	OrderIndex   map[string]string
	NoOrderIndex map[string]string

	// AGG hints
	HashAgg      bool
	StreamAgg    bool
	MPP1PhaseAgg bool
	MPP2PhaseAgg bool

	// Subquery hints
	SemiJoinRewrite bool
	NoDecorrelate   bool
	UseTOJA         bool

	// Global hints
	QBName                string
	MaxExecutionTime      time.Duration
	MemoryQuota           int64
	ReadConsistentReplica bool
	ResourceGroup         string
}

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

type ParserError struct {
	SQL     string `json:"sql"`
	Message string `json:"message"`
	Pos     int    `json:"pos"`
}

type ProcedureInfo struct {
	Name    string
	Params  []ProcedureParam
	Body    *BlockStmt
	Returns []ColumnInfo // 返回值(用于函数)
}

type ProcedureParam struct {
	Name      string
	ParamType ParamType
	DataType  string
}

type ProcedureStmt struct {
	Name   string
	Params []ProcedureParam
	Body   *BlockStmt
}

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

type QueryHandler struct{}

type QueryResult struct {
	Type    string
	Tables  []string
	Columns []string
	Stmt    *ast.SelectStmt
}

type RecommendIndexConfig struct {
	MaxNumIndexes   int
	MaxIndexColumns int
	MaxNumQuery     int
	Timeout         int // 秒
}

type RecommendIndexParser struct{}

type RecommendIndexStatement struct {
	Action string // "RUN", "SHOW", "SET"

	// RUN 参数
	Query    string // SQL 查询字符串
	ForQuery bool   // 是否指定了 FOR 子句

	// SET 参数
	OptionName  string // 选项名称
	OptionValue string // 选项值
}

type ReturnStmt struct {
	Expression Expression // 返回值(可选)
}

type RevokeStatement struct {
	Privileges []string `json:"privileges"`
	On         string   `json:"on"`   // e.g., "db.*", "db.table"
	From       string   `json:"from"` // e.g., "'user'@'host'"
}

type Row map[string]interface{}

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

type SQLInfo struct {
	Tables       []string      // 涉及的表名
	Columns      []string      // 涉及的列名
	Databases    []string      // 涉及的数据库名
	WhereExpr    ast.ExprNode  // WHERE 条件表达式
	LimitExpr    *ast.Limit    // LIMIT 表达式
	OrderByItems []*ast.ByItem // ORDER BY 子句
	GroupByItems []*ast.ByItem // GROUP BY 子句
	Having       ast.ExprNode  // HAVING 子句
	IsSelect     bool
	IsInsert     bool
	IsUpdate     bool
	IsDelete     bool
	IsDDL        bool
}

type SQLStatement struct {
	Type       SQLType              `json:"type"`
	RawSQL     string               `json:"raw_sql"`
	Select     *SelectStatement     `json:"select,omitempty"`
	Insert     *InsertStatement     `json:"insert,omitempty"`
	Update     *UpdateStatement     `json:"update,omitempty"`
	Delete     *DeleteStatement     `json:"delete,omitempty"`
	Create     *CreateStatement     `json:"create,omitempty"`
	CreateView *CreateViewStatement `json:"create_view,omitempty"`
	Drop       *DropStatement       `json:"drop,omitempty"`
	DropView   *DropViewStatement   `json:"drop_view,omitempty"`
	Alter      *AlterStatement      `json:"alter,omitempty"`
	// Note: TiDB does not support ALTER VIEW, so AlterView field is deprecated
	// AlterView    *AlterViewStatement    `json:"alter_view,omitempty"`
	CreateIndex *CreateIndexStatement `json:"create_index,omitempty"`
	DropIndex   *DropIndexStatement   `json:"drop_index,omitempty"`
	Show        *ShowStatement        `json:"show,omitempty"`
	Describe    *DescribeStatement    `json:"describe,omitempty"`
	Explain     *ExplainStatement     `json:"explain,omitempty"`
	Begin       *TransactionStatement `json:"begin,omitempty"`
	Commit      *TransactionStatement `json:"commit,omitempty"`
	Rollback    *TransactionStatement `json:"rollback,omitempty"`
	Use         *UseStatement         `json:"use,omitempty"`
	CreateUser  *CreateUserStatement  `json:"create_user,omitempty"`
	DropUser    *DropUserStatement    `json:"drop_user,omitempty"`
	Grant       *GrantStatement       `json:"grant,omitempty"`
	Revoke      *RevokeStatement      `json:"revoke,omitempty"`
	SetPassword *SetPasswordStatement `json:"set_password,omitempty"`
}

type SQLType string

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

type SelectColumn struct {
	Name       string      `json:"name"`
	Alias      string      `json:"alias,omitempty"`
	Table      string      `json:"table,omitempty"`
	Expr       *Expression `json:"expr,omitempty"`
	IsWildcard bool        `json:"is_wildcard"` // 是否是 *
}

type SelectStatement struct {
	Distinct bool           `json:"distinct"`
	Columns  []SelectColumn `json:"columns"`
	From     string         `json:"from"`
	Joins    []JoinInfo     `json:"joins,omitempty"`
	Where    *Expression    `json:"where,omitempty"`
	GroupBy  []string       `json:"group_by,omitempty"`
	Having   *Expression    `json:"having,omitempty"`
	OrderBy  []OrderByItem  `json:"order_by,omitempty"`
	Limit    *int64         `json:"limit,omitempty"`
	Offset   *int64         `json:"offset,omitempty"`
	Hints    string         `json:"hints,omitempty"` // Raw hints string from SQL comment
}

type SetHandler struct{}

type SetPasswordStatement struct {
	Username    string `json:"username"`
	Host        string `json:"host,omitempty"` // Default is '%'
	NewPassword string `json:"new_password"`   // PASSWORD('password')
}

type SetResult struct {
	Type  string
	Vars  map[string]interface{}
	Count int
}

type SetStmt struct {
	Variable string
	Value    Expression
}

type ShowHandler struct{}

type ShowResult struct {
	Type   string
	ShowTp string
	Tables []string
}

type ShowStatement struct {
	Type  string `json:"type"` // TABLES, DATABASES, COLUMNS, PROCESSLIST, etc.
	Table string `json:"table,omitempty"`
	Where string `json:"where,omitempty"`
	Like  string `json:"like,omitempty"`
	Full  bool   `json:"full,omitempty"` // SHOW FULL PROCESSLIST
}

type Statement interface{}

type StmtHandler interface {
	Handle(stmt ast.StmtNode) (result interface{}, err error)
}

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

type TransactionStatement struct {
	Level string `json:"level,omitempty"` // 隔离级别：READ UNCOMMITTED, READ COMMITTED, REPEATABLE READ, SERIALIZABLE
}

type UpdateStatement struct {
	Table   string                 `json:"table"`
	Set     map[string]interface{} `json:"set"`
	Where   *Expression            `json:"where,omitempty"`
	OrderBy []OrderByItem          `json:"order_by,omitempty"`
	Limit   *int64                 `json:"limit,omitempty"`
}

type UseHandler struct{}

type UseResult struct {
	Type     string
	Database string
}

type UseStatement struct {
	Database string `json:"database"` // 数据库名
}

type WhileStmt struct {
	Condition Expression
	Body      *BlockStmt
}

type WindowExpression struct {
	FuncName string       // 函数名
	Args     []Expression // 函数参数
	Spec     *WindowSpec  // 窗口规范
	Distinct bool         // DISTINCT标记
}

type WindowFrame struct {
	Mode  FrameMode   // 帧模式(ROWS/RANGE)
	Start FrameBound  // 起始边界
	End   *FrameBound // 结束边界(可为空)
}

type WindowSpec struct {
	Name        string       // 窗口名称(如果有)
	PartitionBy []Expression // PARTITION BY表达式
	OrderBy     []OrderItem  // ORDER BY表达式
	Frame       *WindowFrame // 窗口帧定义
}

type WindowType int

type WithClause struct {
	CTEs        []*CTEInfo // CTE列表
	IsRecursive bool       // 是否递归
}