From e88e10b68e80fd37c99de3396e4da5c0df49849b Mon Sep 17 00:00:00 2001
From: TudbuT <tudbut@tudbut.de>
Date: Fri, 4 Aug 2023 22:18:25 +0200
Subject: [PATCH] fix @rust functions

---
 grammar.js | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/grammar.js b/grammar.js
index 6152b51..13334e1 100644
--- a/grammar.js
+++ b/grammar.js
@@ -29,8 +29,7 @@ module.exports = grammar({
     function_definition: $ => choice($.func, $.block),
     func: $ => seq(
       'func', $._spacing,
-      $.identifier, optional($._spacing, $.identifier), $._spacing,
-      $.block,
+      $.identifier, $._spacing, choice(seq('@', $.identifier, $.block), $.block),
     ),
     block: $ => seq(
       optional('!'), '{', optional($._spacing, repeat(seq(/[^ \n\r\t|]+/, $._spacing)), '|'),