feat: improve MATERIALIZED VIEW support

grammar/keywords.js

@@ -110,6 +110,7 @@ export default {
   keyword_no: _ => make_keyword("no"),
   keyword_data: _ => make_keyword("data"),
   keyword_type: _ => make_keyword("type"),
+  keyword_refresh: _ => make_keyword("refresh"),
   keyword_rename: _ => make_keyword("rename"),
   keyword_to: _ => make_keyword("to"),
   keyword_database: _ => make_keyword("database"),

grammar/statements/alter.js

@@ -6,6 +6,7 @@ export default {
     choice(
       $.alter_table,
       $.alter_view,
+      $.alter_materialized_view,
       $.alter_schema,
       $.alter_type,
       $.alter_index,
@@ -211,6 +212,19 @@ export default {
     ),
   ),
 
+  alter_materialized_view: $ => seq(
+    $.keyword_alter,
+    $.keyword_materialized,
+    $.keyword_view,
+    optional($._if_exists),
+    $.object_reference,
+    choice(
+      $.rename_object,
+      $.set_schema,
+      $.change_ownership,
+    ),
+  ),
+
   alter_schema: $ => seq(
     $.keyword_alter,
     $.keyword_schema,

grammar/statements/drop.js

@@ -6,6 +6,7 @@ export default {
     choice(
       $.drop_table,
       $.drop_view,
+      $.drop_materialized_view,
       $.drop_index,
       $.drop_type,
       $.drop_schema,
@@ -34,6 +35,15 @@ export default {
     optional($._drop_behavior),
   ),
 
+  drop_materialized_view: $ => seq(
+    $.keyword_drop,
+    $.keyword_materialized,
+    $.keyword_view,
+    optional($._if_exists),
+    $.object_reference,
+    optional($._drop_behavior),
+  ),
+
   drop_schema: $ => seq(
     $.keyword_drop,
     $.keyword_schema,

grammar/statements/index.js

@@ -14,6 +14,7 @@ import truncate_rules from "./truncate.js";
 import copy_rules from "./copy.js";
 import select_rules from "./select.js";
 import set_rules from "./set.js";
+import refresh_rules from "./refresh.js";
 
 export default {
 
@@ -79,6 +80,7 @@ export default {
     $._rename_statement,
     $._optimize_statement,
     $._merge_statement,
+    $._refresh_statement,
     $.comment_statement,
     $.set_statement,
     $.reset_statement,
@@ -90,6 +92,7 @@ export default {
   ...rename_rules,
   ...optimize_rules,
   ...merge_rules,
+  ...refresh_rules,
   ...comment_rules,
 
   _dml_write: $ => seq(

grammar/statements/refresh.js

@@ -0,0 +1,20 @@
+export default {
+  _refresh_statement: ($) => $.refresh_materialized_view,
+
+  refresh_materialized_view: ($) =>
+    prec.right(
+      seq(
+        $.keyword_refresh,
+        $.keyword_materialized,
+        $.keyword_view,
+        optional($.keyword_concurrently),
+        $.object_reference,
+        optional(
+          choice(
+            seq($.keyword_with, $.keyword_data),
+            seq($.keyword_with, $.keyword_no, $.keyword_data),
+          ),
+        ),
+      ),
+    ),
+};

queries/highlights.scm

@@ -169,6 +169,7 @@
   (keyword_data)
   (keyword_type)
   (keyword_rename)
+  (keyword_refresh)
   (keyword_to)
   (keyword_schema)
   (keyword_owner)

test/corpus/alter.txt

@@ -1800,3 +1800,114 @@ ALTER POLICY my_policy ON my_schema.my_table
                 name: (identifier))))
           (int
             (keyword_int)))))))
+
+================================================================================
+Alter materialized view rename
+================================================================================
+
+ALTER MATERIALIZED VIEW my_view RENAME TO new_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (alter_materialized_view
+      (keyword_alter)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        name: (identifier))
+      (rename_object
+        (keyword_rename)
+        (keyword_to)
+        (object_reference
+          name: (identifier))))))
+
+================================================================================
+Alter materialized view if exists rename
+================================================================================
+
+ALTER MATERIALIZED VIEW IF EXISTS my_view RENAME TO new_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (alter_materialized_view
+      (keyword_alter)
+      (keyword_materialized)
+      (keyword_view)
+      (keyword_if)
+      (keyword_exists)
+      (object_reference
+        name: (identifier))
+      (rename_object
+        (keyword_rename)
+        (keyword_to)
+        (object_reference
+          name: (identifier))))))
+
+================================================================================
+Alter materialized view set schema
+================================================================================
+
+ALTER MATERIALIZED VIEW my_view SET SCHEMA new_schema;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (alter_materialized_view
+      (keyword_alter)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        name: (identifier))
+      (set_schema
+        (keyword_set)
+        (keyword_schema)
+        schema: (identifier)))))
+
+================================================================================
+Alter materialized view change owner
+================================================================================
+
+ALTER MATERIALIZED VIEW my_view OWNER TO someone_else;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (alter_materialized_view
+      (keyword_alter)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        name: (identifier))
+      (change_ownership
+        (keyword_owner)
+        (keyword_to)
+        (identifier)))))
+
+================================================================================
+Alter materialized view with schema qualification
+================================================================================
+
+ALTER MATERIALIZED VIEW myschema.my_view RENAME TO new_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (alter_materialized_view
+      (keyword_alter)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        schema: (identifier)
+        name: (identifier))
+      (rename_object
+        (keyword_rename)
+        (keyword_to)
+        (object_reference
+          name: (identifier))))))

test/corpus/drop.txt

@@ -291,3 +291,77 @@ DROP EXTENSION my_extension;
       (keyword_drop)
       (keyword_extension)
       (identifier))))
+
+================================================================================
+Drop materialized view
+================================================================================
+
+DROP MATERIALIZED VIEW my_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (drop_materialized_view
+      (keyword_drop)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        name: (identifier)))))
+
+================================================================================
+Drop materialized view if exists
+================================================================================
+
+DROP MATERIALIZED VIEW IF EXISTS my_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (drop_materialized_view
+      (keyword_drop)
+      (keyword_materialized)
+      (keyword_view)
+      (keyword_if)
+      (keyword_exists)
+      (object_reference
+        name: (identifier)))))
+
+================================================================================
+Drop materialized view cascade
+================================================================================
+
+DROP MATERIALIZED VIEW IF EXISTS my_view CASCADE;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (drop_materialized_view
+      (keyword_drop)
+      (keyword_materialized)
+      (keyword_view)
+      (keyword_if)
+      (keyword_exists)
+      (object_reference
+        name: (identifier))
+      (keyword_cascade))))
+
+================================================================================
+Drop materialized view with schema
+================================================================================
+
+DROP MATERIALIZED VIEW myschema.my_view;
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (drop_materialized_view
+      (keyword_drop)
+      (keyword_materialized)
+      (keyword_view)
+      (object_reference
+        schema: (identifier)
+        name: (identifier)))))