From 1982fda6779273412158c364836e6c54f79e1906 Mon Sep 17 00:00:00 2001
From: Will Hawkins <hawkinsw@obs.cr>
Date: Fri, 13 Mar 2026 09:40:45 -0400
Subject: [PATCH] Array Identifier Should Be Expression

Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
---
 tree-sitter-p4/grammar.js                  | 2 +-
 tree-sitter-p4/test/corpus/expressions.txt | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tree-sitter-p4/grammar.js b/tree-sitter-p4/grammar.js
index a7370ce..a921c8a 100644
--- a/tree-sitter-p4/grammar.js
+++ b/tree-sitter-p4/grammar.js
@@ -94,7 +94,7 @@ export default grammar({
         transitionSelectionExpression: $ => choice($.identifier, $.selectExpression),
         keysetExpression: $ => $.expression,
         binaryOperatorExpression: $ => choice($.binaryEqualOperatorExpression),
-        arrayAccessExpression: $=> seq($.identifier, $.open_bracket, $.expression, $.close_bracket),
+        arrayAccessExpression: $=> seq($.expression, $.open_bracket, $.expression, $.close_bracket),
 
         // Binary Operations
         binaryEqualOperatorExpression: $ => prec.left(2, seq($.expression, $.double_equal, $.expression)),
diff --git a/tree-sitter-p4/test/corpus/expressions.txt b/tree-sitter-p4/test/corpus/expressions.txt
index 61a710b..29993f1 100644
--- a/tree-sitter-p4/test/corpus/expressions.txt
+++ b/tree-sitter-p4/test/corpus/expressions.txt
@@ -144,7 +144,9 @@ parser simple() {
        (expressionStatement
         (expression
          (arrayAccessExpression
-          (identifier)
+          (expression
+           (identifier)
+          )
           (open_bracket)
           (expression
            (integer)