@graphql-tools/batch-delegate/es5/index.mjs

import { __assign, __awaiter, __generator, __read } from 'tslib';
import { GraphQLList, getNamedType } from 'graphql';
import DataLoader from 'dataloader';
import { delegateToSchema } from '@graphql-tools/delegate/es5';
import { relocatedError } from '@graphql-tools/utils/es5';

var cache1 = new WeakMap();
function createBatchFn(options) {
    var _this = this;
    var _a, _b;
    var argsFromKeys = (_a = options.argsFromKeys) !== null && _a !== void 0 ? _a : (function (keys) { return ({ ids: keys }); });
    var fieldName = (_b = options.fieldName) !== null && _b !== void 0 ? _b : options.info.fieldName;
    var valuesFromResults = options.valuesFromResults, lazyOptionsFn = options.lazyOptionsFn;
    return function (keys) { return __awaiter(_this, void 0, void 0, function () {
        var results, values;
        return __generator(this, function (_a) {
            switch (_a.label) {
                case 0: return [4 /*yield*/, delegateToSchema(__assign({ returnType: new GraphQLList(getNamedType(options.info.returnType)), onLocatedError: function (originalError) {
                            if (originalError.path == null) {
                                return originalError;
                            }
                            var _a = __read(originalError.path, 2), pathFieldName = _a[0], pathNumber = _a[1];
                            if (pathFieldName !== fieldName) {
                                return originalError;
                            }
                            var pathNumberType = typeof pathNumber;
                            if (pathNumberType !== 'number') {
                                return originalError;
                            }
                            return relocatedError(originalError, originalError.path.slice(0, 0).concat(originalError.path.slice(2)));
                        }, args: argsFromKeys(keys) }, (lazyOptionsFn == null ? options : lazyOptionsFn(options))))];
                case 1:
                    results = _a.sent();
                    if (results instanceof Error) {
                        return [2 /*return*/, keys.map(function () { return results; })];
                    }
                    values = valuesFromResults == null ? results : valuesFromResults(results, keys);
                    return [2 /*return*/, Array.isArray(values) ? values : keys.map(function () { return values; })];
            }
        });
    }); };
}
var cacheKeyFn = function (key) { return (typeof key === 'object' ? JSON.stringify(key) : key); };
function getLoader(options) {
    var _a;
    var fieldName = (_a = options.fieldName) !== null && _a !== void 0 ? _a : options.info.fieldName;
    var cache2 = cache1.get(options.info.fieldNodes);
    // Prevents the keys to be passed with the same structure
    var dataLoaderOptions = __assign({ cacheKeyFn: cacheKeyFn }, options.dataLoaderOptions);
    if (cache2 === undefined) {
        cache2 = new WeakMap();
        cache1.set(options.info.fieldNodes, cache2);
        var loaders_1 = Object.create(null);
        cache2.set(options.schema, loaders_1);
        var batchFn_1 = createBatchFn(options);
        var loader_1 = new DataLoader(function (keys) { return batchFn_1(keys); }, dataLoaderOptions);
        loaders_1[fieldName] = loader_1;
        return loader_1;
    }
    var loaders = cache2.get(options.schema);
    if (loaders === undefined) {
        loaders = Object.create(null);
        cache2.set(options.schema, loaders);
        var batchFn_2 = createBatchFn(options);
        var loader_2 = new DataLoader(function (keys) { return batchFn_2(keys); }, dataLoaderOptions);
        loaders[fieldName] = loader_2;
        return loader_2;
    }
    var loader = loaders[fieldName];
    if (loader === undefined) {
        var batchFn_3 = createBatchFn(options);
        loader = new DataLoader(function (keys) { return batchFn_3(keys); }, dataLoaderOptions);
        loaders[fieldName] = loader;
    }
    return loader;
}

function batchDelegateToSchema(options) {
    var key = options.key;
    if (key == null) {
        return null;
    }
    else if (Array.isArray(key) && !key.length) {
        return [];
    }
    var loader = getLoader(options);
    return Array.isArray(key) ? loader.loadMany(key) : loader.load(key);
}

function createBatchDelegateFn(optionsOrArgsFromKeys, lazyOptionsFn, dataLoaderOptions, valuesFromResults) {
    return typeof optionsOrArgsFromKeys === 'function'
        ? createBatchDelegateFnImpl({
            argsFromKeys: optionsOrArgsFromKeys,
            lazyOptionsFn: lazyOptionsFn,
            dataLoaderOptions: dataLoaderOptions,
            valuesFromResults: valuesFromResults,
        })
        : createBatchDelegateFnImpl(optionsOrArgsFromKeys);
}
function createBatchDelegateFnImpl(options) {
    return function (batchDelegateOptions) {
        var loader = getLoader(__assign(__assign({}, options), batchDelegateOptions));
        return loader.load(batchDelegateOptions.key);
    };
}

export { batchDelegateToSchema, createBatchDelegateFn };