// Package parser implements a parser for ClickHouse SQL.
package parser

import (
	"context"
	"fmt"
	"io"
	"strconv"
	"strings"

	"github.com/sqlc-dev/doubleclick/ast"
	"github.com/sqlc-dev/doubleclick/lexer"
	"github.com/sqlc-dev/doubleclick/token"
)

// intervalUnits contains valid SQL interval unit names
var intervalUnits = map[string]bool{
	"YEAR": true, "YEARS": true,
	"QUARTER": true, "QUARTERS": true,
	"MONTH": true, "MONTHS": true,
	"WEEK": true, "WEEKS": true, "W": true,
	"DAY": true, "DAYS": true, "D": true,
	"HOUR": true, "HOURS": true, "H": true,
	"MINUTE": true, "MINUTES": true, "M": true,
	"SECOND": true, "SECONDS": true, "S": true,
	"MILLISECOND": true, "MILLISECONDS": true, "MS": true,
	"MICROSECOND": true, "MICROSECONDS": true, "US": true,
	"NANOSECOND": true, "NANOSECONDS": true, "NS": true,
}

// isIntervalUnit checks if the given string is a valid interval unit name
func isIntervalUnit(s string) bool {
	return intervalUnits[strings.ToUpper(s)]
}

// Parser parses ClickHouse SQL statements.
type Parser struct {
	lexer    *lexer.Lexer
	current  lexer.Item
	peek     lexer.Item
	peekPeek lexer.Item // Third lookahead token for special cases
	errors   []error
}

// New creates a new Parser from an io.Reader.
func New(r io.Reader) *Parser {
	p := &Parser{
		lexer: lexer.New(r),
	}
	// Read three tokens to initialize current, peek, and peekPeek
	p.nextToken()
	p.nextToken()
	p.nextToken()
	return p
}

func (p *Parser) nextToken() {
	p.current = p.peek
	p.peek = p.peekPeek
	for {
		p.peekPeek = p.lexer.NextToken()
		// Skip whitespace and comments
		if p.peekPeek.Token == token.WHITESPACE || p.peekPeek.Token == token.LINE_COMMENT {
			continue
		}
		break
	}
}

func (p *Parser) currentIs(t token.Token) bool {
	return p.current.Token == t
}

func (p *Parser) peekIs(t token.Token) bool {
	return p.peek.Token == t
}

func (p *Parser) peekPeekIs(t token.Token) bool {
	return p.peekPeek.Token == t
}

// isColumnsFunction checks if current token is COLUMNS function (for column expressions)
func (p *Parser) isColumnsFunction() bool {
	return p.currentIs(token.COLUMNS) && p.peekIs(token.LPAREN)
}

// peekPeekIsIntervalUnit checks if the third lookahead token is an interval unit
// This is used for distinguishing "INTERVAL '2' AS n minute" patterns
func (p *Parser) peekPeekIsIntervalUnit() bool {
	return isIntervalUnit(p.peekPeek.Value)
}

// isExplainFollowedByStatement checks if EXPLAIN is followed by tokens that indicate
// an EXPLAIN statement (SELECT, WITH, AST, SYNTAX, etc.) rather than being used as an identifier
func (p *Parser) isExplainFollowedByStatement() bool {
	// EXPLAIN can be followed by:
	// - SELECT, WITH (for EXPLAIN SELECT ...)
	// - QUERY, AST, SYNTAX, PLAN, PIPELINE, ESTIMATE, TABLE, CURRENT (explain types)
	// - Identifier for explain options like "header = 1"
	// If followed by comparison operators (LIKE, =, etc.) or logical operators, it's being used as identifier
	switch p.peek.Token {
	case token.SELECT, token.WITH:
		return true
	case token.IDENT:
		// Check if it's an EXPLAIN type or option
		upperValue := strings.ToUpper(p.peek.Value)
		switch upperValue {
		case "QUERY", "AST", "SYNTAX", "PLAN", "PIPELINE", "ESTIMATE", "TABLE", "CURRENT":
			return true
		case "HEADER", "ACTIONS", "DESCRIPTION", "JSON", "GRAPH", "COMPACT", "INDEXES", "SORTING", "AGGREGATION":
			// These are explain options
			return true
		}
		return false
	default:
		// If followed by operators like LIKE, =, <, >, etc., it's being used as identifier
		return false
	}
}

func (p *Parser) expect(t token.Token) bool {
	if p.currentIs(t) {
		p.nextToken()
		return true
	}
	p.errors = append(p.errors, fmt.Errorf("expected %s, got %s at line %d, column %d",
		t, p.current.Token, p.current.Pos.Line, p.current.Pos.Column))
	return false
}

func (p *Parser) expectPeek(t token.Token) bool {
	if p.peekIs(t) {
		p.nextToken()
		return true
	}
	p.errors = append(p.errors, fmt.Errorf("expected %s, got %s at line %d, column %d",
		t, p.peek.Token, p.peek.Pos.Line, p.peek.Pos.Column))
	return false
}

// Parse parses SQL statements from the input.
func Parse(ctx context.Context, r io.Reader) ([]ast.Statement, error) {
	p := New(r)
	return p.ParseStatements(ctx)
}

// ParseStatements parses multiple SQL statements.
func (p *Parser) ParseStatements(ctx context.Context) ([]ast.Statement, error) {
	var statements []ast.Statement

	for !p.currentIs(token.EOF) {
		select {
		case <-ctx.Done():
			return statements, ctx.Err()
		default:
		}

		// Skip leading semicolons (empty statements)
		for p.currentIs(token.SEMICOLON) {
			p.nextToken()
		}
		if p.currentIs(token.EOF) {
			break
		}

		stmt := p.parseStatement()
		if stmt != nil {
			// Check for PARALLEL WITH to chain statements
			if p.currentIs(token.PARALLEL) && p.peekIs(token.WITH) {
				stmt = p.parseParallelWith(stmt)
			}
			statements = append(statements, stmt)
		}

		// Skip semicolons between statements
		for p.currentIs(token.SEMICOLON) {
			p.nextToken()
		}
	}

	if len(p.errors) > 0 {
		return statements, fmt.Errorf("parse errors: %v", p.errors)
	}
	return statements, nil
}

// parseParallelWith parses PARALLEL WITH clauses to chain statements
func (p *Parser) parseParallelWith(first ast.Statement) *ast.ParallelWithQuery {
	parallel := &ast.ParallelWithQuery{
		Position:   first.Pos(),
		Statements: []ast.Statement{first},
	}

	for p.currentIs(token.PARALLEL) && p.peekIs(token.WITH) {
		p.nextToken() // skip PARALLEL
		p.nextToken() // skip WITH
		stmt := p.parseStatement()
		if stmt != nil {
			parallel.Statements = append(parallel.Statements, stmt)
		}
	}

	return parallel
}

func (p *Parser) parseStatement() ast.Statement {
	switch p.current.Token {
	case token.SELECT:
		return p.parseSelectWithUnion()
	case token.WITH:
		// WITH can precede SELECT or INSERT in ClickHouse
		return p.parseWithStatement()
	case token.FROM:
		// FROM ... SELECT syntax (ClickHouse extension)
		return p.parseFromSelectSyntax()
	case token.LPAREN:
		// Parenthesized SELECT at statement level: (SELECT 1)
		return p.parseParenthesizedSelect()
	case token.INSERT:
		return p.parseInsert()
	case token.CREATE:
		return p.parseCreate()
	case token.REPLACE:
		// REPLACE TABLE is equivalent to CREATE OR REPLACE TABLE
		return p.parseReplace()
	case token.DROP:
		// Check for DROP SETTINGS PROFILE
		if p.peekIs(token.SETTINGS) {
			return p.parseDropSettingsProfile()
		}
		// Check for DROP PROFILE
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "PROFILE" {
			return p.parseDropSettingsProfile()
		}
		// Check for DROP ROW POLICY or DROP POLICY
		if p.peek.Token == token.IDENT && (strings.ToUpper(p.peek.Value) == "ROW" || strings.ToUpper(p.peek.Value) == "POLICY") {
			return p.parseDropRowPolicy()
		}
		// Check for DROP ROLE
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "ROLE" {
			return p.parseDropRole()
		}
		// Check for DROP RESOURCE
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "RESOURCE" {
			return p.parseDropResource()
		}
		// Check for DROP WORKLOAD
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "WORKLOAD" {
			return p.parseDropWorkload()
		}
		// Check for DROP NAMED COLLECTION
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "NAMED" {
			return p.parseDropNamedCollection()
		}
		return p.parseDrop()
	case token.ALTER:
		// Check for ALTER USER
		if p.peekIs(token.USER) {
			return p.parseAlterUser()
		}
		// Check for ALTER SETTINGS PROFILE
		if p.peekIs(token.SETTINGS) {
			return p.parseAlterSettingsProfile()
		}
		// Check for ALTER PROFILE
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "PROFILE" {
			return p.parseAlterSettingsProfile()
		}
		// Check for ALTER ROW POLICY or ALTER POLICY
		if p.peek.Token == token.IDENT && (strings.ToUpper(p.peek.Value) == "ROW" || strings.ToUpper(p.peek.Value) == "POLICY") {
			return p.parseAlterRowPolicy()
		}
		// Check for ALTER ROLE
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "ROLE" {
			return p.parseAlterRole()
		}
		// Check for ALTER NAMED COLLECTION
		if p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "NAMED" {
			return p.parseAlterNamedCollection()
		}
		return p.parseAlter()
	case token.TRUNCATE:
		return p.parseTruncate()
	case token.UNDROP:
		return p.parseUndrop()
	case token.USE:
		return p.parseUse()
	case token.DESCRIBE, token.DESC:
		return p.parseDescribe()
	case token.SHOW:
		return p.parseShow()
	case token.EXPLAIN:
		return p.parseExplain()
	case token.SET:
		// Check for SET TRANSACTION SNAPSHOT
		if p.peekIs(token.TRANSACTION) {
			return p.parseTransactionControl()
		}
		// Check for SET DEFAULT ROLE
		if p.peekIs(token.DEFAULT) {
			return p.parseSetRole()
		}
		return p.parseSet()
	case token.UPDATE:
		return p.parseUpdate()
	case token.DELETE:
		return p.parseDelete()
	case token.OPTIMIZE:
		return p.parseOptimize()
	case token.SYSTEM:
		return p.parseSystem()
	case token.RENAME:
		return p.parseRename()
	case token.EXCHANGE:
		return p.parseExchange()
	case token.EXISTS:
		// EXISTS table_name syntax (check if table exists)
		return p.parseExistsStatement()
	case token.DETACH:
		return p.parseDetach()
	case token.ATTACH:
		return p.parseAttach()
	case token.CHECK:
		return p.parseCheck()
	case token.GRANT:
		return p.parseGrant()
	case token.REVOKE:
		return p.parseRevoke()
	case token.BEGIN:
		return p.parseTransactionControl()
	case token.COMMIT:
		return p.parseTransactionControl()
	case token.ROLLBACK:
		return p.parseTransactionControl()
	case token.BACKUP:
		return p.parseBackup()
	case token.RESTORE:
		return p.parseRestore()
	case token.KILL:
		return p.parseKill()
	default:
		p.errors = append(p.errors, fmt.Errorf("unexpected token %s at line %d, column %d",
			p.current.Token, p.current.Pos.Line, p.current.Pos.Column))
		p.nextToken()
		return nil
	}
}

// parseWithStatement parses WITH ... (SELECT|INSERT) statements
// WITH clause can precede both SELECT and INSERT in ClickHouse
func (p *Parser) parseWithStatement() ast.Statement {
	// Save position to check for WITH ... INSERT later
	pos := p.current.Pos

	// Peek ahead to see if this is WITH ... INSERT
	// We need to parse the WITH clause first to check what follows
	p.nextToken() // skip WITH

	// Skip RECURSIVE keyword if present
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RECURSIVE" {
		p.nextToken()
	}

	// Parse the WITH clause
	with := p.parseWithClause()

	// Now check what follows: INSERT or SELECT
	if p.currentIs(token.INSERT) {
		// WITH ... INSERT ... SELECT syntax
		ins := p.parseInsert()
		if ins != nil {
			// Store the WITH clause in InsertQuery.With for explain to handle
			// Don't propagate to SelectQuery.With - the explain code will output
			// the inherited WITH at the end of each SelectQuery's children
			ins.With = with
		}
		return ins
	}

	// For SELECT, we use parseSelectWithParsedWith to continue with normal parsing
	// but with the already-parsed WITH clause
	return p.parseSelectWithUnionWithParsedWith(pos, with)
}

// parseSelectWithUnionWithParsedWith parses a SELECT with an already-parsed WITH clause
func (p *Parser) parseSelectWithUnionWithParsedWith(pos token.Position, with []ast.Expression) *ast.SelectWithUnionQuery {
	query := &ast.SelectWithUnionQuery{
		Position: pos,
	}

	// Parse first select with the pre-parsed WITH clause
	sel := p.parseSelectWithParsedWith(with)
	if sel == nil {
		return nil
	}

	// Check for INTERSECT/EXCEPT
	if p.isIntersectExceptWithWrapper() {
		stmts := []ast.Statement{sel}
		var ops []string

		for p.isIntersectExceptWithWrapper() {
			var op string
			if p.currentIs(token.EXCEPT) {
				op = "EXCEPT"
			} else {
				op = "INTERSECT"
			}
			p.nextToken()

			if p.currentIs(token.ALL) {
				op += " ALL"
				p.nextToken()
			} else if p.currentIs(token.DISTINCT) {
				op += " DISTINCT"
				p.nextToken()
			}
			ops = append(ops, op)

			var nextStmt ast.Statement
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				nested := p.parseSelectWithUnion()
				if nested == nil {
					break
				}
				p.expect(token.RPAREN)
				nextStmt = nested
			} else {
				nextSel := p.parseSelect()
				if nextSel == nil {
					break
				}
				nextStmt = nextSel
			}
			stmts = append(stmts, nextStmt)
		}

		result := buildIntersectExceptTree(stmts, ops)
		query.Selects = append(query.Selects, result)

		// Handle UNION after INTERSECT/EXCEPT
		for p.currentIs(token.UNION) {
			p.nextToken()
			mode := "ALL"
			if p.currentIs(token.ALL) {
				p.nextToken()
			} else if p.currentIs(token.DISTINCT) {
				mode = "DISTINCT"
				p.nextToken()
			}
			query.UnionModes = append(query.UnionModes, mode)

			var nextStmt ast.Statement
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				nested := p.parseSelectWithUnion()
				if nested == nil {
					break
				}
				p.expect(token.RPAREN)
				nextStmt = nested
			} else {
				nextSel := p.parseSelect()
				if nextSel == nil {
					break
				}
				nextStmt = nextSel
			}
			query.Selects = append(query.Selects, nextStmt)
		}

		// Parse union-level SETTINGS and FORMAT
		var formatParsed bool
		for p.currentIs(token.SETTINGS) || p.currentIs(token.FORMAT) {
			if p.currentIs(token.SETTINGS) {
				p.nextToken()
				settings := p.parseSettingsList()
				query.Settings = settings
				if formatParsed {
					query.SettingsAfterFormat = true
				} else {
					query.SettingsBeforeFormat = true
				}
			} else if p.currentIs(token.FORMAT) {
				p.nextToken()
				formatParsed = true
				if len(query.Selects) > 0 {
					if sq, ok := query.Selects[0].(*ast.SelectQuery); ok {
						if p.currentIs(token.NULL) {
							sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{"Null"}}
							p.nextToken()
						} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
							sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{p.current.Value}}
							p.nextToken()
						}
					}
				}
			}
		}

		return query
	}

	query.Selects = append(query.Selects, sel)

	// Handle UNION
	for p.currentIs(token.UNION) {
		p.nextToken()
		mode := "ALL"
		if p.currentIs(token.ALL) {
			mode = "ALL"
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			mode = "DISTINCT"
			p.nextToken()
		}
		query.UnionModes = append(query.UnionModes, mode)

		var nextStmt ast.Statement
		if p.currentIs(token.LPAREN) {
			p.nextToken()
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			nextStmt = nested
		} else {
			nextSelect := p.parseSelect()
			if nextSelect == nil {
				break
			}
			nextStmt = nextSelect
		}
		query.Selects = append(query.Selects, nextStmt)
	}

	// Parse union-level SETTINGS and FORMAT
	var formatParsed bool
	for p.currentIs(token.SETTINGS) || p.currentIs(token.FORMAT) {
		if p.currentIs(token.SETTINGS) {
			p.nextToken()
			settings := p.parseSettingsList()
			query.Settings = settings
			if formatParsed {
				query.SettingsAfterFormat = true
			} else {
				query.SettingsBeforeFormat = true
			}
		} else if p.currentIs(token.FORMAT) {
			p.nextToken()
			formatParsed = true
			if len(query.Selects) > 0 {
				if sq, ok := query.Selects[0].(*ast.SelectQuery); ok {
					if p.currentIs(token.NULL) {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{"Null"}}
						p.nextToken()
					} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{p.current.Value}}
						p.nextToken()
					}
				}
			}
		}
	}

	return query
}

// parseSelectWithParsedWith parses a SELECT statement with an already-parsed WITH clause
func (p *Parser) parseSelectWithParsedWith(with []ast.Expression) *ast.SelectQuery {
	// Use the internal helper that does the actual parsing
	return p.parseSelectInternal(with)
}

// parseSelectWithUnion parses SELECT ... UNION/INTERSECT/EXCEPT ... queries
func (p *Parser) parseSelectWithUnion() *ast.SelectWithUnionQuery {
	query := &ast.SelectWithUnionQuery{
		Position: p.current.Pos,
	}

	// Parse first select item (could be parenthesized or direct SELECT)
	var firstItem ast.Statement
	var firstWasParenthesized bool

	if p.currentIs(token.LPAREN) {
		firstWasParenthesized = true
		p.nextToken() // skip (
		nested := p.parseSelectWithUnion()
		if nested == nil {
			return nil
		}
		p.expect(token.RPAREN)
		firstItem = nested
	} else {
		// Parse first SELECT
		sel := p.parseSelect()
		if sel == nil {
			return nil
		}
		firstItem = sel
	}

	// Check if this is INTERSECT/EXCEPT that needs SelectIntersectExceptQuery wrapper
	// Only INTERSECT ALL and EXCEPT ALL are treated like UNION ALL (flattened into ExpressionList)
	if p.isIntersectExceptWithWrapper() {
		// Collect all operands and operators first, then apply precedence
		// INTERSECT has higher precedence than EXCEPT

		// Start with first operand (statements list) and operators list
		stmts := []ast.Statement{firstItem}
		var ops []string

		// Parse all INTERSECT/EXCEPT clauses and collect them
		for p.isIntersectExceptWithWrapper() {
			// Record the operator type
			var op string
			if p.currentIs(token.EXCEPT) {
				op = "EXCEPT"
			} else {
				op = "INTERSECT"
			}
			p.nextToken() // skip INTERSECT/EXCEPT

			// Handle ALL or DISTINCT if present
			if p.currentIs(token.ALL) {
				op += " ALL"
				p.nextToken()
			} else if p.currentIs(token.DISTINCT) {
				op += " DISTINCT"
				p.nextToken()
			}
			ops = append(ops, op)

			// Parse the next operand
			// UNION has LOWER precedence than INTERSECT/EXCEPT, so don't consume UNION here
			var nextStmt ast.Statement
			if p.currentIs(token.LPAREN) {
				p.nextToken() // skip (
				nested := p.parseSelectWithUnion()
				if nested == nil {
					break
				}
				p.expect(token.RPAREN)
				nextStmt = nested
			} else {
				// Parse just a SELECT (don't consume UNION which has lower precedence)
				sel := p.parseSelect()
				if sel == nil {
					break
				}
				nextStmt = sel
			}
			stmts = append(stmts, nextStmt)
		}

		// Now apply precedence: INTERSECT binds tighter than EXCEPT
		result := buildIntersectExceptTree(stmts, ops)

		query.Selects = append(query.Selects, result)
		// Don't return yet - there might be a UNION ALL following the INTERSECT/EXCEPT chain
		// Fall through to the UNION parsing section
	} else {
		// Handle regular case (UNION, INTERSECT ALL, EXCEPT ALL, or single SELECT)
		if firstWasParenthesized {
			if nested, ok := firstItem.(*ast.SelectWithUnionQuery); ok {
				for _, s := range nested.Selects {
					query.Selects = append(query.Selects, s)
				}
			}
		} else {
			query.Selects = append(query.Selects, firstItem)
		}
	}

	// Parse UNION/INTERSECT ALL/EXCEPT ALL clauses
	for p.currentIs(token.UNION) || p.currentIs(token.EXCEPT) || p.currentIs(token.INTERSECT) {
		// Check if we hit INTERSECT/EXCEPT that should use wrapper (not ALL)
		// If so, we need to wrap the current UNION result as the first operand
		if p.isIntersectExceptWithWrapper() {
			// Wrap current query as first operand of INTERSECT/EXCEPT
			return p.parseIntersectExceptWithFirstOperand(query)
		}

		var setOp string
		if p.currentIs(token.UNION) {
			setOp = "UNION"
		} else if p.currentIs(token.EXCEPT) {
			setOp = "EXCEPT"
		} else {
			setOp = "INTERSECT"
		}
		p.nextToken() // skip UNION/INTERSECT/EXCEPT

		var mode string
		if p.currentIs(token.ALL) {
			query.UnionAll = true
			mode = "ALL"
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			mode = "DISTINCT"
			p.nextToken()
		}
		query.UnionModes = append(query.UnionModes, setOp+" "+mode)

		// Handle parenthesized subqueries
		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			// Keep parenthesized union as nested SelectWithUnionQuery
			// This allows proper grouping in the explain phase
			query.Selects = append(query.Selects, nested)
		} else {
			sel := p.parseSelect()
			if sel == nil {
				break
			}
			query.Selects = append(query.Selects, sel)
		}
	}

	// Parse union-level SETTINGS and FORMAT (for queries like SELECT ... SETTINGS ... SETTINGS ... FORMAT ...)
	// These come after the individual SELECTs have been parsed
	var formatParsed bool
	for p.currentIs(token.SETTINGS) || p.currentIs(token.FORMAT) {
		if p.currentIs(token.SETTINGS) {
			p.nextToken()
			settings := p.parseSettingsList()
			query.Settings = settings
			if formatParsed {
				query.SettingsAfterFormat = true
			} else {
				query.SettingsBeforeFormat = true
			}
		} else if p.currentIs(token.FORMAT) {
			p.nextToken()
			formatParsed = true
			// Get the format name and attach to first SELECT
			if len(query.Selects) > 0 {
				if sq, ok := query.Selects[0].(*ast.SelectQuery); ok {
					if p.currentIs(token.NULL) {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{"Null"}}
						p.nextToken()
					} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{p.current.Value}}
						p.nextToken()
					}
				}
			}
		}
	}

	return query
}

// parseIntersectExceptWithFirstOperand handles the case where UNION is followed by INTERSECT/EXCEPT
// Precedence: INTERSECT > UNION > EXCEPT
// So: "A UNION ALL B INTERSECT C" = "A UNION ALL (B INTERSECT C)" - pop last SELECT for INTERSECT
//     "A UNION ALL B EXCEPT C" = "(A UNION ALL B) EXCEPT C" - use entire UNION for EXCEPT
func (p *Parser) parseIntersectExceptWithFirstOperand(unionQuery *ast.SelectWithUnionQuery) *ast.SelectWithUnionQuery {
	// Check if we're starting with INTERSECT or EXCEPT to determine precedence behavior
	startsWithIntersect := p.currentIs(token.INTERSECT)

	var firstOperand ast.Statement
	if startsWithIntersect {
		// INTERSECT has higher precedence than UNION
		// Pop the last select from unionQuery to be the first operand of INTERSECT
		firstOperand = unionQuery.Selects[len(unionQuery.Selects)-1]
		unionQuery.Selects = unionQuery.Selects[:len(unionQuery.Selects)-1]
		// Also remove the last union mode if it exists
		if len(unionQuery.UnionModes) > 0 {
			unionQuery.UnionModes = unionQuery.UnionModes[:len(unionQuery.UnionModes)-1]
		}
	} else {
		// EXCEPT has lower precedence than UNION
		// Use the entire union as the first operand
		firstOperand = unionQuery
		// Create a new query to hold the result
		unionQuery = &ast.SelectWithUnionQuery{
			Position: unionQuery.Position,
		}
	}

	// Collect operands starting with the first operand
	stmts := []ast.Statement{firstOperand}
	var ops []string

	// Parse all INTERSECT/EXCEPT clauses
	for p.isIntersectExceptWithWrapper() {
		var op string
		if p.currentIs(token.EXCEPT) {
			op = "EXCEPT"
		} else {
			op = "INTERSECT"
		}
		p.nextToken() // skip INTERSECT/EXCEPT

		// Handle ALL or DISTINCT if present
		if p.currentIs(token.ALL) {
			op += " ALL"
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			op += " DISTINCT"
			p.nextToken()
		}
		ops = append(ops, op)

		// Parse the next select
		var nextStmt ast.Statement
		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			nextStmt = nested
		} else {
			sel := p.parseSelect()
			if sel == nil {
				break
			}
			nextStmt = sel
		}
		stmts = append(stmts, nextStmt)
	}

	// Build the tree with proper precedence
	result := buildIntersectExceptTree(stmts, ops)

	// Add the result to the union query
	unionQuery.Selects = append(unionQuery.Selects, result)

	// Continue parsing any UNION/UNION ALL that follows the INTERSECT/EXCEPT chain
	for p.currentIs(token.UNION) || p.currentIs(token.EXCEPT) || p.currentIs(token.INTERSECT) {
		// If we hit another INTERSECT/EXCEPT, we need to handle it recursively
		if p.isIntersectExceptWithWrapper() {
			return p.parseIntersectExceptWithFirstOperand(unionQuery)
		}

		p.nextToken() // skip UNION

		var mode string
		if p.currentIs(token.ALL) {
			unionQuery.UnionAll = true
			mode = "ALL"
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			mode = "DISTINCT"
			p.nextToken()
		}
		unionQuery.UnionModes = append(unionQuery.UnionModes, "UNION "+mode)

		// Handle parenthesized subqueries
		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			for _, s := range nested.Selects {
				unionQuery.Selects = append(unionQuery.Selects, s)
			}
		} else {
			sel := p.parseSelect()
			if sel == nil {
				break
			}
			unionQuery.Selects = append(unionQuery.Selects, sel)
		}
	}

	return unionQuery
}

// parseSelectWithUnionOnly parses SELECT with UNION/UNION ALL but stops at INTERSECT/EXCEPT.
// This is used for parsing operands in EXCEPT expressions where UNION has higher precedence.
func (p *Parser) parseSelectWithUnionOnly() ast.Statement {
	// Parse first SELECT
	sel := p.parseSelect()
	if sel == nil {
		return nil
	}

	// Check if followed by UNION (but not INTERSECT/EXCEPT which end this operand)
	if !p.currentIs(token.UNION) {
		return sel
	}

	// Build SelectWithUnionQuery for UNION/UNION ALL chain
	query := &ast.SelectWithUnionQuery{
		Position: sel.Pos(),
		Selects:  []ast.Statement{sel},
	}

	// Parse UNION/UNION ALL clauses
	for p.currentIs(token.UNION) {
		p.nextToken() // skip UNION

		var mode string
		if p.currentIs(token.ALL) {
			query.UnionAll = true
			mode = "ALL"
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			mode = "DISTINCT"
			p.nextToken()
		}
		query.UnionModes = append(query.UnionModes, "UNION "+mode)

		// Handle parenthesized subqueries
		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			for _, s := range nested.Selects {
				query.Selects = append(query.Selects, s)
			}
		} else {
			nextSel := p.parseSelect()
			if nextSel == nil {
				break
			}
			query.Selects = append(query.Selects, nextSel)
		}
	}

	return query
}

// isIntersectExceptWithWrapper checks if the current token is INTERSECT or EXCEPT
// that should use a SelectIntersectExceptQuery wrapper.
// All INTERSECT and EXCEPT variants (including ALL and DISTINCT) use the wrapper.
func (p *Parser) isIntersectExceptWithWrapper() bool {
	return p.currentIs(token.EXCEPT) || p.currentIs(token.INTERSECT)
}

// isIntersectOp checks if the operator is an INTERSECT variant (not EXCEPT)
func isIntersectOp(op string) bool {
	return strings.HasPrefix(op, "INTERSECT")
}

// buildIntersectExceptTree builds the AST tree respecting operator precedence.
// INTERSECT has higher precedence than EXCEPT, so:
// "a EXCEPT b INTERSECT c" becomes "a EXCEPT (b INTERSECT c)"
// "a INTERSECT b EXCEPT c" becomes "(a INTERSECT b) EXCEPT c"
//
// EXCEPT is left-associative and creates binary trees:
// "a EXCEPT b EXCEPT c" becomes "((a) EXCEPT b) EXCEPT c"
//
// stmts has n elements, ops has n-1 elements where ops[i] is the operator between stmts[i] and stmts[i+1]
func buildIntersectExceptTree(stmts []ast.Statement, ops []string) ast.Statement {
	if len(stmts) == 1 {
		return stmts[0]
	}

	// First pass: group consecutive INTERSECT operations (higher precedence)
	// Result will be a list of statements/groups connected by EXCEPT operators
	var groups []ast.Statement
	var exceptOps []string

	i := 0
	for i < len(stmts) {
		// Start a new group with stmts[i]
		groupStmts := []ast.Statement{stmts[i]}
		var groupOps []string

		// Collect consecutive INTERSECTs - look at the operator AFTER the current statement
		for i < len(ops) && isIntersectOp(ops[i]) {
			groupOps = append(groupOps, ops[i])
			i++
			groupStmts = append(groupStmts, stmts[i])
		}

		// Create the group
		var groupStmt ast.Statement
		if len(groupStmts) == 1 {
			// Single statement, no grouping needed
			groupStmt = groupStmts[0]
		} else {
			// Multiple statements connected by INTERSECT
			groupStmt = &ast.SelectIntersectExceptQuery{
				Selects:   groupStmts,
				Operators: groupOps,
			}
		}
		groups = append(groups, groupStmt)

		// Check if there's an EXCEPT connecting to the next group
		if i < len(ops) && !isIntersectOp(ops[i]) {
			exceptOps = append(exceptOps, ops[i])
			i++ // Move past the EXCEPT operator to start next group
		} else if i < len(stmts)-1 {
			// No more operators but still have stmts - shouldn't happen with valid input
			i++
		} else {
			// We've processed all statements
			break
		}
	}

	// Now all groups are connected by EXCEPT - build the final tree
	if len(groups) == 1 {
		return groups[0]
	}

	// Build left-associative binary tree for EXCEPT operations
	// "a EXCEPT b EXCEPT c" becomes "((a) EXCEPT b) EXCEPT c"
	result := groups[0]
	for j := 0; j < len(exceptOps); j++ {
		result = &ast.SelectIntersectExceptQuery{
			Selects:   []ast.Statement{result, groups[j+1]},
			Operators: []string{exceptOps[j]},
		}
	}
	return result
}

func (p *Parser) parseSelect() *ast.SelectQuery {
	return p.parseSelectInternal(nil)
}

// parseSelectInternal parses a SELECT query with an optional pre-parsed WITH clause
func (p *Parser) parseSelectInternal(preParsedWith []ast.Expression) *ast.SelectQuery {
	sel := &ast.SelectQuery{
		Position: p.current.Pos,
		With:     preParsedWith,
	}

	// Handle WITH clause only if not pre-parsed
	if preParsedWith == nil && p.currentIs(token.WITH) {
		p.nextToken()
		// Skip RECURSIVE keyword if present
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RECURSIVE" {
			p.nextToken()
		}
		sel.With = p.parseWithClause()
	}

	// Handle FROM ... SELECT syntax (ClickHouse extension)
	// This can come after WITH clause: WITH 1 as n FROM ... SELECT ...
	if p.currentIs(token.FROM) {
		p.nextToken() // skip FROM
		sel.From = p.parseTablesInSelect()
		// Now expect SELECT
		if !p.expect(token.SELECT) {
			return nil
		}
	} else if !p.expect(token.SELECT) {
		return nil
	}

	// Handle DISTINCT or ALL
	if p.currentIs(token.DISTINCT) {
		sel.Distinct = true
		p.nextToken()
		// Check for DISTINCT ON (col1, col2, ...)
		if p.currentIs(token.ON) {
			p.nextToken() // skip ON
			if p.expect(token.LPAREN) {
				sel.DistinctOn = p.parseExpressionList()
				p.expect(token.RPAREN)
			}
		}
	} else if p.currentIs(token.ALL) {
		// ALL is the default, just skip it
		p.nextToken()
	}

	// Handle TOP
	if p.currentIs(token.TOP) {
		p.nextToken()
		// Use MUL_PREC to stop at * (which would be parsed as column selector, not multiplication)
		sel.Top = p.parseExpression(MUL_PREC)
		// WITH TIES modifier after TOP
		if p.currentIs(token.WITH) && p.peekIs(token.TIES) {
			p.nextToken() // skip WITH
			p.nextToken() // skip TIES
		}
	}

	// Parse column list
	sel.Columns = p.parseExpressionList()

	// Parse FROM clause
	if p.currentIs(token.FROM) {
		p.nextToken()
		sel.From = p.parseTablesInSelect()
	}

	// Parse ARRAY JOIN clause
	if p.currentIs(token.ARRAY) || (p.currentIs(token.LEFT) && p.peekIs(token.ARRAY)) {
		sel.ArrayJoin = p.parseArrayJoin()
	}

	// Parse PREWHERE clause
	if p.currentIs(token.PREWHERE) {
		p.nextToken()
		sel.PreWhere = p.parseExpression(LOWEST)
	}

	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken()
		sel.Where = p.parseExpression(LOWEST)
	}

	// Parse GROUP BY clause
	if p.currentIs(token.GROUP) {
		p.nextToken()
		if !p.expect(token.BY) {
			return nil
		}

		// Handle GROUPING SETS, ROLLUP(...), CUBE(...) as special expressions
		if p.currentIs(token.GROUPING) && p.peekIs(token.SETS) {
			// GROUPING SETS ((a), (b), (a, b))
			p.nextToken() // skip GROUPING
			p.nextToken() // skip SETS
			sel.GroupBy = p.parseGroupingSets()
			sel.GroupingSets = true
		} else if p.currentIs(token.ROLLUP) && p.peekIs(token.LPAREN) {
			// ROLLUP(a, b, c)
			p.nextToken() // skip ROLLUP
			p.nextToken() // skip (
			sel.GroupBy = p.parseExpressionList()
			p.expect(token.RPAREN)
			sel.WithRollup = true
		} else if p.currentIs(token.CUBE) && p.peekIs(token.LPAREN) {
			// CUBE(a, b, c)
			p.nextToken() // skip CUBE
			p.nextToken() // skip (
			sel.GroupBy = p.parseExpressionList()
			p.expect(token.RPAREN)
			sel.WithCube = true
		} else if p.currentIs(token.ALL) {
			// GROUP BY ALL - special ClickHouse syntax
			sel.GroupByAll = true
			sel.GroupBy = p.parseExpressionList() // Still parse it, but mark as GroupByAll
		} else {
			sel.GroupBy = p.parseExpressionList()
		}

		// WITH ROLLUP
		if p.currentIs(token.WITH) && p.peekIs(token.ROLLUP) {
			p.nextToken()
			p.nextToken()
			sel.WithRollup = true
		}

		// WITH CUBE
		if p.currentIs(token.WITH) && p.peekIs(token.CUBE) {
			p.nextToken()
			p.nextToken()
			sel.WithCube = true
		}

		// WITH TOTALS
		if p.currentIs(token.WITH) && p.peekIs(token.TOTALS) {
			p.nextToken()
			p.nextToken()
			sel.WithTotals = true
		}
	}

	// Parse HAVING clause
	if p.currentIs(token.HAVING) {
		p.nextToken()
		sel.Having = p.parseExpression(LOWEST)
	}

	// Parse QUALIFY clause (window function filter)
	if p.currentIs(token.QUALIFY) {
		p.nextToken()
		sel.Qualify = p.parseExpression(LOWEST)
	}

	// Parse WINDOW clause for named windows
	if p.currentIs(token.WINDOW) {
		p.nextToken()
		sel.Window = p.parseWindowDefinitions()
	}

	// Parse ORDER BY clause
	if p.currentIs(token.ORDER) {
		p.nextToken()
		if !p.expect(token.BY) {
			return nil
		}
		sel.OrderBy = p.parseOrderByList()
	}

	// Parse INTERPOLATE clause (comes after ORDER BY ... WITH FILL)
	if p.currentIs(token.INTERPOLATE) {
		p.nextToken()
		sel.Interpolate = p.parseInterpolateList()
	}

	// Parse LIMIT clause
	if p.currentIs(token.LIMIT) {
		p.nextToken()
		sel.Limit = p.parseExpression(LOWEST)

		// LIMIT n, m syntax (offset, limit)
		if p.currentIs(token.COMMA) {
			p.nextToken()
			sel.Offset = sel.Limit
			sel.Limit = p.parseExpression(LOWEST)
		}

		// LIMIT BY clause (ClickHouse specific: LIMIT n BY expr1, expr2, ...)
		if p.currentIs(token.BY) {
			p.nextToken()
			// If we had comma syntax (LIMIT offset, count BY ...), save values for LIMIT BY
			// Otherwise just LIMIT n BY ... uses n as the count
			if sel.Offset != nil {
				// LIMIT offset, count BY ... -> LimitByOffset=offset, LimitByLimit=count
				sel.LimitByOffset = sel.Offset
				sel.LimitByLimit = sel.Limit
				sel.Offset = nil
				sel.Limit = nil
			} else {
				// LIMIT n BY ... -> LimitByLimit=n
				sel.LimitByLimit = sel.Limit
				sel.Limit = nil
			}
			// Parse LIMIT BY expressions
			for !p.isEndOfExpression() {
				expr := p.parseExpression(LOWEST)
				sel.LimitBy = append(sel.LimitBy, expr)
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			// After LIMIT BY, there can be another LIMIT for overall output
			if p.currentIs(token.LIMIT) {
				p.nextToken()
				sel.Limit = p.parseExpression(LOWEST)
				sel.LimitByHasLimit = true
				// Handle LIMIT offset, count syntax
				if p.currentIs(token.COMMA) {
					p.nextToken()
					sel.Offset = sel.Limit
					sel.Limit = p.parseExpression(LOWEST)
				}
			}
		}

		// WITH TIES modifier
		if p.currentIs(token.WITH) && p.peekIs(token.TIES) {
			p.nextToken() // skip WITH
			p.nextToken() // skip TIES
		}
	}

	// Parse OFFSET clause
	if p.currentIs(token.OFFSET) {
		p.nextToken()
		sel.Offset = p.parseExpression(LOWEST)
		// Skip optional ROW/ROWS keyword (SQL standard)
		if p.currentIs(token.IDENT) && (strings.ToUpper(p.current.Value) == "ROW" || strings.ToUpper(p.current.Value) == "ROWS") {
			p.nextToken()
		}
		// LIMIT n OFFSET m BY expr syntax - handle BY after OFFSET
		if p.currentIs(token.BY) && sel.Limit != nil && len(sel.LimitBy) == 0 {
			p.nextToken()
			// Move Limit and Offset to LimitByLimit and LimitByOffset
			sel.LimitByLimit = sel.Limit
			sel.LimitByOffset = sel.Offset
			sel.Limit = nil
			sel.Offset = nil
			// Parse LIMIT BY expressions
			for !p.isEndOfExpression() {
				expr := p.parseExpression(LOWEST)
				sel.LimitBy = append(sel.LimitBy, expr)
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
		}
	}

	// Parse FETCH FIRST ... ROW ONLY (SQL standard syntax)
	if p.currentIs(token.FETCH) {
		p.nextToken()
		// Skip FIRST or NEXT
		if p.currentIs(token.FIRST) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "NEXT") {
			p.nextToken()
		}
		// Parse the limit count
		if !p.currentIs(token.IDENT) || strings.ToUpper(p.current.Value) != "ROW" {
			sel.Limit = p.parseExpression(LOWEST)
		}
		// Skip ROW/ROWS
		if p.currentIs(token.IDENT) && (strings.ToUpper(p.current.Value) == "ROW" || strings.ToUpper(p.current.Value) == "ROWS") {
			p.nextToken()
		}
		// Skip ONLY
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ONLY" {
			p.nextToken()
		}
		// Skip WITH TIES
		if p.currentIs(token.WITH) {
			p.nextToken()
			if p.currentIs(token.TIES) {
				p.nextToken()
			}
		}
	}

	// Parse WITH TOTALS (can appear after GROUP BY or at end of SELECT)
	if p.currentIs(token.WITH) && p.peekIs(token.TOTALS) {
		p.nextToken()
		p.nextToken()
		sel.WithTotals = true
		// HAVING can follow WITH TOTALS
		if p.currentIs(token.HAVING) {
			p.nextToken()
			sel.Having = p.parseExpression(LOWEST)
		}
	}

	// Parse QUALIFY clause (window function filtering)
	if p.currentIs(token.QUALIFY) {
		p.nextToken()
		sel.Qualify = p.parseExpression(LOWEST)
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		sel.Settings = p.parseSettingsList()
	}

	// Parse INTO OUTFILE clause
	if p.currentIs(token.INTO) {
		p.nextToken()
		if p.currentIs(token.OUTFILE) {
			p.nextToken()
			if p.currentIs(token.STRING) {
				sel.IntoOutfile = &ast.IntoOutfileClause{
					Position: p.current.Pos,
					Filename: p.current.Value,
				}
				p.nextToken()
				// Parse optional TRUNCATE
				if p.currentIs(token.TRUNCATE) {
					sel.IntoOutfile.Truncate = true
					p.nextToken()
				}
			}
		}
	}

	// Only parse FORMAT if there were no SETTINGS before it
	// If SETTINGS was parsed above, FORMAT belongs at union level (parseSelectWithUnion will handle it)
	if p.currentIs(token.FORMAT) && len(sel.Settings) == 0 {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.currentIs(token.NULL) || p.current.Token.IsKeyword() {
			sel.Format = &ast.Identifier{
				Position: p.current.Pos,
				Parts:    []string{p.current.Value},
			}
			p.nextToken()
		}
		// Skip any inline data after FORMAT (e.g., FORMAT JSONEachRow {"x": 1}, {"y": 2})
		// This can happen in INSERT ... SELECT ... FORMAT ... statements
		for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.SETTINGS) {
			p.nextToken()
		}
		// Parse SETTINGS clause that comes AFTER FORMAT (belongs to this SELECT)
		if p.currentIs(token.SETTINGS) {
			p.nextToken()
			sel.Settings = p.parseSettingsList()
			sel.SettingsAfterFormat = true
		}
	}

	return sel
}

func (p *Parser) parseWithClause() []ast.Expression {
	var elements []ast.Expression

	for {
		elem := &ast.WithElement{
			Position: p.current.Pos,
		}

		// Check if it's the "name AS (subquery)" syntax (standard SQL CTE)
		// or "expr AS name" syntax (ClickHouse scalar)
		// Allow keywords as CTE names (e.g., WITH table AS (SELECT 1))
		// But exclude NULL/TRUE/FALSE which have special literal meanings
		isNameToken := p.currentIs(token.IDENT) ||
			(p.current.Token.IsKeyword() && !p.currentIs(token.NULL) && !p.currentIs(token.TRUE) && !p.currentIs(token.FALSE))
		if isNameToken && p.peekIs(token.AS) {
			// This could be "name AS (subquery)" or "ident AS alias" for scalar
			// Need to look ahead to determine: if IDENT AS LPAREN (SELECT...) -> CTE
			// If IDENT AS IDENT -> scalar WITH (first ident is expression, second is alias)
			name := p.current.Value
			pos := p.current.Pos
			p.nextToken() // skip identifier
			p.nextToken() // skip AS

			if p.currentIs(token.LPAREN) {
				// Could be CTE: name AS (subquery) OR could be name AS (expr)
				p.nextToken()
				if p.currentIs(token.SELECT) || p.currentIs(token.WITH) {
					// Standard CTE: name AS (SELECT...)
					subquery := p.parseSelectWithUnion()
					if !p.expect(token.RPAREN) {
						return nil
					}
					elem.Name = name
					elem.Query = &ast.Subquery{Query: subquery}
				} else {
					// It's an expression in parentheses, use name as alias
					// e.g., WITH x AS (1 + 2)
					expr := p.parseExpression(LOWEST)
					p.expect(token.RPAREN)
					elem.Name = name
					elem.Query = expr
				}
			} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				// Scalar: IDENT AS IDENT (e.g., WITH number AS k)
				// The first identifier is a column reference, second is the alias
				alias := p.current.Value
				p.nextToken()
				elem.Name = alias
				elem.Query = &ast.Identifier{Position: pos, Parts: []string{name}}
			} else {
				// Scalar expression where the first identifier is used directly
				// This is likely "name AS name" which means the CTE name is name with scalar value name
				elem.Name = name
				elem.Query = &ast.Identifier{Position: pos, Parts: []string{name}}
			}
		} else {
			// Scalar WITH: expr AS name (ClickHouse style)
			// This handles various forms:
			// - WITH 1 AS x, WITH 'hello' AS s, WITH func() AS f
			// - WITH (SELECT ...) AS name (subquery expression)
			// - WITH (SELECT ...) + (SELECT ...) AS name (binary expression of subqueries)
			// - WITH x -> toString(x) AS lambda_1 (lambda expression)
			// Arrow has OR_PREC precedence, so it gets parsed with ALIAS_PREC
			// Note: AS name is optional in ClickHouse, e.g., WITH 1 SELECT 1 is valid
			elem.ScalarWith = true
			elem.Query = p.parseExpression(ALIAS_PREC) // Use ALIAS_PREC to stop before AS

			// AS name is optional
			if p.currentIs(token.AS) {
				p.nextToken()
				// Alias can be IDENT or certain keywords (KEY, VALUES, etc.)
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					elem.Name = p.current.Value
					p.nextToken()
				}
			}
		}

		elements = append(elements, elem)

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	return elements
}

func (p *Parser) parseTablesInSelect() *ast.TablesInSelectQuery {
	tables := &ast.TablesInSelectQuery{
		Position: p.current.Pos,
	}

	// Parse first table
	elem := p.parseTableElement()
	if elem == nil {
		return nil
	}
	tables.Tables = append(tables.Tables, elem)

	// Parse JOINs
	for p.isJoinKeyword() {
		elem := p.parseTableElementWithJoin()
		if elem == nil {
			break
		}
		tables.Tables = append(tables.Tables, elem)
	}

	return tables
}

func (p *Parser) isJoinKeyword() bool {
	switch p.current.Token {
	case token.JOIN, token.INNER, token.LEFT, token.RIGHT, token.FULL, token.CROSS,
		token.GLOBAL, token.ANY, token.ALL, token.ASOF, token.SEMI, token.ANTI, token.PASTE,
		token.ARRAY:
		return true
	case token.COMMA:
		return true
	}
	return false
}

func (p *Parser) parseTableElement() *ast.TablesInSelectQueryElement {
	elem := &ast.TablesInSelectQueryElement{
		Position: p.current.Pos,
	}

	elem.Table = p.parseTableExpression()
	return elem
}

func (p *Parser) parseTableElementWithJoin() *ast.TablesInSelectQueryElement {
	elem := &ast.TablesInSelectQueryElement{
		Position: p.current.Pos,
	}

	// Handle comma join (implicit cross join)
	if p.currentIs(token.COMMA) {
		p.nextToken()
		elem.Table = p.parseTableExpression()
		// ClickHouse adds an empty TableJoin node for comma joins
		if elem.Table != nil {
			elem.Join = &ast.TableJoin{
				Position: elem.Position,
			}
		}
		return elem
	}

	// Handle ARRAY JOIN or LEFT ARRAY JOIN
	if p.currentIs(token.ARRAY) || (p.currentIs(token.LEFT) && p.peekIs(token.ARRAY)) {
		elem.ArrayJoin = p.parseArrayJoin()
		return elem
	}

	// Parse JOIN
	join := &ast.TableJoin{
		Position: p.current.Pos,
	}

	// Parse join modifiers
	if p.currentIs(token.GLOBAL) {
		join.Global = true
		p.nextToken()
	}

	// Parse strictness
	switch p.current.Token {
	case token.ANY:
		join.Strictness = ast.JoinStrictAny
		p.nextToken()
	case token.ALL:
		join.Strictness = ast.JoinStrictAll
		p.nextToken()
	case token.ASOF:
		join.Strictness = ast.JoinStrictAsof
		p.nextToken()
	case token.SEMI:
		join.Strictness = ast.JoinStrictSemi
		p.nextToken()
	case token.ANTI:
		join.Strictness = ast.JoinStrictAnti
		p.nextToken()
	}

	// Parse join type
	switch p.current.Token {
	case token.INNER:
		join.Type = ast.JoinInner
		p.nextToken()
	case token.LEFT:
		join.Type = ast.JoinLeft
		p.nextToken()
		if p.currentIs(token.OUTER) {
			p.nextToken()
		}
	case token.RIGHT:
		join.Type = ast.JoinRight
		p.nextToken()
		if p.currentIs(token.OUTER) {
			p.nextToken()
		}
	case token.FULL:
		join.Type = ast.JoinFull
		p.nextToken()
		if p.currentIs(token.OUTER) {
			p.nextToken()
		}
	case token.CROSS:
		join.Type = ast.JoinCross
		p.nextToken()
	case token.PASTE:
		join.Type = ast.JoinPaste
		p.nextToken()
	default:
		join.Type = ast.JoinInner
	}

	// Parse strictness after type if not already parsed (e.g., RIGHT ANTI JOIN)
	if join.Strictness == "" {
		switch p.current.Token {
		case token.ANY:
			join.Strictness = ast.JoinStrictAny
			p.nextToken()
		case token.ALL:
			join.Strictness = ast.JoinStrictAll
			p.nextToken()
		case token.ASOF:
			join.Strictness = ast.JoinStrictAsof
			p.nextToken()
		case token.SEMI:
			join.Strictness = ast.JoinStrictSemi
			p.nextToken()
		case token.ANTI:
			join.Strictness = ast.JoinStrictAnti
			p.nextToken()
		}
	}

	if !p.expect(token.JOIN) {
		return nil
	}

	elem.Table = p.parseTableExpression()

	// Parse ON or USING clause
	if p.currentIs(token.ON) {
		p.nextToken()
		join.On = p.parseExpression(LOWEST)
	} else if p.currentIs(token.USING) {
		p.nextToken()
		if p.currentIs(token.LPAREN) {
			p.nextToken()
			exprs := p.parseExpressionList()
			if exprs == nil {
				// Empty USING () - use empty non-nil slice to distinguish from no USING
				exprs = []ast.Expression{}
			}
			join.Using = exprs
			p.expect(token.RPAREN)
		} else {
			join.Using = p.parseExpressionList()
		}
	}

	elem.Join = join
	return elem
}

func (p *Parser) parseTableExpression() *ast.TableExpression {
	expr := &ast.TableExpression{
		Position: p.current.Pos,
	}

	// Handle subquery
	if p.currentIs(token.LPAREN) {
		p.nextToken()
		if p.currentIs(token.SELECT) || p.currentIs(token.WITH) || p.currentIs(token.LPAREN) {
			// SELECT, WITH, or nested (SELECT...) for UNION queries like ((SELECT 1) UNION ALL SELECT 2)
			subquery := p.parseSelectWithUnion()
			expr.Table = &ast.Subquery{Query: subquery}
		} else if p.currentIs(token.FROM) {
			// FROM ... SELECT (ClickHouse extension) - e.g., FROM (FROM numbers(1) SELECT *)
			subquery := p.parseFromSelectSyntax()
			expr.Table = &ast.Subquery{Query: subquery}
		} else if p.currentIs(token.EXPLAIN) {
			// EXPLAIN as subquery in FROM clause
			explain := p.parseExplain()
			expr.Table = &ast.Subquery{Query: explain}
		} else {
			// Table function or expression
			expr.Table = p.parseExpression(LOWEST)
		}
		p.expect(token.RPAREN)
	} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() || p.currentIs(token.NUMBER) {
		// Table identifier or function (keywords can be table names like "system")
		// Table names can also start with numbers in ClickHouse
		pos := p.current.Pos
		ident := p.parseIdentifierName()

		if p.currentIs(token.LPAREN) {
			// Table function
			expr.Table = p.parseFunctionCall(ident, pos)
		} else if p.currentIs(token.DOT) {
			// database.table
			p.nextToken()
			tableName := p.parseIdentifierName()
			expr.Table = &ast.TableIdentifier{
				Position: pos,
				Database: ident,
				Table:    tableName,
			}
		} else {
			expr.Table = &ast.TableIdentifier{
				Position: pos,
				Table:    ident,
			}
		}
	}

	// Handle alias (keywords like LEFT, RIGHT, FIRST can be used as aliases after AS,
	// or without AS if they're not clause keywords)
	// In ClickHouse, alias comes BEFORE FINAL: FROM table t FINAL
	if p.currentIs(token.AS) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			expr.Alias = p.current.Value
			p.nextToken()
		}
	} else if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && !p.isKeywordForClause() && !p.currentIs(token.FINAL) && !p.currentIs(token.SAMPLE) {
		// Don't consume PARALLEL as alias if followed by WITH (parallel query syntax)
		if p.currentIs(token.PARALLEL) && p.peekIs(token.WITH) {
			return expr
		}
		// Don't consume FINAL or SAMPLE as alias
		expr.Alias = p.current.Value
		p.nextToken()
	}

	// Handle FINAL (after alias)
	if p.currentIs(token.FINAL) {
		expr.Final = true
		p.nextToken()
	}

	// Handle SAMPLE
	if p.currentIs(token.SAMPLE) {
		p.nextToken()
		expr.Sample = &ast.SampleClause{
			Position: p.current.Pos,
			Ratio:    p.parseExpression(LOWEST),
		}
		if p.currentIs(token.OFFSET) {
			p.nextToken()
			expr.Sample.Offset = p.parseExpression(LOWEST)
		}
	}

	return expr
}

func (p *Parser) isKeywordForClause() bool {
	switch p.current.Token {
	case token.WHERE, token.GROUP, token.HAVING, token.QUALIFY, token.ORDER, token.LIMIT,
		token.OFFSET, token.UNION, token.EXCEPT, token.SETTINGS, token.FORMAT,
		token.PREWHERE, token.JOIN, token.LEFT, token.RIGHT, token.INNER,
		token.FULL, token.CROSS, token.PASTE, token.ON, token.USING, token.GLOBAL,
		token.ANY, token.ALL, token.SEMI, token.ANTI, token.ASOF, token.ARRAY,
		token.WINDOW, token.WITH, token.INTERSECT, token.SELECT:
		return true
	}
	// Handle TOTALS as a clause keyword when used in "WITH TOTALS"
	if p.current.Token == token.IDENT && strings.ToUpper(p.current.Value) == "TOTALS" {
		return true
	}
	return false
}

func (p *Parser) isEndOfExpression() bool {
	switch p.current.Token {
	case token.EOF, token.RPAREN, token.RBRACKET, token.SEMICOLON,
		token.UNION, token.EXCEPT, token.ORDER, token.LIMIT,
		token.OFFSET, token.SETTINGS, token.FORMAT, token.INTO,
		token.WITH:
		return true
	}
	return false
}

func (p *Parser) parseOrderByList() []*ast.OrderByElement {
	var elements []*ast.OrderByElement

	for {
		elem := &ast.OrderByElement{
			Position:   p.current.Pos,
			Expression: p.parseExpression(LOWEST),
		}

		// Handle ASC/DESC
		if p.currentIs(token.ASC) {
			p.nextToken()
		} else if p.currentIs(token.DESC) {
			elem.Descending = true
			p.nextToken()
		}

		// Handle NULLS FIRST/LAST
		if p.currentIs(token.NULLS) {
			p.nextToken()
			if p.currentIs(token.FIRST) {
				t := true
				elem.NullsFirst = &t
				p.nextToken()
			} else if p.currentIs(token.LAST) {
				f := false
				elem.NullsFirst = &f
				p.nextToken()
			}
		}

		// Handle COLLATE
		if p.currentIs(token.COLLATE) {
			p.nextToken()
			if p.currentIs(token.STRING) || p.currentIs(token.IDENT) {
				elem.Collate = p.current.Value
				p.nextToken()
			}
		}

		// Handle WITH FILL
		if p.currentIs(token.WITH) && p.peekIs(token.FILL) {
			p.nextToken() // skip WITH
			p.nextToken() // skip FILL
			elem.WithFill = true

			// Handle FROM
			if p.currentIs(token.FROM) {
				p.nextToken()
				elem.FillFrom = p.parseExpression(LOWEST)
			}

			// Handle TO
			if p.currentIs(token.TO) {
				p.nextToken()
				elem.FillTo = p.parseExpression(LOWEST)
			}

			// Handle STEP
			if p.currentIs(token.STEP) {
				p.nextToken()
				elem.FillStep = p.parseExpression(LOWEST)
			}

			// Handle STALENESS
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STALENESS" {
				p.nextToken()
				elem.FillStaleness = p.parseExpression(LOWEST)
			}
		}

		elements = append(elements, elem)

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	return elements
}

// parseInterpolateList parses INTERPOLATE (col1 AS expr1, col2, col3 AS expr3)
func (p *Parser) parseInterpolateList() []*ast.InterpolateElement {
	var elements []*ast.InterpolateElement

	// Expect opening parenthesis
	if !p.currentIs(token.LPAREN) {
		return elements
	}
	p.nextToken()

	for {
		if p.currentIs(token.RPAREN) {
			break
		}

		// Column name
		if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
			break
		}

		elem := &ast.InterpolateElement{
			Position: p.current.Pos,
			Column:   p.current.Value,
		}
		p.nextToken()

		// Optional AS expression
		if p.currentIs(token.AS) {
			p.nextToken()
			elem.Value = p.parseExpression(LOWEST)
		}

		elements = append(elements, elem)

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	// Expect closing parenthesis
	if p.currentIs(token.RPAREN) {
		p.nextToken()
	}

	return elements
}

func (p *Parser) parseSettingsList() []*ast.SettingExpr {
	var settings []*ast.SettingExpr

	for {
		// Setting names can be identifiers or keywords (like 'limit')
		if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
			break
		}

		setting := &ast.SettingExpr{
			Position: p.current.Pos,
			Name:     p.current.Value,
		}
		p.nextToken()

		// Settings can have optional value (bool settings can be just name)
		if p.currentIs(token.EQ) {
			p.nextToken()
			// Use ALIAS_PREC to stop before AS (for AS SELECT in CREATE TABLE AS SELECT)
			setting.Value = p.parseExpression(ALIAS_PREC)
		} else {
			// Boolean setting without value - defaults to true
			setting.Value = &ast.Literal{
				Position: setting.Position,
				Type:     ast.LiteralBoolean,
				Value:    true,
			}
		}
		settings = append(settings, setting)

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	return settings
}

func (p *Parser) parseInsert() *ast.InsertQuery {
	ins := &ast.InsertQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip INSERT

	if !p.expect(token.INTO) {
		return nil
	}

	// Skip optional TABLE keyword
	if p.currentIs(token.TABLE) {
		p.nextToken()
	}

	// Handle INSERT INTO FUNCTION
	if p.currentIs(token.FUNCTION) {
		p.nextToken()
		// Parse the function call
		funcName := p.parseIdentifierName()
		if funcName != "" && p.currentIs(token.LPAREN) {
			ins.Function = p.parseFunctionCall(funcName, p.current.Pos)
		}
	} else {
		// Parse table name (can start with a number in ClickHouse)
		tableName := p.parseIdentifierName()
		if tableName != "" {
			if p.currentIs(token.DOT) {
				p.nextToken()
				ins.Database = tableName
				ins.Table = p.parseIdentifierName()
			} else {
				ins.Table = tableName
			}
		}
	}

	// Parse column list
	if p.currentIs(token.LPAREN) {
		p.nextToken()
		// Check for special column expressions (*, table.*, COLUMNS(...), with EXCEPT/APPLY/REPLACE)
		if p.currentIs(token.ASTERISK) || p.isColumnsFunction() ||
			(p.currentIs(token.IDENT) && p.peekIs(token.DOT) && p.peekPeekIs(token.ASTERISK)) {
			// Parse as expression to handle EXCEPT/APPLY/REPLACE transformers
			expr := p.parseExpression(LOWEST)
			if expr != nil {
				ins.ColumnExpressions = append(ins.ColumnExpressions, expr)
			}
			// Handle comma-separated expressions
			for p.currentIs(token.COMMA) {
				p.nextToken()
				expr = p.parseExpression(LOWEST)
				if expr != nil {
					ins.ColumnExpressions = append(ins.ColumnExpressions, expr)
				}
			}
		} else {
			// Regular column names
			for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
				pos := p.current.Pos
				colName := p.parseIdentifierName()
				if colName != "" {
					// Handle dotted column names like ip4Map.value (for nested columns)
					for p.currentIs(token.DOT) {
						p.nextToken()
						nextPart := p.parseIdentifierName()
						if nextPart != "" {
							colName = colName + "." + nextPart
						}
					}
					ins.Columns = append(ins.Columns, &ast.Identifier{
						Position: pos,
						Parts:    []string{colName},
					})
				}
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
		}
		p.expect(token.RPAREN)
	}

	// Parse PARTITION BY (for INSERT INTO FUNCTION)
	if p.currentIs(token.PARTITION) {
		p.nextToken()
		if p.currentIs(token.BY) {
			p.nextToken()
			ins.PartitionBy = p.parseExpression(LOWEST)
		}
	}

	// Parse SETTINGS before VALUES
	if p.currentIs(token.SETTINGS) {
		ins.HasSettings = true
		p.nextToken()
		// Just parse and skip the settings
		p.parseSettingsList()
	}

	// Parse FROM INFILE clause (for INSERT ... FROM INFILE '...' COMPRESSION 'gz')
	if p.currentIs(token.FROM) {
		p.nextToken()
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "INFILE" {
			p.nextToken()
			// Store the file path
			if p.currentIs(token.STRING) {
				ins.Infile = p.current.Value
				p.nextToken()
			}
			// Handle COMPRESSION clause
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "COMPRESSION" {
				p.nextToken()
				if p.currentIs(token.STRING) {
					ins.Compression = p.current.Value
					p.nextToken()
				}
			}
		}
	}

	// Parse VALUES or SELECT
	if p.currentIs(token.VALUES) {
		p.nextToken()
		// Parse VALUES rows: (expr, expr, ...), (expr, expr, ...), ...
		for {
			if !p.currentIs(token.LPAREN) {
				break
			}
			p.nextToken() // skip (
			var row []ast.Expression
			for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
				expr := p.parseExpression(LOWEST)
				if expr != nil {
					row = append(row, expr)
				}
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			if p.currentIs(token.RPAREN) {
				p.nextToken() // skip )
			}
			ins.Values = append(ins.Values, row)
			// Check for more rows
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
	} else if p.currentIs(token.SELECT) || p.currentIs(token.WITH) {
		ins.Select = p.parseSelectWithUnion()
		// If the SELECT has settings, mark the INSERT as having settings too
		if ins.Select != nil {
			if sel, ok := ins.Select.(*ast.SelectWithUnionQuery); ok && sel != nil && len(sel.Selects) > 0 {
				lastSel := sel.Selects[len(sel.Selects)-1]
				if lastSel != nil {
					if selQuery, ok := lastSel.(*ast.SelectQuery); ok && selQuery != nil && len(selQuery.Settings) > 0 {
						ins.HasSettings = true
					}
				}
			}
		}
	}

	// Parse FORMAT (format names can be keywords like Null, JSON, etc.)
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.currentIs(token.NULL) || p.current.Token.IsKeyword() {
			ins.Format = &ast.Identifier{
				Position: p.current.Pos,
				Parts:    []string{p.current.Value},
			}
			p.nextToken()
		}
		// Skip any inline data after FORMAT (e.g., FORMAT JSONEachRow {"x": 1}, {"y": 2})
		// The data is raw and should not be parsed as SQL
		for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
			p.nextToken()
		}
	}

	return ins
}

func (p *Parser) parseCreate() ast.Statement {
	pos := p.current.Pos
	p.nextToken() // skip CREATE

	// Handle CREATE [UNIQUE] INDEX
	if p.currentIs(token.INDEX) {
		return p.parseCreateIndex(pos)
	}
	// Handle CREATE UNIQUE INDEX
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UNIQUE" {
		p.nextToken() // skip UNIQUE
		if p.currentIs(token.INDEX) {
			return p.parseCreateIndex(pos)
		}
	}

	create := &ast.CreateQuery{
		Position: pos,
	}

	// Handle OR REPLACE
	if p.currentIs(token.OR) {
		p.nextToken()
		if p.currentIs(token.REPLACE) {
			create.OrReplace = true
			p.nextToken()
		}
	}

	// Handle TEMPORARY
	if p.currentIs(token.TEMPORARY) {
		create.Temporary = true
		p.nextToken()
	}

	// Handle MATERIALIZED
	if p.currentIs(token.MATERIALIZED) {
		create.Materialized = true
		p.nextToken()
	}

	// Handle WINDOW (for WINDOW VIEW)
	if p.currentIs(token.WINDOW) {
		create.WindowView = true
		p.nextToken()
	}

	// What are we creating?
	switch p.current.Token {
	case token.TABLE:
		p.nextToken()
		p.parseCreateTable(create)
	case token.DATABASE:
		create.CreateDatabase = true
		p.nextToken()
		p.parseCreateDatabase(create)
	case token.VIEW:
		p.nextToken()
		p.parseCreateView(create)
	case token.FUNCTION:
		// CREATE FUNCTION name AS lambda_expr
		create.CreateFunction = true
		p.nextToken()
		p.parseCreateFunction(create)
	case token.USER:
		// CREATE USER name ...
		create.CreateUser = true
		p.nextToken()
		p.parseCreateUser(create)
	case token.SETTINGS:
		// CREATE SETTINGS PROFILE
		return p.parseCreateSettingsProfile(pos)
	case token.IDENT:
		// Handle CREATE DICTIONARY, CREATE RESOURCE, CREATE WORKLOAD, CREATE NAMED COLLECTION, etc.
		identUpper := strings.ToUpper(p.current.Value)
		switch identUpper {
		case "DICTIONARY":
			create.CreateDictionary = true
			p.nextToken()
			p.parseCreateDictionary(create)
		case "NAMED":
			// CREATE NAMED COLLECTION name AS key=value, ...
			return p.parseCreateNamedCollection(pos)
		case "PROFILE":
			// CREATE PROFILE (without SETTINGS keyword)
			return p.parseCreateSettingsProfile(pos)
		case "ROW":
			// CREATE ROW POLICY
			return p.parseCreateRowPolicy(pos)
		case "POLICY":
			// CREATE POLICY (without ROW keyword)
			return p.parseCreateRowPolicy(pos)
		case "ROLE":
			// CREATE ROLE
			return p.parseCreateRole(pos)
		case "RESOURCE":
			// CREATE RESOURCE
			return p.parseCreateResource(pos)
		case "WORKLOAD":
			// CREATE WORKLOAD
			return p.parseCreateWorkload(pos)
		case "QUOTA":
			// CREATE QUOTA
			return p.parseCreateQuota(pos)
		default:
			p.errors = append(p.errors, fmt.Errorf("expected TABLE, DATABASE, VIEW, FUNCTION, USER after CREATE"))
			return nil
		}
	default:
		p.errors = append(p.errors, fmt.Errorf("expected TABLE, DATABASE, VIEW, FUNCTION, USER after CREATE"))
		return nil
	}

	// Handle FORMAT clause (for things like CREATE TABLE ... FORMAT Null)
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		// Store format name (Null, etc.)
		if p.currentIs(token.NULL) {
			create.Format = "Null"
			p.nextToken()
		} else if p.currentIs(token.IDENT) {
			create.Format = p.current.Value
			p.nextToken()
		}
	}

	return create
}

// parseReplace handles REPLACE TABLE/DICTIONARY syntax, which is equivalent to CREATE OR REPLACE
func (p *Parser) parseReplace() ast.Statement {
	pos := p.current.Pos
	p.nextToken() // skip REPLACE

	// REPLACE TABLE name ...
	if p.currentIs(token.TABLE) {
		p.nextToken() // skip TABLE

		create := &ast.CreateQuery{
			Position:  pos,
			OrReplace: true, // REPLACE TABLE implies OR REPLACE
		}

		p.parseCreateTable(create)
		return create
	}

	// REPLACE DICTIONARY name ...
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		p.nextToken() // skip DICTIONARY

		create := &ast.CreateQuery{
			Position:         pos,
			OrReplace:        true,
			CreateDictionary: true,
		}

		p.parseCreateDictionary(create)
		return create
	}

	return nil
}

func (p *Parser) parseCreateIndex(pos token.Position) *ast.CreateIndexQuery {
	p.nextToken() // skip INDEX

	query := &ast.CreateIndexQuery{
		Position: pos,
	}

	// Skip IF NOT EXISTS if present (comes before index name)
	if p.currentIs(token.IF) {
		p.nextToken() // IF
		if p.currentIs(token.NOT) {
			p.nextToken() // NOT
		}
		if p.currentIs(token.EXISTS) {
			p.nextToken() // EXISTS
		}
	}

	// Parse index name
	query.IndexName = p.parseIdentifierName()

	// Expect ON
	if p.currentIs(token.ON) {
		p.nextToken()
	}

	// Parse table name
	query.Table = p.parseIdentifierName()
	if p.currentIs(token.DOT) {
		p.nextToken()
		query.Table = p.parseIdentifierName()
	}

	// Parse column expression - can be in parentheses or directly after table name
	if p.currentIs(token.LPAREN) {
		query.ColumnsParenthesized = true
		p.nextToken() // skip (

		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			col := p.parseExpression(0)
			query.Columns = append(query.Columns, col)

			// Skip ASC/DESC modifiers
			if p.currentIs(token.ASC) || p.currentIs(token.DESC) {
				p.nextToken()
			}

			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}

		if p.currentIs(token.RPAREN) {
			p.nextToken() // skip )
		}
	} else if !p.currentIs(token.SEMICOLON) && !p.currentIs(token.EOF) &&
		!(p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE") {
		// Expression directly after table name without parentheses
		// e.g., CREATE INDEX idx ON tbl date(ts) TYPE MinMax
		col := p.parseExpression(0)
		query.Columns = append(query.Columns, col)
	}

	// Parse TYPE clause
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
		p.nextToken() // skip TYPE
		query.Type = p.parseIdentifierName()
	}

	// Parse GRANULARITY clause
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "GRANULARITY" {
		p.nextToken() // skip GRANULARITY
		if p.currentIs(token.NUMBER) {
			val, _ := strconv.Atoi(p.current.Value)
			query.Granularity = val
			p.nextToken()
		}
	}

	return query
}

func (p *Parser) parseCreateTable(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse table name (can start with a number in ClickHouse)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			create.Database = tableName
			create.Table = p.parseIdentifierName()
		} else {
			create.Table = tableName
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle CLONE AS source_table
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CLONE" {
		p.nextToken() // skip CLONE
		if p.currentIs(token.AS) {
			p.nextToken() // skip AS
			create.CloneAs = p.parseIdentifierName()
		}
	}

	// Handle UUID clause (CREATE TABLE name UUID 'uuid-value' ...)
	// The UUID is not shown in EXPLAIN AST output, but we need to skip it
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UUID" {
		p.nextToken() // skip UUID
		if p.currentIs(token.STRING) {
			p.nextToken() // skip the UUID value
		}
	}

	// Parse column definitions and indexes
	if p.currentIs(token.LPAREN) {
		p.nextToken()
		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			// Handle INDEX definition
			if p.currentIs(token.INDEX) {
				idx := p.parseIndexDefinition()
				if idx != nil {
					create.Indexes = append(create.Indexes, idx)
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
				// Parse PROJECTION definitions: PROJECTION name (SELECT ...)
				p.nextToken() // skip PROJECTION
				proj := p.parseProjection()
				if proj != nil {
					create.Projections = append(create.Projections, proj)
				}
			} else if p.currentIs(token.CONSTRAINT) {
				// Parse CONSTRAINT name CHECK/ASSUME (expression)
				p.nextToken() // skip CONSTRAINT
				constraintName := p.parseIdentifierName() // constraint name
				if p.currentIs(token.CHECK) {
					p.nextToken() // skip CHECK
					constraint := &ast.Constraint{
						Position:   p.current.Pos,
						Name:       constraintName,
						Expression: p.parseExpression(LOWEST),
					}
					create.Constraints = append(create.Constraints, constraint)
				} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ASSUME" {
					p.nextToken() // skip ASSUME
					constraint := &ast.Constraint{
						Position:   p.current.Pos,
						Name:       constraintName,
						Expression: p.parseExpression(LOWEST),
					}
					create.Constraints = append(create.Constraints, constraint)
				} else {
					// Skip other constraint types we don't know about
					for !p.currentIs(token.COMMA) && !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
						p.nextToken()
					}
				}
			} else if p.currentIs(token.PRIMARY) {
				// Handle PRIMARY KEY as table constraint: PRIMARY KEY (col1, col2) or PRIMARY KEY col
				p.nextToken() // skip PRIMARY
				if p.currentIs(token.KEY) {
					p.nextToken() // skip KEY
				}
				// Parse the primary key column(s) into create.PrimaryKey
				if p.currentIs(token.LPAREN) {
					p.nextToken() // skip (
					for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
						expr := p.parseExpression(LOWEST)
						if expr != nil {
							create.ColumnsPrimaryKey = append(create.ColumnsPrimaryKey, expr)
						}
						if p.currentIs(token.COMMA) {
							p.nextToken()
						} else {
							break
						}
					}
					p.expect(token.RPAREN)
					// If no columns were added, mark that we saw an empty PRIMARY KEY ()
					if len(create.ColumnsPrimaryKey) == 0 {
						create.HasEmptyColumnsPrimaryKey = true
					}
				} else {
					// Single column: PRIMARY KEY col
					expr := p.parseExpression(LOWEST)
					if expr != nil {
						create.ColumnsPrimaryKey = append(create.ColumnsPrimaryKey, expr)
					}
				}
			} else {
				col := p.parseColumnDeclaration()
				if col != nil {
					create.Columns = append(create.Columns, col)
				}
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
		p.expect(token.RPAREN)
	}

	// Parse ENGINE
	if p.currentIs(token.ENGINE) {
		p.nextToken()
		if p.currentIs(token.EQ) {
			p.nextToken()
		}
		create.Engine = p.parseEngineClause()
	}

	// Parse table options in flexible order (PARTITION BY, ORDER BY, PRIMARY KEY, etc.)
	p.parseTableOptions(create)

	// Parse AS SELECT or AS (subquery) or AS table_function() or AS database.table
	if p.currentIs(token.AS) {
		p.nextToken()
		if p.currentIs(token.SELECT) || p.currentIs(token.WITH) || p.currentIs(token.LPAREN) {
			// AS SELECT... or AS (SELECT...) INTERSECT ...
			create.AsSelect = p.parseSelectWithUnion()
		} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			// AS table_function(...) or AS database.table
			name := p.parseIdentifierName()
			if p.currentIs(token.DOT) {
				// AS database.table - skip the table name
				p.nextToken()
				p.parseIdentifierName()
			} else if p.currentIs(token.LPAREN) {
				// AS function(...) - parse as a function call
				fn := &ast.FunctionCall{Name: name}
				p.nextToken() // skip (
				if !p.currentIs(token.RPAREN) {
					fn.Arguments = p.parseExpressionList()
				}
				if p.currentIs(token.RPAREN) {
					p.nextToken()
				}
				create.AsTableFunction = fn
			}
			_ = name // Use name for future AS table support
		}
	}

	// Parse ENGINE after AS (for CREATE TABLE x AS y ENGINE=z syntax)
	if create.Engine == nil && p.currentIs(token.ENGINE) {
		p.nextToken()
		if p.currentIs(token.EQ) {
			p.nextToken()
		}
		create.Engine = p.parseEngineClause()
	}

	// Parse table options after AS ... ENGINE (PARTITION BY, ORDER BY, etc.)
	p.parseTableOptions(create)
}

// parseTableOptions parses table options: PARTITION BY, ORDER BY, PRIMARY KEY, SAMPLE BY, TTL, SETTINGS
func (p *Parser) parseTableOptions(create *ast.CreateQuery) {
	for {
		switch {
		case p.currentIs(token.PARTITION):
			p.nextToken()
			if p.expect(token.BY) {
				// Use ALIAS_PREC to avoid consuming AS keyword (for AS SELECT)
				create.PartitionBy = p.parseExpression(ALIAS_PREC)
			}
		case p.currentIs(token.ORDER):
			p.nextToken()
			if p.expect(token.BY) {
				if p.currentIs(token.LPAREN) {
					pos := p.current.Pos
					p.nextToken()
					exprs, hasModifier := p.parseCreateOrderByExpressions()
					p.expect(token.RPAREN)
					// Track if any ASC/DESC modifiers were present
					create.OrderByHasModifiers = hasModifier
					// Store tuple literal for ORDER BY with multiple exprs, empty tuple, or any with ASC/DESC modifiers
					if len(exprs) == 0 || len(exprs) > 1 || hasModifier {
						create.OrderBy = []ast.Expression{&ast.Literal{
							Position: pos,
							Type:     ast.LiteralTuple,
							Value:    exprs,
						}}
					} else {
						// Single expression in parentheses without modifiers
						// Check if there's a binary operator continuing the expression (e.g., (a + b) * c)
						expr := exprs[0]
						if isBinaryOperatorToken(p.current.Token) {
							// Mark the expression as parenthesized and continue parsing
							if binExpr, ok := expr.(*ast.BinaryExpr); ok {
								binExpr.Parenthesized = true
							}
							// Continue parsing from this expression as left operand
							expr = p.parseExpressionFrom(expr, LOWEST)
						}
						create.OrderBy = []ast.Expression{expr}
					}
				} else {
					// Use ALIAS_PREC to avoid consuming AS keyword (for AS SELECT)
					expr := p.parseExpression(ALIAS_PREC)
					create.OrderBy = []ast.Expression{expr}
					// Handle ASC/DESC modifier after single non-parenthesized ORDER BY expression
					if p.currentIs(token.ASC) || p.currentIs(token.DESC) {
						create.OrderByHasModifiers = true
						p.nextToken()
					}
				}
			}
		case p.currentIs(token.PRIMARY):
			p.nextToken()
			if p.expect(token.KEY) {
				if p.currentIs(token.LPAREN) {
					pos := p.current.Pos
					p.nextToken()
					exprs := p.parseExpressionList()
					p.expect(token.RPAREN)
					// Store tuple literal for PRIMARY KEY (expr1, expr2, ...) or PRIMARY KEY ()
					if len(exprs) == 0 || len(exprs) > 1 {
						create.PrimaryKey = []ast.Expression{&ast.Literal{
							Position: pos,
							Type:     ast.LiteralTuple,
							Value:    exprs,
						}}
					} else {
						// Single expression in parentheses - just extract it
						create.PrimaryKey = exprs
					}
				} else {
					// Use ALIAS_PREC to avoid consuming AS keyword (for AS SELECT)
					create.PrimaryKey = []ast.Expression{p.parseExpression(ALIAS_PREC)}
				}
			}
		case p.currentIs(token.SAMPLE):
			p.nextToken()
			if p.expect(token.BY) {
				// Use ALIAS_PREC to avoid consuming AS keyword (for AS SELECT)
				create.SampleBy = p.parseExpression(ALIAS_PREC)
			}
		case p.currentIs(token.TTL):
			p.nextToken()
			create.TTL = &ast.TTLClause{
				Position: p.current.Pos,
			}
			// Parse TTL elements (comma-separated)
			for {
				elem := p.parseTTLElement()
				create.TTL.Elements = append(create.TTL.Elements, elem)
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			// Keep backward compatibility with Expression/Expressions fields
			if len(create.TTL.Elements) > 0 {
				create.TTL.Expression = create.TTL.Elements[0].Expr
				for i := 1; i < len(create.TTL.Elements); i++ {
					create.TTL.Expressions = append(create.TTL.Expressions, create.TTL.Elements[i].Expr)
				}
			}
		case p.currentIs(token.SETTINGS):
			// Track if SETTINGS comes before COMMENT
			if create.Comment == "" && len(create.Settings) == 0 {
				create.SettingsBeforeComment = true
			}
			p.nextToken()
			settings := p.parseSettingsList()
			// If Settings is already set, this is a second SETTINGS clause (query-level)
			if len(create.Settings) > 0 {
				create.QuerySettings = settings
			} else {
				create.Settings = settings
			}
		case p.currentIs(token.COMMENT):
			p.nextToken()
			if p.currentIs(token.STRING) {
				create.Comment = p.current.Value
				p.nextToken()
			}
			// If we see COMMENT but Settings wasn't set yet, clear the flag
			if len(create.Settings) == 0 {
				create.SettingsBeforeComment = false
			}
		default:
			return
		}
	}
}

func (p *Parser) parseCreateDatabase(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse database name (can start with a number in ClickHouse)
	create.Database = p.parseIdentifierName()

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse ENGINE
	if p.currentIs(token.ENGINE) {
		p.nextToken()
		if p.currentIs(token.EQ) {
			p.nextToken()
		}
		create.Engine = p.parseEngineClause()
	}

	// Handle ORDER BY clause (ClickHouse allows ORDER BY in CREATE DATABASE)
	// This is stored as OrderBy for output in Storage definition
	if p.currentIs(token.ORDER) {
		p.nextToken() // skip ORDER
		if p.currentIs(token.BY) {
			p.nextToken() // skip BY
		}
		create.OrderBy = []ast.Expression{p.parseExpression(LOWEST)}
	}

	// Handle SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		create.Settings = p.parseSettingsList()
	}
}

func (p *Parser) parseCreateView(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse view name (can start with a number in ClickHouse)
	viewName := p.parseIdentifierName()
	if viewName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			create.Database = viewName
			create.View = p.parseIdentifierName()
		} else {
			create.View = viewName
		}
	}

	// Handle UUID clause (CREATE MATERIALIZED VIEW name UUID 'uuid-value' ...)
	// The UUID is not shown in EXPLAIN AST output, but we need to skip it
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UUID" {
		p.nextToken() // skip UUID
		if p.currentIs(token.STRING) {
			p.nextToken() // skip the UUID value
		}
	}

	// Handle ON CLUSTER (can appear before or after column definitions)
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle REFRESH clause for materialized views (REFRESH AFTER/EVERY interval APPEND TO target)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "REFRESH" {
		p.nextToken() // skip REFRESH
		create.HasRefresh = true

		// Parse refresh timing: AFTER interval or EVERY interval
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			if upper == "AFTER" || upper == "EVERY" {
				create.RefreshType = upper
				p.nextToken()
				// Parse interval value and unit
				create.RefreshInterval = p.parseExpression(AND_PREC)
				// Parse interval unit if present as identifier
				if p.currentIs(token.IDENT) {
					unitUpper := strings.ToUpper(p.current.Value)
					if unitUpper == "SECOND" || unitUpper == "MINUTE" || unitUpper == "HOUR" ||
						unitUpper == "DAY" || unitUpper == "WEEK" || unitUpper == "MONTH" || unitUpper == "YEAR" {
						create.RefreshUnit = unitUpper
						p.nextToken()
					}
				}
			}
		}

		// Handle APPEND TO target - different from regular TO, part of REFRESH strategy
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "APPEND" {
			p.nextToken() // skip APPEND
			create.RefreshAppend = true
			if p.currentIs(token.TO) {
				p.nextToken() // skip TO
				toName := p.parseIdentifierName()
				if p.currentIs(token.DOT) {
					p.nextToken()
					create.ToDatabase = toName
					create.To = p.parseIdentifierName()
				} else {
					create.To = toName
				}
			}
		}

		// For REFRESH ... APPEND TO target (columns), column definitions come after
		if p.currentIs(token.LPAREN) && len(create.Columns) == 0 {
			p.nextToken()
			for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
				col := p.parseColumnDeclaration()
				if col != nil {
					create.Columns = append(create.Columns, col)
				}
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			p.expect(token.RPAREN)
		}

		// Handle EMPTY keyword
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "EMPTY" {
			create.Empty = true
			p.nextToken()
		}
	}

	// Parse column definitions (e.g., CREATE VIEW v (x UInt64) AS SELECT ...)
	// For MATERIALIZED VIEW, this can also include INDEX, PROJECTION, and PRIMARY KEY
	if p.currentIs(token.LPAREN) {
		p.nextToken()
		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			// Handle INDEX definition
			if p.currentIs(token.INDEX) {
				idx := p.parseIndexDefinition()
				if idx != nil {
					create.Indexes = append(create.Indexes, idx)
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
				// Parse PROJECTION definitions: PROJECTION name (SELECT ...)
				p.nextToken() // skip PROJECTION
				proj := p.parseProjection()
				if proj != nil {
					create.Projections = append(create.Projections, proj)
				}
			} else if p.currentIs(token.PRIMARY) {
				// PRIMARY KEY in column list
				p.nextToken() // skip PRIMARY
				if p.currentIs(token.KEY) {
					p.nextToken() // skip KEY
					expr := p.parseExpression(LOWEST)
					if expr != nil {
						create.ColumnsPrimaryKey = append(create.ColumnsPrimaryKey, expr)
					}
				}
			} else {
				col := p.parseColumnDeclaration()
				if col != nil {
					create.Columns = append(create.Columns, col)
				}
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
		p.expect(token.RPAREN)
	}

	// Handle ON CLUSTER (if it appears after column definitions instead of before)
	if create.OnCluster == "" && p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle TO (target table for materialized views and window views)
	// TO clause is not valid for regular views - only for MATERIALIZED VIEW or WINDOW VIEW
	if p.currentIs(token.TO) {
		if !create.Materialized && !create.WindowView {
			p.errors = append(p.errors, fmt.Errorf("TO clause is only valid for MATERIALIZED VIEW or WINDOW VIEW, not VIEW"))
			return
		}
		p.nextToken()
		toName := p.parseIdentifierName()
		if p.currentIs(token.DOT) {
			p.nextToken()
			create.ToDatabase = toName
			create.To = p.parseIdentifierName()
		} else {
			create.To = toName
		}

		// For MATERIALIZED VIEW ... TO target (columns) syntax,
		// column definitions can come after the TO target
		if p.currentIs(token.LPAREN) && len(create.Columns) == 0 {
			p.nextToken()
			for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
				col := p.parseColumnDeclaration()
				if col != nil {
					create.Columns = append(create.Columns, col)
				}
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			p.expect(token.RPAREN)
		}
	}

	// Parse INNER ENGINE (for window views) - comes before regular ENGINE
	if p.currentIs(token.INNER) {
		p.nextToken() // skip INNER
		if p.currentIs(token.ENGINE) {
			p.nextToken() // skip ENGINE
			if p.currentIs(token.EQ) {
				p.nextToken()
			}
			create.InnerEngine = p.parseEngineClause()
		}
	}

	// Parse ENGINE (for materialized views)
	if p.currentIs(token.ENGINE) {
		p.nextToken()
		if p.currentIs(token.EQ) {
			p.nextToken()
		}
		create.Engine = p.parseEngineClause()
	}

	// Parse table options (ORDER BY, PRIMARY KEY, etc.) for materialized views
	p.parseTableOptions(create)

	// Parse POPULATE (for materialized views)
	if p.currentIs(token.POPULATE) {
		create.Populate = true
		p.nextToken()
	}

	// Parse AS SELECT or AS (subquery) INTERSECT/UNION (subquery)
	if p.currentIs(token.AS) {
		p.nextToken()
		if p.currentIs(token.SELECT) || p.currentIs(token.WITH) || p.currentIs(token.LPAREN) {
			create.AsSelect = p.parseSelectWithUnion()
			// Extract FORMAT from inner SelectQuery and move it to CreateQuery
			// For CREATE VIEW/MATERIALIZED VIEW, FORMAT should be at CreateQuery level
			if swu, ok := create.AsSelect.(*ast.SelectWithUnionQuery); ok && swu != nil {
				for _, sel := range swu.Selects {
					if sq, ok := sel.(*ast.SelectQuery); ok && sq != nil && sq.Format != nil {
						create.Format = sq.Format.Name()
						sq.Format = nil
						break
					}
				}
			}
		}
	}
}

func (p *Parser) parseCreateFunction(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse function name
	create.FunctionName = p.parseIdentifierName()

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse AS lambda_expression
	if p.currentIs(token.AS) {
		p.nextToken()
		create.FunctionBody = p.parseExpression(LOWEST)
	}
}

func (p *Parser) parseCreateUser(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse user name
	create.UserName = p.parseIdentifierName()

	// Look for authentication data (NOT IDENTIFIED or IDENTIFIED)
	// Scan through tokens looking for these keywords
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		// Check for NOT IDENTIFIED
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "IDENTIFIED" {
				create.HasAuthenticationData = true
				p.nextToken()
			}
			continue
		}
		// Check for IDENTIFIED (without NOT)
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "IDENTIFIED" {
			create.HasAuthenticationData = true
			p.nextToken()
			// Parse authentication method and value
			// Forms: IDENTIFIED BY 'password'
			//        IDENTIFIED WITH method BY 'password'
			//        IDENTIFIED WITH method BY 'password', method BY 'password', ...
			for {
				// Skip WITH if present (auth method follows)
				if p.currentIs(token.WITH) {
					p.nextToken()
				}
				// Check for ssh_key authentication method
				isSSHKey := p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "SSH_KEY"
				// Skip auth method name (plaintext_password, sha256_password, etc.)
				// Stop at BY (token), comma, or next section keywords
				gotAuthValue := false
				for p.currentIs(token.IDENT) {
					ident := strings.ToUpper(p.current.Value)
					// Stop at HOST, SETTINGS, DEFAULT, GRANTEES - don't consume these
					if ident == "HOST" || ident == "SETTINGS" || ident == "DEFAULT" || ident == "GRANTEES" {
						break
					}
					p.nextToken()
					// Handle REALM/SERVER string values (for kerberos/ldap) - capture them!
					if p.currentIs(token.STRING) && (ident == "REALM" || ident == "SERVER") {
						create.AuthenticationValues = append(create.AuthenticationValues, p.current.Value)
						gotAuthValue = true
						p.nextToken()
					}
				}
				// Check for BY 'value' or BY KEY ... TYPE ... (SSH key auth)
				if p.currentIs(token.BY) {
					p.nextToken()
					if isSSHKey {
						// Parse SSH key format: BY KEY 'key' TYPE 'type' [, KEY 'key' TYPE 'type' ...]
						for {
							if p.currentIs(token.KEY) {
								p.nextToken()
								if p.currentIs(token.STRING) {
									p.nextToken() // skip key value
								}
								// Skip TYPE 'algorithm'
								if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
									p.nextToken()
									if p.currentIs(token.STRING) {
										p.nextToken() // skip type value
									}
								}
								create.SSHKeyCount++
							}
							// Check for comma (multiple keys)
							if p.currentIs(token.COMMA) {
								p.nextToken()
								continue
							}
							break
						}
						gotAuthValue = true
					} else if p.currentIs(token.STRING) {
						create.AuthenticationValues = append(create.AuthenticationValues, p.current.Value)
						gotAuthValue = true
						p.nextToken()
					}
				}
				_ = gotAuthValue // suppress unused variable warning if any
				// Check for comma (multiple auth methods)
				if p.currentIs(token.COMMA) {
					p.nextToken()
					continue
				}
				break
			}
			continue
		}
		p.nextToken()
	}
}

func (p *Parser) parseAlterUser() *ast.CreateQuery {
	create := &ast.CreateQuery{
		Position:   p.current.Pos,
		CreateUser: true,
		AlterUser:  true,
	}

	p.nextToken() // skip ALTER
	p.nextToken() // skip USER

	// Parse user name
	create.UserName = p.parseIdentifierName()

	// Scan for authentication data (NOT IDENTIFIED or IDENTIFIED)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		// Check for NOT IDENTIFIED
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "IDENTIFIED" {
				create.HasAuthenticationData = true
				p.nextToken()
			}
			continue
		}
		// Check for IDENTIFIED (without NOT)
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "IDENTIFIED" {
			create.HasAuthenticationData = true
			p.nextToken()
			// Parse authentication method and value
			for {
				// Skip WITH if present
				if p.currentIs(token.WITH) {
					p.nextToken()
				}
				// Skip auth method name
				for p.currentIs(token.IDENT) {
					ident := strings.ToUpper(p.current.Value)
					if ident == "HOST" || ident == "SETTINGS" || ident == "DEFAULT" || ident == "GRANTEES" {
						break
					}
					p.nextToken()
					// Handle REALM/SERVER string values (for kerberos/ldap) - capture them!
					if p.currentIs(token.STRING) && (ident == "REALM" || ident == "SERVER") {
						create.AuthenticationValues = append(create.AuthenticationValues, p.current.Value)
						p.nextToken()
					}
				}
				// Check for BY 'value'
				if p.currentIs(token.BY) {
					p.nextToken()
					if p.currentIs(token.STRING) {
						create.AuthenticationValues = append(create.AuthenticationValues, p.current.Value)
						p.nextToken()
					}
				}
				// Check for comma (multiple auth methods)
				if p.currentIs(token.COMMA) {
					p.nextToken()
					continue
				}
				break
			}
			continue
		}
		p.nextToken()
	}

	return create
}

func (p *Parser) parseCreateGeneric(create *ast.CreateQuery) {
	// Parse name
	name := p.parseIdentifierName()
	if name != "" {
		create.Table = name // Reuse Table field for generic name
	}

	// Skip the rest of the statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}
}

func (p *Parser) parseCreateSettingsProfile(pos token.Position) *ast.CreateSettingsProfileQuery {
	query := &ast.CreateSettingsProfileQuery{
		Position: pos,
	}

	// Skip SETTINGS if present (CREATE SETTINGS PROFILE vs CREATE PROFILE)
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
	}

	// Skip PROFILE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE" {
		p.nextToken()
	}

	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
		}
		if p.currentIs(token.EXISTS) {
			p.nextToken()
		}
	}

	// Parse profile names (can be multiple: s1, s2, s3)
	for {
		name := p.parseIdentifierName()
		if name != "" {
			query.Names = append(query.Names, name)
		}
		if p.currentIs(token.COMMA) {
			p.nextToken()
			continue
		}
		break
	}

	// Skip the rest of the statement (SETTINGS, TO, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropSettingsProfile() *ast.DropSettingsProfileQuery {
	query := &ast.DropSettingsProfileQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Skip SETTINGS if present (DROP SETTINGS PROFILE vs DROP PROFILE)
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
	}

	// Skip PROFILE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE" {
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			query.IfExists = true
			p.nextToken()
		}
	}

	// Parse profile names (can be multiple: s1, s2, s3)
	for {
		name := p.parseIdentifierName()
		if name != "" {
			query.Names = append(query.Names, name)
		}
		if p.currentIs(token.COMMA) {
			p.nextToken()
			continue
		}
		break
	}

	// Skip the rest of the statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseAlterSettingsProfile() *ast.AlterSettingsProfileQuery {
	query := &ast.AlterSettingsProfileQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip ALTER

	// Skip SETTINGS if present (ALTER SETTINGS PROFILE vs ALTER PROFILE)
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
	}

	// Skip PROFILE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE" {
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			p.nextToken()
		}
	}

	// Parse profile names (can be multiple: s1, s2, s3)
	for {
		name := p.parseIdentifierName()
		if name != "" {
			query.Names = append(query.Names, name)
		}
		if p.currentIs(token.COMMA) {
			p.nextToken()
			continue
		}
		break
	}

	// Skip the rest of the statement (SETTINGS, RENAME TO, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseShowCreateSettingsProfile(pos token.Position) *ast.ShowCreateSettingsProfileQuery {
	query := &ast.ShowCreateSettingsProfileQuery{
		Position: pos,
	}

	// Skip SETTINGS if present (SHOW CREATE SETTINGS PROFILE vs SHOW CREATE PROFILE)
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
	}

	// Skip PROFILE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE" {
		p.nextToken()
	}

	// Parse profile names (can be multiple: s1, s2, s3)
	for {
		name := p.parseIdentifierName()
		if name != "" {
			query.Names = append(query.Names, name)
		}
		if p.currentIs(token.COMMA) {
			p.nextToken()
			continue
		}
		break
	}

	// Skip tokens until FORMAT or end of statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
		p.nextToken()
	}

	// Parse FORMAT clause if present
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			query.Format = p.current.Value
			p.nextToken()
		}
	}

	return query
}

func (p *Parser) parseCreateRowPolicy(pos token.Position) *ast.CreateRowPolicyQuery {
	query := &ast.CreateRowPolicyQuery{
		Position: pos,
	}

	// Skip ROW if present (CREATE ROW POLICY vs CREATE POLICY)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROW" {
		p.nextToken()
	}

	// Skip POLICY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "POLICY" {
		p.nextToken()
	}

	// Skip the rest of the statement (policy names, ON table, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropRowPolicy() *ast.DropRowPolicyQuery {
	query := &ast.DropRowPolicyQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Skip ROW if present (DROP ROW POLICY vs DROP POLICY)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROW" {
		p.nextToken()
	}

	// Skip POLICY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "POLICY" {
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			query.IfExists = true
			p.nextToken()
		}
	}

	// Skip the rest of the statement (policy names, ON table, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseAlterRowPolicy() *ast.CreateRowPolicyQuery {
	query := &ast.CreateRowPolicyQuery{
		Position: p.current.Pos,
		IsAlter:  true,
	}

	p.nextToken() // skip ALTER

	// Skip ROW if present (ALTER ROW POLICY vs ALTER POLICY)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROW" {
		p.nextToken()
	}

	// Skip POLICY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "POLICY" {
		p.nextToken()
	}

	// Skip the rest of the statement (policy names, ON table, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseShowCreateRowPolicy(pos token.Position) *ast.ShowCreateRowPolicyQuery {
	query := &ast.ShowCreateRowPolicyQuery{
		Position: pos,
	}

	// Skip ROW if present (SHOW CREATE ROW POLICY vs SHOW CREATE POLICY)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROW" {
		p.nextToken()
	}

	// Skip POLICY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "POLICY" {
		p.nextToken()
	}

	// Skip tokens until FORMAT or end of statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
		p.nextToken()
	}

	// Parse FORMAT clause if present
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			query.Format = p.current.Value
			p.nextToken()
		}
	}

	return query
}

func (p *Parser) parseCreateRole(pos token.Position) *ast.CreateRoleQuery {
	query := &ast.CreateRoleQuery{
		Position: pos,
	}

	// Skip ROLE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROLE" {
		p.nextToken()
	}

	// Skip the rest of the statement (role names, SETTINGS, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropRole() *ast.DropRoleQuery {
	query := &ast.DropRoleQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Skip ROLE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROLE" {
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			query.IfExists = true
			p.nextToken()
		}
	}

	// Skip the rest of the statement (role names, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseAlterRole() *ast.CreateRoleQuery {
	query := &ast.CreateRoleQuery{
		Position: p.current.Pos,
		IsAlter:  true,
	}

	p.nextToken() // skip ALTER

	// Skip ROLE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROLE" {
		p.nextToken()
	}

	// Skip the rest of the statement (role names, SETTINGS, RENAME TO, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseCreateQuota(pos token.Position) *ast.CreateQuotaQuery {
	query := &ast.CreateQuotaQuery{
		Position: pos,
	}

	// Skip QUOTA keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "QUOTA" {
		p.nextToken()
	}

	// Parse quota name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Skip the rest of the statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseCreateNamedCollection(pos token.Position) *ast.CreateNamedCollectionQuery {
	query := &ast.CreateNamedCollectionQuery{
		Position: pos,
	}

	// Skip NAMED keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "NAMED" {
		p.nextToken()
	}

	// Skip COLLECTION keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "COLLECTION" {
		p.nextToken()
	}

	// Parse collection name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() || p.currentIs(token.STRING) {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Skip the rest of the statement (AS key=value, ...)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseAlterNamedCollection() *ast.AlterNamedCollectionQuery {
	pos := p.current.Pos
	p.nextToken() // skip ALTER

	query := &ast.AlterNamedCollectionQuery{
		Position: pos,
	}

	// Skip NAMED keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "NAMED" {
		p.nextToken()
	}

	// Skip COLLECTION keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "COLLECTION" {
		p.nextToken()
	}

	// Parse collection name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() || p.currentIs(token.STRING) {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Skip the rest of the statement (DELETE key, SET key=value, ...)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropNamedCollection() *ast.DropNamedCollectionQuery {
	pos := p.current.Pos
	p.nextToken() // skip DROP

	query := &ast.DropNamedCollectionQuery{
		Position: pos,
	}

	// Skip NAMED keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "NAMED" {
		p.nextToken()
	}

	// Skip COLLECTION keyword
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "COLLECTION" {
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			query.IfExists = true
			p.nextToken()
		}
	}

	// Parse collection name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() || p.currentIs(token.STRING) {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Skip the rest of the statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseShowCreateRole(pos token.Position) *ast.ShowCreateRoleQuery {
	query := &ast.ShowCreateRoleQuery{
		Position:  pos,
		RoleCount: 1, // Default to 1 role
	}

	// Skip ROLE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROLE" {
		p.nextToken()
	}

	// Count role names (separated by commas)
	// Skip first role name
	for p.currentIs(token.IDENT) || p.currentIs(token.STRING) || p.current.Token.IsKeyword() {
		p.nextToken()
		// Handle role@host syntax
		if p.currentIs(token.IDENT) && strings.HasPrefix(p.current.Value, "@") {
			p.nextToken()
		}
		break
	}

	// Count additional roles
	for p.currentIs(token.COMMA) {
		query.RoleCount++
		p.nextToken()
		// Skip role name
		for p.currentIs(token.IDENT) || p.currentIs(token.STRING) || p.current.Token.IsKeyword() {
			p.nextToken()
			// Handle role@host syntax
			if p.currentIs(token.IDENT) && strings.HasPrefix(p.current.Value, "@") {
				p.nextToken()
			}
			break
		}
	}

	// Skip tokens until FORMAT or end of statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
		p.nextToken()
	}

	// Parse FORMAT clause if present
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			query.Format = p.current.Value
			p.nextToken()
		}
	}

	return query
}

func (p *Parser) parseCreateResource(pos token.Position) *ast.CreateResourceQuery {
	query := &ast.CreateResourceQuery{
		Position: pos,
	}

	// Skip RESOURCE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RESOURCE" {
		p.nextToken()
	}

	// Get resource name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Skip the rest of the statement (resource definition, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropResource() *ast.DropResourceQuery {
	query := &ast.DropResourceQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Skip RESOURCE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RESOURCE" {
		p.nextToken()
	}

	// Skip the rest of the statement (IF EXISTS, name, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseCreateWorkload(pos token.Position) *ast.CreateWorkloadQuery {
	query := &ast.CreateWorkloadQuery{
		Position: pos,
	}

	// Skip WORKLOAD
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "WORKLOAD" {
		p.nextToken()
	}

	// Check for IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				p.nextToken()
			}
		}
	}

	// Get workload name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		query.Name = p.current.Value
		p.nextToken()
	}

	// Check for IN (parent workload)
	if p.currentIs(token.IN) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			query.Parent = p.current.Value
			p.nextToken()
		}
	}

	// Skip the rest of the statement (SETTINGS, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseDropWorkload() *ast.DropWorkloadQuery {
	query := &ast.DropWorkloadQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Skip WORKLOAD
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "WORKLOAD" {
		p.nextToken()
	}

	// Skip the rest of the statement (IF EXISTS, name, etc.)
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseCreateDictionary(create *ast.CreateQuery) {
	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				create.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse dictionary name (possibly database.name)
	name := p.parseIdentifierName()
	if p.currentIs(token.DOT) {
		create.Database = name
		p.nextToken()
		name = p.parseIdentifierName()
	}
	create.Table = name

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CLUSTER" {
			p.nextToken()
			create.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse column definitions (attributes) if present
	if p.currentIs(token.LPAREN) {
		p.nextToken() // skip (
		create.DictionaryAttrs = p.parseDictionaryAttributes()
		if p.currentIs(token.RPAREN) {
			p.nextToken() // skip )
		}
	}

	// Initialize dictionary definition
	dictDef := &ast.DictionaryDefinition{
		Position: p.current.Pos,
	}

	// Parse PRIMARY KEY, SOURCE, LIFETIME, LAYOUT, RANGE, SETTINGS
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		// Handle PRIMARY as a keyword token
		if p.currentIs(token.PRIMARY) {
			p.nextToken() // skip PRIMARY
			if p.currentIs(token.KEY) {
				p.nextToken() // skip KEY
				dictDef.PrimaryKey = p.parseDictionaryPrimaryKey()
			}
			continue
		}
		// Handle SETTINGS as a keyword token
		if p.currentIs(token.SETTINGS) {
			p.nextToken() // skip SETTINGS
			// Parse dictionary settings: SETTINGS(key=value, ...) or SETTINGS key=value, ...
			if p.currentIs(token.LPAREN) {
				p.nextToken() // skip (
				dictDef.Settings = p.parseSettingsList()
				p.expect(token.RPAREN)
			} else {
				dictDef.Settings = p.parseSettingsList()
			}
			continue
		}
		// Handle COMMENT as a keyword token
		if p.currentIs(token.COMMENT) {
			p.nextToken() // skip COMMENT
			if p.currentIs(token.STRING) {
				create.Comment = p.current.Value
				p.nextToken()
			}
			continue
		}
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			switch upper {
			case "PRIMARY":
				p.nextToken() // skip PRIMARY
				if p.currentIs(token.KEY) {
					p.nextToken() // skip KEY
					dictDef.PrimaryKey = p.parseDictionaryPrimaryKey()
				}
			case "SOURCE":
				p.nextToken() // skip SOURCE
				dictDef.Source = p.parseDictionarySource()
			case "LIFETIME":
				p.nextToken() // skip LIFETIME
				dictDef.Lifetime = p.parseDictionaryLifetime()
			case "LAYOUT":
				p.nextToken() // skip LAYOUT
				dictDef.Layout = p.parseDictionaryLayout()
			case "RANGE":
				p.nextToken() // skip RANGE
				dictDef.Range = p.parseDictionaryRange()
			case "SETTINGS":
				p.nextToken() // skip SETTINGS
				// Parse dictionary settings: SETTINGS(key=value, ...) or SETTINGS key=value, ...
				if p.currentIs(token.LPAREN) {
					p.nextToken() // skip (
					dictDef.Settings = p.parseSettingsList()
					p.expect(token.RPAREN)
				} else {
					dictDef.Settings = p.parseSettingsList()
				}
			case "COMMENT":
				p.nextToken() // skip COMMENT
				if p.currentIs(token.STRING) {
					create.Comment = p.current.Value
					p.nextToken()
				}
			default:
				p.nextToken()
			}
		} else {
			p.nextToken()
		}
	}

	// Only set dictionary definition if it has any content
	if len(dictDef.PrimaryKey) > 0 || dictDef.Source != nil || dictDef.Lifetime != nil || dictDef.Layout != nil || dictDef.Range != nil || len(dictDef.Settings) > 0 {
		create.DictionaryDef = dictDef
	}
}

func (p *Parser) isDictionaryClauseKeyword() bool {
	if !p.currentIs(token.IDENT) {
		return false
	}
	upper := strings.ToUpper(p.current.Value)
	switch upper {
	case "PRIMARY", "SOURCE", "LIFETIME", "LAYOUT", "RANGE", "SETTINGS", "COMMENT":
		return true
	}
	return false
}

func (p *Parser) parseDictionaryAttributes() []*ast.DictionaryAttributeDeclaration {
	var attrs []*ast.DictionaryAttributeDeclaration

	for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
		attr := p.parseDictionaryAttribute()
		if attr != nil {
			attrs = append(attrs, attr)
		}

		// Handle comma between attributes
		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	return attrs
}

func (p *Parser) parseDictionaryAttribute() *ast.DictionaryAttributeDeclaration {
	attr := &ast.DictionaryAttributeDeclaration{
		Position: p.current.Pos,
	}

	// Parse attribute name
	attr.Name = p.parseIdentifierName()
	if attr.Name == "" {
		return nil
	}

	// Parse type
	if !p.currentIs(token.COMMA) && !p.currentIs(token.RPAREN) {
		attr.Type = p.parseDataType()
	}

	// Parse optional clauses: DEFAULT, EXPRESSION, HIERARCHICAL, INJECTIVE, IS_OBJECT_ID
	for !p.currentIs(token.EOF) && !p.currentIs(token.COMMA) && !p.currentIs(token.RPAREN) {
		if p.currentIs(token.DEFAULT) {
			p.nextToken()
			attr.Default = p.parseExpression(LOWEST)
		} else if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			switch upper {
			case "EXPRESSION":
				p.nextToken()
				attr.Expression = p.parseExpression(LOWEST)
			case "HIERARCHICAL":
				attr.Hierarchical = true
				p.nextToken()
			case "INJECTIVE":
				attr.Injective = true
				p.nextToken()
			case "IS_OBJECT_ID":
				attr.IsObjectID = true
				p.nextToken()
			default:
				p.nextToken()
			}
		} else {
			p.nextToken()
		}
	}

	return attr
}

func (p *Parser) parseDictionaryPrimaryKey() []ast.Expression {
	var keys []ast.Expression

	// Can be single identifier or tuple (id1, id2)
	if p.currentIs(token.LPAREN) {
		p.nextToken() // skip (
		for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
			expr := p.parseExpression(LOWEST)
			if expr != nil {
				keys = append(keys, expr)
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
		if p.currentIs(token.RPAREN) {
			p.nextToken() // skip )
		}
	} else {
		// Can be comma-separated identifiers: PRIMARY KEY id, id_key
		for {
			expr := p.parseExpression(LOWEST)
			if expr != nil {
				keys = append(keys, expr)
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
	}

	return keys
}

func (p *Parser) parseDictionarySource() *ast.DictionarySource {
	source := &ast.DictionarySource{
		Position: p.current.Pos,
	}

	if !p.currentIs(token.LPAREN) {
		return source
	}
	p.nextToken() // skip (

	// Parse source type (e.g., CLICKHOUSE, MYSQL, FILE, NULL)
	if p.currentIs(token.IDENT) || p.currentIs(token.NULL) || p.current.Token.IsKeyword() {
		source.Type = strings.ToUpper(p.current.Value)
		p.nextToken()
	}

	// Parse key-value arguments in parentheses
	if p.currentIs(token.LPAREN) {
		p.nextToken() // skip (
		source.Args = p.parseKeyValuePairs()
		if p.currentIs(token.RPAREN) {
			p.nextToken() // skip )
		}
	}

	if p.currentIs(token.RPAREN) {
		p.nextToken() // skip )
	}

	return source
}

func (p *Parser) parseKeyValuePairs() []*ast.KeyValuePair {
	var pairs []*ast.KeyValuePair

	for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
		pair := &ast.KeyValuePair{
			Position: p.current.Pos,
		}

		// Parse key (the key is not included in EXPLAIN output, just the value)
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			pair.Key = p.current.Value
			p.nextToken()
		}

		// Parse value (can be various types - string, number, function call, identifier)
		// Value is present if current token is:
		// - A string, number, or other literal
		// - An identifier followed by LPAREN (function call)
		// - An identifier NOT followed by an identifier (key-only pairs have adjacent identifiers)
		if !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				// Check if this is the value or the start of the next key
				// If peek is an identifier/keyword, this identifier is the value
				// If peek is RPAREN/EOF, this identifier is the value
				// If peek is LPAREN, this is a function call value
				if p.peekIs(token.IDENT) || (p.peek.Token.IsKeyword() && !p.peekIs(token.LPAREN)) {
					// This identifier is followed by another identifier/keyword, treat as value
					pair.Value = &ast.Identifier{Position: p.current.Pos, Parts: []string{p.current.Value}}
					p.nextToken()
				} else {
					// Either a function call, or this identifier is the last thing before )
					pair.Value = p.parseExpression(LOWEST)
				}
			} else {
				// Non-identifier value (string, number, etc.)
				pair.Value = p.parseExpression(LOWEST)
			}
		}

		pairs = append(pairs, pair)
	}

	return pairs
}

func (p *Parser) parseDictionaryLifetime() *ast.DictionaryLifetime {
	lifetime := &ast.DictionaryLifetime{
		Position: p.current.Pos,
	}

	if !p.currentIs(token.LPAREN) {
		return lifetime
	}
	p.nextToken() // skip (

	// Parse MIN and MAX or just a single value
	for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			if upper == "MIN" {
				p.nextToken()
				lifetime.Min = p.parseExpression(LOWEST)
			} else if upper == "MAX" {
				p.nextToken()
				lifetime.Max = p.parseExpression(LOWEST)
			} else {
				p.nextToken()
			}
		} else {
			p.nextToken()
		}
	}

	if p.currentIs(token.RPAREN) {
		p.nextToken() // skip )
	}

	return lifetime
}

func (p *Parser) parseDictionaryLayout() *ast.DictionaryLayout {
	layout := &ast.DictionaryLayout{
		Position: p.current.Pos,
	}

	if !p.currentIs(token.LPAREN) {
		return layout
	}
	p.nextToken() // skip (

	// Parse layout type (e.g., FLAT, HASHED, COMPLEX_KEY_HASHED)
	if p.currentIs(token.IDENT) {
		layout.Type = strings.ToUpper(p.current.Value)
		p.nextToken()
	}

	// Parse optional arguments in parentheses
	if p.currentIs(token.LPAREN) {
		p.nextToken() // skip (
		layout.Args = p.parseKeyValuePairs()
		if p.currentIs(token.RPAREN) {
			p.nextToken() // skip )
		}
	}

	if p.currentIs(token.RPAREN) {
		p.nextToken() // skip )
	}

	return layout
}

func (p *Parser) parseDictionaryRange() *ast.DictionaryRange {
	dictRange := &ast.DictionaryRange{
		Position: p.current.Pos,
	}

	if !p.currentIs(token.LPAREN) {
		return dictRange
	}
	p.nextToken() // skip (

	// Parse MIN and MAX
	for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			if upper == "MIN" {
				p.nextToken()
				dictRange.Min = p.parseExpression(LOWEST)
			} else if upper == "MAX" {
				p.nextToken()
				dictRange.Max = p.parseExpression(LOWEST)
			} else {
				p.nextToken()
			}
		} else {
			p.nextToken()
		}
	}

	if p.currentIs(token.RPAREN) {
		p.nextToken() // skip )
	}

	return dictRange
}

func (p *Parser) parseIndexDefinition() *ast.IndexDefinition {
	idx := &ast.IndexDefinition{
		Position: p.current.Pos,
	}

	p.nextToken() // skip INDEX

	// Parse index name
	idx.Name = p.parseIdentifierName()

	// Parse expression (the column or expression being indexed)
	idx.Expression = p.parseExpression(LOWEST)

	// Parse TYPE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
		p.nextToken()
		// Type is a function call like bloom_filter(0.025) or minmax
		pos := p.current.Pos
		typeName := p.parseIdentifierName()
		if typeName != "" {
			idx.Type = &ast.FunctionCall{
				Position: pos,
				Name:     typeName,
			}
			// Check for parentheses (type parameters)
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				if !p.currentIs(token.RPAREN) {
					idx.Type.Arguments = p.parseExpressionList()
				}
				p.expect(token.RPAREN)
			}
		}
	}

	// Parse GRANULARITY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "GRANULARITY" {
		p.nextToken()
		idx.Granularity = p.parseExpression(LOWEST)
	}

	return idx
}

func (p *Parser) parseColumnDeclaration() *ast.ColumnDeclaration {
	col := &ast.ColumnDeclaration{
		Position: p.current.Pos,
	}

	// Parse column name (can be identifier or keyword like KEY)
	// Also handles nested column names like n.y
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		col.Name = p.current.Value
		p.nextToken()
		// Handle nested column names (e.g., n.y for nested columns)
		for p.currentIs(token.DOT) {
			p.nextToken() // skip .
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				col.Name += "." + p.current.Value
				p.nextToken()
			} else {
				break
			}
		}
	} else {
		return nil
	}

	// Check if next token indicates type is omitted
	// DEFAULT/MATERIALIZED/ALIAS indicate we go straight to default expression
	// CODEC indicates we go straight to codec specification (no type)
	// COMMENT indicates we go straight to comment (no type)
	isCodec := p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CODEC"
	if p.currentIs(token.DEFAULT) || p.currentIs(token.MATERIALIZED) || p.currentIs(token.ALIAS) || isCodec || p.currentIs(token.COMMENT) {
		// Type is omitted, skip to parsing below
	} else {
		// Parse data type
		col.Type = p.parseDataType()
	}

	// Parse STATISTICS clause (e.g., STATISTICS(tdigest, uniq))
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
		p.nextToken()
		col.Statistics = p.parseStatisticsExpr()
	}

	// Handle COLLATE clause (MySQL compatibility, e.g., varchar(255) COLLATE binary)
	if p.currentIs(token.COLLATE) {
		p.nextToken()
		// Skip collation name
		if p.currentIs(token.IDENT) || p.currentIs(token.STRING) {
			p.nextToken()
		}
	}

	// Handle NOT NULL / NULL constraint
	if p.currentIs(token.NOT) {
		p.nextToken()
		if p.currentIs(token.NULL) {
			notNull := false
			col.Nullable = &notNull
			p.nextToken()
		}
	} else if p.currentIs(token.NULL) {
		// NULL is explicit nullable (default)
		nullable := true
		col.Nullable = &nullable
		p.nextToken()
	}

	// Parse DEFAULT/MATERIALIZED/ALIAS/EPHEMERAL
	switch p.current.Token {
	case token.DEFAULT:
		col.DefaultKind = "DEFAULT"
		p.nextToken()
		col.Default = p.parseExpression(LOWEST)
	case token.MATERIALIZED:
		col.DefaultKind = "MATERIALIZED"
		p.nextToken()
		col.Default = p.parseExpression(LOWEST)
	case token.ALIAS:
		col.DefaultKind = "ALIAS"
		p.nextToken()
		col.Default = p.parseExpression(LOWEST)
	}

	// Handle EPHEMERAL (can be EPHEMERAL or EPHEMERAL default_value)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "EPHEMERAL" {
		col.DefaultKind = "EPHEMERAL"
		p.nextToken()
		// Optional default value - but don't parse column keywords (CODEC, COMMENT, TTL, etc.) as expressions
		if !p.currentIs(token.COMMA) && !p.currentIs(token.RPAREN) && !p.currentIs(token.IDENT) &&
			!p.currentIs(token.COMMENT) && !p.currentIs(token.TTL) && !p.currentIs(token.PRIMARY) &&
			!p.currentIs(token.SETTINGS) {
			col.Default = p.parseExpression(LOWEST)
		}
	}

	// Handle NOT NULL / NULL after DEFAULT (ClickHouse allows DEFAULT expr NOT NULL)
	if p.currentIs(token.NOT) {
		p.nextToken()
		if p.currentIs(token.NULL) {
			notNull := false
			col.Nullable = &notNull
			p.nextToken()
		}
	} else if p.currentIs(token.NULL) && col.Nullable == nil {
		// NULL is explicit nullable (default)
		nullable := true
		col.Nullable = &nullable
		p.nextToken()
	}

	// Parse CODEC
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CODEC" {
		p.nextToken()
		col.Codec = p.parseCodecExpr()
	}

	// Parse TTL
	if p.currentIs(token.TTL) {
		p.nextToken()
		col.TTL = p.parseExpression(LOWEST)
	}

	// Parse PRIMARY KEY (column constraint)
	if p.currentIs(token.PRIMARY) {
		p.nextToken()
		if p.currentIs(token.KEY) {
			col.PrimaryKey = true
			p.nextToken()
		}
	}

	// Parse COMMENT
	if p.currentIs(token.COMMENT) {
		p.nextToken()
		if p.currentIs(token.STRING) {
			col.Comment = p.current.Value
			p.nextToken()
		}
	}

	// Parse column-level SETTINGS (key = value, ...)
	// Only parse if SETTINGS is followed by LPAREN (column-level settings use parentheses)
	// If no LPAREN, leave SETTINGS for the parent (ALTER/CREATE) to handle
	if p.currentIs(token.SETTINGS) && p.peek.Token == token.LPAREN {
		p.nextToken() // skip SETTINGS
		p.nextToken() // skip (
		col.Settings = p.parseSettingsList()
		p.expect(token.RPAREN)
	}

	return col
}

func (p *Parser) parseDataType() *ast.DataType {
	// Type names can be identifiers or keywords (Array, Nested, Key, etc.)
	if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
		return nil
	}

	dt := &ast.DataType{
		Position: p.current.Pos,
		Name:     p.current.Value,
	}
	p.nextToken()

	// For MySQL-compatible INT types, handle display width and UNSIGNED/SIGNED
	upperName := strings.ToUpper(dt.Name)
	isMySQLIntType := upperName == "INT" || upperName == "INT1" || upperName == "TINYINT" || upperName == "SMALLINT" ||
		upperName == "MEDIUMINT" || upperName == "BIGINT" || upperName == "INTEGER"

	if isMySQLIntType && p.currentIs(token.LPAREN) {
		// Skip the display width parameter (e.g., INT(11))
		p.nextToken() // skip (
		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			p.nextToken()
		}
		p.expect(token.RPAREN)
	}

	// Handle UNSIGNED/SIGNED modifiers for MySQL INT types
	if isMySQLIntType && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) {
		modifier := strings.ToUpper(p.current.Value)
		if modifier == "UNSIGNED" || modifier == "SIGNED" {
			dt.Name = dt.Name + " " + p.current.Value
			p.nextToken()
		}
	}

	// Handle DOUBLE PRECISION
	if upperName == "DOUBLE" && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) {
		if strings.ToUpper(p.current.Value) == "PRECISION" {
			dt.Name = dt.Name + " " + p.current.Value
			p.nextToken()
		}
	}

	// Handle multi-word character types
	// CHAR VARYING, CHAR LARGE OBJECT, CHARACTER VARYING, CHARACTER LARGE OBJECT
	// NCHAR VARYING, NCHAR LARGE OBJECT
	if upperName == "CHAR" || upperName == "CHARACTER" || upperName == "NCHAR" {
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			modifier := strings.ToUpper(p.current.Value)
			if modifier == "VARYING" {
				dt.Name = dt.Name + " " + p.current.Value
				p.nextToken()
			} else if modifier == "LARGE" {
				dt.Name = dt.Name + " " + p.current.Value
				p.nextToken()
				// Expect OBJECT
				if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && strings.ToUpper(p.current.Value) == "OBJECT" {
					dt.Name = dt.Name + " " + p.current.Value
					p.nextToken()
				}
			}
		}
	}

	// Handle BINARY VARYING, BINARY LARGE OBJECT
	if upperName == "BINARY" && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) {
		modifier := strings.ToUpper(p.current.Value)
		if modifier == "VARYING" {
			dt.Name = dt.Name + " " + p.current.Value
			p.nextToken()
		} else if modifier == "LARGE" {
			dt.Name = dt.Name + " " + p.current.Value
			p.nextToken()
			// Expect OBJECT
			if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && strings.ToUpper(p.current.Value) == "OBJECT" {
				dt.Name = dt.Name + " " + p.current.Value
				p.nextToken()
			}
		}
	}

	// Handle NATIONAL CHAR, NATIONAL CHARACTER, etc.
	if upperName == "NATIONAL" && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) {
		// NATIONAL can be followed by CHAR or CHARACTER
		nextWord := strings.ToUpper(p.current.Value)
		if nextWord == "CHAR" || nextWord == "CHARACTER" {
			dt.Name = dt.Name + " " + p.current.Value
			p.nextToken()
			// Can be followed by VARYING or LARGE OBJECT
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				modifier := strings.ToUpper(p.current.Value)
				if modifier == "VARYING" {
					dt.Name = dt.Name + " " + p.current.Value
					p.nextToken()
				} else if modifier == "LARGE" {
					dt.Name = dt.Name + " " + p.current.Value
					p.nextToken()
					// Expect OBJECT
					if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && strings.ToUpper(p.current.Value) == "OBJECT" {
						dt.Name = dt.Name + " " + p.current.Value
						p.nextToken()
					}
				}
			}
		}
	}

	// Parse type parameters
	if p.currentIs(token.LPAREN) {
		dt.HasParentheses = true
		p.nextToken()

		// Determine if this type uses named parameters (Nested, Tuple, JSON)
		upperName := strings.ToUpper(dt.Name)
		usesNamedParams := upperName == "NESTED" || upperName == "TUPLE" || upperName == "JSON" || upperName == "OBJECT"
		// JSON and OBJECT types wrap their parameters in ObjectTypeArgument
		isObjectType := upperName == "JSON" || upperName == "OBJECT"

		// Parse type parameters, but stop on keywords that can't be part of type params
		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) && !p.currentIs(token.COLLATE) {
			var param ast.Expression

			// Special handling for SKIP in JSON/OBJECT types: SKIP path or SKIP REGEXP 'pattern'
			if isObjectType && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && strings.ToUpper(p.current.Value) == "SKIP" {
				pos := p.current.Pos
				p.nextToken() // consume SKIP

				// Check for SKIP REGEXP 'pattern'
				if p.currentIs(token.REGEXP) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "REGEXP") {
					p.nextToken() // consume REGEXP
					// Parse the pattern string
					if p.currentIs(token.STRING) {
						pattern := p.current.Value
						p.nextToken()
						param = &ast.FunctionCall{
							Position:  pos,
							Name:      "SKIP REGEXP",
							Arguments: []ast.Expression{&ast.Literal{Position: pos, Value: pattern, Type: ast.LiteralString}},
						}
					}
				} else {
					// Parse dotted path: a, a.b, a.b.c, etc.
					var pathParts []string
					for {
						if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
							pathParts = append(pathParts, p.current.Value)
							p.nextToken()
						}
						if p.currentIs(token.DOT) {
							p.nextToken() // consume dot
						} else {
							break
						}
					}
					if len(pathParts) > 0 {
						param = &ast.FunctionCall{
							Position:  pos,
							Name:      "SKIP",
							Arguments: []ast.Expression{&ast.Identifier{Position: pos, Parts: pathParts}},
						}
					}
				}
				// Wrap in ObjectTypeArgument
				if param != nil {
					param = &ast.ObjectTypeArgument{
						Position: param.Pos(),
						Expr:     param,
					}
					dt.Parameters = append(dt.Parameters, param)
				}
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
				continue
			}

			// Check if this is a named parameter: identifier followed by a type name
			// e.g., "a UInt32" where "a" is the name and "UInt32" is the type
			isNamedParam := false
			if usesNamedParams && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) {
				// Check if current is NOT a type name and peek IS a type name or LPAREN follows for complex types
				// But NOT if peek is '=' which indicates an expression like max_dynamic_paths=8
				if !p.isDataTypeName(p.current.Value) && !p.peekIs(token.EQ) {
					// Current is a name (not a type), next should be a type
					isNamedParam = true
				} else if !p.peekIs(token.EQ) && (p.peekIs(token.IDENT) || p.peekIs(token.LPAREN)) {
					// Current looks like a type name but is followed by another identifier
					// This happens with things like "a Tuple(...)" where "a" looks like it could be a type
					// Check if peek is a known type name
					if p.peekIs(token.IDENT) && p.isDataTypeName(p.peek.Value) {
						isNamedParam = true
					} else if p.peekIs(token.LPAREN) {
						// Could be a function-like type or named with parenthesized type
						// Check if current is a valid type name - if so, it's a type, not a name
						if !p.isDataTypeName(p.current.Value) {
							isNamedParam = true
						}
					}
				}
			}


			if isNamedParam {
				// Parse as name + type pair
				// For JSON/OBJECT types, the name can be a dotted path like a.b.c
				pos := p.current.Pos
				var nameParts []string
				nameParts = append(nameParts, p.current.Value)
				p.nextToken()
				// Parse additional dotted parts if this is a JSON/OBJECT type
				if isObjectType {
					for p.currentIs(token.DOT) {
						p.nextToken() // consume dot
						if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
							nameParts = append(nameParts, p.current.Value)
							p.nextToken()
						} else {
							break
						}
					}
				}
				paramName := strings.Join(nameParts, ".")
				// Parse the type for this parameter
				paramType := p.parseDataType()
				if paramType != nil {
					param = &ast.NameTypePair{
						Position: pos,
						Name:     paramName,
						Type:     paramType,
					}
				}
			} else if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && p.isDataTypeName(p.current.Value) {
				// It's a type name, parse as data type
				param = p.parseDataType()
			} else {
				// Parse as expression (for things like Decimal(10, 2))
				param = p.parseExpression(LOWEST)
			}

			// Wrap in ObjectTypeArgument for JSON/OBJECT types
			if param != nil {
				if isObjectType {
					param = &ast.ObjectTypeArgument{
						Position: param.Pos(),
						Expr:     param,
					}
				}
				dt.Parameters = append(dt.Parameters, param)
			}

			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
		p.expect(token.RPAREN)
	}

	return dt
}

func (p *Parser) isDataTypeName(name string) bool {
	upper := strings.ToUpper(name)
	types := []string{
		"INT", "INT8", "INT16", "INT32", "INT64", "INT128", "INT256",
		"UINT8", "UINT16", "UINT32", "UINT64", "UINT128", "UINT256",
		"FLOAT32", "FLOAT64", "FLOAT", "DOUBLE", "BFLOAT16",
		"DECIMAL", "DECIMAL32", "DECIMAL64", "DECIMAL128", "DECIMAL256", "DEC",
		"STRING", "FIXEDSTRING",
		"UUID", "DATE", "DATE32", "DATETIME", "DATETIME64",
		"ENUM", "ENUM8", "ENUM16",
		"ARRAY", "TUPLE", "MAP", "NESTED",
		"NULLABLE", "LOWCARDINALITY",
		"BOOL", "BOOLEAN",
		"IPV4", "IPV6",
		"NOTHING", "INTERVAL",
		"JSON", "OBJECT", "VARIANT",
		"AGGREGATEFUNCTION", "SIMPLEAGGREGATEFUNCTION",
		"POINT", "RING", "POLYGON", "MULTIPOLYGON",
		"TIME64", "TIME",
		"DYNAMIC",
		"QBIT",
	}
	for _, t := range types {
		if upper == t {
			return true
		}
	}
	return false
}

func (p *Parser) parseCodecExpr() *ast.CodecExpr {
	codec := &ast.CodecExpr{
		Position: p.current.Pos,
	}

	if !p.expect(token.LPAREN) {
		return nil
	}

	for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
		// Accept IDENT or keywords as codec names (e.g., "Default" is a keyword)
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			name := p.current.Value
			pos := p.current.Pos
			p.nextToken()

			fn := &ast.FunctionCall{
				Position: pos,
				Name:     name,
			}

			if p.currentIs(token.LPAREN) {
				p.nextToken()
				if !p.currentIs(token.RPAREN) {
					fn.Arguments = p.parseExpressionList()
				}
				p.expect(token.RPAREN)
			}

			codec.Codecs = append(codec.Codecs, fn)
		}

		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	p.expect(token.RPAREN)
	return codec
}

func (p *Parser) parseStatisticsExpr() []*ast.FunctionCall {
	var stats []*ast.FunctionCall

	if !p.expect(token.LPAREN) {
		return nil
	}

	for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
		if p.currentIs(token.IDENT) {
			name := p.current.Value
			pos := p.current.Pos
			p.nextToken()

			fn := &ast.FunctionCall{
				Position: pos,
				Name:     name,
			}

			// Statistics types can have optional parameters: e.g., tdigest(100)
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				if !p.currentIs(token.RPAREN) {
					fn.Arguments = p.parseExpressionList()
				}
				p.expect(token.RPAREN)
			}

			stats = append(stats, fn)
		}

		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	p.expect(token.RPAREN)
	return stats
}

// parseStatisticsColumnList parses comma-separated column names for ALTER STATISTICS commands
func (p *Parser) parseStatisticsColumnList() []string {
	var columns []string

	for p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		columns = append(columns, p.current.Value)
		p.nextToken()

		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	return columns
}

// parseStatisticsTypeList parses comma-separated statistics type names for ALTER STATISTICS TYPE clause
func (p *Parser) parseStatisticsTypeList() []*ast.FunctionCall {
	var types []*ast.FunctionCall

	for p.currentIs(token.IDENT) {
		name := p.current.Value
		pos := p.current.Pos
		p.nextToken()

		fn := &ast.FunctionCall{
			Position: pos,
			Name:     name,
		}

		// Statistics types can have optional parameters
		if p.currentIs(token.LPAREN) {
			p.nextToken()
			if !p.currentIs(token.RPAREN) {
				fn.Arguments = p.parseExpressionList()
			}
			p.expect(token.RPAREN)
		}

		types = append(types, fn)

		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	return types
}

func (p *Parser) parseEngineClause() *ast.EngineClause {
	engine := &ast.EngineClause{
		Position: p.current.Pos,
	}

	// Engine name can be identifier or keyword (Null, Join, Memory, etc.)
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		engine.Name = p.current.Value
		p.nextToken()
	}

	if p.currentIs(token.LPAREN) {
		engine.HasParentheses = true
		p.nextToken()
		if !p.currentIs(token.RPAREN) {
			// Engine parameters should not parse implicit aliases
			// e.g., Distributed('cluster', database, table) - table is NOT an alias for database
			engine.Parameters = p.parseEngineParameters()
		}
		p.expect(token.RPAREN)
	}

	return engine
}

// parseEngineParameters parses comma-separated expressions for engine clauses
// without treating identifiers as implicit aliases
func (p *Parser) parseEngineParameters() []ast.Expression {
	var exprs []ast.Expression

	if p.currentIs(token.RPAREN) || p.currentIs(token.EOF) {
		return exprs
	}

	expr := p.parseExpression(LOWEST)
	if expr != nil {
		exprs = append(exprs, expr)
	}

	for p.currentIs(token.COMMA) {
		p.nextToken()
		expr := p.parseExpression(LOWEST)
		if expr != nil {
			exprs = append(exprs, expr)
		}
	}

	return exprs
}

func (p *Parser) parseDrop() *ast.DropQuery {
	drop := &ast.DropQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DROP

	// Handle TEMPORARY
	if p.currentIs(token.TEMPORARY) {
		drop.Temporary = true
		p.nextToken()
	}

	// What are we dropping?
	dropUser := false
	dropFunction := false
	dropDictionary := false
	switch p.current.Token {
	case token.TABLE:
		p.nextToken()
	case token.DATABASE:
		drop.DropDatabase = true
		p.nextToken()
	case token.VIEW:
		p.nextToken()
	case token.USER:
		dropUser = true
		p.nextToken()
	case token.FUNCTION:
		dropFunction = true
		p.nextToken()
	case token.INDEX:
		drop.Index = "_pending_"
		p.nextToken()
	case token.SETTINGS:
		// DROP SETTINGS PROFILE
		p.nextToken() // skip SETTINGS
		// Skip "PROFILE" if present
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE" {
			p.nextToken()
		}
		drop.SettingsProfile = "_pending_"
	default:
		// Handle multi-word DROP types: ROW POLICY, NAMED COLLECTION, DICTIONARY
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			switch upper {
			case "DICTIONARY":
				dropDictionary = true
				p.nextToken()
			case "ROW":
				// DROP ROW POLICY
				p.nextToken() // skip ROW
				if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "POLICY" {
					p.nextToken() // skip POLICY
				}
				// Mark as row policy drop - name will be set later
				drop.RowPolicy = "_pending_"
			case "POLICY":
				// DROP POLICY
				p.nextToken()
				drop.Policy = "_pending_"
			case "QUOTA":
				p.nextToken()
				drop.Quota = "_pending_"
			case "ROLE":
				p.nextToken()
				drop.Role = "_pending_"
			case "NAMED":
				// DROP NAMED COLLECTION - skip tokens
				for p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					if p.currentIs(token.IF) {
						break // Hit IF EXISTS
					}
					p.nextToken()
				}
			default:
				p.nextToken() // skip unknown token
			}
		} else {
			p.nextToken() // skip unknown token
		}
	}

	// Handle IF EXISTS or IF EMPTY
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			drop.IfExists = true
			p.nextToken()
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "EMPTY" {
			// IF EMPTY - skip the EMPTY keyword
			p.nextToken()
		}
	}

	// Parse name (can start with a number in ClickHouse)
	name := p.parseIdentifierName()
	if name != "" {
		var database, tableName string
		if p.currentIs(token.DOT) {
			p.nextToken()
			database = name
			tableName = p.parseIdentifierName()
		} else {
			tableName = name
		}

		if dropUser {
			drop.User = tableName
			// Handle user@host syntax
			if p.currentIs(token.IDENT) && p.current.Value == "@" {
				p.nextToken() // skip @
				// Hostname can be identifier, string, or IP in quotes
				if p.currentIs(token.IDENT) || p.currentIs(token.STRING) || p.current.Token.IsKeyword() {
					drop.User = drop.User + "@" + p.current.Value
					p.nextToken()
				}
			}
		} else if dropFunction {
			drop.Function = tableName
		} else if drop.Role == "_pending_" {
			drop.Role = tableName
		} else if drop.Quota == "_pending_" {
			drop.Quota = tableName
		} else if drop.Policy == "_pending_" {
			drop.Policy = tableName
		} else if drop.RowPolicy == "_pending_" {
			drop.RowPolicy = tableName
		} else if drop.SettingsProfile == "_pending_" {
			drop.SettingsProfile = tableName
		} else if drop.Index == "_pending_" {
			drop.Index = tableName
			// For DROP INDEX, parse ON table_name
			if p.currentIs(token.ON) {
				p.nextToken() // skip ON
				tableNamePart := p.parseIdentifierName()
				if p.currentIs(token.DOT) {
					p.nextToken()
					drop.Database = tableNamePart
					drop.Table = p.parseIdentifierName()
				} else {
					drop.Table = tableNamePart
				}
			}
		} else if dropDictionary {
			drop.Dictionary = tableName
			// Also set Table/Tables for backward compatibility with AST JSON
			drop.Tables = append(drop.Tables, &ast.TableIdentifier{
				Position: drop.Position,
				Database: database,
				Table:    tableName,
			})
			drop.Table = tableName
			if database != "" {
				drop.Database = database
			}
		} else if drop.DropDatabase {
			drop.Database = tableName
		} else {
			// First table - add to Tables list
			drop.Tables = append(drop.Tables, &ast.TableIdentifier{
				Position: drop.Position,
				Database: database,
				Table:    tableName,
			})
			drop.Table = tableName // Keep for backward compatibility
			if database != "" {
				drop.Database = database
			}
		}
	}

	// Handle multiple tables/users (DROP TABLE IF EXISTS t1, t2, t3 or DROP USER u1, u2@host)
	for p.currentIs(token.COMMA) {
		p.nextToken()
		pos := p.current.Pos
		name := p.parseIdentifierName()
		var database, tableName string
		if p.currentIs(token.DOT) {
			p.nextToken()
			database = name
			tableName = p.parseIdentifierName()
		} else {
			tableName = name
		}
		// Handle user@host syntax for additional users
		if dropUser && p.currentIs(token.IDENT) && p.current.Value == "@" {
			p.nextToken() // skip @
			if p.currentIs(token.IDENT) || p.currentIs(token.STRING) || p.current.Token.IsKeyword() {
				tableName = tableName + "@" + p.current.Value
				p.nextToken()
			}
		}
		if tableName != "" {
			drop.Tables = append(drop.Tables, &ast.TableIdentifier{
				Position: pos,
				Database: database,
				Table:    tableName,
			})
		}
	}

	// Handle PARALLEL WITH (drop multiple tables in parallel)
	// Syntax: DROP TABLE IF EXISTS t1 PARALLEL WITH DROP TABLE IF EXISTS t2
	for p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PARALLEL" {
		p.nextToken() // skip PARALLEL
		if p.currentIs(token.WITH) {
			p.nextToken() // skip WITH
			// Parse the next DROP statement
			if p.currentIs(token.DROP) {
				p.nextToken() // skip DROP
				// Handle TEMPORARY
				if p.currentIs(token.TEMPORARY) {
					p.nextToken()
				}
				// Skip TABLE/DATABASE/etc
				if p.currentIs(token.TABLE) || p.currentIs(token.DATABASE) || p.currentIs(token.VIEW) {
					p.nextToken()
				}
				// Handle IF EXISTS
				if p.currentIs(token.IF) {
					p.nextToken()
					if p.currentIs(token.EXISTS) {
						p.nextToken()
					}
				}
				// Parse table name
				pos := p.current.Pos
				name := p.parseIdentifierName()
				var database, tableName string
				if p.currentIs(token.DOT) {
					p.nextToken()
					database = name
					tableName = p.parseIdentifierName()
				} else {
					tableName = name
				}
				if tableName != "" {
					drop.Tables = append(drop.Tables, &ast.TableIdentifier{
						Position: pos,
						Database: database,
						Table:    tableName,
					})
				}
			}
		}
	}

	// Handle ON table or ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			if p.currentIs(token.IDENT) || p.currentIs(token.STRING) {
				drop.OnCluster = p.current.Value
				p.nextToken()
			}
		} else {
			// ON table_name (for DROP ROW POLICY, etc.)
			// Skip the table reference - can be db.table or db.* (wildcard)
			p.parseIdentifierName()
			if p.currentIs(token.DOT) {
				p.nextToken()
				// Handle wildcard (*) or table name
				if p.currentIs(token.ASTERISK) {
					p.nextToken()
				} else {
					p.parseIdentifierName()
				}
			}
		}
	}

	// Handle second ON CLUSTER (can appear after ON table)
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			if p.currentIs(token.IDENT) || p.currentIs(token.STRING) {
				drop.OnCluster = p.current.Value
				p.nextToken()
			}
		}
	}

	// Handle SYNC (can appear before or after FORMAT)
	if p.currentIs(token.SYNC) {
		drop.Sync = true
		p.nextToken()
	}

	// Handle FORMAT clause (for things like DROP TABLE ... FORMAT Null)
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		// Store format name (Null, etc.)
		if p.currentIs(token.NULL) {
			drop.Format = "Null"
			p.nextToken()
		} else if p.currentIs(token.IDENT) {
			drop.Format = p.current.Value
			p.nextToken()
		}
	}

	// Handle SYNC again (can also appear after FORMAT)
	if p.currentIs(token.SYNC) {
		drop.Sync = true
		p.nextToken()
	}

	// Handle NO DELAY
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "NO" {
		p.nextToken()
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DELAY" {
			p.nextToken()
		}
	}

	// Handle SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken() // skip SETTINGS
		drop.Settings = p.parseSettingsList()
	}

	return drop
}

func (p *Parser) parseAlter() *ast.AlterQuery {
	alter := &ast.AlterQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip ALTER

	// Skip TEMPORARY keyword if present
	if p.currentIs(token.TEMPORARY) {
		p.nextToken()
	}

	if !p.expect(token.TABLE) {
		return nil
	}

	// Parse table name (can start with a number in ClickHouse)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			alter.Database = tableName
			alter.Table = p.parseIdentifierName()
		} else {
			alter.Table = tableName
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			alter.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse commands (can be parenthesized for multiple mutations)
	for {
		// Handle parenthesized command syntax: ALTER TABLE t (DELETE WHERE ...), (UPDATE ...)
		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			cmd := p.parseAlterCommand()
			if cmd != nil {
				alter.Commands = append(alter.Commands, cmd)
			}
			p.expect(token.RPAREN)
		} else {
			cmd := p.parseAlterCommand()
			if cmd == nil {
				break
			}
			alter.Commands = append(alter.Commands, cmd)
		}

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.NULL) {
			alter.Format = "Null"
			p.nextToken()
		} else if p.currentIs(token.IDENT) {
			alter.Format = p.current.Value
			p.nextToken()
		}
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		alter.Settings = p.parseSettingsList()
	}

	return alter
}

func (p *Parser) parseAlterCommand() *ast.AlterCommand {
	cmd := &ast.AlterCommand{
		Position: p.current.Pos,
	}

	switch p.current.Token {
	case token.ADD:
		p.nextToken()
		if p.currentIs(token.COLUMN) {
			cmd.Type = ast.AlterAddColumn
			p.nextToken()
			// Handle IF NOT EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				if p.currentIs(token.NOT) {
					p.nextToken()
					if p.currentIs(token.EXISTS) {
						cmd.IfNotExists = true
						p.nextToken()
					}
				}
			}
			cmd.Column = p.parseColumnDeclaration()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "AFTER" {
				p.nextToken()
				if p.currentIs(token.IDENT) {
					// Handle dotted column names like AddedNested1.B
					afterCol := p.current.Value
					p.nextToken()
					for p.currentIs(token.DOT) {
						p.nextToken() // skip DOT
						if p.currentIs(token.IDENT) {
							afterCol += "." + p.current.Value
							p.nextToken()
						}
					}
					cmd.AfterColumn = afterCol
				}
			}
		} else if p.currentIs(token.INDEX) {
			cmd.Type = ast.AlterAddIndex
			p.nextToken()
			// Parse index name
			idxName := ""
			if p.currentIs(token.IDENT) {
				idxName = p.current.Value
				cmd.Index = idxName
				p.nextToken()
			}
			// Create IndexDef to store full index definition
			idx := &ast.IndexDefinition{
				Position: p.current.Pos,
				Name:     idxName,
			}
			// Parse expression - let parseExpression handle parentheses naturally
			// This allows (a, b, c) to be parsed as a tuple
			if !p.currentIs(token.IDENT) || strings.ToUpper(p.current.Value) != "TYPE" {
				idx.Expression = p.parseExpression(LOWEST)
				cmd.IndexExpr = idx.Expression
			}
			// Parse TYPE
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
				p.nextToken()
				// Type is a function call like bloom_filter(0.025) or vector_similarity('hnsw', 'L2Distance', 1)
				// Note: Index types can be keywords (e.g., SET) so we accept both IDENT and keywords
				pos := p.current.Pos
				typeName := ""
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					typeName = p.current.Value
					cmd.IndexType = typeName
					p.nextToken()
				}
				if typeName != "" {
					idx.Type = &ast.FunctionCall{
						Position: pos,
						Name:     typeName,
					}
					// Check for parentheses (type parameters)
					if p.currentIs(token.LPAREN) {
						p.nextToken()
						if !p.currentIs(token.RPAREN) {
							idx.Type.Arguments = p.parseExpressionList()
						}
						p.expect(token.RPAREN)
					}
				}
			}
			// Parse GRANULARITY
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "GRANULARITY" {
				p.nextToken()
				if p.currentIs(token.NUMBER) {
					granularity, _ := strconv.Atoi(p.current.Value)
					cmd.Granularity = granularity
					p.nextToken()
				}
			}
			// Parse AFTER
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "AFTER" {
				p.nextToken()
				// Index name can be an identifier or keyword
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					cmd.AfterIndex = p.current.Value
					p.nextToken()
				}
			}
			cmd.IndexDef = idx
		} else if p.currentIs(token.CONSTRAINT) {
			cmd.Type = ast.AlterAddConstraint
			p.nextToken()
			// Parse constraint name
			if p.currentIs(token.IDENT) {
				cmd.ConstraintName = p.current.Value
				p.nextToken()
			}
			// Parse CHECK or ASSUME
			if p.currentIs(token.CHECK) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ASSUME") {
				p.nextToken()
				cmd.Constraint = &ast.Constraint{
					Position:   p.current.Pos,
					Name:       cmd.ConstraintName,
					Expression: p.parseExpression(LOWEST),
				}
			}
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
			cmd.Type = ast.AlterAddProjection
			p.nextToken()
			cmd.Projection = p.parseProjection()
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
			cmd.Type = ast.AlterAddStatistics
			p.nextToken()
			// Handle IF NOT EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				if p.currentIs(token.NOT) {
					p.nextToken()
					if p.currentIs(token.EXISTS) {
						cmd.IfNotExists = true
						p.nextToken()
					}
				}
			}
			// Parse column list (comma-separated identifiers)
			cmd.StatisticsColumns = p.parseStatisticsColumnList()
			// Parse TYPE clause
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
				p.nextToken()
				cmd.StatisticsTypes = p.parseStatisticsTypeList()
			}
		}
	case token.DROP:
		p.nextToken()
		if p.currentIs(token.COLUMN) {
			cmd.Type = ast.AlterDropColumn
			p.nextToken()
			if p.currentIs(token.IF) {
				p.nextToken()
				p.expect(token.EXISTS)
				cmd.IfExists = true
			}
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				// Handle dotted column names like NestedColumn.A
				colName := p.current.Value
				p.nextToken()
				for p.currentIs(token.DOT) {
					p.nextToken() // skip DOT
					if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
						colName += "." + p.current.Value
						p.nextToken()
					}
				}
				cmd.ColumnName = colName
			}
		} else if p.currentIs(token.INDEX) {
			cmd.Type = ast.AlterDropIndex
			p.nextToken()
			// Handle IF EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				if p.currentIs(token.EXISTS) {
					cmd.IfExists = true
					p.nextToken()
				}
			}
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				cmd.Index = p.current.Value
				p.nextToken()
			}
		} else if p.currentIs(token.CONSTRAINT) {
			cmd.Type = ast.AlterDropConstraint
			p.nextToken()
			if p.currentIs(token.IDENT) {
				cmd.ConstraintName = p.current.Value
				p.nextToken()
			}
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DETACHED" {
			// DROP DETACHED PARTITION
			p.nextToken() // skip DETACHED
			if p.currentIs(token.PARTITION) {
				p.nextToken() // skip PARTITION
				cmd.Type = ast.AlterDropDetachedPartition
				cmd.Partition = p.parseExpression(LOWEST)
			}
		} else if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterDropPartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PART" {
			// DROP PART is treated like DROP PARTITION in ClickHouse
			cmd.Type = ast.AlterDropPartition
			cmd.IsPart = true
			p.nextToken()
			cmd.Partition = p.parseExpression(LOWEST)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
			cmd.Type = ast.AlterDropProjection
			p.nextToken()
			if p.currentIs(token.IF) {
				p.nextToken()
				p.expect(token.EXISTS)
				cmd.IfExists = true
			}
			if p.currentIs(token.IDENT) {
				cmd.ProjectionName = p.current.Value
				p.nextToken()
			}
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
			cmd.Type = ast.AlterDropStatistics
			p.nextToken()
			if p.currentIs(token.IF) {
				p.nextToken()
				p.expect(token.EXISTS)
				cmd.IfExists = true
			}
			cmd.StatisticsColumns = p.parseStatisticsColumnList()
		}
	case token.IDENT:
		// Handle CLEAR, MATERIALIZE
		upper := strings.ToUpper(p.current.Value)
		if upper == "CLEAR" {
			p.nextToken()
			if p.currentIs(token.INDEX) {
				cmd.Type = ast.AlterClearIndex
				p.nextToken()
				if p.currentIs(token.IDENT) {
					cmd.Index = p.current.Value
					p.nextToken()
				}
				// Parse IN PARTITION clause
				if p.currentIs(token.IN) {
					p.nextToken() // skip IN
					if p.currentIs(token.PARTITION) {
						p.nextToken() // skip PARTITION
						cmd.Partition = p.parseExpression(LOWEST)
					}
				}
			} else if p.currentIs(token.COLUMN) {
				cmd.Type = ast.AlterClearColumn
				p.nextToken()
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					cmd.ColumnName = p.current.Value
					p.nextToken()
				}
				// Parse IN PARTITION
				if p.currentIs(token.IN) {
					p.nextToken() // skip IN
					if p.currentIs(token.PARTITION) {
						p.nextToken() // skip PARTITION
						cmd.Partition = p.parseExpression(LOWEST)
					}
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
				cmd.Type = ast.AlterClearProjection
				p.nextToken()
				if p.currentIs(token.IDENT) {
					cmd.ProjectionName = p.current.Value
					p.nextToken()
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
				cmd.Type = ast.AlterClearStatistics
				p.nextToken()
				if p.currentIs(token.IF) {
					p.nextToken()
					p.expect(token.EXISTS)
					cmd.IfExists = true
				}
				cmd.StatisticsColumns = p.parseStatisticsColumnList()
			}
		} else if upper == "MATERIALIZE" {
			p.nextToken()
			if p.currentIs(token.INDEX) {
				cmd.Type = ast.AlterMaterializeIndex
				p.nextToken()
				if p.currentIs(token.IDENT) {
					cmd.Index = p.current.Value
					p.nextToken()
				}
				// Parse IN PARTITION ID clause
				if p.currentIs(token.IN) {
					p.nextToken() // skip IN
					if p.currentIs(token.PARTITION) {
						p.nextToken() // skip PARTITION
						// Check for PARTITION ID 'value' syntax
						if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
							p.nextToken()
							cmd.PartitionIsID = true
						}
						cmd.Partition = p.parseExpression(LOWEST)
					}
				}
			} else if p.currentIs(token.COLUMN) {
				cmd.Type = ast.AlterMaterializeColumn
				p.nextToken()
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					cmd.ColumnName = p.current.Value
					p.nextToken()
				}
				// Parse IN PARTITION clause
				if p.currentIs(token.IN) {
					p.nextToken() // skip IN
					if p.currentIs(token.PARTITION) {
						p.nextToken() // skip PARTITION
						// Check for PARTITION ID 'value' syntax
						if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
							p.nextToken()
							cmd.PartitionIsID = true
						}
						cmd.Partition = p.parseExpression(LOWEST)
					}
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
				cmd.Type = ast.AlterMaterializeProjection
				p.nextToken()
				if p.currentIs(token.IDENT) {
					cmd.ProjectionName = p.current.Value
					p.nextToken()
				}
			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
				cmd.Type = ast.AlterMaterializeStatistics
				p.nextToken()
				if p.currentIs(token.IF) {
					p.nextToken()
					p.expect(token.EXISTS)
					cmd.IfExists = true
				}
				cmd.StatisticsColumns = p.parseStatisticsColumnList()
			} else if p.currentIs(token.TTL) {
				cmd.Type = ast.AlterMaterializeTTL
				p.nextToken()
			}
		} else if upper == "MOVE" {
			p.nextToken()
			if p.currentIs(token.PARTITION) {
				cmd.Type = ast.AlterMovePartition
				p.nextToken()
				// Check for PARTITION ID 'value' syntax
				if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
					p.nextToken()
					cmd.PartitionIsID = true
				}
				cmd.Partition = p.parseExpression(LOWEST)
				// Parse TO TABLE/DISK/VOLUME destination
				if p.currentIs(token.TO) {
					p.nextToken()
					if p.currentIs(token.TABLE) {
						p.nextToken()
						// Parse destination table (can be qualified: database.table)
						destName := p.parseIdentifierName()
						if p.currentIs(token.DOT) {
							p.nextToken()
							cmd.ToDatabase = destName
							cmd.ToTable = p.parseIdentifierName()
						} else {
							cmd.ToTable = destName
						}
					} else if p.currentIs(token.IDENT) && (strings.ToUpper(p.current.Value) == "DISK" || strings.ToUpper(p.current.Value) == "VOLUME") {
						// MOVE PARTITION ... TO DISK 'disk_name' or TO VOLUME 'volume_name'
						p.nextToken() // skip DISK/VOLUME
						if p.currentIs(token.STRING) {
							p.nextToken() // skip the disk/volume name
						}
					}
				}
			}
		} else if upper == "REMOVE" {
			p.nextToken()
			// REMOVE SAMPLE BY
			if p.currentIs(token.SAMPLE) {
				p.nextToken() // skip SAMPLE
				if p.currentIs(token.BY) {
					p.nextToken() // skip BY
				}
				cmd.Type = ast.AlterRemoveSampleBy
			} else if p.currentIs(token.TTL) {
				// REMOVE TTL (table-level TTL)
				p.nextToken() // skip TTL
				cmd.Type = ast.AlterRemoveTTL
			}
		} else if upper == "RESET" {
			p.nextToken() // skip RESET
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "SETTING" {
				p.nextToken() // skip SETTING
				cmd.Type = ast.AlterResetSetting
				// Parse comma-separated list of setting names
				for {
					if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
						cmd.ResetSettings = append(cmd.ResetSettings, p.current.Value)
						p.nextToken()
					}
					if p.currentIs(token.COMMA) {
						p.nextToken()
					} else {
						break
					}
				}
			}
		} else {
			return nil
		}
	case token.MODIFY:
		p.nextToken()
		if p.currentIs(token.COLUMN) {
			cmd.Type = ast.AlterModifyColumn
			p.nextToken()
			// Handle IF EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				if p.currentIs(token.EXISTS) {
					cmd.IfExists = true
					p.nextToken()
				}
			}
			// Handle MODIFY COLUMN name REMOVE ... (e.g., REMOVE COMMENT)
			// Check if the next token after column name is REMOVE
			if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "REMOVE" {
				// Just parse column name without type
				colName := p.current.Value
				p.nextToken() // skip column name
				cmd.Column = &ast.ColumnDeclaration{Name: colName}
				// Skip REMOVE COMMENT etc. but stop at SETTINGS clause
				for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.COMMA) && !p.currentIs(token.SETTINGS) {
					p.nextToken()
				}
			} else if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && p.peek.Token == token.MODIFY {
				// MODIFY COLUMN colname MODIFY SETTING key = value
				colName := p.current.Value
				p.nextToken() // skip column name
				cmd.Column = &ast.ColumnDeclaration{Name: colName}
				p.nextToken() // skip MODIFY
				if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "SETTING" {
					p.nextToken() // skip SETTING
					cmd.Settings = p.parseSettingsList()
				}
			} else if (p.currentIs(token.IDENT) || p.current.Token.IsKeyword()) && p.peek.Token == token.IDENT && strings.ToUpper(p.peek.Value) == "RESET" {
				// MODIFY COLUMN colname RESET SETTING key, key2, ...
				colName := p.current.Value
				p.nextToken() // skip column name
				cmd.Column = &ast.ColumnDeclaration{Name: colName}
				p.nextToken() // skip RESET
				if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "SETTING" {
					p.nextToken() // skip SETTING
					// Parse comma-separated list of setting names
					for {
						if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
							cmd.ResetSettings = append(cmd.ResetSettings, p.current.Value)
							p.nextToken()
						}
						if p.currentIs(token.COMMA) {
							p.nextToken()
						} else {
							break
						}
					}
				}
			} else {
				cmd.Column = p.parseColumnDeclaration()
			}
			// Parse AFTER column_name clause
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "AFTER" {
				p.nextToken() // skip AFTER
				cmd.AfterColumn = p.parseIdentifierName()
			}
		} else if p.currentIs(token.TTL) {
			cmd.Type = ast.AlterModifyTTL
			p.nextToken()
			cmd.TTL = &ast.TTLClause{
				Position: p.current.Pos,
			}
			// Parse TTL elements using parseTTLElement (captures WHERE clause)
			for {
				elem := p.parseTTLElement()
				cmd.TTL.Elements = append(cmd.TTL.Elements, elem)
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
			// Keep backward compatibility with Expression/Expressions fields
			if len(cmd.TTL.Elements) > 0 {
				cmd.TTL.Expression = cmd.TTL.Elements[0].Expr
				for i := 1; i < len(cmd.TTL.Elements); i++ {
					cmd.TTL.Expressions = append(cmd.TTL.Expressions, cmd.TTL.Elements[i].Expr)
				}
			}
		} else if p.currentIs(token.SETTINGS) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "SETTING") {
			// Both SETTINGS and SETTING (singular) are accepted
			cmd.Type = ast.AlterModifySetting
			p.nextToken()
			cmd.Settings = p.parseSettingsList()
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "STATISTICS" {
			cmd.Type = ast.AlterModifyStatistics
			p.nextToken()
			// Parse column list (comma-separated identifiers)
			cmd.StatisticsColumns = p.parseStatisticsColumnList()
			// Parse TYPE clause
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TYPE" {
				p.nextToken()
				cmd.StatisticsTypes = p.parseStatisticsTypeList()
			}
		} else if p.currentIs(token.COMMENT) {
			// MODIFY COMMENT 'comment string'
			cmd.Type = ast.AlterModifyComment
			p.nextToken()
			if p.currentIs(token.STRING) {
				cmd.Comment = p.current.Value
				p.nextToken()
			}
		} else if p.currentIs(token.ORDER) {
			// MODIFY ORDER BY (expr, ...)
			cmd.Type = ast.AlterModifyOrderBy
			p.nextToken() // skip ORDER
			if p.currentIs(token.BY) {
				p.nextToken() // skip BY
			}
			// Parse the order by expression(s)
			if p.currentIs(token.LPAREN) {
				p.nextToken() // skip (
				cmd.OrderByExpr = p.parseExpressionList()
				p.expect(token.RPAREN)
			} else {
				// Single expression without parentheses
				cmd.OrderByExpr = []ast.Expression{p.parseExpression(LOWEST)}
			}
		} else if p.currentIs(token.SAMPLE) {
			// MODIFY SAMPLE BY expr
			cmd.Type = ast.AlterModifySampleBy
			p.nextToken() // skip SAMPLE
			if p.currentIs(token.BY) {
				p.nextToken() // skip BY
			}
			cmd.SampleByExpr = p.parseExpression(LOWEST)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "QUERY" {
			// MODIFY QUERY SELECT ...
			cmd.Type = ast.AlterModifyQuery
			p.nextToken() // skip QUERY
			cmd.Query = p.parseSelectWithUnion()
		}
	case token.RENAME:
		p.nextToken()
		if p.currentIs(token.COLUMN) {
			cmd.Type = ast.AlterRenameColumn
			p.nextToken()
			// Handle IF EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				p.expect(token.EXISTS)
				cmd.IfExists = true
			}
			// Parse column name (can be dotted like n.x for nested columns)
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				cmd.ColumnName = p.parseDottedIdentifier()
			}
			if p.currentIs(token.TO) {
				p.nextToken()
				// Parse new column name (can also be dotted)
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					cmd.NewName = p.parseDottedIdentifier()
				}
			}
		}
	case token.COMMENT:
		p.nextToken()
		if p.currentIs(token.COLUMN) {
			cmd.Type = ast.AlterCommentColumn
			p.nextToken()
			// Handle IF EXISTS
			if p.currentIs(token.IF) {
				p.nextToken()
				p.expect(token.EXISTS)
				cmd.IfExists = true
			}
			// Parse column name
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				cmd.ColumnName = p.current.Value
				p.nextToken()
			}
			// Parse comment string
			if p.currentIs(token.STRING) {
				cmd.Comment = p.current.Value
				p.nextToken()
			}
		}
	case token.DETACH:
		p.nextToken()
		if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterDetachPartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PART" {
			// DETACH PART is displayed as DROP_PARTITION in ClickHouse EXPLAIN
			cmd.Type = ast.AlterDropPartition
			cmd.IsPart = true
			p.nextToken()
			cmd.Partition = p.parseExpression(LOWEST)
		}
	case token.ATTACH:
		p.nextToken()
		if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterAttachPartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
			// Handle FROM table (ATTACH PARTITION ... FROM table)
			if p.currentIs(token.FROM) {
				p.nextToken()
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					cmd.FromTable = p.current.Value
					p.nextToken()
				}
			}
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PART" {
			// ATTACH PART uses ATTACH_PARTITION type in ClickHouse EXPLAIN
			cmd.Type = ast.AlterAttachPartition
			cmd.IsPart = true
			p.nextToken()
			cmd.Partition = p.parseExpression(LOWEST)
		}
	case token.FREEZE:
		p.nextToken()
		if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterFreezePartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
		} else {
			cmd.Type = ast.AlterFreeze
		}
	case token.REPLACE:
		p.nextToken()
		if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterReplacePartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
			if p.currentIs(token.FROM) {
				p.nextToken()
				if p.currentIs(token.IDENT) {
					cmd.FromTable = p.current.Value
					p.nextToken()
				}
			}
		}
	case token.FETCH:
		p.nextToken()
		if p.currentIs(token.PARTITION) {
			cmd.Type = ast.AlterFetchPartition
			p.nextToken()
			// Check for PARTITION ID 'value' syntax
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
				p.nextToken()
				cmd.PartitionIsID = true
			}
			cmd.Partition = p.parseExpression(LOWEST)
			// FROM path
			if p.currentIs(token.FROM) {
				p.nextToken()
				if p.currentIs(token.STRING) {
					cmd.FromPath = p.current.Value
					p.nextToken()
				}
			}
		}
	case token.APPLY:
		// APPLY PATCHES IN PARTITION expr
		// APPLY DELETED MASK [IN PARTITION expr]
		p.nextToken() // skip APPLY
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PATCHES" {
			p.nextToken() // skip PATCHES
			cmd.Type = ast.AlterApplyPatches
			if p.currentIs(token.IN) {
				p.nextToken() // skip IN
				if p.currentIs(token.PARTITION) {
					p.nextToken() // skip PARTITION
					cmd.Partition = p.parseExpression(LOWEST)
				}
			}
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DELETED" {
			p.nextToken() // skip DELETED
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "MASK" {
				p.nextToken() // skip MASK
				cmd.Type = ast.AlterApplyDeletedMask
				if p.currentIs(token.IN) {
					p.nextToken() // skip IN
					if p.currentIs(token.PARTITION) {
						p.nextToken() // skip PARTITION
						// Check for PARTITION ID 'value' syntax
						if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
							p.nextToken()
							cmd.PartitionIsID = true
						}
						cmd.Partition = p.parseExpression(LOWEST)
					}
				}
			}
		}
	case token.DELETE:
		// DELETE WHERE condition - mutation to delete rows
		cmd.Type = ast.AlterDeleteWhere
		p.nextToken() // skip DELETE
		if p.currentIs(token.WHERE) {
			p.nextToken() // skip WHERE
			cmd.Where = p.parseExpression(LOWEST)
		}
	case token.UPDATE:
		// UPDATE col = expr, ... WHERE condition - mutation to update rows
		cmd.Type = ast.AlterUpdate
		p.nextToken() // skip UPDATE
		// Parse assignments
		for {
			// Column name can be IDENT or keyword (e.g., key, value)
			if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
				break
			}
			assign := &ast.Assignment{
				Position: p.current.Pos,
				Column:   p.current.Value,
			}
			p.nextToken() // skip column name
			if p.currentIs(token.EQ) {
				p.nextToken() // skip =
				assign.Value = p.parseExpression(LOWEST)
			}
			cmd.Assignments = append(cmd.Assignments, assign)
			if !p.currentIs(token.COMMA) {
				break
			}
			p.nextToken() // skip comma
		}
		// Handle IN PARTITION (UPDATE ... IN PARTITION <partition> WHERE ...)
		// The expression parser may have incorrectly consumed "expr IN PARTITION" as an InExpression.
		// Check if the last assignment value is an InExpression with right side being "PARTITION".
		if len(cmd.Assignments) > 0 {
			lastAssign := cmd.Assignments[len(cmd.Assignments)-1]
			if inExpr, ok := lastAssign.Value.(*ast.InExpr); ok && len(inExpr.List) == 1 {
				if ident, ok := inExpr.List[0].(*ast.Identifier); ok && strings.ToUpper(ident.Name()) == "PARTITION" {
					// Fix the mis-parse: the actual assignment value is the left side of IN
					lastAssign.Value = inExpr.Expr
					// Check for PARTITION ID 'value' syntax
					if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
						p.nextToken()
						cmd.PartitionIsID = true
					}
					// Current token should be the partition expression (e.g., ALL or '1')
					cmd.Partition = p.parseExpression(LOWEST)
				}
			}
		}
		if p.currentIs(token.IN) {
			p.nextToken() // skip IN
			if p.currentIs(token.PARTITION) {
				p.nextToken() // skip PARTITION
				// Check for PARTITION ID 'value' syntax
				if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
					p.nextToken()
					cmd.PartitionIsID = true
				}
				cmd.Partition = p.parseExpression(LOWEST)
			}
		}
		if p.currentIs(token.WHERE) {
			p.nextToken() // skip WHERE
			cmd.Where = p.parseExpression(LOWEST)
		}
	default:
		return nil
	}

	return cmd
}

func (p *Parser) parseTruncate() *ast.TruncateQuery {
	trunc := &ast.TruncateQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip TRUNCATE

	// Handle TEMPORARY keyword
	if p.currentIs(token.TEMPORARY) {
		trunc.Temporary = true
		p.nextToken()
	}

	// Handle TABLE or DATABASE keyword
	if p.currentIs(token.TABLE) {
		p.nextToken()
	} else if p.currentIs(token.DATABASE) {
		trunc.TruncateDatabase = true
		p.nextToken()
	}

	// Handle IF EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.EXISTS) {
			trunc.IfExists = true
			p.nextToken()
		}
	}

	// Parse table name (can start with a number in ClickHouse)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			trunc.Database = tableName
			trunc.Table = p.parseIdentifierName()
		} else {
			trunc.Table = tableName
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			trunc.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle SETTINGS
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		trunc.Settings = p.parseSettingsList()
	}

	return trunc
}

func (p *Parser) parseUndrop() *ast.UndropQuery {
	undrop := &ast.UndropQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip UNDROP

	if p.currentIs(token.TABLE) {
		p.nextToken()
	}

	// Parse table name (can start with a number in ClickHouse)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			undrop.Database = tableName
			undrop.Table = p.parseIdentifierName()
		} else {
			undrop.Table = tableName
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			undrop.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle UUID
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UUID" {
		p.nextToken()
		if p.currentIs(token.STRING) {
			undrop.UUID = p.current.Value
			p.nextToken()
		}
	}

	// Handle FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.NULL) {
			undrop.Format = "Null"
			p.nextToken()
		} else if p.currentIs(token.IDENT) {
			undrop.Format = p.current.Value
			p.nextToken()
		}
	}

	return undrop
}

func (p *Parser) parseUpdate() *ast.UpdateQuery {
	update := &ast.UpdateQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip UPDATE

	// Parse table name (can be database.table)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			update.Database = tableName
			update.Table = p.parseIdentifierName()
		} else {
			update.Table = tableName
		}
	}

	// Expect SET keyword
	if !p.currentIs(token.SET) {
		return update
	}
	p.nextToken() // skip SET

	// Parse assignments: col = expr, col = expr, ...
	for {
		if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
			break
		}
		assign := &ast.Assignment{
			Position: p.current.Pos,
			Column:   p.current.Value,
		}
		p.nextToken() // skip column name
		if p.currentIs(token.EQ) {
			p.nextToken() // skip =
			assign.Value = p.parseExpression(LOWEST)
		}
		update.Assignments = append(update.Assignments, assign)
		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken() // skip comma
	}

	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken() // skip WHERE
		update.Where = p.parseExpression(LOWEST)
	}

	return update
}

func (p *Parser) parseDelete() *ast.DeleteQuery {
	del := &ast.DeleteQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DELETE

	// Skip optional FROM
	if p.currentIs(token.FROM) {
		p.nextToken()
	}

	// Parse table name (can be database.table)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			del.Database = tableName
			del.Table = p.parseIdentifierName()
		} else {
			del.Table = tableName
		}
	}

	// Parse ON CLUSTER clause
	if p.currentIs(token.ON) {
		p.nextToken() // skip ON
		if p.currentIs(token.CLUSTER) {
			p.nextToken() // skip CLUSTER
			del.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse IN PARTITION clause
	if p.currentIs(token.IN) {
		p.nextToken() // skip IN
		if p.currentIs(token.PARTITION) {
			p.nextToken() // skip PARTITION
			del.Partition = p.parseExpression(LOWEST)
		}
	}

	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken() // skip WHERE
		del.Where = p.parseExpression(LOWEST)
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken() // skip SETTINGS
		del.Settings = p.parseSettingsList()
	}

	return del
}

func (p *Parser) parseUse() *ast.UseQuery {
	use := &ast.UseQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip USE

	// Skip optional DATABASE keyword (USE DATABASE dbname is equivalent to USE dbname)
	// But only if DATABASE is followed by another identifier/keyword (not semicolon or EOF)
	// e.g., "USE DATABASE d1" vs "USE database" where database is the db name
	if p.currentIs(token.DATABASE) && !p.peekIs(token.SEMICOLON) && !p.peekIs(token.EOF) {
		p.nextToken()
	}

	// Database name can be an identifier or a keyword like DEFAULT (can also start with number)
	use.Database = p.parseIdentifierName()

	return use
}

func (p *Parser) parseDescribe() *ast.DescribeQuery {
	desc := &ast.DescribeQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DESCRIBE or DESC

	if p.currentIs(token.TABLE) {
		p.nextToken()
	}

	// Check for subquery: DESCRIBE (SELECT ...)
	if p.currentIs(token.LPAREN) {
		desc.TableExpr = p.parseTableExpression()
	} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		// Parse table name or table function
		// Table functions look like: format(CSV, '...'), url('...'), s3Cluster(...)
		pos := p.current.Pos
		tableName := p.current.Value
		p.nextToken()

		// Check if this is a function call (table function)
		if p.currentIs(token.LPAREN) {
			desc.TableFunction = p.parseFunctionCall(tableName, pos)
		} else if p.currentIs(token.DOT) {
			p.nextToken()
			desc.Database = tableName
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				desc.Table = p.current.Value
				p.nextToken()
			}
		} else {
			desc.Table = tableName
		}
	}

	// Skip FINAL keyword if present (can appear after table function)
	if p.currentIs(token.FINAL) {
		p.nextToken()
	}

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.currentIs(token.NULL) || p.current.Token.IsKeyword() {
			desc.Format = p.current.Value
			p.nextToken()
		}
	}

	// Parse SETTINGS clause (can come after FORMAT)
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		desc.Settings = p.parseSettingsList()
	}

	return desc
}

func (p *Parser) parseShow() ast.Statement {
	pos := p.current.Pos

	p.nextToken() // skip SHOW

	// Handle SHOW PRIVILEGES first - it has its own statement type
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PRIVILEGES" {
		p.nextToken()
		return &ast.ShowPrivilegesQuery{Position: pos}
	}

	// Handle SHOW GRANTS - it has its own statement type
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "GRANTS" {
		query := &ast.ShowGrantsQuery{Position: pos}
		// Skip tokens until FORMAT or end of statement
		for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
			p.nextToken()
		}
		// Parse FORMAT clause if present
		if p.currentIs(token.FORMAT) {
			p.nextToken()
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				query.Format = p.current.Value
				p.nextToken()
			}
		}
		return query
	}

	show := &ast.ShowQuery{
		Position: pos,
	}

	// Handle TEMPORARY keyword (SHOW TEMPORARY TABLES)
	if p.currentIs(token.TEMPORARY) {
		show.Temporary = true
		p.nextToken()
	}

	switch p.current.Token {
	case token.TABLES:
		show.ShowType = ast.ShowTables
		p.nextToken()
	case token.TABLE:
		// SHOW TABLE is equivalent to SHOW CREATE TABLE
		show.ShowType = ast.ShowCreate
		p.nextToken()
	case token.DATABASES:
		show.ShowType = ast.ShowDatabases
		p.nextToken()
	case token.DATABASE:
		// SHOW DATABASE is equivalent to SHOW CREATE DATABASE
		show.ShowType = ast.ShowCreateDB
		p.nextToken()
	case token.COLUMNS:
		show.ShowType = ast.ShowColumns
		p.nextToken()
	case token.INDEX:
		// SHOW INDEX FROM table - treat as ShowColumns (ClickHouse maps to ShowColumns)
		show.ShowType = ast.ShowColumns
		p.nextToken()
	case token.CREATE:
		p.nextToken()
		if p.currentIs(token.DATABASE) {
			show.ShowType = ast.ShowCreateDB
			p.nextToken()
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "QUOTA" {
			// SHOW CREATE QUOTA <name>
			p.nextToken()
			query := &ast.ShowCreateQuotaQuery{Position: pos}
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				query.Name = p.current.Value
				p.nextToken()
			}
			// Parse FORMAT clause if present
			if p.currentIs(token.FORMAT) {
				p.nextToken()
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					query.Format = p.current.Value
					p.nextToken()
				}
			}
			return query
		} else if p.currentIs(token.SETTINGS) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROFILE") {
			// SHOW CREATE SETTINGS PROFILE or SHOW CREATE PROFILE
			return p.parseShowCreateSettingsProfile(pos)
		} else if p.currentIs(token.IDENT) && (strings.ToUpper(p.current.Value) == "ROW" || strings.ToUpper(p.current.Value) == "POLICY") {
			// SHOW CREATE ROW POLICY or SHOW CREATE POLICY
			return p.parseShowCreateRowPolicy(pos)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ROLE" {
			// SHOW CREATE ROLE
			return p.parseShowCreateRole(pos)
		} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
			show.ShowType = ast.ShowCreateDictionary
			p.nextToken()
		} else if p.currentIs(token.VIEW) {
			show.ShowType = ast.ShowCreateView
			p.nextToken()
		} else if p.currentIs(token.USER) {
			show.ShowType = ast.ShowCreateUser
			p.nextToken()
			// Skip user name and host pattern until FORMAT or end
			// Also check for commas to detect multiple users
			for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
				if p.currentIs(token.COMMA) {
					show.MultipleUsers = true
				}
				p.nextToken()
			}
			// Parse FORMAT clause if present
			if p.currentIs(token.FORMAT) {
				p.nextToken()
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					show.Format = p.current.Value
					p.nextToken()
				}
			}
		} else {
			show.ShowType = ast.ShowCreate
			// Handle SHOW CREATE TABLE, SHOW CREATE TEMPORARY TABLE, etc.
			if p.currentIs(token.TEMPORARY) {
				p.nextToken()
			}
			if p.currentIs(token.TABLE) {
				p.nextToken()
			}
		}
	case token.SETTINGS:
		show.ShowType = ast.ShowSettings
		p.nextToken()
	case token.FULL:
		// SHOW FULL COLUMNS/FIELDS FROM table - treat as ShowColumns
		p.nextToken()
		if p.currentIs(token.COLUMNS) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "FIELDS") {
			p.nextToken()
		}
		show.ShowType = ast.ShowColumns
	default:
		// Handle SHOW PROCESSLIST, SHOW DICTIONARIES, SHOW FUNCTIONS, etc.
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			switch upper {
			case "PROCESSLIST":
				show.ShowType = ast.ShowProcesses
			case "DICTIONARIES":
				show.ShowType = ast.ShowDictionaries
			case "FUNCTIONS":
				show.ShowType = ast.ShowFunctions
			case "SETTING":
				show.ShowType = ast.ShowSetting
			case "INDEXES", "INDICES", "KEYS", "FIELDS":
				// SHOW INDEXES/INDICES/KEYS/FIELDS FROM table - treat as ShowColumns
				show.ShowType = ast.ShowColumns
			case "FULL":
				// SHOW FULL COLUMNS/FIELDS FROM table - treat as ShowColumns
				p.nextToken()
				if p.currentIs(token.COLUMNS) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "FIELDS") {
					p.nextToken()
				}
				show.ShowType = ast.ShowColumns
				// Don't consume another token, fall through to FROM parsing
				goto parseFrom
			case "EXTENDED":
				// SHOW EXTENDED INDEX FROM table - treat as ShowColumns
				p.nextToken()
				if p.currentIs(token.INDEX) {
					p.nextToken()
				}
				show.ShowType = ast.ShowColumns
				// Don't consume another token, fall through to FROM parsing
				goto parseFrom
			case "CHANGED":
				// SHOW CHANGED SETTINGS - treat as ShowSettings
				p.nextToken()
				if p.currentIs(token.SETTINGS) {
					show.ShowType = ast.ShowSettings
					p.nextToken()
				}
				goto parseFrom
			}
			p.nextToken()
		}
	}

parseFrom:

	// Parse FROM clause (or table/database name for SHOW CREATE TABLE/DATABASE/DICTIONARY/VIEW)
	showCreateTypes := show.ShowType == ast.ShowCreate || show.ShowType == ast.ShowCreateDB || show.ShowType == ast.ShowCreateDictionary || show.ShowType == ast.ShowCreateView
	if p.currentIs(token.FROM) || (showCreateTypes && (p.currentIs(token.IDENT) || p.current.Token.IsKeyword())) {
		if p.currentIs(token.FROM) {
			p.nextToken()
		}
		// Parse table name which can be database.table or just table
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			name := p.current.Value
			p.nextToken()
			if p.currentIs(token.DOT) {
				p.nextToken()
				show.Database = name
				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
					show.From = p.current.Value
					p.nextToken()
				}
			} else {
				show.From = name
			}
		}
	}

	// Handle SHOW INDEX FROM table FROM database syntax (second FROM for database)
	if p.currentIs(token.FROM) && show.ShowType == ast.ShowColumns {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			show.Database = p.current.Value
			p.nextToken()
		}
	}

	// Parse NOT LIKE, LIKE or ILIKE clause
	if p.currentIs(token.NOT) {
		p.nextToken()
		if p.currentIs(token.LIKE) || p.currentIs(token.ILIKE) {
			p.nextToken()
			if p.currentIs(token.STRING) {
				// NOT LIKE - store the pattern with a prefix to indicate negation
				show.Like = "!" + p.current.Value // Using ! prefix to indicate NOT LIKE
				p.nextToken()
			}
		}
	} else if p.currentIs(token.LIKE) || p.currentIs(token.ILIKE) {
		p.nextToken()
		if p.currentIs(token.STRING) {
			show.Like = p.current.Value
			p.nextToken()
		}
	}

	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken()
		show.Where = p.parseExpression(LOWEST)
	}

	// Parse LIMIT clause
	if p.currentIs(token.LIMIT) {
		p.nextToken()
		show.Limit = p.parseExpression(LOWEST)
	}

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			show.Format = p.current.Value
			p.nextToken()
		}
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		show.HasSettings = true
		// Skip SETTINGS and all settings key-value pairs
		p.nextToken()
		for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) && !p.currentIs(token.FORMAT) {
			p.nextToken()
		}
	}

	return show
}

func (p *Parser) parseExplain() *ast.ExplainQuery {
	explain := &ast.ExplainQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip EXPLAIN

	// Parse explain type
	if p.currentIs(token.IDENT) {
		switch strings.ToUpper(p.current.Value) {
		case "AST":
			explain.ExplainType = ast.ExplainAST
			explain.ExplicitType = true
			p.nextToken()
		case "SYNTAX":
			explain.ExplainType = ast.ExplainSyntax
			explain.ExplicitType = true
			p.nextToken()
		case "PLAN":
			explain.ExplainType = ast.ExplainPlan
			explain.ExplicitType = true
			p.nextToken()
		case "PIPELINE":
			explain.ExplainType = ast.ExplainPipeline
			explain.ExplicitType = true
			p.nextToken()
		case "ESTIMATE":
			explain.ExplainType = ast.ExplainEstimate
			explain.ExplicitType = true
			p.nextToken()
		case "QUERY":
			// EXPLAIN QUERY TREE
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TREE" {
				p.nextToken()
			}
			explain.ExplainType = ast.ExplainQueryTree
			explain.ExplicitType = true
		case "CURRENT":
			// EXPLAIN CURRENT TRANSACTION
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TRANSACTION" {
				p.nextToken()
			}
			explain.ExplainType = ast.ExplainCurrentTransaction
			explain.ExplicitType = true
			return explain // No statement follows CURRENT TRANSACTION
		default:
			explain.ExplainType = ast.ExplainPlan
		}
	}

	// Parse EXPLAIN options (e.g., header = 1, optimize = 0)
	// These come before the actual statement
	// Options can be identifiers or keywords like OPTIMIZE followed by =
	var optionParts []string
	for p.peekIs(token.EQ) && !p.currentIs(token.SELECT) && !p.currentIs(token.WITH) {
		// This is an option (name = value)
		explain.HasSettings = true
		optionName := p.current.Value
		p.nextToken() // skip option name
		p.nextToken() // skip =
		// Get the value
		optionValue := p.current.Value
		if p.currentIs(token.NUMBER) || p.currentIs(token.STRING) || p.currentIs(token.IDENT) {
			optionParts = append(optionParts, optionName+" = "+optionValue)
		}
		p.parseExpression(LOWEST) // skip value expression (may consume more tokens)
		// Skip comma if present
		if p.currentIs(token.COMMA) {
			p.nextToken()
		}
	}
	if len(optionParts) > 0 {
		explain.OptionsString = strings.Join(optionParts, ", ")
	}

	// Parse the statement being explained
	explain.Statement = p.parseStatement()

	return explain
}

func (p *Parser) parseSet() *ast.SetQuery {
	set := &ast.SetQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip SET

	set.Settings = p.parseSettingsList()

	return set
}

func (p *Parser) parseSetRole() *ast.SetRoleQuery {
	query := &ast.SetRoleQuery{
		Position: p.current.Pos,
	}

	// Skip SET DEFAULT ROLE ... TO ...
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return query
}

func (p *Parser) parseOptimize() *ast.OptimizeQuery {
	opt := &ast.OptimizeQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip OPTIMIZE

	if !p.expect(token.TABLE) {
		return nil
	}

	// Parse table name (can start with a number in ClickHouse)
	tableName := p.parseIdentifierName()
	if tableName != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			opt.Database = tableName
			opt.Table = p.parseIdentifierName()
		} else {
			opt.Table = tableName
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			opt.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle PARTITION
	if p.currentIs(token.PARTITION) {
		p.nextToken()
		// Check for PARTITION ID 'value' syntax
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "ID" {
			opt.PartitionByID = true
			p.nextToken()
		}
		opt.Partition = p.parseExpression(LOWEST)
	}

	// Handle FINAL or FORCE (both are equivalent for forcing merge)
	if p.currentIs(token.FINAL) || (p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "FORCE") {
		opt.Final = true
		p.nextToken()
	}

	// Handle CLEANUP
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CLEANUP" {
		opt.Cleanup = true
		p.nextToken()
	}

	// Handle DEDUPLICATE
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DEDUPLICATE" {
		opt.Dedupe = true
		p.nextToken()
	}

	// Handle SETTINGS
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		opt.Settings = p.parseSettingsList()
	}

	return opt
}

func (p *Parser) parseSystem() *ast.SystemQuery {
	sys := &ast.SystemQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip SYSTEM

	// Read the command - can include identifiers and keywords (like TTL, SYNC, etc.)
	// Stop when we see:
	// 1. An IDENT followed by DOT (qualified table name like sqllt.table)
	// 2. A plain IDENT (not a command keyword) followed by end-of-statement (SEMICOLON, EOF, FORMAT),
	//    UNLESS the previous part was FAILPOINT (failpoint names are part of the command)
	var parts []string
	for p.currentIs(token.IDENT) || p.isSystemCommandKeyword() {
		// Check if this IDENT is followed by DOT - if so, it's a table name, not part of the command
		if p.currentIs(token.IDENT) && p.peekIs(token.DOT) {
			break
		}
		// Check if we've completed a command that expects a table name next
		// Commands like STOP MERGES, START MERGES, STOP TTL MERGES, START TTL MERGES, etc.
		if len(parts) > 0 {
			upperCmd := strings.ToUpper(strings.Join(parts, " "))
			if strings.HasSuffix(upperCmd, " MERGES") ||
				strings.HasSuffix(upperCmd, " MOVES") ||
				strings.HasSuffix(upperCmd, " FETCHES") ||
				strings.HasSuffix(upperCmd, " SENDS") ||
				strings.HasSuffix(upperCmd, " MUTATIONS") ||
				upperCmd == "FLUSH DISTRIBUTED" ||
				upperCmd == "STOP DISTRIBUTED SENDS" ||
				upperCmd == "START DISTRIBUTED SENDS" {
				// Next token should be the table name
				break
			}
		}
		// Check if this is a plain IDENT (not a command keyword) followed by end-of-statement
		// This indicates it's likely a table name, not part of the command
		// Exception: after FAILPOINT, the identifier is the failpoint name (part of command)
		if p.currentIs(token.IDENT) && !p.isSystemCommandKeyword() {
			if p.peekIs(token.SEMICOLON) || p.peekIs(token.EOF) || p.peekIs(token.FORMAT) {
				// Check if previous part was FAILPOINT or FOR - these are followed by identifiers that are part of command
				if len(parts) > 0 {
					lastPart := strings.ToUpper(parts[len(parts)-1])
					if lastPart == "FAILPOINT" || lastPart == "FOR" {
						// This is a failpoint name or format name, consume it as part of command
						parts = append(parts, p.current.Value)
						p.nextToken()
						continue
					}
				}
				break
			}
			// Special case: for SYNC REPLICA commands, check if next token is a mode keyword (PULL/LIGHTWEIGHT/STRICT)
			// followed by end-of-statement. If so, current token is the table name.
			if p.peekIs(token.IDENT) {
				nextUpper := strings.ToUpper(p.peek.Value)
				if nextUpper == "PULL" || nextUpper == "LIGHTWEIGHT" || nextUpper == "STRICT" {
					// Look ahead one more token to check if it's followed by end-of-statement
					currentValue := p.current.Value
					p.nextToken() // now at mode keyword
					modeValue := p.current.Value
					if p.peekIs(token.SEMICOLON) || p.peekIs(token.EOF) || p.peekIs(token.FORMAT) {
						// Mode keyword is followed by end-of-statement
						// Include mode in command, but NOT the table name
						parts = append(parts, modeValue)
						p.nextToken() // move past mode keyword
						// Restore table name info for later parsing
						sys.Command = strings.Join(parts, " ")
						sys.Table = currentValue
						// Skip the normal table parsing since we've set it here
						return sys
					}
					// Not followed by end-of-statement, continue normally
					// But we've consumed tokens, so we need to handle this carefully
					// For now, let's add both to parts and continue
					parts = append(parts, currentValue)
					parts = append(parts, modeValue)
					p.nextToken()
					continue
				}
			}
		}
		parts = append(parts, p.current.Value)
		p.nextToken()
	}
	sys.Command = strings.Join(parts, " ")

	// Check for ON CLUSTER clause - comes before the table name
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			sys.OnCluster = p.parseIdentifierName()
		}
	}

	// Parse optional table name for commands like SYNC REPLICA table
	// Table names can be keywords like "system" or dotted like "system.one"
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		tableName := p.current.Value
		p.nextToken()
		if p.currentIs(token.DOT) {
			p.nextToken()
			sys.Database = tableName
			if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
				sys.Table = p.current.Value
				p.nextToken()
			}
		} else {
			// For certain commands, the table name appears as both database and table in EXPLAIN
			// But for FLUSH DISTRIBUTED, use DuplicateTableOutput instead of setting both
			upperCmd := strings.ToUpper(sys.Command)
			if strings.Contains(upperCmd, "RELOAD DICTIONARY") ||
				strings.Contains(upperCmd, "DROP REPLICA") ||
				strings.Contains(upperCmd, "RESTORE REPLICA") ||
				strings.Contains(upperCmd, "STOP DISTRIBUTED SENDS") ||
				strings.Contains(upperCmd, "START DISTRIBUTED SENDS") ||
				strings.Contains(upperCmd, "LOAD PRIMARY KEY") ||
				strings.Contains(upperCmd, "UNLOAD PRIMARY KEY") {
				sys.Database = tableName
				sys.Table = tableName
			} else {
				sys.Table = tableName
			}
		}
	}

	// Set DuplicateTableOutput for commands that need database/table output twice
	upperCmd := strings.ToUpper(sys.Command)
	if strings.Contains(upperCmd, "FLUSH DISTRIBUTED") {
		// FLUSH DISTRIBUTED always outputs the table name twice (even if unqualified)
		if sys.Table != "" {
			sys.DuplicateTableOutput = true
		}
	} else if strings.Contains(upperCmd, "STOP DISTRIBUTED SENDS") ||
		strings.Contains(upperCmd, "START DISTRIBUTED SENDS") ||
		strings.Contains(upperCmd, "RELOAD DICTIONARY") {
		// Only set duplicate if database and table are different (qualified name)
		if sys.Database != sys.Table {
			sys.DuplicateTableOutput = true
		}
	}

	// Parse optional SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken() // skip SETTINGS
		sys.Settings = p.parseSettingsList()
	}

	return sys
}

// isSystemCommandKeyword returns true if current token is a keyword/identifier that is part of SYSTEM command
// and should NOT be treated as a table name
func (p *Parser) isSystemCommandKeyword() bool {
	switch p.current.Token {
	case token.TTL, token.SYNC, token.DROP, token.FORMAT, token.FOR, token.INDEX, token.INSERT,
		token.PRIMARY, token.KEY, token.DISTRIBUTED, token.RESTORE:
		return true
	}
	// Handle identifiers that are part of SYSTEM commands (not table names)
	if p.currentIs(token.IDENT) {
		upper := strings.ToUpper(p.current.Value)
		switch upper {
		case "SCHEMA", "CACHE", "QUEUE",
			// FAILPOINT names are part of the command, not table names
			"FAILPOINT",
			// These are common parts of SYSTEM commands that end with identifiers
			"ENABLE", "DISABLE", "FLUSH", "RELOAD", "RESTART", "STOP", "START",
			// Parts of STOP/START commands
			"LISTEN", "LISTENING", "MOVES", "MERGES", "TTL", "SENDS", "FETCHES", "PULLING",
			"REPLICATED", "DISTRIBUTED", "CLEANUP",
			// RELOAD targets
			"DICTIONARIES", "DICTIONARY", "MODELS", "MODEL", "FUNCTIONS", "FUNCTION",
			"EMBEDDED", "CONFIG", "SYMBOLS", "ASYNCHRONOUS", "METRICS",
			// FLUSH/DROP targets
			"LOGS", "ASYNC", "UNCOMPRESSED", "COMPILED", "MARK", "QUERY", "MMAP",
			"DNS", "FILESYSTEM", "S3",
			// Other command parts
			"MUTATIONS", "REPLICATION", "QUEUES", "DDL", "REPLICAS", "REPLICA":
			return true
		}
	}
	return false
}

func (p *Parser) parseRename() *ast.RenameQuery {
	rename := &ast.RenameQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip RENAME

	// Handle RENAME TABLE, RENAME DICTIONARY, or RENAME DATABASE
	if p.currentIs(token.TABLE) {
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		p.nextToken()
	} else if p.currentIs(token.DATABASE) {
		p.nextToken()
		rename.RenameDatabase = true
	} else {
		return nil
	}

	// Handle IF EXISTS after TABLE/DICTIONARY
	if p.currentIs(token.IF) && p.peekIs(token.EXISTS) {
		p.nextToken() // skip IF
		p.nextToken() // skip EXISTS
		rename.IfExists = true
	}

	// Parse rename pairs (can have multiple: t1 TO t2, t3 TO t4, ...)
	for {
		pair := &ast.RenamePair{}

		// Parse from table name (can be qualified: database.table)
		fromName := p.parseIdentifierName()
		if p.currentIs(token.DOT) {
			p.nextToken()
			pair.FromDatabase = fromName
			pair.FromTable = p.parseIdentifierName()
		} else {
			pair.FromTable = fromName
		}

		if !p.expect(token.TO) {
			break
		}

		// Parse to table name (can be qualified: database.table)
		toName := p.parseIdentifierName()
		if p.currentIs(token.DOT) {
			p.nextToken()
			pair.ToDatabase = toName
			pair.ToTable = p.parseIdentifierName()
		} else {
			pair.ToTable = toName
		}

		rename.Pairs = append(rename.Pairs, pair)

		// Check for more pairs
		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	// Set legacy From/To fields for backward compatibility (first pair)
	if len(rename.Pairs) > 0 {
		first := rename.Pairs[0]
		if first.FromDatabase != "" {
			rename.From = first.FromDatabase + "." + first.FromTable
		} else {
			rename.From = first.FromTable
		}
		if first.ToDatabase != "" {
			rename.To = first.ToDatabase + "." + first.ToTable
		} else {
			rename.To = first.ToTable
		}
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			rename.OnCluster = p.parseIdentifierName()
		}
	}

	// Handle SETTINGS
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		rename.Settings = p.parseSettingsList()
	}

	return rename
}

func (p *Parser) parseExchange() *ast.ExchangeQuery {
	exchange := &ast.ExchangeQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip EXCHANGE

	// Handle EXCHANGE TABLES or EXCHANGE DICTIONARIES
	if p.currentIs(token.TABLES) {
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARIES" {
		p.nextToken()
	} else {
		return nil
	}

	// Parse first table name (can be database.table)
	name1 := p.parseIdentifierName()
	if p.currentIs(token.DOT) {
		p.nextToken()
		exchange.Database1 = name1
		exchange.Table1 = p.parseIdentifierName()
	} else {
		exchange.Table1 = name1
	}

	if !p.expect(token.AND) {
		return nil
	}

	// Parse second table name (can be database.table)
	name2 := p.parseIdentifierName()
	if p.currentIs(token.DOT) {
		p.nextToken()
		exchange.Database2 = name2
		exchange.Table2 = p.parseIdentifierName()
	} else {
		exchange.Table2 = name2
	}

	// Handle ON CLUSTER
	if p.currentIs(token.ON) {
		p.nextToken()
		if p.currentIs(token.CLUSTER) {
			p.nextToken()
			exchange.OnCluster = p.parseIdentifierName()
		}
	}

	return exchange
}

func (p *Parser) parseDetach() *ast.DetachQuery {
	detach := &ast.DetachQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip DETACH

	// Check for DATABASE, TABLE, or DICTIONARY keyword
	isDatabase := false
	isDictionary := false
	if p.currentIs(token.DATABASE) {
		isDatabase = true
		p.nextToken()
	} else if p.currentIs(token.TABLE) {
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		isDictionary = true
		p.nextToken()
	}

	// Parse name (can be qualified: database.table for TABLE, database.dict for DICTIONARY)
	name := p.parseIdentifierName()
	if p.currentIs(token.DOT) && !isDatabase {
		p.nextToken()
		detach.Database = name
		if isDictionary {
			detach.Dictionary = p.parseIdentifierName()
		} else {
			detach.Table = p.parseIdentifierName()
		}
	} else if isDatabase {
		detach.Database = name
	} else if isDictionary {
		detach.Dictionary = name
	} else {
		detach.Table = name
	}

	return detach
}

func (p *Parser) parseAttach() *ast.AttachQuery {
	attach := &ast.AttachQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip ATTACH

	// Check for DATABASE, TABLE, DICTIONARY, or MATERIALIZED VIEW keyword
	isDatabase := false
	isDictionary := false
	isMaterializedView := false
	if p.currentIs(token.DATABASE) {
		isDatabase = true
		p.nextToken()
	} else if p.currentIs(token.TABLE) {
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		isDictionary = true
		p.nextToken()
	} else if p.currentIs(token.MATERIALIZED) {
		p.nextToken()
		if p.currentIs(token.VIEW) {
			isMaterializedView = true
			attach.IsMaterializedView = true
			p.nextToken()
		}
	}

	// Handle IF NOT EXISTS
	if p.currentIs(token.IF) {
		p.nextToken()
		if p.currentIs(token.NOT) {
			p.nextToken()
			if p.currentIs(token.EXISTS) {
				attach.IfNotExists = true
				p.nextToken()
			}
		}
	}

	// Parse name (can be qualified: database.table for TABLE, database.dict for DICTIONARY)
	name := p.parseIdentifierName()
	if p.currentIs(token.DOT) && !isDatabase {
		p.nextToken()
		attach.Database = name
		if isDictionary {
			attach.Dictionary = p.parseIdentifierName()
		} else {
			attach.Table = p.parseIdentifierName()
		}
	} else if isDatabase {
		attach.Database = name
	} else if isDictionary {
		attach.Dictionary = name
	} else {
		attach.Table = name
	}

	// Parse UUID clause (for ATTACH MATERIALIZED VIEW mv UUID 'uuid' ...)
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UUID" {
		p.nextToken()
		if p.currentIs(token.STRING) {
			attach.UUID = p.current.Value
			p.nextToken()
		}
	}

	// Parse TO INNER UUID clause (for ATTACH MATERIALIZED VIEW mv UUID 'uuid' TO INNER UUID 'inner_uuid' ...)
	if p.currentIs(token.TO) {
		p.nextToken()
		if p.currentIs(token.INNER) {
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "UUID" {
				p.nextToken()
				if p.currentIs(token.STRING) {
					attach.InnerUUID = p.current.Value
					p.nextToken()
				}
			}
		}
	}

	_ = isMaterializedView

	// Parse FROM clause: ATTACH TABLE name FROM 'path'
	if p.currentIs(token.FROM) {
		p.nextToken() // skip FROM
		if p.currentIs(token.STRING) {
			attach.FromPath = p.current.Value
			p.nextToken()
		}
	}

	// Parse column definitions for ATTACH TABLE name(col1 type, ...)
	if !isDatabase && p.currentIs(token.LPAREN) {
		p.nextToken()
		for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
			if p.currentIs(token.PRIMARY) {
				// Handle PRIMARY KEY as table constraint
				p.nextToken() // skip PRIMARY
				if p.currentIs(token.KEY) {
					p.nextToken() // skip KEY
				}
				if p.currentIs(token.LPAREN) {
					p.nextToken() // skip (
					for !p.currentIs(token.RPAREN) && !p.currentIs(token.EOF) {
						expr := p.parseExpression(LOWEST)
						if expr != nil {
							attach.ColumnsPrimaryKey = append(attach.ColumnsPrimaryKey, expr)
						}
						if p.currentIs(token.COMMA) {
							p.nextToken()
						} else {
							break
						}
					}
					p.expect(token.RPAREN)
				} else {
					expr := p.parseExpression(LOWEST)
					if expr != nil {
						attach.ColumnsPrimaryKey = append(attach.ColumnsPrimaryKey, expr)
					}
				}
			} else if p.currentIs(token.INDEX) {
				// Handle INDEX definition
				idx := p.parseIndexDefinition()
				if idx != nil {
					attach.Indexes = append(attach.Indexes, idx)
				}
			} else {
				col := p.parseColumnDeclaration()
				if col != nil {
					attach.Columns = append(attach.Columns, col)
				}
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
		p.expect(token.RPAREN)
	}

	// Parse ENGINE
	if p.currentIs(token.ENGINE) {
		p.nextToken()
		if p.currentIs(token.EQ) {
			p.nextToken()
		}
		attach.Engine = p.parseEngineClause()
	}

	// Parse table options (ORDER BY, PRIMARY KEY, PARTITION BY, AS SELECT)
	for {
		switch {
		case p.currentIs(token.PARTITION):
			p.nextToken()
			if p.expect(token.BY) {
				attach.PartitionBy = p.parseExpression(ALIAS_PREC)
			}
		case p.currentIs(token.ORDER):
			p.nextToken()
			if p.expect(token.BY) {
				if p.currentIs(token.LPAREN) {
					pos := p.current.Pos
					p.nextToken()
					exprs := p.parseExpressionList()
					p.expect(token.RPAREN)
					if len(exprs) == 0 || len(exprs) > 1 {
						attach.OrderBy = []ast.Expression{&ast.Literal{
							Position: pos,
							Type:     ast.LiteralTuple,
							Value:    exprs,
						}}
					} else {
						attach.OrderBy = exprs
					}
				} else {
					attach.OrderBy = []ast.Expression{p.parseExpression(ALIAS_PREC)}
				}
			}
		case p.currentIs(token.PRIMARY):
			p.nextToken()
			if p.expect(token.KEY) {
				if p.currentIs(token.LPAREN) {
					pos := p.current.Pos
					p.nextToken()
					exprs := p.parseExpressionList()
					p.expect(token.RPAREN)
					if len(exprs) == 0 || len(exprs) > 1 {
						attach.PrimaryKey = []ast.Expression{&ast.Literal{
							Position: pos,
							Type:     ast.LiteralTuple,
							Value:    exprs,
						}}
					} else {
						attach.PrimaryKey = exprs
					}
				} else {
					attach.PrimaryKey = []ast.Expression{p.parseExpression(ALIAS_PREC)}
				}
			}
		case p.currentIs(token.AS):
			// AS SELECT clause for materialized views
			p.nextToken()
			if p.currentIs(token.SELECT) {
				attach.SelectQuery = p.parseSelectWithUnion()
			}
		case p.currentIs(token.SETTINGS):
			// SETTINGS clause
			p.nextToken()
			attach.Settings = p.parseSettingsList()
		default:
			return attach
		}
	}
}

func (p *Parser) parseCheck() *ast.CheckQuery {
	check := &ast.CheckQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip CHECK

	// Skip optional TABLE keyword
	if p.currentIs(token.TABLE) {
		p.nextToken()
	}

	// Parse table name (can be qualified: database.table)
	tableName := p.parseIdentifierName()
	if p.currentIs(token.DOT) {
		p.nextToken()
		check.Database = tableName
		check.Table = p.parseIdentifierName()
	} else {
		check.Table = tableName
	}

	// Parse optional PARTITION clause
	if p.currentIs(token.PARTITION) {
		p.nextToken() // skip PARTITION
		check.Partition = p.parseExpression(LOWEST)
	}

	// Parse optional PART clause
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PART" {
		p.nextToken() // skip PART
		check.Part = p.parseExpression(LOWEST)
	}

	// Parse optional FORMAT
	if p.currentIs(token.FORMAT) {
		p.nextToken() // skip FORMAT
		check.Format = p.parseIdentifierName()
	}

	// Parse optional SETTINGS
	if p.currentIs(token.SETTINGS) {
		p.nextToken() // skip SETTINGS
		check.Settings = p.parseSettingsList()
	}

	return check
}

func (p *Parser) parseArrayJoin() *ast.ArrayJoinClause {
	aj := &ast.ArrayJoinClause{
		Position: p.current.Pos,
	}

	// Check for LEFT ARRAY JOIN
	if p.currentIs(token.LEFT) {
		aj.Left = true
		p.nextToken()
	}

	if !p.expect(token.ARRAY) {
		return nil
	}

	if !p.expect(token.JOIN) {
		return nil
	}

	// Parse array expressions
	aj.Columns = p.parseExpressionList()

	return aj
}

func (p *Parser) parseWindowDefinitions() []*ast.WindowDefinition {
	var defs []*ast.WindowDefinition

	for {
		def := &ast.WindowDefinition{
			Position: p.current.Pos,
		}

		// Parse window name
		if p.currentIs(token.IDENT) {
			def.Name = p.current.Value
			p.nextToken()
		}

		if !p.expect(token.AS) {
			break
		}

		if !p.expect(token.LPAREN) {
			break
		}

		// Parse window specification
		spec := &ast.WindowSpec{
			Position: p.current.Pos,
		}

		// Check for named window reference (e.g., w1 as (w0 ORDER BY ...))
		if p.currentIs(token.IDENT) {
			upper := strings.ToUpper(p.current.Value)
			if upper != "PARTITION" && upper != "ORDER" && upper != "ROWS" && upper != "RANGE" && upper != "GROUPS" {
				spec.Name = p.current.Value
				p.nextToken()
			}
		}

		// Parse PARTITION BY
		if p.currentIs(token.PARTITION) {
			p.nextToken()
			if p.expect(token.BY) {
				spec.PartitionBy = p.parseExpressionList()
			}
		}

		// Parse ORDER BY
		if p.currentIs(token.ORDER) {
			p.nextToken()
			if p.expect(token.BY) {
				spec.OrderBy = p.parseOrderByList()
			}
		}

		// Parse frame specification (ROWS/RANGE/GROUPS)
		if p.currentIs(token.IDENT) {
			frameType := strings.ToUpper(p.current.Value)
			if frameType == "ROWS" || frameType == "RANGE" || frameType == "GROUPS" {
				spec.Frame = p.parseWindowFrame()
			}
		}

		p.expect(token.RPAREN)
		def.Spec = spec
		defs = append(defs, def)

		if !p.currentIs(token.COMMA) {
			break
		}
		p.nextToken()
	}

	return defs
}

// parseIdentifierName parses an identifier name that may start with a number.
// In ClickHouse, table and database names can start with digits (e.g., 03657_test).
// When such names are lexed, they produce NUMBER + IDENT tokens that need to be combined.
func (p *Parser) parseIdentifierName() string {
	var name string

	// Handle identifier or keyword used as name
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		name = p.current.Value
		p.nextToken()
		return name
	}

	// Handle parameterized identifiers like {CLICKHOUSE_DATABASE:Identifier}
	if p.currentIs(token.PARAM) {
		name = "{" + p.current.Value + "}"
		p.nextToken()
		return name
	}

	// Handle name starting with number (e.g., 03657_test)
	if p.currentIs(token.NUMBER) {
		name = p.current.Value
		p.nextToken()
		// Check if followed by identifier (underscore connects them)
		if p.currentIs(token.IDENT) {
			name += p.current.Value
			p.nextToken()
		}
		return name
	}

	// Handle string (e.g., for cluster names)
	if p.currentIs(token.STRING) {
		name = p.current.Value
		p.nextToken()
		return name
	}

	return ""
}

// parseDottedIdentifier parses an identifier that may contain dots (e.g., n.x for nested columns)
func (p *Parser) parseDottedIdentifier() string {
	if !p.currentIs(token.IDENT) && !p.current.Token.IsKeyword() {
		return ""
	}

	parts := []string{p.current.Value}
	p.nextToken()

	// Continue parsing if followed by dots
	for p.currentIs(token.DOT) {
		p.nextToken() // skip .
		if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			parts = append(parts, p.current.Value)
			p.nextToken()
		} else {
			break
		}
	}

	return strings.Join(parts, ".")
}

// parseFromSelectSyntax handles ClickHouse's FROM ... SELECT syntax
// e.g., FROM numbers(1) SELECT number
func (p *Parser) parseFromSelectSyntax() *ast.SelectWithUnionQuery {
	query := &ast.SelectWithUnionQuery{
		Position: p.current.Pos,
	}

	sel := &ast.SelectQuery{
		Position: p.current.Pos,
	}

	// Skip FROM
	p.nextToken()

	// Parse table expression
	sel.From = p.parseTablesInSelect()

	// Parse SELECT
	if !p.expect(token.SELECT) {
		return nil
	}

	// Handle DISTINCT
	if p.currentIs(token.DISTINCT) {
		sel.Distinct = true
		p.nextToken()
		// Check for DISTINCT ON (col1, col2, ...)
		if p.currentIs(token.ON) {
			p.nextToken() // skip ON
			if p.expect(token.LPAREN) {
				sel.DistinctOn = p.parseExpressionList()
				p.expect(token.RPAREN)
			}
		}
	}

	// Parse column list
	sel.Columns = p.parseExpressionList()

	// Continue parsing the rest of SELECT (WHERE, GROUP BY, etc.)
	p.parseSelectRemainder(sel)

	query.Selects = append(query.Selects, sel)
	return query
}

// parseSelectRemainder parses the remainder of a SELECT after columns
func (p *Parser) parseSelectRemainder(sel *ast.SelectQuery) {
	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken()
		sel.Where = p.parseExpression(LOWEST)
	}

	// Parse GROUP BY clause
	if p.currentIs(token.GROUP) {
		p.nextToken()
		if p.expect(token.BY) {
			sel.GroupBy = p.parseExpressionList()
		}
	}

	// Parse HAVING clause
	if p.currentIs(token.HAVING) {
		p.nextToken()
		sel.Having = p.parseExpression(LOWEST)
	}

	// Parse ORDER BY clause
	if p.currentIs(token.ORDER) {
		p.nextToken()
		if p.expect(token.BY) {
			sel.OrderBy = p.parseOrderByList()
		}
	}

	// Parse LIMIT clause
	if p.currentIs(token.LIMIT) {
		p.nextToken()
		sel.Limit = p.parseExpression(LOWEST)
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		sel.Settings = p.parseSettingsList()
	}
}

// parseParenthesizedSelect handles (SELECT ...) at statement level
func (p *Parser) parseParenthesizedSelect() *ast.SelectWithUnionQuery {
	pos := p.current.Pos
	p.nextToken() // skip (

	// Check if this is actually a SELECT statement or nested parentheses
	if !p.currentIs(token.SELECT) && !p.currentIs(token.WITH) && !p.currentIs(token.LPAREN) {
		// Not a SELECT and not nested parens, just skip until we find closing paren
		depth := 1
		for depth > 0 && !p.currentIs(token.EOF) {
			if p.currentIs(token.LPAREN) {
				depth++
			} else if p.currentIs(token.RPAREN) {
				depth--
			}
			if depth > 0 {
				p.nextToken()
			}
		}
		if p.currentIs(token.RPAREN) {
			p.nextToken()
		}
		return &ast.SelectWithUnionQuery{Position: pos}
	}

	// Parse the inner query
	inner := p.parseSelectWithUnion()

	p.expect(token.RPAREN)

	// Wrap the result as first operand
	firstItem := inner

	// Check for UNION/EXCEPT/INTERSECT after the parenthesized select
	if p.isIntersectExceptWithWrapper() {
		// Handle INTERSECT/EXCEPT that creates SelectIntersectExceptQuery
		stmts := []ast.Statement{firstItem}
		var ops []string

		for p.isIntersectExceptWithWrapper() {
			var op string
			if p.currentIs(token.EXCEPT) {
				op = "EXCEPT"
			} else {
				op = "INTERSECT"
			}
			p.nextToken() // skip INTERSECT/EXCEPT

			if p.currentIs(token.ALL) {
				op += " ALL"
				p.nextToken()
			} else if p.currentIs(token.DISTINCT) {
				op += " DISTINCT"
				p.nextToken()
			}
			ops = append(ops, op)

			// Parse the next operand
			var nextStmt ast.Statement
			if p.currentIs(token.LPAREN) {
				p.nextToken() // skip (
				nested := p.parseSelectWithUnion()
				if nested == nil {
					break
				}
				p.expect(token.RPAREN)
				nextStmt = nested
			} else {
				sel := p.parseSelect()
				if sel == nil {
					break
				}
				nextStmt = sel
			}
			stmts = append(stmts, nextStmt)
		}

		result := buildIntersectExceptTree(stmts, ops)
		return &ast.SelectWithUnionQuery{
			Position: pos,
			Selects:  []ast.Statement{result},
		}
	}

	// Check for UNION ALL / UNION
	query := &ast.SelectWithUnionQuery{
		Position: pos,
	}
	if inner != nil {
		for _, s := range inner.Selects {
			query.Selects = append(query.Selects, s)
		}
		query.UnionModes = inner.UnionModes
		query.UnionAll = inner.UnionAll
	}

	// Handle UNION after parenthesized select
	for p.currentIs(token.UNION) {
		p.nextToken() // skip UNION
		mode := ""
		if p.currentIs(token.ALL) {
			mode = "ALL"
			query.UnionAll = true
			p.nextToken()
		} else if p.currentIs(token.DISTINCT) {
			mode = "DISTINCT"
			p.nextToken()
		}
		query.UnionModes = append(query.UnionModes, mode)

		if p.currentIs(token.LPAREN) {
			p.nextToken() // skip (
			nested := p.parseSelectWithUnion()
			if nested == nil {
				break
			}
			p.expect(token.RPAREN)
			// Keep parenthesized union as nested SelectWithUnionQuery
			query.Selects = append(query.Selects, nested)
		} else {
			sel := p.parseSelect()
			if sel == nil {
				break
			}
			query.Selects = append(query.Selects, sel)
		}
	}

	// Parse FORMAT and SETTINGS in either order after UNION
	// SETTINGS may come before FORMAT: (SELECT ...) UNION ALL (SELECT ...) SETTINGS x=1 FORMAT TSV
	// FORMAT may come before SETTINGS: (SELECT ...) UNION ALL (SELECT ...) FORMAT TSV SETTINGS x=1
	var formatParsed bool
	for p.currentIs(token.FORMAT) || p.currentIs(token.SETTINGS) {
		if p.currentIs(token.FORMAT) {
			p.nextToken()
			formatParsed = true
			// Get the format name and attach to first SELECT
			if len(query.Selects) > 0 {
				if sq, ok := query.Selects[0].(*ast.SelectQuery); ok {
					if p.currentIs(token.NULL) {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{"Null"}}
						p.nextToken()
					} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
						sq.Format = &ast.Identifier{Position: p.current.Pos, Parts: []string{p.current.Value}}
						p.nextToken()
					}
				}
			}
		} else if p.currentIs(token.SETTINGS) {
			p.nextToken()
			settings := p.parseSettingsList()
			// Store union-level settings in the SelectWithUnionQuery
			query.Settings = settings
			if formatParsed {
				query.SettingsAfterFormat = true
			} else {
				query.SettingsBeforeFormat = true
			}
		}
	}

	return query
}

// parseExistsStatement handles EXISTS table_name syntax
func (p *Parser) parseExistsStatement() *ast.ExistsQuery {
	exists := &ast.ExistsQuery{
		Position:   p.current.Pos,
		ExistsType: ast.ExistsTable, // default to TABLE
	}

	p.nextToken() // skip EXISTS

	// Check for TEMPORARY keyword
	if p.currentIs(token.TEMPORARY) {
		exists.Temporary = true
		p.nextToken()
	}

	// Check for DICTIONARY, DATABASE, VIEW, or TABLE keyword
	if p.currentIs(token.TABLE) {
		exists.ExistsType = ast.ExistsTable
		p.nextToken()
	} else if p.currentIs(token.DATABASE) {
		exists.ExistsType = ast.ExistsDatabase
		p.nextToken()
	} else if p.currentIs(token.VIEW) {
		exists.ExistsType = ast.ExistsView
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		exists.ExistsType = ast.ExistsDictionary
		p.nextToken()
	}

	// Parse table/database/dictionary/view name
	name := p.parseIdentifierName()
	if name != "" {
		if p.currentIs(token.DOT) {
			p.nextToken()
			exists.Database = name
			exists.Table = p.parseIdentifierName()
		} else {
			exists.Table = name
		}
	}

	// Handle SETTINGS
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		exists.Settings = p.parseSettingsList()
	}

	return exists
}

// parseProjection parses a projection definition: name (SELECT ... [ORDER BY col] [GROUP BY ...])
func (p *Parser) parseProjection() *ast.Projection {
	proj := &ast.Projection{
		Position: p.current.Pos,
	}

	// Parse projection name (can be identifier or keyword like VALUES)
	if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
		proj.Name = p.current.Value
		p.nextToken()
	}

	// Parse (SELECT ...)
	if !p.currentIs(token.LPAREN) {
		return proj
	}
	p.nextToken() // skip (

	proj.Select = &ast.ProjectionSelectQuery{
		Position: p.current.Pos,
	}

	// Parse WITH clause if present
	if p.currentIs(token.WITH) {
		p.nextToken() // skip WITH
		proj.Select.With = p.parseWithClause()
	}

	// Parse SELECT keyword (optional in projection)
	if p.currentIs(token.SELECT) {
		p.nextToken()
	}

	// Parse column list
	for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
		// Check for GROUP BY or ORDER BY
		if p.currentIs(token.GROUP) || p.currentIs(token.ORDER) {
			break
		}

		col := p.parseExpression(LOWEST)
		if col != nil {
			// Handle implicit alias (identifier without AS)
			col = p.parseImplicitAlias(col)
			proj.Select.Columns = append(proj.Select.Columns, col)
		}

		if p.currentIs(token.COMMA) {
			p.nextToken()
		} else {
			break
		}
	}

	// Parse GROUP BY
	if p.currentIs(token.GROUP) {
		p.nextToken() // GROUP
		if p.currentIs(token.BY) {
			p.nextToken() // BY
		}
		for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) && !p.currentIs(token.ORDER) {
			expr := p.parseExpression(LOWEST)
			if expr != nil {
				proj.Select.GroupBy = append(proj.Select.GroupBy, expr)
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
	}

	// Parse ORDER BY
	if p.currentIs(token.ORDER) {
		p.nextToken() // ORDER
		if p.currentIs(token.BY) {
			p.nextToken() // BY
		}
		// Parse ORDER BY columns (comma-separated expressions)
		for !p.currentIs(token.EOF) && !p.currentIs(token.RPAREN) {
			expr := p.parseExpression(LOWEST)
			if expr != nil {
				proj.Select.OrderBy = append(proj.Select.OrderBy, expr)
			} else {
				break
			}
			if p.currentIs(token.COMMA) {
				p.nextToken()
			} else {
				break
			}
		}
	}

	// Skip closing paren
	if p.currentIs(token.RPAREN) {
		p.nextToken()
	}

	return proj
}

// parseGrant handles GRANT statements
func (p *Parser) parseGrant() *ast.GrantQuery {
	grant := &ast.GrantQuery{
		Position: p.current.Pos,
		IsRevoke: false,
	}

	// Skip all tokens until end of statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return grant
}

// parseRevoke handles REVOKE statements
func (p *Parser) parseRevoke() *ast.GrantQuery {
	grant := &ast.GrantQuery{
		Position: p.current.Pos,
		IsRevoke: true,
	}

	// Skip all tokens until end of statement
	for !p.currentIs(token.EOF) && !p.currentIs(token.SEMICOLON) {
		p.nextToken()
	}

	return grant
}

func (p *Parser) parseBackup() *ast.BackupQuery {
	backup := &ast.BackupQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip BACKUP

	// Parse what to backup: TABLE, DATABASE, DICTIONARY, ALL, TEMPORARY
	if p.currentIs(token.TABLE) {
		p.nextToken()
		name := p.parseIdentifierName()
		if p.currentIs(token.DOT) {
			p.nextToken()
			backup.Database = name
			backup.Table = p.parseIdentifierName()
		} else {
			backup.Table = name
		}
	} else if p.currentIs(token.DATABASE) {
		p.nextToken()
		backup.Database = p.parseIdentifierName()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		p.nextToken()
		backup.Dictionary = p.parseIdentifierName()
	} else if p.currentIs(token.ALL) {
		backup.All = true
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TEMPORARY" {
		backup.Temporary = true
		p.nextToken()
		if p.currentIs(token.TABLE) {
			p.nextToken()
		}
	}

	// Parse TO clause
	if p.currentIs(token.TO) {
		p.nextToken()
		// Parse target - it's a function call like Null or Disk('path')
		if p.currentIs(token.NULL) || p.currentIs(token.IDENT) {
			name := p.current.Value
			p.nextToken()
			fn := &ast.FunctionCall{
				Position: backup.Position,
				Name:     name,
			}
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				if !p.currentIs(token.RPAREN) {
					fn.Arguments = p.parseExpressionList()
				}
				p.expect(token.RPAREN)
			}
			backup.Target = fn
		}
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		backup.Settings = p.parseSettingsList()
	}

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		backup.Format = p.parseIdentifierName()
	}

	return backup
}

func (p *Parser) parseRestore() *ast.RestoreQuery {
	restore := &ast.RestoreQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip RESTORE

	// Parse what to restore: TABLE, DATABASE, DICTIONARY, ALL, TEMPORARY
	if p.currentIs(token.TABLE) {
		p.nextToken()
		name := p.parseIdentifierName()
		if p.currentIs(token.DOT) {
			p.nextToken()
			restore.Database = name
			restore.Table = p.parseIdentifierName()
		} else {
			restore.Table = name
		}
	} else if p.currentIs(token.DATABASE) {
		p.nextToken()
		restore.Database = p.parseIdentifierName()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "DICTIONARY" {
		p.nextToken()
		restore.Dictionary = p.parseIdentifierName()
	} else if p.currentIs(token.ALL) {
		restore.All = true
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "TEMPORARY" {
		restore.Temporary = true
		p.nextToken()
		if p.currentIs(token.TABLE) {
			p.nextToken()
		}
	}

	// Parse FROM clause
	if p.currentIs(token.FROM) {
		p.nextToken()
		// Parse source - it's a function call like Null or Disk('path')
		if p.currentIs(token.NULL) || p.currentIs(token.IDENT) {
			name := p.current.Value
			p.nextToken()
			fn := &ast.FunctionCall{
				Position: restore.Position,
				Name:     name,
			}
			if p.currentIs(token.LPAREN) {
				p.nextToken()
				if !p.currentIs(token.RPAREN) {
					fn.Arguments = p.parseExpressionList()
				}
				p.expect(token.RPAREN)
			}
			restore.Source = fn
		}
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken()
		restore.Settings = p.parseSettingsList()
	}

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		restore.Format = p.parseIdentifierName()
	}

	return restore
}

// parseTransactionControl handles BEGIN, COMMIT, ROLLBACK, and SET TRANSACTION SNAPSHOT statements
func (p *Parser) parseTransactionControl() *ast.TransactionControlQuery {
	query := &ast.TransactionControlQuery{
		Position: p.current.Pos,
	}

	switch p.current.Token {
	case token.BEGIN:
		query.Action = "BEGIN"
		p.nextToken() // skip BEGIN
		// Skip optional TRANSACTION keyword
		if p.currentIs(token.TRANSACTION) {
			p.nextToken()
		}
	case token.COMMIT:
		query.Action = "COMMIT"
		p.nextToken() // skip COMMIT
	case token.ROLLBACK:
		query.Action = "ROLLBACK"
		p.nextToken() // skip ROLLBACK
	case token.SET:
		p.nextToken() // skip SET
		if p.currentIs(token.TRANSACTION) {
			p.nextToken() // skip TRANSACTION
			if p.currentIs(token.SNAPSHOT) {
				p.nextToken() // skip SNAPSHOT
				query.Action = "SET_SNAPSHOT"
				// Parse snapshot number
				if p.currentIs(token.NUMBER) {
					// Parse the number value
					val, err := strconv.ParseInt(p.current.Value, 10, 64)
					if err == nil {
						query.Snapshot = val
					}
					p.nextToken()
				}
			}
		}
	}

	return query
}

// parseKill handles KILL QUERY/MUTATION statements
func (p *Parser) parseKill() *ast.KillQuery {
	query := &ast.KillQuery{
		Position: p.current.Pos,
	}

	p.nextToken() // skip KILL

	// Parse QUERY or MUTATION
	if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "QUERY" {
		query.Type = "QUERY"
		p.nextToken()
	} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "MUTATION" {
		query.Type = "MUTATION"
		p.nextToken()
	}

	// Parse WHERE clause
	if p.currentIs(token.WHERE) {
		p.nextToken() // skip WHERE
		query.Where = p.parseExpression(LOWEST)
	}

	// Parse SYNC/ASYNC/TEST
	// SYNC can be a keyword token or IDENT
	for p.currentIs(token.IDENT) || p.currentIs(token.SYNC) {
		upper := strings.ToUpper(p.current.Value)
		switch upper {
		case "SYNC":
			query.Sync = true
			p.nextToken()
		case "ASYNC":
			query.Sync = false
			p.nextToken()
		case "TEST":
			query.Test = true
			p.nextToken()
		default:
			// Exit loop for unknown keywords
			goto endModifiers
		}
	}
endModifiers:

	// Parse FORMAT clause
	if p.currentIs(token.FORMAT) {
		p.nextToken()
		if p.currentIs(token.NULL) {
			query.Format = "Null"
			p.nextToken()
		} else if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
			query.Format = p.current.Value
			p.nextToken()
		}
	}

	// Parse SETTINGS clause
	if p.currentIs(token.SETTINGS) {
		p.nextToken() // skip SETTINGS
		query.Settings = p.parseSettingsList()
	}

	return query
}

// parseTTLElement parses a single TTL element: expression [DELETE] [WHERE condition] [GROUP BY ...] [SET ...]
func (p *Parser) parseTTLElement() *ast.TTLElement {
	elem := &ast.TTLElement{
		Position: p.current.Pos,
		Expr:     p.parseExpression(ALIAS_PREC),
	}
	// Skip RECOMPRESS CODEC(...), DELETE, TO DISK, TO VOLUME (but not WHERE)
	p.skipTTLModifiersExceptWhere()
	// Handle WHERE clause for this TTL element (conditional deletion)
	if p.currentIs(token.WHERE) {
		p.nextToken()
		elem.Where = p.parseExpression(ALIAS_PREC)
	}
	// Handle GROUP BY x SET y = max(y) syntax (skip for now, already parsed in Where or just skip)
	if p.currentIs(token.GROUP) {
		p.nextToken()
		if p.currentIs(token.BY) {
			p.nextToken()
			for {
				p.parseExpression(ALIAS_PREC)
				if p.currentIs(token.COMMA) {
					p.nextToken()
				} else {
					break
				}
			}
		}
	}
	// Handle SET clause - assignments are comma separated (id = expr, id = expr, ...)
	// We need to distinguish between:
	// - Comma continuing SET: followed by IDENT = pattern
	// - Comma starting new TTL: followed by expression (like d + toIntervalYear(...))
	if p.currentIs(token.SET) {
		p.nextToken()
		for {
			// Parse assignment expression: id = expr
			p.parseExpression(ALIAS_PREC)
			// Check for comma
			if p.currentIs(token.COMMA) {
				// Check if this is a SET continuation (COMMA IDENT EQ pattern)
				// We can check using peek (what follows comma) and peekPeek (what follows that)
				// without consuming any tokens
				isSetContinuation := false
				if p.peekIs(token.IDENT) || p.peek.Token.IsKeyword() {
					if p.peekPeekIs(token.EQ) {
						// It's another SET assignment (id = expr)
						isSetContinuation = true
					}
				}
				if isSetContinuation {
					// Consume comma and continue parsing SET assignments
					p.nextToken()
					continue
				}
				// Not a SET assignment - break and let caller handle the comma
				break
			}
			// No comma, end of SET clause
			break
		}
	}
	return elem
}

// skipTTLModifiersExceptWhere skips TTL modifiers but stops at WHERE
func (p *Parser) skipTTLModifiersExceptWhere() {
	for {
		// Skip RECOMPRESS CODEC(...)
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RECOMPRESS" {
			p.nextToken()
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CODEC" {
				p.nextToken()
				if p.currentIs(token.LPAREN) {
					depth := 1
					p.nextToken()
					for depth > 0 && !p.currentIs(token.EOF) {
						if p.currentIs(token.LPAREN) {
							depth++
						} else if p.currentIs(token.RPAREN) {
							depth--
						}
						p.nextToken()
					}
				}
			}
			continue
		}
		// Skip DELETE (TTL ... DELETE)
		if p.currentIs(token.DELETE) {
			p.nextToken()
			continue
		}
		// Skip TO DISK 'name' or TO VOLUME 'name'
		if p.currentIs(token.TO) {
			p.nextToken()
			if p.currentIs(token.IDENT) {
				upper := strings.ToUpper(p.current.Value)
				if upper == "DISK" || upper == "VOLUME" {
					p.nextToken()
					if p.currentIs(token.STRING) {
						p.nextToken()
					}
					continue
				}
			}
		}
		break
	}
}

// skipTTLModifiers skips TTL modifiers like RECOMPRESS CODEC(...), DELETE, TO DISK, TO VOLUME
func (p *Parser) skipTTLModifiers() {
	for {
		// Skip RECOMPRESS CODEC(...)
		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "RECOMPRESS" {
			p.nextToken() // skip RECOMPRESS
			if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "CODEC" {
				p.nextToken() // skip CODEC
				if p.currentIs(token.LPAREN) {
					// Skip the entire CODEC(...) call
					depth := 1
					p.nextToken() // skip (
					for depth > 0 && !p.currentIs(token.EOF) {
						if p.currentIs(token.LPAREN) {
							depth++
						} else if p.currentIs(token.RPAREN) {
							depth--
						}
						p.nextToken()
					}
				}
			}
			continue
		}
		// Skip DELETE (TTL ... DELETE)
		if p.currentIs(token.DELETE) {
			p.nextToken()
			continue
		}
		// Skip TO DISK 'name' or TO VOLUME 'name'
		if p.currentIs(token.TO) {
			p.nextToken()
			if p.currentIs(token.IDENT) {
				upper := strings.ToUpper(p.current.Value)
				if upper == "DISK" || upper == "VOLUME" {
					p.nextToken()
					if p.currentIs(token.STRING) {
						p.nextToken()
					}
					continue
				}
			}
		}
		break
	}
}