fix(core): sets info parameter as required ElementInfo for Collection…

….toImplementation
rafamel · Nov 24, 2019 · 6317dca · 6317dca
1 parent 69ca8ff
commit 6317dca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/packages/core/src/classes/Collection/definitions.ts b/packages/core/src/classes/Collection/definitions.ts
@@ -67,7 +67,7 @@ export type CollectionImplementationInputFn<
   Q extends QueryServiceUnion = QueryServiceUnion,
   M extends MutationServiceUnion = MutationServiceUnion,
   S extends SubscriptionServiceUnion = SubscriptionServiceUnion
-> = (service: Q | M | S, info: ElementInfo) => ServiceImplementation;
+> = (service: Q | M | S, info: Required<ElementInfo>) => ServiceImplementation;
 
 /* Options */
 export interface CollectionInterceptOptions {

diff --git a/packages/core/src/classes/Collection/helpers/to.ts b/packages/core/src/classes/Collection/helpers/to.ts
@@ -26,11 +26,12 @@ export function toImplementation<
   collection: AbstractCollectionTree<Q, M, S>,
   fn: CollectionImplementationInputFn
 ): CollectionTreeImplementation {
-  return replace(collection, (element, info, next) => {
+  return replace(collection, (element, { path, route }, next) => {
     element = next(element);
-    return isElementService(element)
-      ? (fn(element, info) as AbstractElement<Q, M, S>)
-      : element;
+    if (!isElementService(element)) return element;
+
+    if (!route) throw Error(`Route expected at path: ${path}`);
+    return fn(element, { path, route }) as AbstractElement<Q, M, S>;
   }) as CollectionTreeImplementation;
 }