refactor(gg): share resolver return type renderer

prisma-labs · Jan 25, 2019 · 6b66923 · 6b66923
1 parent eea3e40
commit 6b66923
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/packages/graphqlgen/src/generators/common.ts b/packages/graphqlgen/src/generators/common.ts
@@ -230,6 +230,9 @@ export const union = (types: string[]): string => {
   return types.join(' | ')
 }
 
+export const resolverReturnType = (returnType: string): string =>
+  union([returnType, `Promise<${returnType}>`])
+
 type FieldPrintOptions = {
   isReturn?: boolean
 }

diff --git a/packages/graphqlgen/src/generators/flow-generator.ts b/packages/graphqlgen/src/generators/flow-generator.ts
@@ -23,6 +23,7 @@ import {
   UnionsMap,
   createInterfacesMap,
   createUnionsMap,
+  resolverReturnType,
 } from './common'
 import { renderTypeResolveTypeResolver } from './ts-generator'
 
@@ -352,13 +353,15 @@ function renderResolverFunctionInterface(
     return `
     export type ${resolverName} = {|
       subscribe: ${resolverDefinition} => AsyncIterator<${returnType}> | Promise<AsyncIterator<${returnType}>>,
-      resolve?: ${resolverDefinition} => ${returnType} | Promise<${returnType}>
+      resolve?: ${resolverDefinition} => ${resolverReturnType(returnType)}
     |}
     `
   }
 
   return `
-  export type ${resolverName} = ${resolverDefinition} => ${returnType} | Promise<${returnType}>
+  export type ${resolverName} = ${resolverDefinition} => ${resolverReturnType(
+    returnType,
+  )}
   `
 }
 
@@ -416,14 +419,12 @@ function renderResolverTypeInterfaceFunction(
     return `
     ${field.name}: {|
       subscribe: ${resolverDefinition} => AsyncIterator<${returnType}> | Promise<AsyncIterator<${returnType}>>,
-      resolve?: ${resolverDefinition} => ${returnType} | Promise<${returnType}>
+      resolve?: ${resolverDefinition} => ${resolverReturnType(returnType)}
     |}
     `
   }
   return `
-  ${
-    field.name
-  }: ${resolverDefinition} => ${returnType} | Promise<${returnType}>,
+  ${field.name}: ${resolverDefinition} => ${resolverReturnType(returnType)},
   `
 }
 

diff --git a/packages/graphqlgen/src/generators/ts-generator.ts b/packages/graphqlgen/src/generators/ts-generator.ts
@@ -24,6 +24,7 @@ import {
   createInterfacesMap,
   createUnionsMap,
   union,
+  resolverReturnType,
 } from './common'
 import { TypeAliasDefinition } from '../introspection/types'
 import { upperFirst } from '../utils'
@@ -502,9 +503,6 @@ function renderResolverFunctionInterface(
   `
 }
 
-const resolverReturnType = (returnType: string): string =>
-  union([returnType, `Promise<${returnType}>`])
-
 function renderResolverTypeInterface(
   type: GraphQLTypeObject | GraphQLInterfaceObject,
   modelMap: ModelMap,