feat: add generic context type

plantain-00 · Nov 11, 2018 · d6fc17d · d6fc17d
1 parent 8443c25
commit d6fc17d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/demo/root-type.ts b/demo/root-type.ts
@@ -2,10 +2,10 @@ import { GraphQLResolveInfo } from 'graphql'
 
 import { MutationResult, CreateInput, GetResult } from './cases'
 
-export interface Root {
- create(input: { input: CreateInput }, context: any, info: GraphQLResolveInfo): MutationResult | Promise<MutationResult>
- user(input: { id: string }, context: any, info: GraphQLResolveInfo): GetResult | Promise<GetResult>
- users(input: {}, context: any, info: GraphQLResolveInfo): GetResult | Promise<GetResult>
+export interface Root<TContext = any> {
+ create(input: { input: CreateInput }, context: TContext, info: GraphQLResolveInfo): MutationResult | Promise<MutationResult>
+ user(input: { id: string }, context: TContext, info: GraphQLResolveInfo): GetResult | Promise<GetResult>
+ users(input: {}, context: TContext, info: GraphQLResolveInfo): GetResult | Promise<GetResult>
 }
 
 export interface ResolveResult {

diff --git a/src/graphql-root-type-generator.ts b/src/graphql-root-type-generator.ts
@@ -12,15 +12,15 @@ export function generateGraphqlRootType(declarations: TypeDeclaration[], graphql
  for (const member of typeDeclaration.members) {
  const memberType = getMemberType(member.type, referenceTypes)
  const parameters = getMemberParameters(referenceTypes, member.parameters)
- rootTypes.push(` ${member.name}(${parameters}, context: any, info: GraphQLResolveInfo): ${memberType} | Promise<${memberType}>`)
+ rootTypes.push(` ${member.name}(${parameters}, context: TContext, info: GraphQLResolveInfo): ${memberType} | Promise<${memberType}>`)
  resolveResults.push(` ${member.name}: ${memberType}`)
  }
  }
  }
  const referenceTypeImports = getReferenceTypeImports(referenceTypes, graphqlRootTypePath)
  return `import { GraphQLResolveInfo } from 'graphql'
 
-` + referenceTypeImports + `export interface Root {
+` + referenceTypeImports + `export interface Root<TContext = any> {
 ${rootTypes.join('\n')}
 }