feat: different type exports

Author: reconbot

lib/index.ts

@@ -1,11 +1,11 @@
-import { ServerArgs, ServerClosure } from './types'
+import { ServerArgs, ServerClosure, ServerInstance } from './types'
 import { publish } from './pubsub/publish'
 import { complete } from './pubsub/complete'
 import { handleGatewayEvent } from './gateway'
 import { handleStateMachineEvent } from './stepFunctionHandler'
 import { makeServerClosure } from './makeServerClosure'
 
-export const createInstance = (opts: ServerArgs) => {
+export const createInstance = (opts: ServerArgs): ServerInstance => {
   const closure: ServerClosure = makeServerClosure(opts)
 
   return {
@@ -17,7 +17,6 @@ export const createInstance = (opts: ServerArgs) => {
 }
 
 export * from './pubsub/subscribe'
-
 export * from './types'
 export { Subscription } from './model/Subscription'
 export { Connection } from './model/Connection'

lib/types.ts

@@ -1,18 +1,15 @@
 /* eslint-disable @typescript-eslint/ban-types */
-import {
-  ConnectionInitMessage,
-  PingMessage,
-  PongMessage,
-} from 'graphql-ws'
+import { ConnectionInitMessage, PingMessage, PongMessage } from 'graphql-ws'
 import { DataMapper } from '@aws/dynamodb-data-mapper'
-import {
-  APIGatewayEventRequestContext,
-  APIGatewayProxyEvent,
-} from 'aws-lambda'
+import { APIGatewayEventRequestContext, APIGatewayProxyEvent } from 'aws-lambda'
 import { GraphQLResolveInfo, GraphQLSchema } from 'graphql'
 import { DynamoDB } from 'aws-sdk'
 import { Subscription } from './model/Subscription'
 import { Connection } from './model/Connection'
+import { publish } from './pubsub/publish'
+import { complete } from './pubsub/complete'
+import { handleGatewayEvent } from './gateway'
+import { handleStateMachineEvent } from './stepFunctionHandler'
 
 export type ServerArgs = {
   schema: GraphQLSchema
@@ -53,6 +50,13 @@ export type ServerClosure = {
   }
 } & Omit<ServerArgs, 'tableNames'>
 
+export interface ServerInstance {
+    gatewayHandler: ReturnType<typeof handleGatewayEvent>
+    stateMachineHandler:ReturnType<typeof handleStateMachineEvent>
+    publish:ReturnType<typeof publish>
+    complete: ReturnType<typeof complete>
+}
+
 export type TableNames = {
   connections: string
   subscriptions: string
@@ -108,7 +112,6 @@ export type PubSubEvent = {
 
 export type MessageHandler<T> = (arg: { server: ServerClosure, event: APIGatewayWebSocketEvent, message: T }) => Promise<void>
 
-
 /*
   Matches the ApiGatewayManagementApi class from aws-sdk but only provides the methods we use
 */