graphile-build-pg
Version:
Build a GraphQL schema by reflection over a PostgreSQL schema. Easy to customize since it's built with plugins on graphile-build
129 lines (127 loc) • 4.91 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = void 0;
const base64 = str => Buffer.from(String(str)).toString("base64");
var PgScalarFunctionConnectionPlugin = function PgScalarFunctionConnectionPlugin(builder) {
builder.hook("init", (_, build) => {
const {
newWithHooks,
pgIntrospectionResultsByKind: introspectionResultsByKind,
getTypeByName,
pgGetGqlTypeByTypeIdAndModifier,
graphql: {
GraphQLObjectType,
GraphQLNonNull,
GraphQLList,
GraphQLString
},
inflection,
pgOmit: omit,
describePgEntity,
sqlCommentByAddingTags,
pgField
} = build;
const Cursor = getTypeByName("Cursor");
introspectionResultsByKind.procedure.forEach(proc => {
// PERFORMANCE: These used to be .filter(...) calls
if (!proc.returnsSet) return;
if (!proc.namespace) return;
if (omit(proc, "execute")) return;
const returnType = introspectionResultsByKind.typeById[proc.returnTypeId];
const returnTypeTable = introspectionResultsByKind.classById[returnType.classId];
if (returnTypeTable) {
// Just use the standard table connection from PgTablesPlugin
return;
}
if (returnType.id === "2249") {
// Defer handling to PgRecordFunctionConnectionPlugin
return;
}
// TODO: PG10 doesn't support the equivalent of pg_attribute.atttypemod
// on function arguments and return types, however maybe a later
// version of PG will?
const NodeType = pgGetGqlTypeByTypeIdAndModifier(returnType.id, null) || GraphQLString;
const EdgeType = newWithHooks(GraphQLObjectType, {
name: inflection.scalarFunctionEdge(proc),
description: build.wrapDescription(`A \`${NodeType.name}\` edge in the connection.`, "type"),
fields: ({
fieldWithHooks
}) => {
return {
cursor: fieldWithHooks("cursor", ({
addDataGenerator
}) => {
addDataGenerator(() => ({
usesCursor: [true]
}));
return {
description: build.wrapDescription("A cursor for use in pagination.", "field"),
type: Cursor,
resolve(data) {
return base64(JSON.stringify(data.__cursor));
}
};
}, {
isCursorField: true
}),
node: {
description: build.wrapDescription(`The \`${NodeType.name}\` at the end of the edge.`, "field"),
type: NodeType,
resolve(data) {
return data.value;
}
}
};
}
}, {
__origin: `Adding function result edge type for ${describePgEntity(proc)}. You can rename the function's GraphQL field (and its dependent types) via a 'Smart Comment':\n\n ${sqlCommentByAddingTags(proc, {
name: "newNameHere"
})}`,
isEdgeType: true,
nodeType: NodeType,
pgIntrospection: proc
});
/*const ConnectionType = */
newWithHooks(GraphQLObjectType, {
name: inflection.scalarFunctionConnection(proc),
description: build.wrapDescription(`A connection to a list of \`${NodeType.name}\` values.`, "type"),
fields: ({
fieldWithHooks
}) => {
return {
nodes: pgField(build, fieldWithHooks, "nodes", {
description: build.wrapDescription(`A list of \`${NodeType.name}\` objects.`, "field"),
type: new GraphQLNonNull(new GraphQLList(NodeType)),
resolve(data) {
return data.data.map(entry => entry.value);
}
}),
edges: pgField(build, fieldWithHooks, "edges", {
description: build.wrapDescription(`A list of edges which contains the \`${NodeType.name}\` and cursor to aid in pagination.`, "field"),
type: new GraphQLNonNull(new GraphQLList(new GraphQLNonNull(EdgeType))),
resolve(data) {
return data.data;
}
}, {}, false, {
hoistCursor: true
})
};
}
}, {
__origin: `Adding function connection type for ${describePgEntity(proc)}. You can rename the function's GraphQL field (and its dependent types) via a 'Smart Comment':\n\n ${sqlCommentByAddingTags(proc, {
name: "newNameHere"
})}`,
isConnectionType: true,
isPgRowConnectionType: true,
edgeType: EdgeType,
nodeType: NodeType,
pgIntrospection: proc
});
});
return _;
}, ["PgScalarFunctionConnection"], [], ["PgTypes"]);
};
exports.default = PgScalarFunctionConnectionPlugin;
//# sourceMappingURL=PgScalarFunctionConnectionPlugin.js.map