Set up our own Plugin discovery for Yoga/NestJS

Author: CarsonF

src/core/graphql/driver.ts

@@ -1,14 +1,28 @@
+import { DiscoveryService } from '@golevelup/nestjs-discovery';
 import { YogaDriver, YogaDriverConfig } from '@graphql-yoga/nestjs';
 import { Injectable } from '@nestjs/common';
 import { HttpAdapter } from '../http';
+import { Plugin } from './plugin.decorator';
 
 @Injectable()
 export class Driver extends YogaDriver<'fastify'> {
-  constructor(private readonly http: HttpAdapter) {
+  constructor(
+    private readonly discovery: DiscoveryService,
+    private readonly http: HttpAdapter,
+  ) {
     super();
   }
 
   async start(options: YogaDriverConfig<'fastify'>) {
+    // Do our plugin discovery / registration
+    const discoveredPlugins = await this.discovery.providersWithMetaAtKey(
+      Plugin.KEY,
+    );
+    options.plugins = [
+      ...(options.plugins ?? []),
+      ...discoveredPlugins.map((cls) => cls.discoveredClass.instance),
+    ];
+
     await super.start(options);
 
     // Setup file upload handling

src/core/graphql/index.ts

@@ -4,3 +4,4 @@ export {
   isGqlContext,
   NotGraphQLContext,
 } from './gql-context.host';
+export * from './plugin.decorator';

src/core/graphql/plugin.decorator.ts

@@ -0,0 +1,13 @@
+import { YogaDriverServerContext } from '@graphql-yoga/nestjs';
+import { createMetadataDecorator } from '@seedcompany/nest';
+import { Plugin as PluginNoContext } from 'graphql-yoga';
+import { GqlContextType } from '~/common';
+
+export const Plugin = createMetadataDecorator({
+  types: ['class'],
+});
+
+export type Plugin = PluginNoContext<
+  GqlContextType,
+  YogaDriverServerContext<'fastify'>
+>;