From 576e338dcbb7298f5b9c85d46a754d89a8283971 Mon Sep 17 00:00:00 2001
From: phannebohm <phannebohm@hsbi.de>
Date: Wed, 13 Mar 2024 10:50:25 +0100
Subject: [PATCH] Use begin/end rule for class definitions

Fixes #28
---
 syntaxes/modelica.tmGrammar.yaml | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/syntaxes/modelica.tmGrammar.yaml b/syntaxes/modelica.tmGrammar.yaml
index dff9317..238463f 100644
--- a/syntaxes/modelica.tmGrammar.yaml
+++ b/syntaxes/modelica.tmGrammar.yaml
@@ -49,23 +49,23 @@ patterns:
   - match: \b(extends|partial|within)\b
     name: support.type
 
-  # Block definitions
-  - match: \b(model|class|record|block|package)\s+(\w+)\s*(".*")*
-    captures:
+  # Class definitions
+  - begin: \b(model|class|record|block|package|function)\s+(\w+)\s*(".*")*
+    beginCaptures:
       1:
         name: keyword
       2:
         name: entity.name.type
       3:
         name: comment.line
-  - match: \b(function)\s+(\w+)\s*(".*")*
-    captures:
+    end: \b(end)\s+(\2)\s*;
+    endCaptures:
       1:
         name: keyword
       2:
-        name: entity.name.function
-      3:
-        name: comment.line
+        name: entity.name.type
+    patterns:
+      - include: "source.modelica"
 
   # Annotations
   - begin: \b(annotation)\b