feat: add support in buildContext for updating context based on augme…

packages/mercurius/src/buildContext.ts

@@ -1,12 +1,21 @@
-import type { ApiConfig } from "@dzangolab/fastify-config";
-import type { Database } from "@dzangolab/fastify-slonik";
-import type { FastifyRequest } from "fastify";
-
-const buildContext = async (request: FastifyRequest) => {
-  return {
-    config: request.config as ApiConfig,
-    database: request.slonik as Database,
-  };
+import type { FastifyRequest, FastifyReply } from "fastify";
+import type { MercuriusContext } from "mercurius";
+
+const buildContext = async (request: FastifyRequest, reply: FastifyReply) => {
+  const plugins = request.config.mercurius.plugins;
+
+  const context = {
+    config: request.config,
+    database: request.slonik,
+  } as MercuriusContext;
+
+  if (plugins) {
+    for (const plugin of plugins) {
+      await plugin.updateContext(context, request, reply);
+    }
+  }
+
+  return context;
 };
 
 export default buildContext;

packages/mercurius/src/index.ts

@@ -1,3 +1,4 @@
+import type { MercuriusEnabledPlugin } from "./types";
 import type { ApiConfig } from "@dzangolab/fastify-config";
 import type { Database } from "@dzangolab/fastify-slonik";
 import type { MercuriusOptions } from "mercurius";
@@ -13,8 +14,11 @@ declare module "@dzangolab/fastify-config" {
   interface ApiConfig {
     mercurius: MercuriusOptions & {
       enabled?: boolean;
+      plugins?: MercuriusEnabledPlugin[];
     };
   }
 }
 
 export { default } from "./plugin";
+
+export { type MercuriusEnabledPlugin } from "./types";

packages/mercurius/src/types.ts

@@ -0,0 +1,17 @@
+import type {
+  FastifyPluginCallback,
+  FastifyPluginAsync,
+  FastifyRequest,
+  FastifyReply,
+} from "fastify";
+import type { MercuriusContext } from "mercurius";
+
+export interface MercuriusEnabledPlugin
+  extends FastifyPluginAsync,
+    FastifyPluginCallback {
+  updateContext: (
+    context: MercuriusContext,
+    request: FastifyRequest,
+    reply: FastifyReply
+  ) => Promise<void>;
+}