allow for more dimensions

DerekStride · Jul 28, 2023 · c383ca2 · c383ca2
1 parent 3877ef6
commit c383ca2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/grammar.js b/grammar.js
@@ -12,6 +12,7 @@ module.exports = grammar({
     [$.object_reference, $._qualified_field],
     [$.object_reference],
     [$.between_expression, $.binary_expression],
+    [$.array_size_definition],
   ],
 
   precedences: $ => [
@@ -447,7 +448,7 @@ module.exports = grammar({
     array_size_definition: $ => seq(
       choice(
         seq(optional($.keyword_array), $._array_size_definition),
-        seq( $._array_size_definition, $._array_size_definition),
+        repeat1($._array_size_definition),
         $.keyword_array,
       ),
     ),

diff --git a/test/corpus/create.txt b/test/corpus/create.txt
@@ -1566,7 +1566,8 @@ CREATE TABLE tab (
     matrix text[][],
     square integer[3][3],
     array4 integer ARRAY[4],
-    array_ integer ARRAY
+    array_ integer ARRAY,
+    multid integer[3][3][3][3]
 );
 
 --------------------------------------------------------------------------------
@@ -1611,4 +1612,13 @@ CREATE TABLE tab (
           (int
             (keyword_int))
           (array_size_definition
-            (keyword_array)))))))
+            (keyword_array)))
+        (column_definition
+          (identifier)
+          (int
+            (keyword_int))
+          (array_size_definition
+            (literal)
+            (literal)
+            (literal)
+            (literal)))))))