From 378d97f41ab9bf1a3dc2136f340bb57d263ea474 Mon Sep 17 00:00:00 2001
From: Kristóf Marussy <kristof@marussy.com>
Date: Fri, 17 May 2024 17:13:21 +0200
Subject: feat: rule parsing

---
 .../frontend/src/language/problemLanguageSupport.ts | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

(limited to 'subprojects/frontend/src/language/problemLanguageSupport.ts')

diff --git a/subprojects/frontend/src/language/problemLanguageSupport.ts b/subprojects/frontend/src/language/problemLanguageSupport.ts
index dd5d6347..ae998d20 100644
--- a/subprojects/frontend/src/language/problemLanguageSupport.ts
+++ b/subprojects/frontend/src/language/problemLanguageSupport.ts
@@ -14,7 +14,12 @@ import {
 } from '@codemirror/language';
 import { styleTags, tags as t } from '@lezer/highlight';
 
-import { foldBlockComment, foldConjunction, foldDeclaration } from './folding';
+import {
+  foldBlockComment,
+  foldConjunction,
+  foldDeclaration,
+  foldWholeNode,
+} from './folding';
 import {
   indentBlockComment,
   indentDeclaration,
@@ -30,11 +35,13 @@ const parserWithMetadata = parser.configure({
       'module problem class enum pred fn scope': t.definitionKeyword,
       'import as declare atom multi': t.definitionKeyword,
       'extern datatype aggregator': t.definitionKeyword,
+      rule: t.definitionKeyword,
       'abstract extends refers contains container opposite': t.modifier,
       default: t.modifier,
+      'propagation decision': t.modifier,
       'true false unknown error': t.keyword,
+      'candidate may must': t.operatorKeyword,
       'count in is': t.operatorKeyword,
-      // 'new delete': t.keyword,
       NotOp: t.operator,
       UnknownOp: t.operator,
       OrOp: t.separator,
@@ -45,7 +52,7 @@ const parserWithMetadata = parser.configure({
       'RelationName/QualifiedName': t.typeName,
       'DatatypeName/QualifiedName': t.keyword,
       'AggregatorName/QualifiedName': t.operatorKeyword,
-      // 'RuleName/QualifiedName': t.typeName,
+      'RuleName/QualifiedName': t.typeName,
       'AtomNodeName/QualifiedName': t.atom,
       'VariableName/QualifiedName': t.variableName,
       'ModuleName/QualifiedName': t.typeName,
@@ -62,7 +69,7 @@ const parserWithMetadata = parser.configure({
       ScopeDeclaration: indentDeclaration,
       PredicateBody: indentPredicateOrRule,
       // FunctionBody: indentPredicateOrRule,
-      // RuleBody: indentPredicateOrRule,
+      RuleBody: indentPredicateOrRule,
       BlockComment: indentBlockComment,
     }),
     foldNodeProp.add({
@@ -71,9 +78,9 @@ const parserWithMetadata = parser.configure({
       ParameterList: foldInside,
       PredicateBody: foldInside,
       // FunctionBody: foldInside,
-      // RuleBody: foldInside,
+      RuleBody: foldInside,
       Conjunction: foldConjunction,
-      // Consequent: foldWholeNode,
+      Consequent: foldWholeNode,
       AtomDeclaration: foldDeclaration,
       NodeDeclaration: foldDeclaration,
       ScopeDeclaration: foldDeclaration,
@@ -92,7 +99,7 @@ const problemLanguage = LRLanguage.define({
       },
       line: '%',
     },
-    indentOnInput: /^\s*(?:\{|\}|\(|\)|->|;|\.)$/,
+    indentOnInput: /^\s*(?:\{|\}|\(|\)|->|==>|;|\.)$/,
   },
 });
 
-- 
cgit v1.2.3-54-g00ecf