@omnigraph/odata/cjs/directives.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.TransportDirective = exports.AbstractTypeDirective = exports.UpdateEntitySetDirective = exports.DeleteEntitySetDirective = exports.CreateEntitySetDirective = exports.EntitySetCountDirective = exports.EntitySetByIdentifierDirective = exports.EntitySetDirective = exports.SingletonDirective = exports.BoundActionDirective = exports.UnboundActionDirective = exports.BoundFunctionDirective = exports.UnboundFunctionDirective = exports.ResolveRootDirective = exports.NavPropDirective = exports.PluralNavDirective = exports.SingularNavDirective = exports.EntityInfoDirective = void 0;
exports.processDirectives = processDirectives;
const graphql_1 = require("graphql");
const cross_helpers_1 = require("@graphql-mesh/cross-helpers");
const string_interpolation_1 = require("@graphql-mesh/string-interpolation");
const utils_1 = require("@graphql-tools/utils");
const getDataloaderFactory_js_1 = require("./getDataloaderFactory.js");
const abstractTypeResolver_js_1 = require("./resolvers/abstractTypeResolver.js");
const boundActionResolver_js_1 = require("./resolvers/boundActionResolver.js");
const boundFunctionResolver_js_1 = require("./resolvers/boundFunctionResolver.js");
const createEntitySetResolver_js_1 = require("./resolvers/createEntitySetResolver.js");
const deleteEntitySetByIdentifierResolver_js_1 = require("./resolvers/deleteEntitySetByIdentifierResolver.js");
const entitySetByIdentifierResolver_js_1 = require("./resolvers/entitySetByIdentifierResolver.js");
const entitySetCountResolver_js_1 = require("./resolvers/entitySetCountResolver.js");
const entitySetResolver_js_1 = require("./resolvers/entitySetResolver.js");
const navPropResolver_js_1 = require("./resolvers/navPropResolver.js");
const rootResolver_js_1 = require("./resolvers/rootResolver.js");
const singletonResolver_js_1 = require("./resolvers/singletonResolver.js");
const singularNavResolver_js_1 = require("./resolvers/singularNavResolver.js");
const unboundActionResolver_js_1 = require("./resolvers/unboundActionResolver.js");
const unboundFunction_js_1 = require("./resolvers/unboundFunction.js");
const updateEntitySetResolver_js_1 = require("./resolvers/updateEntitySetResolver.js");
exports.EntityInfoDirective = new graphql_1.GraphQLDirective({
    name: 'entityInfo',
    args: {
        actualFields: {
            type: new graphql_1.GraphQLList(graphql_1.GraphQLString),
        },
        navigationFields: {
            type: new graphql_1.GraphQLList(graphql_1.GraphQLString),
        },
        identifierFieldName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldTypeRef: {
            type: graphql_1.GraphQLString,
        },
        isOpenType: {
            type: graphql_1.GraphQLBoolean,
        },
    },
    locations: [graphql_1.DirectiveLocation.OBJECT, graphql_1.DirectiveLocation.INTERFACE],
});
exports.SingularNavDirective = new graphql_1.GraphQLDirective({
    name: 'singularNav',
    args: {
        navigationPropertyName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.PluralNavDirective = new graphql_1.GraphQLDirective({
    name: 'pluralNav',
    args: {
        navigationPropertyName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.NavPropDirective = new graphql_1.GraphQLDirective({
    name: 'navProp',
    args: {
        navigationPropertyName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.ResolveRootDirective = new graphql_1.GraphQLDirective({
    name: 'resolveRoot',
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.UnboundFunctionDirective = new graphql_1.GraphQLDirective({
    name: 'unboundFunction',
    args: {
        functionName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.BoundFunctionDirective = new graphql_1.GraphQLDirective({
    name: 'boundFunction',
    args: {
        functionRef: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.UnboundActionDirective = new graphql_1.GraphQLDirective({
    name: 'unboundAction',
    args: {
        actionName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.BoundActionDirective = new graphql_1.GraphQLDirective({
    name: 'boundAction',
    args: {
        actionRef: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.SingletonDirective = new graphql_1.GraphQLDirective({
    name: 'singleton',
    args: {
        singletonName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.EntitySetDirective = new graphql_1.GraphQLDirective({
    name: 'entitySet',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.EntitySetByIdentifierDirective = new graphql_1.GraphQLDirective({
    name: 'entitySetByIdentifier',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldTypeRef: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.EntitySetCountDirective = new graphql_1.GraphQLDirective({
    name: 'entitySetCount',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.CreateEntitySetDirective = new graphql_1.GraphQLDirective({
    name: 'createEntitySet',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.DeleteEntitySetDirective = new graphql_1.GraphQLDirective({
    name: 'deleteEntitySet',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldTypeRef: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.UpdateEntitySetDirective = new graphql_1.GraphQLDirective({
    name: 'updateEntitySet',
    args: {
        entitySetName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldName: {
            type: graphql_1.GraphQLString,
        },
        identifierFieldTypeRef: {
            type: graphql_1.GraphQLString,
        },
    },
    locations: [graphql_1.DirectiveLocation.FIELD_DEFINITION],
});
exports.AbstractTypeDirective = new graphql_1.GraphQLDirective({
    name: 'abstractType',
    args: {
        entityTypeName: {
            type: graphql_1.GraphQLString,
        },
        isAbstract: {
            type: graphql_1.GraphQLBoolean,
        },
        aliasNamespaceMap: {
            type: new graphql_1.GraphQLList(new graphql_1.GraphQLList(graphql_1.GraphQLString)),
        },
        multipleSchemas: {
            type: graphql_1.GraphQLBoolean,
        },
        namespaces: {
            type: new graphql_1.GraphQLList(graphql_1.GraphQLString),
        },
    },
    locations: [graphql_1.DirectiveLocation.INTERFACE, graphql_1.DirectiveLocation.OBJECT],
});
exports.TransportDirective = new graphql_1.GraphQLDirective({
    name: 'transport',
    isRepeatable: true,
    locations: [graphql_1.DirectiveLocation.SCHEMA],
    args: {
        kind: { type: new graphql_1.GraphQLNonNull(graphql_1.GraphQLString) },
        subgraph: { type: new graphql_1.GraphQLNonNull(graphql_1.GraphQLString) },
        location: { type: new graphql_1.GraphQLNonNull(graphql_1.GraphQLString) },
        headers: { type: new graphql_1.GraphQLList(new graphql_1.GraphQLList(graphql_1.GraphQLString)) },
        options: { type: new graphql_1.GraphQLScalarType({ name: 'TransportOptions' }) },
    },
});
function processDirectives({ schema, fetchFn }) {
    const schemaLevelDirectives = (0, utils_1.getDirectiveExtensions)(schema);
    const transportDirective = schemaLevelDirectives?.transport?.[0];
    if (!transportDirective) {
        throw new Error('Missing transport directive');
    }
    const operationHeaders = Array.isArray(transportDirective.headers)
        ? Object.fromEntries(transportDirective.headers)
        : transportDirective.headers;
    const origHeadersFactory = (0, string_interpolation_1.getInterpolatedHeadersFactory)(operationHeaders);
    const headersFactory = (resolverData, method) => {
        const headers = origHeadersFactory(resolverData);
        if (headers.accept == null) {
            headers.accept = 'application/json';
        }
        if (headers['content-type'] == null && method !== 'GET') {
            headers['content-type'] = 'application/json';
        }
        return headers;
    };
    const endpoint = string_interpolation_1.stringInterpolator.parse(transportDirective.location, {
        env: cross_helpers_1.process.env,
    });
    const dataloaderFactory = (0, getDataloaderFactory_js_1.getDataloaderFactory)({
        endpoint,
        fetchFn,
        headersFactory,
        batchMode: transportDirective.options?.batch || 'none',
    });
    return (0, utils_1.mapSchema)(schema, {
        [utils_1.MapperKind.ABSTRACT_TYPE]: type => {
            const typeDirectives = (0, utils_1.getDirectiveExtensions)(type);
            const abstractTypeDirectives = typeDirectives?.abstractType;
            if (abstractTypeDirectives?.length) {
                const abstractTypeDirective = abstractTypeDirectives[0];
                const typePrototype = Object.getPrototypeOf(type);
                return new typePrototype.constructor({
                    ...type.toConfig(),
                    resolveType: (0, abstractTypeResolver_js_1.createAbstractTypeResolver)({
                        entityTypeName: abstractTypeDirective.entityTypeName,
                        isAbstract: abstractTypeDirective.isAbstract,
                        aliasNamespaceMap: new Map(abstractTypeDirective.aliasNamespaceMap),
                        multipleSchemas: abstractTypeDirective.multipleSchemas,
                        namespaces: new Set(abstractTypeDirective.namespaces),
                    }),
                });
            }
        },
        [utils_1.MapperKind.COMPOSITE_FIELD]: fieldConfig => {
            const fieldDirectives = (0, utils_1.getDirectiveExtensions)(fieldConfig);
            if (fieldDirectives?.boundAction?.length) {
                const boundActionDirective = fieldDirectives.boundAction[0];
                fieldConfig.resolve = (0, boundActionResolver_js_1.createBoundActionResolver)({
                    actionRef: boundActionDirective.actionRef,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.boundFunction?.length) {
                const boundFunctionDirective = fieldDirectives.boundFunction[0];
                fieldConfig.resolve = (0, boundFunctionResolver_js_1.createBoundFunctionResolver)({
                    functionRef: boundFunctionDirective.functionRef,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.createEntitySet?.length) {
                const createEntitySetDirective = fieldDirectives.createEntitySet[0];
                fieldConfig.resolve = (0, createEntitySetResolver_js_1.createCreateEntitySetResolver)({
                    endpoint,
                    entitySetName: createEntitySetDirective.entitySetName,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.deleteEntitySet?.length) {
                const deleteEntitySetDirective = fieldDirectives.deleteEntitySet[0];
                fieldConfig.resolve = (0, deleteEntitySetByIdentifierResolver_js_1.createDeleteEntitySetByIdentifierResolver)({
                    endpoint,
                    entitySetName: deleteEntitySetDirective.entitySetName,
                    identifierFieldName: deleteEntitySetDirective.identifierFieldName,
                    identifierFieldTypeRef: deleteEntitySetDirective.identifierFieldTypeRef,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.entitySetByIdentifier?.length) {
                const entitySetByIdentifierDirective = fieldDirectives.entitySetByIdentifier[0];
                fieldConfig.resolve = (0, entitySetByIdentifierResolver_js_1.createEntitySetByIdentifierResolver)({
                    endpoint,
                    entitySetName: entitySetByIdentifierDirective.entitySetName,
                    identifierFieldName: entitySetByIdentifierDirective.identifierFieldName,
                    identifierFieldTypeRef: entitySetByIdentifierDirective.identifierFieldTypeRef,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.entitySetCount?.length) {
                const entitySetCountDirective = fieldDirectives.entitySetCount[0];
                fieldConfig.resolve = (0, entitySetCountResolver_js_1.createEntitySetCountResolver)({
                    endpoint,
                    entitySetName: entitySetCountDirective.entitySetName,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.entitySet?.length) {
                const entitySetDirective = fieldDirectives.entitySet[0];
                fieldConfig.resolve = (0, entitySetResolver_js_1.createEntitySetResolver)({
                    endpoint,
                    entitySetName: entitySetDirective.entitySetName,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.navProp?.length) {
                const navPropDirective = fieldDirectives.navProp[0];
                fieldConfig.resolve = (0, navPropResolver_js_1.createNavPropResolver)({
                    navigationPropertyName: navPropDirective.navigationPropertyName,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.pluralNav?.length) {
                const pluralNavDirective = fieldDirectives.pluralNav[0];
                fieldConfig.resolve = (0, navPropResolver_js_1.createNavPropResolver)({
                    navigationPropertyName: pluralNavDirective.navigationPropertyName,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.resolveRoot?.length) {
                fieldConfig.resolve = rootResolver_js_1.rootResolver;
            }
            else if (fieldDirectives?.singleton?.length) {
                const singletonDirective = fieldDirectives.singleton[0];
                fieldConfig.resolve = (0, singletonResolver_js_1.createSingletonResolver)({
                    singletonName: singletonDirective.singletonName,
                    endpoint,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.singularNav?.length) {
                const singularNavDirective = fieldDirectives.singularNav[0];
                fieldConfig.resolve = (0, singularNavResolver_js_1.createSingularNavResolver)({
                    navigationPropertyName: singularNavDirective.navigationPropertyName,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.unboundAction?.length) {
                const unboundActionDirective = fieldDirectives.unboundAction[0];
                fieldConfig.resolve = (0, unboundActionResolver_js_1.createUnboundActionResolver)({
                    actionName: unboundActionDirective.actionName,
                    endpoint,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.unboundFunction?.length) {
                const unboundFunctionDirective = fieldDirectives.unboundFunction[0];
                fieldConfig.resolve = (0, unboundFunction_js_1.createUnboundFunctionResolver)({
                    functionName: unboundFunctionDirective.functionName,
                    endpoint,
                    expandNavProps: transportDirective.options?.expandNavProps,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            else if (fieldDirectives?.updateEntitySet?.length) {
                const updateEntitySetDirective = fieldDirectives.updateEntitySet[0];
                fieldConfig.resolve = (0, updateEntitySetResolver_js_1.createUpdateEntitySetResolver)({
                    endpoint,
                    entitySetName: updateEntitySetDirective.entitySetName,
                    identifierFieldName: updateEntitySetDirective.identifierFieldName,
                    identifierFieldTypeRef: updateEntitySetDirective.identifierFieldTypeRef,
                    dataloaderFactory,
                    headersFactory,
                });
            }
            return fieldConfig;
        },
    });
}