UNPKG

@graphql-mesh/runtime

Version:
276 lines (275 loc) 11.4 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.getMesh = exports.wrapFetchWithPlugins = void 0; const graphql_1 = require("graphql"); const core_1 = require("@envelop/core"); const extended_validation_1 = require("@envelop/extended-validation"); const cross_helpers_1 = require("@graphql-mesh/cross-helpers"); const utils_1 = require("@graphql-mesh/utils"); const delegate_1 = require("@graphql-tools/delegate"); const utils_2 = require("@graphql-tools/utils"); const fetch_1 = require("@whatwg-node/fetch"); const constants_js_1 = require("./constants.js"); const in_context_sdk_js_1 = require("./in-context-sdk.js"); const useSubschema_js_1 = require("./useSubschema.js"); const memoizedGetEnvelopedFactory = (0, utils_2.memoize1)(function getEnvelopedFactory(plugins) { return (0, core_1.envelop)({ plugins, }); }); const memoizedGetOperationType = (0, utils_2.memoize1)((document) => { const operationAST = (0, graphql_1.getOperationAST)(document, undefined); if (!operationAST) { throw new Error('Must provide document with a valid operation'); } return operationAST.operation; }); function wrapFetchWithPlugins(plugins) { return async function wrappedFetchFn(url, options, context, info) { if (url != null && typeof url !== 'string') { throw new TypeError(`First parameter(url) of 'fetch' must be a string, got ${(0, utils_2.inspect)(url)}`); } if (options != null && typeof options !== 'object') { throw new TypeError(`Second parameter(options) of 'fetch' must be an object, got ${(0, utils_2.inspect)(options)}`); } if (context != null && typeof context !== 'object') { throw new TypeError(`Third parameter(context) of 'fetch' must be an object, got ${(0, utils_2.inspect)(context)}`); } if (info != null && typeof info !== 'object') { throw new TypeError(`Fourth parameter(info) of 'fetch' must be an object, got ${(0, utils_2.inspect)(info)}`); } let fetchFn; const doneHooks = []; for (const plugin of plugins) { if (plugin?.onFetch != null) { const doneHook = await plugin.onFetch({ fetchFn, setFetchFn(newFetchFn) { fetchFn = newFetchFn; }, url, options, context, info, }); if (doneHook) { doneHooks.push(doneHook); } } } let response = await fetchFn(url, options, context, info); for (const doneHook of doneHooks) { await doneHook({ response, setResponse(newResponse) { response = newResponse; }, }); } return response; }; } exports.wrapFetchWithPlugins = wrapFetchWithPlugins; // 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('🕸️ Mesh'), 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, [constants_js_1.MESH_CONTEXT_SYMBOL]: true, })), { onFetch({ setFetchFn }) { setFetchFn(fetchFn); }, }, ...additionalEnvelopPlugins, ]; const wrappedFetchFn = wrapFetchWithPlugins(initialPluginList); await Promise.allSettled(sources.map(async (apiSource) => { const apiName = apiSource.name; const sourceLogger = logger.child(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.push({ 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.error(`Failed to generate the schema`, e); 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$; const subschema = new delegate_1.Subschema(unifiedSubschema); const plugins = [ (0, core_1.useEngine)({ validate: graphql_1.validate, parse: utils_1.parseWithCache, specifiedRules: graphql_1.specifiedRules, }), (0, useSubschema_js_1.useSubschema)(subschema), (0, core_1.useExtendContext)(() => { if (!inContextSDK$) { const onDelegateHooks = []; for (const plugin of initialPluginList) { if (plugin?.onDelegate != null) { onDelegateHooks.push(plugin.onDelegate); } } inContextSDK$ = (0, in_context_sdk_js_1.getInContextSDK)(subschema.transformedSchema, rawSources, logger, onDelegateHooks); } return inContextSDK$; }), (0, extended_validation_1.useExtendedValidation)({ rules: [extended_validation_1.OneOfInputObjectsRule], }), ...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 async 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 executeFn = memoizedGetOperationType(document) === 'subscription' ? subscribe : execute; return executeFn({ schema, document, contextValue: await contextFactory(contextValue), rootValue, variableValues: variableValues, operationName, }); }; } function sdkRequesterFactory(globalContext) { const executor = createExecutor(globalContext); return async function sdkRequester(...args) { const result = await executor(...args); if ((0, utils_2.isAsyncIterable)(result)) { return (0, utils_2.mapAsyncIterator)(result, extractDataOrThrowErrors); } return extractDataOrThrowErrors(result); }; } function meshDestroy() { return pubsub.publish('destroy', undefined); } return { get schema() { return subschema.transformedSchema; }, rawSources, cache, pubsub, destroy: meshDestroy, logger, plugins, get getEnveloped() { return memoizedGetEnvelopedFactory(plugins); }, createExecutor, get execute() { return createExecutor(); }, get subscribe() { return createExecutor(); }, sdkRequesterFactory, }; } exports.getMesh = getMesh; function extractDataOrThrowErrors(result) { if (result.errors) { if (result.errors.length === 1) { throw result.errors[0]; } throw new AggregateError(result.errors); } return result.data; }