Allow access to discovered types and functions in delaractive configu…

…ration
Blazebit · Feb 28, 2022 · 4feb7e2 · 4feb7e2
1 parent f894e47
commit 4feb7e2
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 4 deletions.
diff --git a/...ive/api/src/main/java/com/blazebit/domain/declarative/DeclarativeDomainConfiguration.java b/...ive/api/src/main/java/com/blazebit/domain/declarative/DeclarativeDomainConfiguration.java
@@ -29,6 +29,7 @@
 import java.lang.annotation.Annotation;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 
 /**
  * @author Christian Beikov
@@ -61,6 +62,13 @@ public interface DeclarativeDomainConfiguration extends ServiceProvider {
      */
     DeclarativeDomainConfiguration removeDomainType(Class<?> domainTypeClass);
 
+    /**
+     * Returns the registered domain types.
+     *
+     * @return the registered domain types
+     */
+    Set<Class<?>> getDomainTypes();
+
     /**
      * Adds the given domain functions java class for later analysis to add its functions as domain function to the domain builder.
      *
@@ -86,6 +94,13 @@ public interface DeclarativeDomainConfiguration extends ServiceProvider {
      */
     DeclarativeDomainConfiguration removeDomainFunctions(Class<?> domainFunctionsClass);
 
+    /**
+     * Returns the registered domain functions.
+     *
+     * @return the registered domain functions
+     */
+    Set<Class<?>> getDomainFunctions();
+
     /**
      * Returns the type resolver.
      *

diff --git a/...ain/java/com/blazebit/domain/declarative/impl/spi/DeclarativeDomainConfigurationImpl.java b/...ain/java/com/blazebit/domain/declarative/impl/spi/DeclarativeDomainConfigurationImpl.java
@@ -300,6 +300,11 @@ public DeclarativeDomainConfiguration removeDomainFunctions(Class<?> domainFunct
         return this;
     }
 
+    @Override
+    public Set<Class<?>> getDomainFunctions() {
+        return domainFunctions.keySet();
+    }
+
     @Override
     public DeclarativeDomainConfiguration addDomainType(Class<?> domainTypeClass) {
         domainTypes.put(domainTypeClass, AnnotationUtils.findAnnotation(domainTypeClass, DomainType.class));
@@ -318,6 +323,11 @@ public DeclarativeDomainConfiguration removeDomainType(Class<?> domainTypeClass)
         return this;
     }
 
+    @Override
+    public Set<Class<?>> getDomainTypes() {
+        return domainTypes.keySet();
+    }
+
     @Override
     public Map<String, Object> getProperties() {
         return properties;

diff --git a/typescript/src/main/typescript/index.ts b/typescript/src/main/typescript/index.ts
@@ -646,17 +646,17 @@ export class DomainModel {
     }
 
     getOperationTypeResolver(typeName: string, operator: DomainOperator): DomainOperationTypeResolver {
-        let resolvers = this.operationTypeResolvers[typeName][DomainOperator[operator]];
-        return resolvers == null ? null : resolvers;
+        let resolvers = this.operationTypeResolvers[typeName];
+        return resolvers == null ? null : resolvers[DomainOperator[operator]];
     }
 
     getPredicateTypeResolvers(typeName: string): StringMap<DomainPredicateTypeResolver> {
         return this.predicateTypeResolvers[typeName];
     }
 
     getPredicateTypeResolver(typeName: string, predicate: DomainPredicate): DomainPredicateTypeResolver {
-        let resolvers = this.predicateTypeResolvers[typeName][DomainPredicate[predicate]];
-        return resolvers == null ? null : resolvers;
+        let resolvers = this.predicateTypeResolvers[typeName];
+        return resolvers == null ? null : resolvers[DomainPredicate[predicate]];
     }
 
     private static getType(typeName: string, types: StringMap<DomainType>, collectionTypes: StringMap<CollectionDomainType>): DomainType {