Merge pull request #43 from sqlstack-io/fix/snowflake-alter-policies

fix(syntax): parse Snowflake ALTER of policies and account-level objects

Author: kvlonge

crates/sqlquack-syntax/src/parser/_helpers/ddl/constraints.rs

@@ -1458,7 +1458,7 @@ impl Parser {
     /// Parse a Snowflake DROP of `kind`, already consumed: the name and the options that kind takes.
     fn parse_drop_snowflake_object(&mut self, kind: String) -> Result<Expression> {
         let if_exists = self.match_keywords(&[TokenType::If, TokenType::Exists]);
-        let name = self.parse_table_ref()?;
+        let name = self.parse_object_name()?;
         let (cascade, restrict, force) = match kind.as_str() {
             "APPLICATION" | "OPENFLOW RUNTIME" => {
                 (self.match_token(TokenType::Cascade), false, false)
@@ -1587,7 +1587,7 @@ impl Parser {
             TokenType::View => self.parse_alter_view_with_modifiers(None, None, None),
             TokenType::Index => self.parse_alter_index(),
             TokenType::Sequence => self.parse_alter_sequence(),
-            _ if self.check_identifier("SESSION") => {
+            _ if self.check_identifier("SESSION") && !self.check_next_identifier("POLICY") => {
                 self.skip();
                 match self.parse_alter_session()? {
                     Some(expr) => Ok(expr),

crates/sqlquack-syntax/src/parser/_helpers/ddl/mod.rs

@@ -8,6 +8,7 @@ pub(crate) mod create_table_forms;
 pub(crate) mod explain;
 pub(crate) mod scripting;
 pub(crate) mod snowflake_alter;
+pub(crate) mod snowflake_alter_account;
 pub(crate) mod snowflake_alter_objects;
 pub(crate) mod snowflake_alter_resources;
 pub(crate) mod table_options;

crates/sqlquack-syntax/src/parser/_helpers/ddl/snowflake_alter.rs

@@ -71,6 +71,9 @@ impl Parser {
         if self.match_token(TokenType::Function) || self.match_identifier("FUNCTION") {
             return self.parse_alter_function().map(Some);
         }
+        if let Some(alter) = self.parse_alter_snowflake_group()? {
+            return Ok(Some(alter));
+        }
         if let Some(alter) = self.parse_alter_snowflake_resource()? {
             return Ok(Some(alter));
         }