Basic working version

Author: benjie

PostGraphileLink.js

@@ -0,0 +1,34 @@
+const { ApolloLink, Observable } = require("apollo-link");
+const performQuery = require("./performQuery");
+
+module.exports = class PostGraphileLink extends ApolloLink {
+  constructor({ pool, schema }) {
+    super();
+
+    this.pool = pool;
+    this.schema = schema;
+  }
+
+  request(operation) {
+    return new Observable(observer => {
+      performQuery(
+        this.pool,
+        this.schema,
+        operation.query,
+        operation.variables,
+        operation.operationName
+      )
+        .then(data => {
+          if (!observer.closed) {
+            observer.next(data);
+            observer.complete();
+          }
+        })
+        .catch(error => {
+          if (!observer.closed) {
+            observer.error(error);
+          }
+        });
+    });
+  }
+};

RenamedQueryPlugin.js

@@ -0,0 +1,41 @@
+module.exports = function RenamedQueryPlugin(builder) {
+  builder.hook("build", build =>
+    build.extend(
+      build,
+      {
+        $$isQuery: Symbol("isQuery"),
+      },
+      `Extending Build`
+    )
+  );
+  builder.hook("GraphQLSchema", (schema, build) => {
+    const {
+      $$isQuery,
+      newWithHooks,
+      extend,
+      graphql: { GraphQLObjectType },
+    } = build;
+    const queryType = newWithHooks(
+      GraphQLObjectType,
+      {
+        description:
+          "PostGraphile Query type, gives access to data from PostgreSQL",
+        name: "PostGraphileQuery",
+        isTypeOf: (value, _context, info) =>
+          info.parentType == null || value === $$isQuery,
+        fields: {},
+      },
+      {
+        __origin: `gatsby-source-pg override`,
+        isRootQuery: true,
+      }
+    );
+    return extend(
+      schema,
+      {
+        query: queryType,
+      },
+      `Adding 'query' type to Schema`
+    );
+  });
+};

createLink.js

@@ -0,0 +1,15 @@
+const { Pool } = require("pg");
+
+const createSchema = require("./createSchema");
+const PostGraphileLink = require("./PostGraphileLink");
+
+module.exports = async options => {
+  const { connectionString, schema: postgresSchema, ...rest } = options;
+
+  const pool = new Pool({
+    connectionString,
+  });
+
+  const graphqlSchema = await createSchema(pool, postgresSchema, rest);
+  return new PostGraphileLink({ pool, schema: graphqlSchema });
+};

createSchema.js

@@ -0,0 +1,55 @@
+const { createPostGraphileSchema } = require('postgraphile')
+
+function requireFrom(modules, moduleName) {
+  const path = [...modules, moduleName].join('/node_modules/')
+  try {
+    return require(path); // eslint-disable-line
+  } catch (e) {
+    // Doesn't exist.
+    if (modules.length > 1) {
+      const result = requireFrom(
+        modules.slice(0, modules.length - 1),
+        moduleName
+      )
+      if (result) return result
+    }
+    return (
+      requireFrom(modules.slice(1), moduleName) ||
+      requireFrom(modules.slice(0, modules.length - 1), moduleName)
+    )
+  }
+}
+const graphileBuild = requireFrom(
+  ['postgraphile', 'postgraphile-core'],
+  'graphile-build'
+)
+/*
+const graphileBuildPg = requireFrom(
+  ['postgraphile', 'postgraphile-core'],
+  'postgraphile/node_modules/graphile-build-pg'
+)
+*/
+
+const RenamedQueryPlugin = require('./RenamedQueryPlugin')
+
+const skipPlugins = [
+  graphileBuild.QueryPlugin,
+  graphileBuild.MutationPlugin,
+  graphileBuild.SubscriptionPlugin,
+]
+const prependPlugins = [RenamedQueryPlugin]
+
+module.exports = async (pool, schema, options) => {
+  const schemas = Array.isArray(schema) ? schema : schema.split(',')
+  const graphqlSchema = await createPostGraphileSchema(pool, schemas, {
+    simpleCollections: 'both',
+    dynamicJson: true,
+    showErrorStack: true,
+    extendedErrors: ['hint', 'detail', 'errcode'],
+    legacyRelations: 'omit',
+    ...options,
+    skipPlugins: [...skipPlugins, ...(options.skipPlugins || [])],
+    prependPlugins: [...prependPlugins, ...(options.prependPlugins || [])],
+  })
+  return graphqlSchema
+}

gatsby-node.js

@@ -0,0 +1,14 @@
+const gatsbySourceGraphQLNode = require("gatsby-source-graphql/gatsby-node");
+
+const createLink = require("./createLink");
+
+exports.sourceNodes = async (
+  utils,
+  { typeName = "PostGraphile", fieldName = "postgres", ...options }
+) => {
+  await gatsbySourceGraphQLNode.sourceNodes(utils, {
+    typeName,
+    fieldName,
+    createLink: () => createLink(options),
+  });
+};

index.js

@@ -25,6 +25,13 @@
     "url": "https://github.com/graphile/gatsby-source-pg/issues"
   },
   "homepage": "https://github.com/graphile/gatsby-source-pg#readme",
+  "dependencies": {
+    "apollo-link": "^1.2.3",
+    "gatsby-source-graphql": "^2.0.5",
+    "graphql": "0.13.x",
+    "pg": "^7.6.0",
+    "postgraphile": "^4.1.0-rc.0"
+  },
   "devDependencies": {
     "babel-eslint": "^10.0.1",
     "eslint": "^5.8.0",
@@ -38,5 +45,8 @@
     "eslint-plugin-react": "^7.11.1",
     "eslint_d": "^7.1.1",
     "prettier": "^1.14.3"
+  },
+  "resolutions": {
+    "graphql": "0.13.x"
   }
 }

package.json

@@ -0,0 +1,22 @@
+const { withPostGraphileContext } = require("postgraphile");
+const { graphql, print } = require("graphql");
+
+module.exports = async function performQuery(
+  pgPool,
+  schema,
+  query,
+  variables,
+  operationName
+) {
+  const queryString = typeof query === 'string' ? query : print(query);
+  return withPostGraphileContext({ pgPool }, async context =>
+    graphql(
+      schema, // The schema from `createPostGraphileSchema`
+      queryString,
+      null,
+      { ...context }, // You can add more to context if you like
+      variables,
+      operationName
+    )
+  );
+};