Add support for multiple statement types and enable more tests (#9)

Author: kyleconroy

ast/alter_function_statement.go

@@ -0,0 +1,45 @@
+package ast
+
+// AlterFunctionStatement represents an ALTER FUNCTION statement
+type AlterFunctionStatement struct {
+	Name          *SchemaObjectName
+	Parameters    []*ProcedureParameter
+	ReturnType    FunctionReturnType
+	Options       []*FunctionOption
+	StatementList *StatementList
+}
+
+func (s *AlterFunctionStatement) statement() {}
+func (s *AlterFunctionStatement) node()      {}
+
+// FunctionReturnType is an interface for function return types
+type FunctionReturnType interface {
+	functionReturnTypeNode()
+}
+
+// ScalarFunctionReturnType represents a scalar function return type
+type ScalarFunctionReturnType struct {
+	DataType DataTypeReference
+}
+
+func (r *ScalarFunctionReturnType) functionReturnTypeNode() {}
+
+// TableValuedFunctionReturnType represents a table-valued function return type
+type TableValuedFunctionReturnType struct {
+	// Simplified - will be expanded later
+}
+
+func (r *TableValuedFunctionReturnType) functionReturnTypeNode() {}
+
+// SelectFunctionReturnType represents a SELECT function return type
+type SelectFunctionReturnType struct {
+	// Simplified - will be expanded later
+}
+
+func (r *SelectFunctionReturnType) functionReturnTypeNode() {}
+
+// FunctionOption represents a function option
+type FunctionOption struct {
+	OptionKind  string
+	OptionState string
+}

ast/alter_index_statement.go

@@ -0,0 +1,23 @@
+package ast
+
+// AlterIndexStatement represents ALTER INDEX statement
+type AlterIndexStatement struct {
+	Name           *Identifier
+	All            bool
+	OnName         *SchemaObjectName
+	AlterIndexType string // "Rebuild", "Reorganize", "Disable", "Set", etc.
+	Partition      *PartitionSpecifier
+	IndexOptions   []IndexOption
+}
+
+func (s *AlterIndexStatement) statement() {}
+func (s *AlterIndexStatement) node()      {}
+
+// PartitionSpecifier represents a partition specifier
+type PartitionSpecifier struct {
+	All     bool
+	Number  ScalarExpression
+	Numbers []ScalarExpression
+}
+
+func (p *PartitionSpecifier) node() {}

ast/alter_table_alter_index_statement.go

@@ -11,10 +11,26 @@ type AlterTableAlterIndexStatement struct {
 func (a *AlterTableAlterIndexStatement) node()      {}
 func (a *AlterTableAlterIndexStatement) statement() {}
 
+// IndexOption is an interface for index options
+type IndexOption interface {
+	Node
+	indexOption()
+}
+
+// IndexStateOption represents an ON/OFF index option
+type IndexStateOption struct {
+	OptionKind  string // "PadIndex", "SortInTempDB", "IgnoreDupKey", etc.
+	OptionState string // "On", "Off"
+}
+
+func (o *IndexStateOption) indexOption() {}
+func (o *IndexStateOption) node()        {}
+
 // IndexExpressionOption represents an index option with expression value
 type IndexExpressionOption struct {
 	OptionKind string
 	Expression ScalarExpression
 }
 
-func (i *IndexExpressionOption) node() {}
+func (i *IndexExpressionOption) indexOption() {}
+func (i *IndexExpressionOption) node()        {}

ast/alter_table_constraint_modification_statement.go

@@ -0,0 +1,13 @@
+package ast
+
+// AlterTableConstraintModificationStatement represents ALTER TABLE ... CHECK/NOCHECK CONSTRAINT
+type AlterTableConstraintModificationStatement struct {
+	SchemaObjectName            *SchemaObjectName
+	ExistingRowsCheckEnforcement string // "NotSpecified", "Check", "NoCheck"
+	ConstraintEnforcement       string // "Check", "NoCheck"
+	All                         bool
+	ConstraintNames             []*Identifier
+}
+
+func (s *AlterTableConstraintModificationStatement) statement() {}
+func (s *AlterTableConstraintModificationStatement) node()      {}

ast/alter_table_switch_statement.go

@@ -0,0 +1,38 @@
+package ast
+
+// AlterTableSwitchStatement represents ALTER TABLE ... SWITCH
+type AlterTableSwitchStatement struct {
+	SchemaObjectName  *SchemaObjectName
+	SourcePartition   ScalarExpression
+	TargetTable       *SchemaObjectName
+	TargetPartition   ScalarExpression
+	Options           []TableSwitchOption
+	LowPriorityLockWait *LowPriorityLockWait
+}
+
+func (s *AlterTableSwitchStatement) statement() {}
+func (s *AlterTableSwitchStatement) node()      {}
+
+// TableSwitchOption is an interface for switch options
+type TableSwitchOption interface {
+	Node
+	tableSwitchOption()
+}
+
+// TruncateTargetTableSwitchOption represents TRUNCATE_TARGET option
+type TruncateTargetTableSwitchOption struct {
+	TruncateTarget bool
+	OptionKind     string
+}
+
+func (o *TruncateTargetTableSwitchOption) tableSwitchOption() {}
+func (o *TruncateTargetTableSwitchOption) node()              {}
+
+// LowPriorityLockWait represents LOW_PRIORITY_LOCK_WAIT option
+type LowPriorityLockWait struct {
+	MaxDuration       ScalarExpression
+	MaxDurationUnit   string // "MINUTES", "SECONDS"
+	AfterWaitAbort    string // "NONE", "SELF", "BLOCKERS"
+}
+
+func (l *LowPriorityLockWait) node() {}

ast/alter_table_trigger_modification_statement.go

@@ -0,0 +1,12 @@
+package ast
+
+// AlterTableTriggerModificationStatement represents ALTER TABLE ... ENABLE/DISABLE TRIGGER
+type AlterTableTriggerModificationStatement struct {
+	SchemaObjectName   *SchemaObjectName
+	TriggerEnforcement string // "Enable" or "Disable"
+	All                bool
+	TriggerNames       []*Identifier
+}
+
+func (s *AlterTableTriggerModificationStatement) statement() {}
+func (s *AlterTableTriggerModificationStatement) node()      {}

ast/alter_trigger_statement.go

@@ -0,0 +1,42 @@
+package ast
+
+// AlterTriggerStatement represents an ALTER TRIGGER statement
+type AlterTriggerStatement struct {
+	Name                  *SchemaObjectName
+	TriggerObject         *TriggerObject
+	TriggerType           string // "For", "After", "InsteadOf"
+	TriggerActions        []*TriggerAction
+	Options               []*TriggerOption
+	WithAppend            bool
+	IsNotForReplication   bool
+	MethodSpecifier       *MethodSpecifier
+	StatementList         *StatementList
+}
+
+func (s *AlterTriggerStatement) statement() {}
+func (s *AlterTriggerStatement) node()      {}
+
+// TriggerObject represents the object a trigger is associated with
+type TriggerObject struct {
+	Name         *SchemaObjectName
+	TriggerScope string // "Normal", "AllServer", "Database"
+}
+
+// TriggerAction represents a trigger action
+type TriggerAction struct {
+	TriggerActionType string              // "Insert", "Update", "Delete", "Event", etc.
+	EventTypeGroup    *EventTypeContainer // For database/server events
+}
+
+// TriggerOption represents a trigger option
+type TriggerOption struct {
+	OptionKind  string
+	OptionState string
+}
+
+// MethodSpecifier represents a CLR method specifier
+type MethodSpecifier struct {
+	AssemblyName *Identifier
+	ClassName    *Identifier
+	MethodName   *Identifier
+}

ast/create_aggregate_statement.go

@@ -0,0 +1,18 @@
+package ast
+
+// CreateAggregateStatement represents a CREATE AGGREGATE statement
+type CreateAggregateStatement struct {
+	Name         *SchemaObjectName
+	Parameters   []*ProcedureParameter
+	ReturnType   DataTypeReference
+	AssemblyName *AssemblyName
+}
+
+func (s *CreateAggregateStatement) statement() {}
+func (s *CreateAggregateStatement) node()      {}
+
+// AssemblyName represents an assembly name reference
+type AssemblyName struct {
+	Name      *Identifier
+	ClassName *Identifier
+}

ast/create_columnstore_index_statement.go

@@ -0,0 +1,16 @@
+package ast
+
+// CreateColumnStoreIndexStatement represents a CREATE COLUMNSTORE INDEX statement
+type CreateColumnStoreIndexStatement struct {
+	Name           *Identifier
+	Clustered      bool
+	OnName         *SchemaObjectName
+	Columns        []*ColumnReferenceExpression
+	OrderedColumns []*ColumnReferenceExpression
+	IndexOptions   []IndexOption
+	FilterClause   ScalarExpression
+	OnPartition    *PartitionSpecifier
+}
+
+func (s *CreateColumnStoreIndexStatement) statement() {}
+func (s *CreateColumnStoreIndexStatement) node()      {}

ast/create_trigger_statement.go

@@ -0,0 +1,22 @@
+package ast
+
+// CreateTriggerStatement represents a CREATE TRIGGER statement
+type CreateTriggerStatement struct {
+	Name                *SchemaObjectName
+	TriggerObject       *TriggerObject
+	TriggerType         string // "For", "After", "InsteadOf"
+	TriggerActions      []*TriggerAction
+	Options             []*TriggerOption
+	WithAppend          bool
+	IsNotForReplication bool
+	MethodSpecifier     *MethodSpecifier
+	StatementList       *StatementList
+}
+
+func (s *CreateTriggerStatement) statement() {}
+func (s *CreateTriggerStatement) node()      {}
+
+// EventTypeContainer represents an event type container
+type EventTypeContainer struct {
+	EventType string
+}