feat: added static bindings for subscriptions

Closes #25

Author: kbrandwijk

src/generators/binding-js.ts

@@ -19,6 +19,7 @@ import {
 } from 'graphql'
 
 import { Generator } from '../types'
+import { renderMainMethodFields, renderMainSubscriptionMethodFields } from './graphcool-js'
 
 export const generator: Generator = {
   Main: renderMainMethod,
@@ -44,6 +45,10 @@ ${renderMainMethodFields('query', queryType.getFields())}
       
     this.mutation = {
 ${renderMainMethodFields('mutation', mutationType.getFields())}
+    }`: ''}${subscriptionType ? `
+      
+    this.subscription = {
+${renderMainSubscriptionMethodFields('mutation', subscriptionType.getFields())}
     }`: ''}
   }
   
@@ -53,11 +58,3 @@ ${renderMainMethodFields('mutation', mutationType.getFields())}
 }`
 }
 
-function renderMainMethodFields(operation: string, fields: GraphQLFieldMap<any, any>): string {
-  return Object.keys(fields).map(f => {
-    const field = fields[f]
-    return `      ${field.name}(args, info) { 
-        return self.delegate('${operation}', '${field.name}', args, {}, info)
-      }`
-  }).join(',\n')
-}

src/generators/binding-ts.ts

@@ -19,7 +19,7 @@ import {
 } from 'graphql'
 
 import { Generator } from '../types'
-import { generator as gcgenerator, renderMainMethodFields } from './graphcool-ts'
+import { generator as gcgenerator, renderMainMethodFields, renderMainSubscriptionMethodFields } from './graphcool-ts'
 
 export const generator: Generator = {
   ...gcgenerator,
@@ -45,6 +45,10 @@ ${renderMainMethodFields('query', queryType.getFields())}
 
   mutation: Mutation = {
 ${renderMainMethodFields('mutation', mutationType.getFields())}
+  }`: ''}${subscriptionType ? `
+
+  subscription: Subscription = {
+${renderMainSubscriptionMethodFields(subscriptionType.getFields())}
   }`: ''}
 }`
 }

src/generators/graphcool-js.ts

@@ -51,6 +51,10 @@ ${renderMainMethodFields('query', queryType.getFields())}
       
     this.mutation = {
 ${renderMainMethodFields('mutation', mutationType.getFields())}
+    }`: ''}${subscriptionType ? `
+      
+    this.subscription = {
+${renderMainSubscriptionMethodFields('mutation', subscriptionType.getFields())}
     }`: ''}
   }
   
@@ -83,11 +87,20 @@ export function renderExistsFields(fields: GraphQLFieldMap<any, any>) : string {
     .join(',\n')
 }
 
-function renderMainMethodFields(operation: string, fields: GraphQLFieldMap<any, any>): string {
+export function renderMainMethodFields(operation: string, fields: GraphQLFieldMap<any, any>): string {
   return Object.keys(fields).map(f => {
     const field = fields[f]
     return `      ${field.name}(args, info) { 
         return self.delegate('${operation}', '${field.name}', args, {}, info)
       }`
   }).join(',\n')
 }
+
+export function renderMainSubscriptionMethodFields(operation: string, fields: GraphQLFieldMap<any, any>): string {
+  return Object.keys(fields).map(f => {
+    const field = fields[f]
+    return `      ${field.name}(args, infoOrQuery) { 
+        return self.delegateSubscription('${field.name}', args, infoOrQuery)
+      }`
+  }).join(',\n')
+}

src/generators/graphcool-ts.ts

@@ -28,6 +28,7 @@ export const generator: Generator = {
   GraphQLEnumType: renderEnumType,
   GraphQLInterfaceType: renderObjectType,
   RootType: renderRootType,
+  SubscriptionType: renderSubscriptionType,
   SchemaType: renderSchemaInterface,
   Main: renderMainMethod,
   Header: renderHeader
@@ -63,7 +64,7 @@ function renderMainMethod(
   exists = {
 ${renderExistsFields(queryType.getFields())}
   }
-  
+
   query: Query = {
 ${renderMainMethodFields('query', queryType.getFields())}
   }${
@@ -72,6 +73,14 @@ ${renderMainMethodFields('query', queryType.getFields())}
 
   mutation: Mutation = {
 ${renderMainMethodFields('mutation', mutationType.getFields())}
+  }`
+      : ''
+  }${
+    subscriptionType
+      ? `
+
+  subscription: Subscription = {
+${renderMainSubscriptionMethodFields(subscriptionType.getFields())}
   }`
       : ''
   }
@@ -110,6 +119,15 @@ export function renderMainMethodFields(operation: string, fields: GraphQLFieldMa
     .join(',\n')
 }
 
+export function renderMainSubscriptionMethodFields(fields: GraphQLFieldMap<any, any>): string {
+  return Object.keys(fields)
+    .map(f => {
+      const field = fields[f]
+      return `    ${field.name}: (args, infoOrQuery): Promise<AsyncIterator<${renderFieldType(field.type)}>> => super.delegateSubscription('${field.name}', args, infoOrQuery)`
+    })
+    .join(',\n')
+}
+
 function renderScalarType(type: GraphQLScalarType): string {
   if (type.name === 'ID') {
     return renderIDType(type)
@@ -161,6 +179,21 @@ function renderRootType(type: GraphQLObjectType): string {
   return renderTypeWrapper(type.name, type.description, fieldDefinition)
 }
 
+function renderSubscriptionType(type: GraphQLObjectType): string {
+  const fieldDefinition = Object.keys(type.getFields())
+    .map(f => {
+      const field = type.getFields()[f]
+      return `  ${field.name}: (args: {${field.args.length > 0 ? ' ' : ''}${field.args
+        .map(f => `${renderFieldName(f)}: ${renderFieldType(f.type)}`)
+        .join(', ')}${
+        field.args.length > 0 ? ' ' : ''
+      }}, infoOrQuery?: GraphQLResolveInfo | string) => Promise<AsyncIterator<${renderFieldType(field.type)}>>`
+    })
+    .join('\n')
+
+  return renderTypeWrapper(type.name, type.description, fieldDefinition)
+}
+
 function renderUnionType(type: GraphQLUnionType): string {
   return `${renderDescription(type.description)}export type ${type.name} = ${type
     .getTypes()

src/index.ts

@@ -45,7 +45,11 @@ ${Object.keys(generators).map(k => `'${k}`).join(', ')}`)
     if (generator.RootType) {
       generatedClass.push(generator.RootType(ast.getQueryType()))
       if (ast.getMutationType()) { generatedClass.push(generator.RootType(ast.getMutationType()!)) }
-      if (ast.getSubscriptionType()) { generatedClass.push(generator.RootType(ast.getSubscriptionType()!)) }
+    }
+
+    // Special case 5: subscription type
+    if (generator.SubscriptionType) {
+      if (ast.getSubscriptionType()) { generatedClass.push(generator.SubscriptionType(ast.getSubscriptionType()!)) }
     }
 
     // Special case 3: the main method

src/types.ts

@@ -5,6 +5,7 @@ export interface Generator {
     Header: (schema: string) => string,
     SchemaType?: (queryType: GraphQLObjectType, mutationType?: GraphQLObjectType | null, subscriptionType?: GraphQLObjectType | null) => string,
     RootType?: (type: GraphQLObjectType) => string,
+    SubscriptionType?: (type: GraphQLObjectType) => string,
     GraphQLUnionType?: (type: GraphQLUnionType) => string,
     GraphQLInputObjectType?: (type: GraphQLInputObjectType) => string,
     GraphQLObjectType?: (type: GraphQLObjectType) => string,