UNPKG

@graphql-mesh/runtime

Version:
281 lines (280 loc) • 12.1 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.wrapFetchWithPlugins = wrapFetchWithPlugins; exports.getMesh = getMesh; const graphql_1 = require("graphql"); const core_1 = require("@envelop/core"); const cross_helpers_1 = require("@graphql-mesh/cross-helpers"); const utils_1 = require("@graphql-mesh/utils"); const executor_1 = require("@graphql-tools/executor"); const utils_2 = require("@graphql-tools/utils"); const wrap_1 = require("@graphql-tools/wrap"); const fetch_1 = require("@whatwg-node/fetch"); const promise_helpers_1 = require("@whatwg-node/promise-helpers"); const constants_js_1 = require("./constants.js"); const utils_js_1 = require("./utils.js"); const memoizedGetEnvelopedFactory = (0, utils_2.memoize1)(function getEnvelopedFactory(plugins) { return (0, core_1.envelop)({ plugins, }); }); function wrapFetchWithPlugins(plugins) { const onFetchHooks = []; for (const plugin of plugins) { if (plugin?.onFetch != null) { onFetchHooks.push(plugin.onFetch); } } return (0, utils_1.wrapFetchWithHooks)(onFetchHooks); } // Use in-context-sdk for tracing function createProxyingResolverFactory(apiName, rootTypeMap) { return function createProxyingResolver({ operation }) { const rootType = rootTypeMap.get(operation); return function proxyingResolver(root, args, context, info) { if (!context?.[apiName]?.[rootType.name]?.[info.fieldName]) { throw new Error(`${info.fieldName} couldn't find in ${rootType.name} of ${apiName} as a ${operation}`); } return context[apiName][rootType.name][info.fieldName]({ root, args, context, info }); }; }; } async function getMesh(options) { const rawSources = []; const { pubsub = new utils_1.PubSub(), cache, logger = new utils_1.DefaultLogger(''), additionalEnvelopPlugins = [], sources, merger, additionalResolvers = [], additionalTypeDefs = [], transforms = [], fetchFn = fetch_1.fetch, } = options; const getMeshLogger = logger.child('GetMesh'); getMeshLogger.debug(`Getting subschemas from source handlers`); let failed = false; const initialPluginList = [ // TODO: Not a good practise to expect users to be a Yoga user (0, core_1.useExtendContext)(({ request, req, connectionParams, }) => { // Maybe Node-like environment if (req?.headers) { return { headers: (0, utils_1.getHeadersObj)(req.headers), connectionParams, }; } // Fetch environment if (request?.headers) { return { headers: (0, utils_1.getHeadersObj)(request.headers), connectionParams, }; } return {}; }), (0, core_1.useExtendContext)(() => ({ pubsub, cache, logger, fetch: wrappedFetchFn, [constants_js_1.MESH_CONTEXT_SYMBOL]: true, })), { onFetch({ setFetchFn }) { setFetchFn(fetchFn); }, }, ...additionalEnvelopPlugins, ]; const wrappedFetchFn = wrapFetchWithPlugins(initialPluginList); await Promise.allSettled(sources.map(async (apiSource, index) => { const apiName = apiSource.name; const sourceLogger = logger.child({ source: apiName }); sourceLogger.debug(`Generating the schema`); try { const source = await apiSource.handler.getMeshSource({ fetchFn: wrappedFetchFn, }); sourceLogger.debug(`The schema has been generated successfully`); let apiSchema = source.schema; sourceLogger.debug(`Analyzing transforms`); let transforms; const { wrapTransforms, noWrapTransforms } = (0, utils_1.groupTransforms)(apiSource.transforms); if (!wrapTransforms?.length && noWrapTransforms?.length) { sourceLogger.debug(`${noWrapTransforms.length} bare transforms found and applying`); apiSchema = (0, utils_1.applySchemaTransforms)(apiSchema, source, null, noWrapTransforms); } else { transforms = apiSource.transforms; } const rootTypeMap = (0, utils_2.getRootTypeMap)(apiSchema); rawSources[index] = { name: apiName, schema: apiSchema, executor: source.executor, transforms, contextVariables: source.contextVariables || {}, handler: apiSource.handler, batch: 'batch' in source ? source.batch : true, merge: source.merge, createProxyingResolver: createProxyingResolverFactory(apiName, rootTypeMap), }; } catch (e) { sourceLogger.debug(e); sourceLogger.error(`Failed to generate the schema for the source\n ${e.message}`); failed = true; } })); if (failed) { throw new Error(`Schemas couldn't be generated successfully. Check for the logs by running Mesh${cross_helpers_1.process.env.DEBUG == null ? ' with DEBUG=1 environmental variable to get more verbose output' : ''}.`); } getMeshLogger.debug(`Schemas have been generated by the source handlers`); getMeshLogger.debug(`Merging schemas using the defined merging strategy.`); const unifiedSubschema = await merger.getUnifiedSchema({ rawSources, typeDefs: additionalTypeDefs, resolvers: additionalResolvers, }); unifiedSubschema.transforms = unifiedSubschema.transforms || []; for (const rootLevelTransform of transforms) { if (rootLevelTransform.noWrap) { if (rootLevelTransform.transformSchema) { unifiedSubschema.schema = rootLevelTransform.transformSchema(unifiedSubschema.schema, unifiedSubschema); } } else { unifiedSubschema.transforms.push(rootLevelTransform); } } let inContextSDK; let schema = unifiedSubschema.schema; if (unifiedSubschema.executor != null || unifiedSubschema.transforms?.length) { schema = (0, wrap_1.wrapSchema)(unifiedSubschema); } const plugins = [ (0, core_1.useEngine)({ execute: executor_1.normalizedExecutor, subscribe: executor_1.normalizedExecutor, validate: graphql_1.validate, parse: utils_1.parseWithCache, specifiedRules: graphql_1.specifiedRules, }), (0, core_1.useSchema)(schema), (0, core_1.useExtendContext)(() => { if (!inContextSDK) { const onDelegateHooks = []; for (const plugin of initialPluginList) { if (plugin?.onDelegate != null) { onDelegateHooks.push(plugin.onDelegate); } } inContextSDK = (0, utils_1.getInContextSDK)(schema, rawSources, logger, onDelegateHooks); } return inContextSDK; }), { onExecute() { return { onExecuteDone({ result, setResult }) { if (result.errors) { // Print errors with stack trace in development if (cross_helpers_1.process.env.NODE_ENV === 'production') { for (const error of result.errors) { const origError = (0, utils_js_1.getOriginalError)(error); if (origError) { logger.error(origError); } } } else { setResult({ ...result, errors: result.errors.map(error => { const origError = (0, utils_js_1.getOriginalError)(error); if (origError) { return (0, utils_2.createGraphQLError)(error.message, { ...error, extensions: { ...error.extensions, originalError: { name: origError.name, message: origError.message, stack: origError.stack, }, }, }); } return error; }), }); } } }, }; }, }, ...initialPluginList, ]; const EMPTY_ROOT_VALUE = {}; const EMPTY_CONTEXT_VALUE = {}; const EMPTY_VARIABLES_VALUE = {}; function createExecutor(globalContext = EMPTY_CONTEXT_VALUE) { const getEnveloped = memoizedGetEnvelopedFactory(plugins); const { schema, parse, execute, subscribe, contextFactory } = getEnveloped(globalContext); return function meshExecutor(documentOrSDL, variableValues = EMPTY_VARIABLES_VALUE, contextValue = EMPTY_CONTEXT_VALUE, rootValue = EMPTY_ROOT_VALUE, operationName) { const document = typeof documentOrSDL === 'string' ? parse(documentOrSDL) : documentOrSDL; const operationAST = (0, graphql_1.getOperationAST)(document, operationName); if (!operationAST) { throw new Error(`Cannot execute a request without a valid operation.`); } const isSubscription = operationAST.operation === 'subscription'; const executeFn = isSubscription ? subscribe : execute; return (0, promise_helpers_1.handleMaybePromise)(() => contextFactory(contextValue), contextValue => executeFn({ schema, document, contextValue, rootValue, variableValues, operationName, })); }; } function sdkRequesterFactory(globalContext) { const executor = createExecutor(globalContext); return function sdkRequester(...args) { return (0, promise_helpers_1.handleMaybePromise)(() => executor(...args), function handleExecutorResultForSdk(result) { if ((0, utils_2.isAsyncIterable)(result)) { return (0, utils_2.mapAsyncIterator)(result, extractDataOrThrowErrors); } return extractDataOrThrowErrors(result); }); }; } function meshDestroy() { return pubsub.publish('destroy', undefined); } return (0, utils_1.makeDisposable)({ schema, rawSources, cache, pubsub, destroy: meshDestroy, logger, plugins, get getEnveloped() { return memoizedGetEnvelopedFactory(plugins); }, createExecutor, get execute() { return createExecutor(); }, get subscribe() { return createExecutor(); }, sdkRequesterFactory, }, meshDestroy); } function extractDataOrThrowErrors(result) { if (result.errors) { if (result.errors.length === 1) { throw result.errors[0]; } throw new AggregateError(result.errors); } return result.data; }