improve attributes check on prefilter definition

chriskalmar · Jun 6, 2022 · aa50f13 · aa50f13
1 parent 0a534e9
commit aa50f13
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/src/engine/filter.ts b/src/engine/filter.ts
@@ -8,10 +8,11 @@ import {
   isFunction,
 } from './util';
 import { isObjectDataType, ObjectDataType } from './datatype/ObjectDataType';
+import { DataTypeFunction } from './datatype/DataType';
 
 type PreFilter = {
   [key: string]: {
-    resolve: Function;
+    resolve: () => unknown;
     attributes?: ObjectDataType;
   };
 };
@@ -173,6 +174,22 @@ const isPreFilter = (preFilterDefinition: PreFilter) => {
       ) {
         return true;
       }
+
+      if (isFunction(preFilterDefinition.attributes)) {
+        const dataTypeBuilder =
+          preFilterDefinition.attributes as DataTypeFunction;
+
+        const attributesType = dataTypeBuilder({
+          setup: {
+            name: 'temporary',
+            description: 'temporary',
+          },
+        });
+
+        if (isObjectDataType(attributesType)) {
+          return true;
+        }
+      }
     }
   }