Implement PRINT, THROW, ALTER TABLE DROP, DROP CREDENTIAL, and REVERT statements

ast/alter_table_drop_table_element_statement.go

@@ -0,0 +1,19 @@
+package ast
+
+// AlterTableDropTableElementStatement represents an ALTER TABLE ... DROP statement.
+type AlterTableDropTableElementStatement struct {
+	SchemaObjectName           *SchemaObjectName
+	AlterTableDropTableElements []*AlterTableDropTableElement
+}
+
+func (*AlterTableDropTableElementStatement) node()      {}
+func (*AlterTableDropTableElementStatement) statement() {}
+
+// AlterTableDropTableElement represents an element being dropped from a table.
+type AlterTableDropTableElement struct {
+	TableElementType string
+	Name             *Identifier
+	IsIfExists       bool
+}
+
+func (*AlterTableDropTableElement) node() {}

ast/binary_expression.go

@@ -0,0 +1,11 @@
+package ast
+
+// BinaryExpression represents a binary scalar expression (Add, Subtract, etc.).
+type BinaryExpression struct {
+	BinaryExpressionType string           `json:"BinaryExpressionType,omitempty"`
+	FirstExpression      ScalarExpression `json:"FirstExpression,omitempty"`
+	SecondExpression     ScalarExpression `json:"SecondExpression,omitempty"`
+}
+
+func (*BinaryExpression) node()             {}
+func (*BinaryExpression) scalarExpression() {}

ast/drop_credential_statement.go

@@ -0,0 +1,11 @@
+package ast
+
+// DropCredentialStatement represents a DROP CREDENTIAL statement.
+type DropCredentialStatement struct {
+	IsDatabaseScoped bool
+	Name             *Identifier
+	IsIfExists       bool
+}
+
+func (*DropCredentialStatement) node()      {}
+func (*DropCredentialStatement) statement() {}

ast/print_statement.go

@@ -0,0 +1,9 @@
+package ast
+
+// PrintStatement represents a PRINT statement.
+type PrintStatement struct {
+	Expression ScalarExpression
+}
+
+func (*PrintStatement) node()      {}
+func (*PrintStatement) statement() {}

ast/revert_statement.go

@@ -0,0 +1,9 @@
+package ast
+
+// RevertStatement represents a REVERT statement.
+type RevertStatement struct {
+	Cookie ScalarExpression
+}
+
+func (*RevertStatement) node()      {}
+func (*RevertStatement) statement() {}

ast/throw_statement.go

@@ -0,0 +1,11 @@
+package ast
+
+// ThrowStatement represents a THROW statement.
+type ThrowStatement struct {
+	ErrorNumber ScalarExpression
+	Message     ScalarExpression
+	State       ScalarExpression
+}
+
+func (*ThrowStatement) node()      {}
+func (*ThrowStatement) statement() {}

ast/variable_reference.go

@@ -0,0 +1,9 @@
+package ast
+
+// VariableReference represents a reference to a variable (e.g., @var).
+type VariableReference struct {
+	Name string `json:"Name,omitempty"`
+}
+
+func (*VariableReference) node()             {}
+func (*VariableReference) scalarExpression() {}

parser/lexer.go

@@ -38,6 +38,18 @@ const (
 	TokenOption
 	TokenAll
 	TokenDistinct
+	TokenPrint
+	TokenThrow
+	TokenAlter
+	TokenTable
+	TokenDrop
+	TokenIndex
+	TokenRevert
+	TokenWith
+	TokenCookie
+	TokenDatabase
+	TokenScoped
+	TokenCredential
 )
 
 // Token represents a lexical token.
@@ -281,6 +293,18 @@ var keywords = map[string]TokenType{
 	"OPTION":   TokenOption,
 	"ALL":      TokenAll,
 	"DISTINCT": TokenDistinct,
+	"PRINT":    TokenPrint,
+	"THROW":    TokenThrow,
+	"ALTER":    TokenAlter,
+	"TABLE":    TokenTable,
+	"DROP":       TokenDrop,
+	"INDEX":      TokenIndex,
+	"REVERT":     TokenRevert,
+	"WITH":       TokenWith,
+	"COOKIE":     TokenCookie,
+	"DATABASE":   TokenDatabase,
+	"SCOPED":     TokenScoped,
+	"CREDENTIAL": TokenCredential,
 }
 
 func lookupKeyword(ident string) TokenType {