Make it so you can pass pgClient into execution

Author: benjie

__tests__/helpers.js

@@ -1,5 +1,8 @@
 // @ts-check
-const { makePgService } = require("postgraphile/adaptors/pg");
+const {
+  makePgService,
+  makeWithPgClientViaPgClientAlreadyInTransaction,
+} = require("postgraphile/adaptors/pg");
 const pg = require("pg");
 const { readFile } = require("fs/promises");
 const { default: PgOrderByRelatedPlugin } = require("../dist/index.js");
@@ -19,10 +22,28 @@ afterAll(() => {
   pool?.end();
 });
 
+/** @type {GraphileConfig.Plugin} */
+const ShoveClientIntoContextPlugin = {
+  name: "ShoveClientIntoContextPlugin",
+
+  grafast: {
+    middleware: {
+      prepareArgs(next, event) {
+        const pgClient = event.args.contextValue.pgClient;
+        if (pgClient) {
+          event.args.contextValue.withPgClient =
+            makeWithPgClientViaPgClientAlreadyInTransaction(pgClient, true);
+        }
+        return next();
+      },
+    },
+  },
+};
+
 /** @type {(schemas: string[], options: import("postgraphile/presets/v4").V4Options) => GraphileConfig.Preset} */
 const makePreset = (schemas, options) => ({
   extends: [makeV4Preset(options)],
-  plugins: [PgOrderByRelatedPlugin],
+  plugins: [PgOrderByRelatedPlugin, ShoveClientIntoContextPlugin],
   pgServices: [
     makePgService({
       pool,

__tests__/integration/queries.test.js

@@ -69,6 +69,10 @@ beforeAll(() => {
           const result = await grafast({
             schema,
             source,
+            variableValues: {},
+            contextValue: {
+              pgClient,
+            },
             resolvedPreset,
             requestContext: {},
           });

__tests__/types.d.ts

@@ -0,0 +1,10 @@
+declare global {
+  namespace Grafast {
+    interface Context {
+      pgClient?: import("pg").PoolClient;
+    }
+  }
+}
+
+// TypeScript hack since this has to be a module
+export const hack = 1;