Add ALTER DATABASE SET statement support, enable 2 more tests (114→116)

New features:
- ALTER DATABASE ... SET statement with database options
- ACCELERATED_DATABASE_RECOVERY option support
- AlterDatabaseSetStatement and AcceleratedDatabaseRecoveryDatabaseOption AST types

Author: claude

ast/alter_database_set_statement.go

@@ -0,0 +1,27 @@
+package ast
+
+// AlterDatabaseSetStatement represents ALTER DATABASE ... SET statement
+type AlterDatabaseSetStatement struct {
+	DatabaseName      *Identifier
+	UseCurrent        bool
+	WithManualCutover bool
+	Options           []DatabaseOption
+}
+
+func (a *AlterDatabaseSetStatement) node()      {}
+func (a *AlterDatabaseSetStatement) statement() {}
+
+// DatabaseOption is an interface for database options
+type DatabaseOption interface {
+	Node
+	databaseOption()
+}
+
+// AcceleratedDatabaseRecoveryDatabaseOption represents ACCELERATED_DATABASE_RECOVERY option
+type AcceleratedDatabaseRecoveryDatabaseOption struct {
+	OptionKind  string // "AcceleratedDatabaseRecovery"
+	OptionState string // "On" or "Off"
+}
+
+func (a *AcceleratedDatabaseRecoveryDatabaseOption) node()           {}
+func (a *AcceleratedDatabaseRecoveryDatabaseOption) databaseOption() {}

parser/parser.go

@@ -376,9 +376,65 @@ func (p *Parser) parseAlterDatabaseStatement() (ast.Statement, error) {
 		}
 	}
 
+	// Parse database name followed by SET
+	if p.curTok.Type == TokenIdent || p.curTok.Type == TokenLBracket {
+		dbName := p.parseIdentifier()
+
+		// Expect SET
+		if p.curTok.Type == TokenSet {
+			return p.parseAlterDatabaseSetStatement(dbName)
+		}
+	}
+
 	return nil, fmt.Errorf("unexpected token after ALTER DATABASE: %s", p.curTok.Literal)
 }
 
+func (p *Parser) parseAlterDatabaseSetStatement(dbName *ast.Identifier) (*ast.AlterDatabaseSetStatement, error) {
+	// Consume SET
+	p.nextToken()
+
+	stmt := &ast.AlterDatabaseSetStatement{
+		DatabaseName: dbName,
+	}
+
+	// Parse options
+	for {
+		optionName := strings.ToUpper(p.curTok.Literal)
+		p.nextToken()
+
+		if p.curTok.Type != TokenEquals {
+			return nil, fmt.Errorf("expected = after %s, got %s", optionName, p.curTok.Literal)
+		}
+		p.nextToken()
+
+		// Parse option value
+		optionValue := strings.ToUpper(p.curTok.Literal)
+		p.nextToken()
+
+		switch optionName {
+		case "ACCELERATED_DATABASE_RECOVERY":
+			opt := &ast.AcceleratedDatabaseRecoveryDatabaseOption{
+				OptionKind:  "AcceleratedDatabaseRecovery",
+				OptionState: capitalizeFirst(optionValue),
+			}
+			stmt.Options = append(stmt.Options, opt)
+		}
+
+		if p.curTok.Type == TokenComma {
+			p.nextToken()
+		} else {
+			break
+		}
+	}
+
+	// Skip optional semicolon
+	if p.curTok.Type == TokenSemicolon {
+		p.nextToken()
+	}
+
+	return stmt, nil
+}
+
 func (p *Parser) parseAlterDatabaseScopedCredentialStatement() (*ast.AlterCredentialStatement, error) {
 	// Consume CREDENTIAL
 	p.nextToken()
@@ -5869,6 +5925,8 @@ func statementToJSON(stmt ast.Statement) jsonNode {
 		return createPartitionSchemeStatementToJSON(s)
 	case *ast.AlterCredentialStatement:
 		return alterCredentialStatementToJSON(s)
+	case *ast.AlterDatabaseSetStatement:
+		return alterDatabaseSetStatementToJSON(s)
 	case *ast.RevertStatement:
 		return revertStatementToJSON(s)
 	case *ast.DropCredentialStatement:
@@ -6172,6 +6230,38 @@ func alterCredentialStatementToJSON(s *ast.AlterCredentialStatement) jsonNode {
 	return node
 }
 
+func alterDatabaseSetStatementToJSON(s *ast.AlterDatabaseSetStatement) jsonNode {
+	node := jsonNode{
+		"$type":             "AlterDatabaseSetStatement",
+		"WithManualCutover": s.WithManualCutover,
+		"UseCurrent":        s.UseCurrent,
+	}
+	if s.DatabaseName != nil {
+		node["DatabaseName"] = identifierToJSON(s.DatabaseName)
+	}
+	if len(s.Options) > 0 {
+		opts := make([]jsonNode, len(s.Options))
+		for i, opt := range s.Options {
+			opts[i] = databaseOptionToJSON(opt)
+		}
+		node["Options"] = opts
+	}
+	return node
+}
+
+func databaseOptionToJSON(opt ast.DatabaseOption) jsonNode {
+	switch o := opt.(type) {
+	case *ast.AcceleratedDatabaseRecoveryDatabaseOption:
+		return jsonNode{
+			"$type":       "AcceleratedDatabaseRecoveryDatabaseOption",
+			"OptionKind":  o.OptionKind,
+			"OptionState": o.OptionState,
+		}
+	default:
+		return jsonNode{"$type": "UnknownDatabaseOption"}
+	}
+}
+
 func indexDefinitionToJSON(idx *ast.IndexDefinition) jsonNode {
 	node := jsonNode{
 		"$type":  "IndexDefinition",

parser/testdata/AcceleratedDatabaseRecoveryTests150/metadata.json

@@ -1 +1 @@
-{"skip": true}
+{"skip": false}

parser/testdata/Baselines150_AcceleratedDatabaseRecoveryTests150/metadata.json

@@ -1 +1 @@
-{"skip": true}
+{"skip": false}