From 5c0bea2246d530c13269993d25fc9e7891894e49 Mon Sep 17 00:00:00 2001
From: Andreas <38031952+AnHeuermann@users.noreply.github.com>
Date: Fri, 15 Mar 2024 13:50:36 +0100
Subject: [PATCH] Enumeration an Type declarations (#43)

* Highlight metamodelica enumerations
* Handling type declarations
---
 syntaxes/metamodelica.tmGrammar.yaml  | 31 +++++++++++++++++++++++++++
 test/metamodelica/Enumeration.test.mo | 20 +++++++++++++++++
 test/modelica/Comments.test.mo        |  6 +++---
 3 files changed, 54 insertions(+), 3 deletions(-)
 create mode 100644 test/metamodelica/Enumeration.test.mo

diff --git a/syntaxes/metamodelica.tmGrammar.yaml b/syntaxes/metamodelica.tmGrammar.yaml
index c11cdc8..e9040d4 100644
--- a/syntaxes/metamodelica.tmGrammar.yaml
+++ b/syntaxes/metamodelica.tmGrammar.yaml
@@ -17,8 +17,28 @@ patterns:
     name: constant.numeric
 
   # Storage
+  ## Enumeration
+  - begin: \b(enumeration)\b\s*\(
+    beginCaptures:
+      1:
+        name: storage.type.enumeration
+    end: \)\s*;
+    patterns:
+      - include: '#enumeration'
+
+  ## Type declaration
+  - match: \b(type)\b\s+\b(\w+)\b\s+(=)
+    captures:
+      1:
+        name: storage.type
+      2:
+        name: entity.name.type
+      3:
+        name: keyword.operator.assignment
+
   - match: \b(Real|Integer|Boolean|String|enumeration|type)\b
     name: storage.type
+
   ## Variability Prefix
   - match: \b(parameter|constant)\b
     name: storage.modifier
@@ -110,4 +130,15 @@ repository:
     match: '\\(x\h{2}|[0-2][0-7]{0,2}|3[0-6][0-7]|37[0-7]?|[4-7][0-7]?|.)'
     name: constant.character.escape
 
+  enumeration:
+    patterns:
+      - match: \b(\w+)\b\s*((?<!\\)"([^"]*|\\")*(?<!\\)")?
+        captures:
+          1:
+            name: entity.name.type
+          2:
+            name: comment.line
+      - include: "source.modelica"
+
+
 scopeName: source.metamodelica
diff --git a/test/metamodelica/Enumeration.test.mo b/test/metamodelica/Enumeration.test.mo
new file mode 100644
index 0000000..914fc43
--- /dev/null
+++ b/test/metamodelica/Enumeration.test.mo
@@ -0,0 +1,20 @@
+// SYNTAX TEST "source.metamodelica" "Enumeration"
+
+type FavouriteMinionFood = enumeration(
+//<--- source.metamodelica storage.type
+//   ^^^^^^^^^^^^^^^^^^^ source.metamodelica entity.name.type
+//                       ^ source.metamodelica keyword.operator.assignment
+//                         ^^^^^^^^^^^ source.metamodelica storage.type.enumeration
+  BANANA  "A banana",
+//^^^^^^ source.metamodelica entity.name.type
+//        ^^^^^^^^^^ source.metamodelica comment.line
+  APPLE   "An apple",
+//^^^^^ source.metamodelica entity.name.type
+//        ^^^^^^^^^^ source.metamodelica comment.line
+  GELATO  "Ice-cream from the italian \"gelato\"",
+//^^^^^^ source.metamodelica entity.name.type
+//        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ source.metamodelica comment.line
+  PAPAYA  "Wikipedia tells me the fruit is a berry. I don't agree."
+//^^^^^^ source.metamodelica entity.name.type
+//        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ source.metamodelica comment.line
+);
diff --git a/test/modelica/Comments.test.mo b/test/modelica/Comments.test.mo
index 93eb638..20bf03c 100644
--- a/test/modelica/Comments.test.mo
+++ b/test/modelica/Comments.test.mo
@@ -12,9 +12,9 @@ model M "Documentation"
 //                    ^^^^^^^^^^^^^ source.modelica comment.line
   String z2 = "hello";
 //            ^^^^^^^ source.modelica string.quoted.double
-  Foo foo(x=1) "Variable foo";
-//    ^^^ source.modelica entity.name.function
-//             ^^^^^^^^^^^^^^ source.modelica comment.line
+  Foo f = foo(x=1) "Variable foo";
+//        ^^^ source.modelica entity.name.function
+//                 ^^^^^^^^^^^^^^ source.modelica comment.line
   Real a = foo("hello") "Function foo";
 //         ^^^ source.modelica entity.name.function
 //             ^^^^^^^ source.modelica string.quoted.double