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 graphql_jit_1 = require("@envelop/graphql-jit"); const cross_helpers_1 = require("@graphql-mesh/cross-helpers"); const utils_1 = require("@graphql-mesh/utils"); const delegate_1 = require("@graphql-tools/delegate"); const executor_1 = require("@graphql-tools/executor"); 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 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); } 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, 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) => { 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; let subschema; if (unifiedSubschema.executor != null || unifiedSubschema.transforms?.length) { subschema = new delegate_1.Subschema(unifiedSubschema); } const plugins = [ (0, core_1.useEngine)({ execute: executor_1.normalizedExecutor, validate: graphql_1.validate, parse: utils_1.parseWithCache, specifiedRules: graphql_1.specifiedRules, }), ...(subschema ? [(0, useSubschema_js_1.useSubschema)(new delegate_1.Subschema(unifiedSubschema))] : [ (0, core_1.useSchema)(unifiedSubschema.schema), (0, graphql_jit_1.useGraphQlJit)({ // TODO: Disable for now customJSONSerializer: false, disableLeafSerialization: true, }, { enableIf(args) { return !(0, utils_js_1.isStreamOperation)(args.document); }, }), ]), (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 ? subschema.transformedSchema : unifiedSubschema.schema, 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 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 contextValue$ = contextFactory(contextValue); 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; if ((0, utils_2.isPromise)(contextValue$)) { return contextValue$.then(contextValue => executeFn({ schema, document, contextValue, rootValue, variableValues: variableValues, operationName, })); } return executeFn({ schema, document, contextValue: contextValue$, rootValue, variableValues: variableValues, operationName, }); }; } function sdkRequesterFactory(globalContext) { const executor = createExecutor(globalContext); return function sdkRequester(...args) { const result$ = executor(...args); if ((0, utils_2.isPromise)(result$)) { return result$.then(handleExecutorResultForSdk); } return handleExecutorResultForSdk(result$); }; } function meshDestroy() { return pubsub.publish('destroy', undefined); } return { get schema() { return subschema ? subschema.transformedSchema : unifiedSubschema.schema; }, 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 handleExecutorResultForSdk(result) { if ((0, utils_2.isAsyncIterable)(result)) { return (0, utils_2.mapAsyncIterator)(result, extractDataOrThrowErrors); } return extractDataOrThrowErrors(result); } function extractDataOrThrowErrors(result) { if (result.errors) { if (result.errors.length === 1) { throw result.errors[0]; } throw new AggregateError(result.errors); } return result.data; }