1 files changed, 53 insertions, 4 deletions

diff --git a/subprojects/language/src/main/java/tools/refinery/language/utils/BuiltinSymbols.java b/subprojects/language/src/main/java/tools/refinery/language/utils/BuiltinSymbols.java
index c87fa044..72f23e85 100644
--- a/subprojects/language/src/main/java/tools/refinery/language/utils/BuiltinSymbols.java
+++ b/subprojects/language/src/main/java/tools/refinery/language/utils/BuiltinSymbols.java
@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
+ * SPDX-FileCopyrightText: 2021-2024 The Refinery Authors <https://refinery.tools/>
  *
  * SPDX-License-Identifier: EPL-2.0
  */
@@ -7,7 +7,56 @@ package tools.refinery.language.utils;
 
 import tools.refinery.language.model.problem.*;
 
-public record BuiltinSymbols(Problem problem, ClassDeclaration node, PredicateDefinition equals,
-                                                         PredicateDefinition exists, ClassDeclaration contained, PredicateDefinition contains,
-                                                         PredicateDefinition invalidContainer) {
+public final class BuiltinSymbols {
+    private final Problem problem;
+    private final ClassDeclaration node;
+    private final PredicateDefinition equals;
+    private final PredicateDefinition exists;
+    private final ClassDeclaration contained;
+    private final PredicateDefinition contains;
+    private final PredicateDefinition invalidContainer;
+
+    public BuiltinSymbols(Problem problem) {
+        this.problem = problem;
+                node = getDeclaration(ClassDeclaration.class, "node");
+                equals = getDeclaration(PredicateDefinition.class, "equals");
+                exists = getDeclaration(PredicateDefinition.class, "exists");
+                contained = getDeclaration(ClassDeclaration.class, "contained");
+                contains = getDeclaration(PredicateDefinition.class, "contains");
+                invalidContainer = getDeclaration(PredicateDefinition.class, "invalidContainer");
+    }
+
+    public Problem problem() {
+        return problem;
+    }
+
+    public ClassDeclaration node() {
+        return node;
+    }
+
+    public PredicateDefinition equals() {
+        return equals;
+    }
+
+    public PredicateDefinition exists() {
+        return exists;
+    }
+
+    public ClassDeclaration contained() {
+        return contained;
+    }
+
+    public PredicateDefinition contains() {
+        return contains;
+    }
+
+    public PredicateDefinition invalidContainer() {
+        return invalidContainer;
+    }
+
+        private <T extends Statement & NamedElement> T getDeclaration(Class<T> type, String name) {
+                return problem.getStatements().stream().filter(type::isInstance).map(type::cast)
+                                .filter(declaration -> name.equals(declaration.getName())).findFirst()
+                                .orElseThrow(() -> new IllegalArgumentException("Built-in declaration " + name + " was not found"));
+        }
 }