feat(language-ide): content assist filtering

2 files changed, 20 insertions, 2 deletions

diff --git a/subprojects/language/src/main/java/tools/refinery/language/resource/ProblemResourceDescriptionStrategy.java b/subprojects/language/src/main/java/tools/refinery/language/resource/ProblemResourceDescriptionStrategy.java
index a16f77eb..cac1f265 100644
--- a/subprojects/language/src/main/java/tools/refinery/language/resource/ProblemResourceDescriptionStrategy.java
+++ b/subprojects/language/src/main/java/tools/refinery/language/resource/ProblemResourceDescriptionStrategy.java
@@ -24,8 +24,9 @@ import java.util.Map;
 
 @Singleton
 public class ProblemResourceDescriptionStrategy extends DefaultResourceDescriptionStrategy {
-        public static final String ERROR_PREDICATE = "tools.refinery.language.resource" +
-                        ".ProblemResourceDescriptionStrategy.ERROR_PREDICATE";
+        private static final String DATA_PREFIX = "tools.refinery.language.resource.ProblemResourceDescriptionStrategy.";
+        public static final String ARITY = DATA_PREFIX + "ARITY";
+        public static final String ERROR_PREDICATE = DATA_PREFIX + "ERROR_PREDICATE";
         public static final String ERROR_PREDICATE_TRUE = "true";
 
         @Inject
@@ -97,6 +98,10 @@ public class ProblemResourceDescriptionStrategy extends DefaultResourceDescripti
 
         protected Map<String, String> getUserData(EObject eObject) {
                 var builder = ImmutableMap.<String, String>builder();
+                if (eObject instanceof Relation relation) {
+                        int arity = ProblemUtil.getArity(relation);
+                        builder.put(ARITY, Integer.toString(arity));
+                }
                 if (eObject instanceof PredicateDefinition predicateDefinition && predicateDefinition.isError()) {
                         builder.put(ERROR_PREDICATE, ERROR_PREDICATE_TRUE);
                 }
diff --git a/subprojects/language/src/main/java/tools/refinery/language/utils/ProblemUtil.java b/subprojects/language/src/main/java/tools/refinery/language/utils/ProblemUtil.java
index a9efc4bb..b3c93b07 100644
--- a/subprojects/language/src/main/java/tools/refinery/language/utils/ProblemUtil.java
+++ b/subprojects/language/src/main/java/tools/refinery/language/utils/ProblemUtil.java
@@ -83,6 +83,19 @@ public final class ProblemUtil {
                 return true;
         }
 
+        public static int getArity(Relation relation) {
+                if (relation instanceof ClassDeclaration || relation instanceof EnumDeclaration) {
+                        return 1;
+                }
+                if (relation instanceof ReferenceDeclaration) {
+                        return 2;
+                }
+                if (relation instanceof PredicateDefinition predicateDefinition) {
+                        return predicateDefinition.getParameters().size();
+                }
+                throw new IllegalArgumentException("Unknown Relation: " + relation);
+        }
+
         private static URI getLibraryUri(String libraryName) {
                 return URI.createURI(ProblemUtil.class.getClassLoader()
                                 .getResource("tools/refinery/language/%s.problem".formatted(libraryName)).toString());