feat(core/generate): generate functions take a collection or a collec…

…tion promise
rafamel · Oct 24, 2019 · 379bc11 · 379bc11
1 parent 5ce0ffc
commit 379bc11
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/packages/core/src/generate/client.ts b/packages/core/src/generate/client.ts
@@ -34,7 +34,7 @@ export interface ClientGenerateOptions {
 }
 
 export async function client(
-  collection: CollectionTree,
+  collection: CollectionTree | Promise<CollectionTree>,
   resolver: (service: Service, info: ElementInfo) => string,
   options?: ClientGenerateOptions
 ): Promise<string> {
@@ -48,6 +48,7 @@ export async function client(
   };
 
   let content = '';
+  collection = await collection;
 
   if (opts.typescript) {
     content += await typings(collection, {

diff --git a/packages/core/src/generate/typings.ts b/packages/core/src/generate/typings.ts
@@ -15,7 +15,7 @@ export interface TypingsGenerateOptions {
 }
 
 export async function typings(
-  collection: CollectionTree,
+  collection: CollectionTree | Promise<CollectionTree>,
   options?: TypingsGenerateOptions
 ): Promise<string> {
   const opts = { write: null, headComments: true, ...options };
@@ -29,7 +29,7 @@ export async function typings(
       '/* This file was automatically generated. DO NOT MODIFY IT BY HAND. */\n\n';
   }
 
-  const { types } = normalize(collection);
+  const { types } = normalize(await collection);
   for (const [key, value] of Object.entries(types)) {
     if (value.kind !== 'error') {
       content += await compile(value.schema, key, { bannerComment: '' });