made mercurius's implicitly injected pubsub subscribe/publish methods…

… type safe (#1115)

* make topics/payloads of default pubsub module type safe and extendable

* fix incorrect use of generics in `pubsub.subscribe` method

index.d.ts

@@ -23,9 +23,21 @@ import { Readable } from 'stream'
 type Mercurius = typeof mercurius
 
 declare namespace mercurius {
-  export interface PubSub {
-    subscribe<TResult = any>(topics: string | string[]): Promise<Readable & AsyncIterableIterator<TResult>>;
-    publish<TResult = any>(event: { topic: string; payload: TResult }, callback?: () => void): void;
+  export interface PubSubTopics {
+    [topic: string]: any;
+  }
+
+  export interface PubSub<TPubSubTopics extends PubSubTopics = PubSubTopics> {
+    subscribe<TTopic extends Extract<keyof TPubSubTopics, string>>(
+      topics: TTopic | TTopic[],
+    ): Promise<Readable & AsyncIterableIterator<TPubSubTopics[TTopic]>>;
+    publish<TTopic extends Extract<keyof TPubSubTopics, string>>(
+      event: {
+        topic: TTopic;
+        payload: TPubSubTopics[TTopic];
+      },
+      callback?: () => void,
+    ): void;
   }
 
   export interface MercuriusContext {

test/types/index.ts

@@ -398,7 +398,7 @@ app.graphql.pubsub.publish({
 })
 
 async () => {
-  const subscription = await app.graphql.pubsub.subscribe<{ newNotification: string }>('topic')
+  const subscription = await app.graphql.pubsub.subscribe('topic')
 
   subscription.on('data', (chunk) => {
     console.log(chunk)