grammar: Add apply to control declaration

Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
2026-04-20 02:27:36 -04:00
parent 6cb28bc873
commit 902dc945fd
2 changed files with 7 additions and 1 deletions
@@ -83,7 +83,7 @@ export default grammar({
        variableDeclaration: $ => seq(optional($.annotations), $.typeRef, field('variable_name', $.identifier), optional(seq($.assignment, $.expression)), $._semicolon),

        // Control declarations
-        control_declaration: $ => seq($.control, $.identifier, $.parameters, '{', repeat(choice($.table_declaration, $.action_declaration)), '}'),
+        control_declaration: $ => seq($.control, $.identifier, $.parameters, '{', repeat(choice($.table_declaration, $.action_declaration)), $.apply, $.statement, '}'),
        action_declaration: $ => seq($.action, $.identifier, $.parameters, $.statement),
        table_declaration: $ => seq($.table, $.identifier, '{', optional($.table_property_list), '}'),