parser: add array_access

Fixes #66
usagi-coffee · Sep 13, 2024 · b21aff0 · b21aff0
1 parent 986f95f
commit b21aff0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/grammar.js b/grammar.js
@@ -170,6 +170,17 @@ module.exports = grammar({
         $.logical_expression
       ),
 
+    array_access: ($) =>
+      prec.right(
+        1,
+        seq(
+          field("array", choice($.identifier, $.object_access)),
+          "[",
+          $._expression,
+          "]"
+        )
+      ),
+
     /// Includes
     include_argument: ($) =>
       choice(
@@ -1235,6 +1246,7 @@ module.exports = grammar({
         $._binary_expression,
         $.qualified_name,
         $.object_access,
+        $.array_access,
         $.function_call,
         $.ternary_expression,
         $.available_expression,

diff --git a/test/corpus/basic.txt b/test/corpus/basic.txt
@@ -1124,6 +1124,8 @@ OBJECT ACCESSORS statement
 DISPLAY some_object:root.
 DISPLAY some_object:chained_property:chained_another:method().
 
+PUT some_object:array[5].
+
 --------------------------------------------------------------------------------
 
 (source_code
@@ -1139,7 +1141,14 @@ DISPLAY some_object:chained_property:chained_another:method().
         object: (identifier)
         property: (identifier)
         property: (identifier)
-        property: (identifier)))))
+        property: (identifier))))
+  (abl_statement
+    statement: (identifier)
+    (array_access
+      array: (object_access
+        object: (identifier)
+        property: (identifier))
+      (number_literal))))
 
 ================================================================================
 TERNARY EXPRESSION statement