fix(frontend): module name highlighting

2 files changed, 5 insertions, 2 deletions

diff --git a/subprojects/frontend/src/language/problem.grammar b/subprojects/frontend/src/language/problem.grammar
index e39ce102..b69ee73f 100644
--- a/subprojects/frontend/src/language/problem.grammar
+++ b/subprojects/frontend/src/language/problem.grammar
@@ -23,7 +23,7 @@
 
 statement {
   ImportStatement {
-    kw<"import"> QualifiedName (kw<"as"> identifier)? "."
+    kw<"import"> ModuleName (kw<"as"> ModuleName)? "."
   } |
   Assertion {
     kw<"default">? (NotOp | UnknownOp)? RelationName
@@ -31,7 +31,7 @@ statement {
     (":" Expr)? "."
   } |
   ProblemDeclaration {
-    (ckw<"module"> | kw<"problem">) QualifiedName "."
+    (ckw<"module"> | kw<"problem">) ModuleName "."
   } |
   ClassDefinition {
     kw<"abstract">? kw<"class"> RelationName
@@ -173,6 +173,8 @@ VariableName[@dynamicPrecedence=10] { QualifiedName ~name }
 
 NodeName { QualifiedName }
 
+ModuleName { QualifiedName }
+
 QualifiedName[implicitCompletion=true] { "::"? identifier (QualifiedNameSeparator "::" identifier)* }
 
 kw<term> { @specialize[@name={term},implicitCompletion=true]<identifier, term> }
diff --git a/subprojects/frontend/src/language/problemLanguageSupport.ts b/subprojects/frontend/src/language/problemLanguageSupport.ts
index aebccfa4..14826363 100644
--- a/subprojects/frontend/src/language/problemLanguageSupport.ts
+++ b/subprojects/frontend/src/language/problemLanguageSupport.ts
@@ -47,6 +47,7 @@ const parserWithMetadata = parser.configure({
       // 'RuleName/QualifiedName': t.typeName,
       'AtomNodeName/QualifiedName': t.atom,
       'VariableName/QualifiedName': t.variableName,
+      'ModuleName/QualifiedName': t.typeName,
       '{ }': t.brace,
       '( )': t.paren,
       '[ ]': t.squareBracket,