Add source positional information to the AST (#63)

Author: kyleconroy

TsqlAstParser/Program.cs

@@ -94,6 +94,16 @@ class AstToJsonConverter
             ["$type"] = node.GetType().Name
         };
 
+        // Emit positional information for fragments that have real tokens.
+        // Synthesized fragments (no source tokens) have StartOffset == -1.
+        if (node.StartOffset >= 0)
+        {
+            result["StartOffset"] = node.StartOffset;
+            result["FragmentLength"] = node.FragmentLength;
+            result["StartLine"] = node.StartLine;
+            result["StartColumn"] = node.StartColumn;
+        }
+
         var type = node.GetType();
         var properties = type.GetProperties(BindingFlags.Public | BindingFlags.Instance);
 

ast/adhoc_table_reference.go

@@ -4,6 +4,7 @@ package ast
 // Syntax: OPENDATASOURCE('provider', 'connstr').'object'
 // Uses AdHocDataSource from execute_statement.go
 type AdHocTableReference struct {
+	Fragment
 	DataSource *AdHocDataSource                   `json:"DataSource,omitempty"`
 	Object     *SchemaObjectNameOrValueExpression `json:"Object,omitempty"`
 	Alias      *Identifier                        `json:"Alias,omitempty"`
@@ -15,6 +16,7 @@ func (*AdHocTableReference) tableReference() {}
 
 // SchemaObjectNameOrValueExpression represents either a schema object name or a value expression
 type SchemaObjectNameOrValueExpression struct {
+	Fragment
 	SchemaObjectName *SchemaObjectName `json:"SchemaObjectName,omitempty"`
 	ValueExpression  ScalarExpression  `json:"ValueExpression,omitempty"`
 }

ast/alter_authorization_statement.go

@@ -2,6 +2,7 @@ package ast
 
 // AlterAuthorizationStatement represents an ALTER AUTHORIZATION statement
 type AlterAuthorizationStatement struct {
+	Fragment
 	SecurityTargetObject *SecurityTargetObject
 	ToSchemaOwner        bool
 	PrincipalName        *Identifier

ast/alter_availability_group_statement.go

@@ -2,12 +2,13 @@ package ast
 
 // AlterAvailabilityGroupStatement represents ALTER AVAILABILITY GROUP statement
 type AlterAvailabilityGroupStatement struct {
-	Name              *Identifier
-	StatementType     string // "Action", "AddDatabase", "RemoveDatabase", "AddReplica", "ModifyReplica", "RemoveReplica", "Set"
-	Action            AvailabilityGroupAction
-	Databases         []*Identifier
-	Replicas          []*AvailabilityReplica
-	Options           []AvailabilityGroupOption
+	Fragment
+	Name          *Identifier
+	StatementType string // "Action", "AddDatabase", "RemoveDatabase", "AddReplica", "ModifyReplica", "RemoveReplica", "Set"
+	Action        AvailabilityGroupAction
+	Databases     []*Identifier
+	Replicas      []*AvailabilityReplica
+	Options       []AvailabilityGroupOption
 }
 
 func (s *AlterAvailabilityGroupStatement) node()      {}
@@ -21,6 +22,7 @@ type AvailabilityGroupAction interface {
 
 // AlterAvailabilityGroupAction represents simple actions like JOIN, ONLINE, OFFLINE
 type AlterAvailabilityGroupAction struct {
+	Fragment
 	ActionType string // "Join", "ForceFailoverAllowDataLoss", "Online", "Offline"
 }
 
@@ -29,6 +31,7 @@ func (a *AlterAvailabilityGroupAction) availabilityGroupAction() {}
 
 // AlterAvailabilityGroupFailoverAction represents FAILOVER action with options
 type AlterAvailabilityGroupFailoverAction struct {
+	Fragment
 	ActionType string // "Failover"
 	Options    []*AlterAvailabilityGroupFailoverOption
 }
@@ -38,6 +41,7 @@ func (a *AlterAvailabilityGroupFailoverAction) availabilityGroupAction() {}
 
 // AlterAvailabilityGroupFailoverOption represents an option for failover action
 type AlterAvailabilityGroupFailoverOption struct {
+	Fragment
 	OptionKind string           // "Target"
 	Value      ScalarExpression // StringLiteral for target server
 }

ast/alter_credential_statement.go

@@ -2,6 +2,7 @@ package ast
 
 // AlterCredentialStatement represents an ALTER CREDENTIAL statement.
 type AlterCredentialStatement struct {
+	Fragment
 	Name             *Identifier
 	Identity         ScalarExpression
 	Secret           ScalarExpression