Add parser features and enable 8 more tests (#16)

Author: kyleconroy

ast/alter_database_set_statement.go

@@ -45,8 +45,11 @@ func (a *AlterDatabaseAddFileStatement) statement() {}
 
 // AlterDatabaseAddFileGroupStatement represents ALTER DATABASE ... ADD FILEGROUP statement
 type AlterDatabaseAddFileGroupStatement struct {
-	DatabaseName  *Identifier
-	FileGroupName *Identifier
+	DatabaseName              *Identifier
+	FileGroupName             *Identifier
+	ContainsFileStream        bool
+	ContainsMemoryOptimizedData bool
+	UseCurrent                bool
 }
 
 func (a *AlterDatabaseAddFileGroupStatement) node()      {}
@@ -62,8 +65,11 @@ func (a *AlterDatabaseModifyFileStatement) statement() {}
 
 // AlterDatabaseModifyFileGroupStatement represents ALTER DATABASE ... MODIFY FILEGROUP statement
 type AlterDatabaseModifyFileGroupStatement struct {
-	DatabaseName  *Identifier
-	FileGroupName *Identifier
+	DatabaseName       *Identifier
+	FileGroupName      *Identifier
+	MakeDefault        bool
+	UpdatabilityOption string // "ReadOnly", "ReadWrite", "ReadOnlyOld", "ReadWriteOld", or ""
+	NewFileGroupName   *Identifier
 }
 
 func (a *AlterDatabaseModifyFileGroupStatement) node()      {}
@@ -95,3 +101,20 @@ type AlterDatabaseRemoveFileGroupStatement struct {
 
 func (a *AlterDatabaseRemoveFileGroupStatement) node()      {}
 func (a *AlterDatabaseRemoveFileGroupStatement) statement() {}
+
+// AlterDatabaseScopedConfigurationClearStatement represents ALTER DATABASE SCOPED CONFIGURATION CLEAR statement
+type AlterDatabaseScopedConfigurationClearStatement struct {
+	Option    *DatabaseConfigurationClearOption
+	Secondary bool
+}
+
+func (a *AlterDatabaseScopedConfigurationClearStatement) node()      {}
+func (a *AlterDatabaseScopedConfigurationClearStatement) statement() {}
+
+// DatabaseConfigurationClearOption represents a CLEAR option
+type DatabaseConfigurationClearOption struct {
+	OptionKind string           // "ProcedureCache"
+	PlanHandle ScalarExpression // Optional binary plan handle
+}
+
+func (d *DatabaseConfigurationClearOption) node() {}

ast/alter_resource_governor_statement.go

@@ -0,0 +1,9 @@
+package ast
+
+type AlterResourceGovernorStatement struct {
+	Command            string
+	ClassifierFunction *SchemaObjectName
+}
+
+func (s *AlterResourceGovernorStatement) node()      {}
+func (s *AlterResourceGovernorStatement) statement() {}

ast/alter_simple_statements.go

@@ -34,7 +34,13 @@ func (s *AlterServiceStatement) statement() {}
 
 // AlterCertificateStatement represents an ALTER CERTIFICATE statement.
 type AlterCertificateStatement struct {
-	Name *Identifier `json:"Name,omitempty"`
+	Name               *Identifier    `json:"Name,omitempty"`
+	Kind               string         `json:"Kind,omitempty"` // RemovePrivateKey, WithActiveForBeginDialog, WithPrivateKey, RemoveAttestedOption, AttestedBy
+	ActiveForBeginDialog string       `json:"ActiveForBeginDialog,omitempty"` // NotSet, On, Off
+	PrivateKeyPath     *StringLiteral `json:"PrivateKeyPath,omitempty"`
+	DecryptionPassword *StringLiteral `json:"DecryptionPassword,omitempty"`
+	EncryptionPassword *StringLiteral `json:"EncryptionPassword,omitempty"`
+	AttestedBy         *StringLiteral `json:"AttestedBy,omitempty"`
 }
 
 func (s *AlterCertificateStatement) node()      {}
@@ -86,12 +92,20 @@ func (s *AlterPartitionFunctionStatement) statement() {}
 
 // AlterFulltextCatalogStatement represents an ALTER FULLTEXT CATALOG statement.
 type AlterFulltextCatalogStatement struct {
-	Name *Identifier `json:"Name,omitempty"`
+	Name    *Identifier                   `json:"Name,omitempty"`
+	Action  string                        `json:"Action,omitempty"` // Rebuild, Reorganize, AsDefault
+	Options []*OnOffFullTextCatalogOption `json:"Options,omitempty"`
 }
 
 func (s *AlterFulltextCatalogStatement) node()      {}
 func (s *AlterFulltextCatalogStatement) statement() {}
 
+// OnOffFullTextCatalogOption represents an option for ALTER FULLTEXT CATALOG
+type OnOffFullTextCatalogOption struct {
+	OptionKind  string `json:"OptionKind,omitempty"`  // AccentSensitivity
+	OptionState string `json:"OptionState,omitempty"` // On, Off
+}
+
 // AlterFulltextIndexStatement represents an ALTER FULLTEXT INDEX statement.
 type AlterFulltextIndexStatement struct {
 	OnName *SchemaObjectName `json:"OnName,omitempty"`
@@ -117,3 +131,14 @@ type AlterServiceMasterKeyStatement struct {
 
 func (s *AlterServiceMasterKeyStatement) node()      {}
 func (s *AlterServiceMasterKeyStatement) statement() {}
+
+// RenameEntityStatement represents a RENAME statement (Azure SQL DW/Synapse).
+type RenameEntityStatement struct {
+	RenameEntityType string            `json:"RenameEntityType,omitempty"` // Object, Database
+	SeparatorType    string            `json:"SeparatorType,omitempty"`    // DoubleColon (only when :: is used)
+	OldName          *SchemaObjectName `json:"OldName,omitempty"`
+	NewName          *Identifier       `json:"NewName,omitempty"`
+}
+
+func (s *RenameEntityStatement) node()      {}
+func (s *RenameEntityStatement) statement() {}

ast/cryptographic_provider_statement.go

@@ -0,0 +1,29 @@
+package ast
+
+// CreateCryptographicProviderStatement represents CREATE CRYPTOGRAPHIC PROVIDER statement
+type CreateCryptographicProviderStatement struct {
+	Name *Identifier
+	File ScalarExpression
+}
+
+func (s *CreateCryptographicProviderStatement) node()      {}
+func (s *CreateCryptographicProviderStatement) statement() {}
+
+// AlterCryptographicProviderStatement represents ALTER CRYPTOGRAPHIC PROVIDER statement
+type AlterCryptographicProviderStatement struct {
+	Name   *Identifier
+	Option string // "None", "Enable", "Disable"
+	File   ScalarExpression
+}
+
+func (s *AlterCryptographicProviderStatement) node()      {}
+func (s *AlterCryptographicProviderStatement) statement() {}
+
+// DropCryptographicProviderStatement represents DROP CRYPTOGRAPHIC PROVIDER statement
+type DropCryptographicProviderStatement struct {
+	Name       *Identifier
+	IsIfExists bool
+}
+
+func (s *DropCryptographicProviderStatement) node()      {}
+func (s *DropCryptographicProviderStatement) statement() {}

ast/federation_statements.go

@@ -0,0 +1,22 @@
+package ast
+
+// CreateFederationStatement represents CREATE FEDERATION statement
+type CreateFederationStatement struct {
+	Name             *Identifier
+	DistributionName *Identifier
+	DataType         DataTypeReference
+}
+
+func (s *CreateFederationStatement) node()      {}
+func (s *CreateFederationStatement) statement() {}
+
+// AlterFederationStatement represents ALTER FEDERATION statement
+type AlterFederationStatement struct {
+	Name             *Identifier
+	Kind             string // "Split", "DropLow", "DropHigh"
+	DistributionName *Identifier
+	Boundary         ScalarExpression
+}
+
+func (s *AlterFederationStatement) node()      {}
+func (s *AlterFederationStatement) statement() {}

ast/use_federation_statement.go

@@ -0,0 +1,12 @@
+package ast
+
+// UseFederationStatement represents USE FEDERATION statement
+type UseFederationStatement struct {
+	FederationName   *Identifier
+	DistributionName *Identifier
+	Value            ScalarExpression
+	Filtering        bool
+}
+
+func (s *UseFederationStatement) node()      {}
+func (s *UseFederationStatement) statement() {}