feat: handle Promise type reference

plantain-00 · Nov 7, 2018 · f3fa881 · f3fa881
1 parent bf087cb
commit f3fa881
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/demo/cases.ts b/demo/cases.ts
@@ -356,7 +356,7 @@ export type EntryType = {
 }
 
 export interface Mutation {
-  create(input: CreateInput): MutationResult
+  create(input: CreateInput): Promise<MutationResult>
 }
 
 interface CreateInput {

diff --git a/demo/debug.json b/demo/debug.json
@@ -2752,7 +2752,7 @@
           "position": {
             "file": "demo/cases.ts",
             "line": 358,
-            "character": 30
+            "character": 38
           }
         },
         "parameters": [

diff --git a/online/variables.ts b/online/variables.ts
@@ -363,7 +363,7 @@ export type EntryType = {
 }
 
 export interface Mutation {
-  create(input: CreateInput): MutationResult
+  create(input: CreateInput): Promise<MutationResult>
 }
 
 interface CreateInput {

diff --git a/src/parser.ts b/src/parser.ts
@@ -587,6 +587,12 @@ export class Parser {
       if (reference.typeName.text === 'Array') {
         return this.getTypeOfArrayTypeReference(reference, sourceFile)
       }
+      if (reference.typeName.text === 'Promise' && reference.typeArguments && reference.typeArguments.length > 0) {
+        const typeArgument = reference.typeArguments[0]
+        if (typeArgument.kind === ts.SyntaxKind.TypeReference) {
+          return this.getTypeOfTypeReference(typeArgument as ts.TypeReferenceNode, sourceFile)
+        }
+      }
       return {
         kind: 'reference',
         name: reference.typeName.text,