Enable PhaseOne_AlterDatabaseModifyFileStatementTest

ast/alter_database_set_statement.go

@@ -6,8 +6,18 @@ type AlterDatabaseSetStatement struct {
 	UseCurrent        bool
 	WithManualCutover bool
 	Options           []DatabaseOption
+	Termination       *AlterDatabaseTermination
 }
 
+// AlterDatabaseTermination represents the termination clause (WITH NO_WAIT, WITH ROLLBACK AFTER N, WITH ROLLBACK IMMEDIATE)
+type AlterDatabaseTermination struct {
+	NoWait            bool
+	ImmediateRollback bool
+	RollbackAfter     ScalarExpression
+}
+
+func (a *AlterDatabaseTermination) node() {}
+
 func (a *AlterDatabaseSetStatement) node()      {}
 func (a *AlterDatabaseSetStatement) statement() {}
 
@@ -82,6 +92,7 @@ type SimpleDatabaseOption struct {
 
 func (d *SimpleDatabaseOption) node()                 {}
 func (d *SimpleDatabaseOption) createDatabaseOption() {}
+func (d *SimpleDatabaseOption) databaseOption()       {}
 
 // MaxSizeDatabaseOption represents a MAXSIZE option.
 type MaxSizeDatabaseOption struct {
@@ -104,6 +115,16 @@ func (l *LiteralDatabaseOption) node()                 {}
 func (l *LiteralDatabaseOption) databaseOption()       {}
 func (l *LiteralDatabaseOption) createDatabaseOption() {}
 
+// ElasticPoolSpecification represents SERVICE_OBJECTIVE = ELASTIC_POOL(name = poolname)
+type ElasticPoolSpecification struct {
+	ElasticPoolName *Identifier
+	OptionKind      string // "ServiceObjective"
+}
+
+func (e *ElasticPoolSpecification) node()                 {}
+func (e *ElasticPoolSpecification) databaseOption()       {}
+func (e *ElasticPoolSpecification) createDatabaseOption() {}
+
 // AlterDatabaseAddFileStatement represents ALTER DATABASE ... ADD FILE statement
 type AlterDatabaseAddFileStatement struct {
 	DatabaseName     *Identifier
@@ -145,11 +166,23 @@ type AlterDatabaseModifyFileGroupStatement struct {
 	MakeDefault        bool
 	UpdatabilityOption string // "ReadOnly", "ReadWrite", "ReadOnlyOld", "ReadWriteOld", or ""
 	NewFileGroupName   *Identifier
+	Termination        *AlterDatabaseTermination
+	UseCurrent         bool
 }
 
 func (a *AlterDatabaseModifyFileGroupStatement) node()      {}
 func (a *AlterDatabaseModifyFileGroupStatement) statement() {}
 
+// AlterDatabaseRebuildLogStatement represents ALTER DATABASE ... REBUILD LOG statement
+type AlterDatabaseRebuildLogStatement struct {
+	DatabaseName    *Identifier
+	FileDeclaration *FileDeclaration
+	UseCurrent      bool
+}
+
+func (a *AlterDatabaseRebuildLogStatement) node()      {}
+func (a *AlterDatabaseRebuildLogStatement) statement() {}
+
 // AlterDatabaseModifyNameStatement represents ALTER DATABASE ... MODIFY NAME statement
 type AlterDatabaseModifyNameStatement struct {
 	DatabaseName *Identifier
@@ -279,3 +312,68 @@ type ChangeRetentionChangeTrackingOptionDetail struct {
 
 func (c *ChangeRetentionChangeTrackingOptionDetail) node()                        {}
 func (c *ChangeRetentionChangeTrackingOptionDetail) changeTrackingOptionDetail() {}
+
+// RecoveryDatabaseOption represents RECOVERY database option
+type RecoveryDatabaseOption struct {
+	OptionKind string // "Recovery"
+	Value      string // "Full", "BulkLogged", "Simple"
+}
+
+func (r *RecoveryDatabaseOption) node()           {}
+func (r *RecoveryDatabaseOption) databaseOption() {}
+
+// CursorDefaultDatabaseOption represents CURSOR_DEFAULT database option
+type CursorDefaultDatabaseOption struct {
+	OptionKind string // "CursorDefault"
+	IsLocal    bool   // true for LOCAL, false for GLOBAL
+}
+
+func (c *CursorDefaultDatabaseOption) node()           {}
+func (c *CursorDefaultDatabaseOption) databaseOption() {}
+
+// PageVerifyDatabaseOption represents PAGE_VERIFY database option
+type PageVerifyDatabaseOption struct {
+	OptionKind string // "PageVerify"
+	Value      string // "Checksum", "None", "TornPageDetection"
+}
+
+func (p *PageVerifyDatabaseOption) node()           {}
+func (p *PageVerifyDatabaseOption) databaseOption() {}
+
+// PartnerDatabaseOption represents PARTNER database mirroring option
+type PartnerDatabaseOption struct {
+	OptionKind    string           // "Partner"
+	PartnerServer ScalarExpression // For PARTNER = 'server'
+	PartnerOption string           // "PartnerServer", "Failover", "ForceServiceAllowDataLoss", "Resume", "SafetyFull", "SafetyOff", "Suspend", "Timeout"
+	Timeout       ScalarExpression // For PARTNER TIMEOUT value
+}
+
+func (p *PartnerDatabaseOption) node()           {}
+func (p *PartnerDatabaseOption) databaseOption() {}
+
+// WitnessDatabaseOption represents WITNESS database mirroring option
+type WitnessDatabaseOption struct {
+	OptionKind    string           // "Witness"
+	WitnessServer ScalarExpression // For WITNESS = 'server'
+	IsOff         bool             // For WITNESS OFF
+}
+
+func (w *WitnessDatabaseOption) node()           {}
+func (w *WitnessDatabaseOption) databaseOption() {}
+
+// ParameterizationDatabaseOption represents PARAMETERIZATION database option
+type ParameterizationDatabaseOption struct {
+	OptionKind string // "Parameterization"
+	IsSimple   bool   // true for SIMPLE, false for FORCED
+}
+
+func (p *ParameterizationDatabaseOption) node()           {}
+func (p *ParameterizationDatabaseOption) databaseOption() {}
+
+// GenericDatabaseOption represents a simple database option with just OptionKind
+type GenericDatabaseOption struct {
+	OptionKind string // e.g., "Emergency", "ErrorBrokerConversations", "EnableBroker", etc.
+}
+
+func (g *GenericDatabaseOption) node()           {}
+func (g *GenericDatabaseOption) databaseOption() {}

ast/alter_procedure_statement.go

@@ -4,6 +4,7 @@ package ast
 type AlterProcedureStatement struct {
 	ProcedureReference *ProcedureReference
 	Parameters         []*ProcedureParameter
+	Options            []ProcedureOptionBase
 	StatementList      *StatementList
 	IsForReplication   bool
 }

ast/alter_table_alter_column_statement.go

@@ -5,10 +5,13 @@ type AlterTableAlterColumnStatement struct {
 	SchemaObjectName            *SchemaObjectName
 	ColumnIdentifier            *Identifier
 	DataType                    DataTypeReference
-	AlterTableAlterColumnOption string // "NoOptionDefined", "AddRowGuidCol", "DropRowGuidCol", "Null", "NotNull", etc.
+	AlterTableAlterColumnOption string // "NoOptionDefined", "AddRowGuidCol", "DropRowGuidCol", "Null", "NotNull", "AddSparse", "DropSparse", etc.
+	StorageOptions              *ColumnStorageOptions
 	IsHidden                    bool
 	Collation                   *Identifier
 	IsMasked                    bool
+	Encryption                  *ColumnEncryptionDefinition
+	MaskingFunction             ScalarExpression
 }
 
 func (a *AlterTableAlterColumnStatement) node()      {}

ast/alter_table_change_tracking_statement.go

@@ -0,0 +1,11 @@
+package ast
+
+// AlterTableChangeTrackingModificationStatement represents ALTER TABLE ... ENABLE/DISABLE CHANGE_TRACKING
+type AlterTableChangeTrackingModificationStatement struct {
+	SchemaObjectName    *SchemaObjectName
+	IsEnable            bool   // true for ENABLE, false for DISABLE
+	TrackColumnsUpdated string // "NotSet", "On", "Off"
+}
+
+func (s *AlterTableChangeTrackingModificationStatement) node()      {}
+func (s *AlterTableChangeTrackingModificationStatement) statement() {}

ast/alter_table_rebuild_statement.go

@@ -0,0 +1,11 @@
+package ast
+
+// AlterTableRebuildStatement represents ALTER TABLE ... REBUILD statement
+type AlterTableRebuildStatement struct {
+	SchemaObjectName *SchemaObjectName
+	Partition        *PartitionSpecifier
+	IndexOptions     []IndexOption
+}
+
+func (s *AlterTableRebuildStatement) node()      {}
+func (s *AlterTableRebuildStatement) statement() {}

ast/alter_table_set_statement.go

@@ -44,3 +44,30 @@ type MemoryOptimizedTableOption struct {
 
 func (o *MemoryOptimizedTableOption) tableOption() {}
 func (o *MemoryOptimizedTableOption) node()        {}
+
+// DurabilityTableOption represents a DURABILITY table option
+type DurabilityTableOption struct {
+	OptionKind                string // "Durability"
+	DurabilityTableOptionKind string // "SchemaOnly", "SchemaAndData"
+}
+
+func (o *DurabilityTableOption) tableOption() {}
+func (o *DurabilityTableOption) node()        {}
+
+// LockEscalationTableOption represents LOCK_ESCALATION option
+type LockEscalationTableOption struct {
+	OptionKind string // "LockEscalation"
+	Value      string // "Auto", "Table", "Disable"
+}
+
+func (o *LockEscalationTableOption) tableOption() {}
+func (o *LockEscalationTableOption) node()        {}
+
+// FileStreamOnTableOption represents FILESTREAM_ON option
+type FileStreamOnTableOption struct {
+	OptionKind string // "FileStreamOn"
+	Value      *IdentifierOrValueExpression
+}
+
+func (o *FileStreamOnTableOption) tableOption() {}
+func (o *FileStreamOnTableOption) node()        {}

ast/backup_statement.go

@@ -6,7 +6,7 @@ type BackupDatabaseStatement struct {
 	DatabaseName   *IdentifierOrValueExpression
 	MirrorToClauses []*MirrorToClause
 	Devices        []*DeviceInfo
-	Options        []*BackupOption
+	Options        []BackupOptionBase
 }
 
 // MirrorToClause represents a MIRROR TO clause in a BACKUP statement
@@ -22,19 +22,37 @@ func (s *BackupDatabaseStatement) node()          {}
 type BackupTransactionLogStatement struct {
 	DatabaseName *IdentifierOrValueExpression
 	Devices      []*DeviceInfo
-	Options      []*BackupOption
+	Options      []BackupOptionBase
 }
 
 func (s *BackupTransactionLogStatement) statementNode() {}
 func (s *BackupTransactionLogStatement) statement()     {}
 func (s *BackupTransactionLogStatement) node()          {}
 
+// BackupOptionBase is an interface for backup options
+type BackupOptionBase interface {
+	backupOption()
+}
+
 // BackupOption represents a backup option
 type BackupOption struct {
 	OptionKind string // Compression, NoCompression, StopOnError, ContinueAfterError, etc.
 	Value      ScalarExpression
 }
 
+func (o *BackupOption) backupOption() {}
+
+// BackupEncryptionOption represents an ENCRYPTION(...) backup option
+type BackupEncryptionOption struct {
+	Algorithm  string           // Aes128, Aes192, Aes256, TripleDes3Key
+	Encryptor  *CryptoMechanism
+	OptionKind string           // typically "None"
+}
+
+func (o *BackupEncryptionOption) backupOption() {}
+
+// CryptoMechanism is defined in create_simple_statements.go
+
 // BackupCertificateStatement represents a BACKUP CERTIFICATE statement
 type BackupCertificateStatement struct {
 	Name                  *Identifier

ast/begin_end_block_statement.go

@@ -38,6 +38,14 @@ type LiteralAtomicBlockOption struct {
 
 func (o *LiteralAtomicBlockOption) atomicBlockOption() {}
 
+// OnOffAtomicBlockOption represents an atomic block option with an ON/OFF value.
+type OnOffAtomicBlockOption struct {
+	OptionKind  string
+	OptionState string // "On" or "Off"
+}
+
+func (o *OnOffAtomicBlockOption) atomicBlockOption() {}
+
 // StatementList is a list of statements.
 type StatementList struct {
 	Statements []Statement `json:"Statements,omitempty"`

ast/column_encryption.go

@@ -0,0 +1,37 @@
+package ast
+
+// ColumnEncryptionDefinition represents the ENCRYPTED WITH specification
+type ColumnEncryptionDefinition struct {
+	Parameters []ColumnEncryptionParameter
+}
+
+func (c *ColumnEncryptionDefinition) node() {}
+
+// ColumnEncryptionParameter is an interface for encryption parameters
+type ColumnEncryptionParameter interface {
+	columnEncryptionParameter()
+}
+
+// ColumnEncryptionKeyNameParameter represents COLUMN_ENCRYPTION_KEY = key_name
+type ColumnEncryptionKeyNameParameter struct {
+	Name          *Identifier
+	ParameterKind string // "ColumnEncryptionKey"
+}
+
+func (c *ColumnEncryptionKeyNameParameter) columnEncryptionParameter() {}
+
+// ColumnEncryptionTypeParameter represents ENCRYPTION_TYPE = DETERMINISTIC|RANDOMIZED
+type ColumnEncryptionTypeParameter struct {
+	EncryptionType string // "Deterministic", "Randomized"
+	ParameterKind  string // "EncryptionType"
+}
+
+func (c *ColumnEncryptionTypeParameter) columnEncryptionParameter() {}
+
+// ColumnEncryptionAlgorithmParameter represents ALGORITHM = 'algorithm_name'
+type ColumnEncryptionAlgorithmParameter struct {
+	EncryptionAlgorithm ScalarExpression // StringLiteral
+	ParameterKind       string           // "Algorithm"
+}
+
+func (c *ColumnEncryptionAlgorithmParameter) columnEncryptionParameter() {}