feat(core/transform): adds toUnary

rafamel · Oct 18, 2019 · 3e4560f · 3e4560f
1 parent f697ba9
commit 3e4560f
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 9 deletions.
diff --git a/packages/core/src/transform/to.ts b/packages/core/src/transform/to.ts
@@ -1,6 +1,15 @@
-import { CollectionTree, CollectionTreeImplementation, Service } from '~/types';
+import {
+  CollectionTree,
+  CollectionTreeImplementation,
+  Service,
+  UnaryCollection
+} from '~/types';
 import { replace } from './replace';
-import { isElementService, isServiceImplementation } from '~/inspect';
+import {
+  isElementService,
+  isServiceImplementation,
+  isServiceSubscription
+} from '~/inspect';
 
 export function toImplementation<T extends CollectionTree>(
   collection: T,
@@ -24,3 +33,28 @@ export function toDeclaration(collection: CollectionTree): CollectionTree {
     return element;
   });
 }
+
+export function toUnary<T extends CollectionTree>(
+  collection: T
+): UnaryCollection<T> {
+  return replace(collection, (element, next) => {
+    element = next(element);
+
+    if (!isElementService(element) || !isServiceSubscription(element)) {
+      return element;
+    }
+
+    if (!isServiceImplementation(element)) {
+      return { ...element, kind: 'query' };
+    }
+
+    const resolve: any = element.resolve;
+    return {
+      ...element,
+      kind: 'query',
+      resolve(...args: any): Promise<any> {
+        return resolve.apply(this, args).toPromise();
+      }
+    };
+  }) as UnaryCollection<T>;
+}
diff --git a/packages/core/src/types/maps.ts b/packages/core/src/types/maps.ts
@@ -10,7 +10,9 @@ import {
 import {
   CollectionTreeImplementation,
   ServiceImplementation,
-  ScopeTreeImplementation
+  ScopeTreeImplementation,
+  QueryServiceImplementation,
+  MutationServiceImplementation
 } from './implementation';
 import { GenericError } from './types';
 
@@ -20,6 +22,22 @@ export type GenericCollection<
   ? CollectionTreeImplementation
   : CollectionTree;
 
+export type UnaryCollection<
+  T extends CollectionTree
+> = T extends CollectionTreeImplementation
+  ? CollectionTree<
+      QueryServiceImplementation,
+      MutationServiceImplementation,
+      never
+    >
+  : CollectionTree<QueryService, MutationService, never>;
+
+export type ApplicationCollection<
+  T extends CollectionTree = CollectionTree
+> = T & {
+  types: { [P in GenericError]: ErrorType };
+};
+
 export type GenericService<
   T extends CollectionTree
 > = T extends CollectionTreeImplementation ? ServiceImplementation : Service;
@@ -33,12 +51,6 @@ export interface ServicesRoutes<T extends CollectionTree = CollectionTree>
   [key: string]: GenericService<T>;
 }
 
-export type ApplicationCollection<
-  T extends CollectionTree = CollectionTree
-> = T & {
-  types: { [P in GenericError]: ErrorType };
-};
-
 export type ScopeCollection<
   T extends CollectionTree,
   N extends string