123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327 |
- function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); keys.push.apply(keys, symbols); } return keys; }
- function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(source, true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(source).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }
- function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
- import flatMap from '../polyfills/flatMap';
- import objectValues from '../polyfills/objectValues';
- import inspect from '../jsutils/inspect';
- import mapValue from '../jsutils/mapValue';
- import invariant from '../jsutils/invariant';
- import devAssert from '../jsutils/devAssert';
- import keyValMap from '../jsutils/keyValMap';
- import { Kind } from '../language/kinds';
- import { isTypeDefinitionNode, isTypeExtensionNode } from '../language/predicates';
- import { assertValidSDLExtension } from '../validation/validate';
- import { GraphQLDirective } from '../type/directives';
- import { isSpecifiedScalarType } from '../type/scalars';
- import { isIntrospectionType } from '../type/introspection';
- import { assertSchema, GraphQLSchema } from '../type/schema';
- import { isScalarType, isObjectType, isInterfaceType, isUnionType, isListType, isNonNullType, isEnumType, isInputObjectType, GraphQLList, GraphQLNonNull, GraphQLScalarType, GraphQLObjectType, GraphQLInterfaceType, GraphQLUnionType, GraphQLEnumType, GraphQLInputObjectType } from '../type/definition';
- import { ASTDefinitionBuilder } from './buildASTSchema';
- export function extendSchema(schema, documentAST, options) {
- assertSchema(schema);
- documentAST && documentAST.kind === Kind.DOCUMENT || devAssert(0, 'Must provide valid Document AST');
- if (!options || !(options.assumeValid || options.assumeValidSDL)) {
- assertValidSDLExtension(documentAST, schema);
- }
- var typeDefs = [];
- var typeExtsMap = Object.create(null);
-
- var directiveDefs = [];
- var schemaDef;
- var schemaExts = [];
- for (var _i2 = 0, _documentAST$definiti2 = documentAST.definitions; _i2 < _documentAST$definiti2.length; _i2++) {
- var def = _documentAST$definiti2[_i2];
- if (def.kind === Kind.SCHEMA_DEFINITION) {
- schemaDef = def;
- } else if (def.kind === Kind.SCHEMA_EXTENSION) {
- schemaExts.push(def);
- } else if (isTypeDefinitionNode(def)) {
- typeDefs.push(def);
- } else if (isTypeExtensionNode(def)) {
- var extendedTypeName = def.name.value;
- var existingTypeExts = typeExtsMap[extendedTypeName];
- typeExtsMap[extendedTypeName] = existingTypeExts ? existingTypeExts.concat([def]) : [def];
- } else if (def.kind === Kind.DIRECTIVE_DEFINITION) {
- directiveDefs.push(def);
- }
- }
-
- if (Object.keys(typeExtsMap).length === 0 && typeDefs.length === 0 && directiveDefs.length === 0 && schemaExts.length === 0 && !schemaDef) {
- return schema;
- }
- var schemaConfig = schema.toConfig();
- var astBuilder = new ASTDefinitionBuilder(options, function (typeName) {
- var type = typeMap[typeName];
- if (type === undefined) {
- throw new Error("Unknown type: \"".concat(typeName, "\"."));
- }
- return type;
- });
- var typeMap = keyValMap(typeDefs, function (node) {
- return node.name.value;
- }, function (node) {
- return astBuilder.buildType(node);
- });
- for (var _i4 = 0, _schemaConfig$types2 = schemaConfig.types; _i4 < _schemaConfig$types2.length; _i4++) {
- var existingType = _schemaConfig$types2[_i4];
- typeMap[existingType.name] = extendNamedType(existingType);
- }
- var operationTypes = {
- query: schemaConfig.query && schemaConfig.query.name,
- mutation: schemaConfig.mutation && schemaConfig.mutation.name,
- subscription: schemaConfig.subscription && schemaConfig.subscription.name
- };
- if (schemaDef) {
- for (var _i6 = 0, _schemaDef$operationT2 = schemaDef.operationTypes; _i6 < _schemaDef$operationT2.length; _i6++) {
- var _ref2 = _schemaDef$operationT2[_i6];
- var operation = _ref2.operation;
- var type = _ref2.type;
- operationTypes[operation] = type.name.value;
- }
- }
- for (var _i8 = 0; _i8 < schemaExts.length; _i8++) {
- var schemaExt = schemaExts[_i8];
- if (schemaExt.operationTypes) {
- for (var _i10 = 0, _schemaExt$operationT2 = schemaExt.operationTypes; _i10 < _schemaExt$operationT2.length; _i10++) {
- var _ref4 = _schemaExt$operationT2[_i10];
- var _operation = _ref4.operation;
- var _type = _ref4.type;
- operationTypes[_operation] = _type.name.value;
- }
- }
- }
- var allowedLegacyNames = schemaConfig.allowedLegacyNames.concat(options && options.allowedLegacyNames || []);
- return new GraphQLSchema({
-
-
-
- query: getMaybeTypeByName(operationTypes.query),
- mutation: getMaybeTypeByName(operationTypes.mutation),
- subscription: getMaybeTypeByName(operationTypes.subscription),
- types: objectValues(typeMap),
- directives: getMergedDirectives(),
- astNode: schemaDef || schemaConfig.astNode,
- extensionASTNodes: schemaConfig.extensionASTNodes.concat(schemaExts),
- allowedLegacyNames: allowedLegacyNames
- });
-
- function replaceType(type) {
- if (isListType(type)) {
- return new GraphQLList(replaceType(type.ofType));
- } else if (isNonNullType(type)) {
- return new GraphQLNonNull(replaceType(type.ofType));
- }
- return replaceNamedType(type);
- }
- function replaceNamedType(type) {
- return typeMap[type.name];
- }
- function getMaybeTypeByName(typeName) {
- return typeName ? typeMap[typeName] : null;
- }
- function getMergedDirectives() {
- var existingDirectives = schema.getDirectives().map(extendDirective);
- existingDirectives || devAssert(0, 'schema must have default directives');
- return existingDirectives.concat(directiveDefs.map(function (node) {
- return astBuilder.buildDirective(node);
- }));
- }
- function extendNamedType(type) {
- if (isIntrospectionType(type) || isSpecifiedScalarType(type)) {
-
- return type;
- } else if (isScalarType(type)) {
- return extendScalarType(type);
- } else if (isObjectType(type)) {
- return extendObjectType(type);
- } else if (isInterfaceType(type)) {
- return extendInterfaceType(type);
- } else if (isUnionType(type)) {
- return extendUnionType(type);
- } else if (isEnumType(type)) {
- return extendEnumType(type);
- } else if (isInputObjectType(type)) {
- return extendInputObjectType(type);
- }
-
- invariant(false, 'Unexpected type: ' + inspect(type));
- }
- function extendDirective(directive) {
- var config = directive.toConfig();
- return new GraphQLDirective(_objectSpread({}, config, {
- args: mapValue(config.args, extendArg)
- }));
- }
- function extendInputObjectType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[config.name] || [];
- var fieldNodes = flatMap(extensions, function (node) {
- return node.fields || [];
- });
- return new GraphQLInputObjectType(_objectSpread({}, config, {
- fields: function fields() {
- return _objectSpread({}, mapValue(config.fields, function (field) {
- return _objectSpread({}, field, {
- type: replaceType(field.type)
- });
- }), {}, keyValMap(fieldNodes, function (field) {
- return field.name.value;
- }, function (field) {
- return astBuilder.buildInputField(field);
- }));
- },
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendEnumType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[type.name] || [];
- var valueNodes = flatMap(extensions, function (node) {
- return node.values || [];
- });
- return new GraphQLEnumType(_objectSpread({}, config, {
- values: _objectSpread({}, config.values, {}, keyValMap(valueNodes, function (value) {
- return value.name.value;
- }, function (value) {
- return astBuilder.buildEnumValue(value);
- })),
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendScalarType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[config.name] || [];
- return new GraphQLScalarType(_objectSpread({}, config, {
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendObjectType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[config.name] || [];
- var interfaceNodes = flatMap(extensions, function (node) {
- return node.interfaces || [];
- });
- var fieldNodes = flatMap(extensions, function (node) {
- return node.fields || [];
- });
- return new GraphQLObjectType(_objectSpread({}, config, {
- interfaces: function interfaces() {
- return [].concat(type.getInterfaces().map(replaceNamedType), interfaceNodes.map(function (node) {
- return astBuilder.getNamedType(node);
- }));
- },
- fields: function fields() {
- return _objectSpread({}, mapValue(config.fields, extendField), {}, keyValMap(fieldNodes, function (node) {
- return node.name.value;
- }, function (node) {
- return astBuilder.buildField(node);
- }));
- },
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendInterfaceType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[config.name] || [];
- var fieldNodes = flatMap(extensions, function (node) {
- return node.fields || [];
- });
- return new GraphQLInterfaceType(_objectSpread({}, config, {
- fields: function fields() {
- return _objectSpread({}, mapValue(config.fields, extendField), {}, keyValMap(fieldNodes, function (node) {
- return node.name.value;
- }, function (node) {
- return astBuilder.buildField(node);
- }));
- },
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendUnionType(type) {
- var config = type.toConfig();
- var extensions = typeExtsMap[config.name] || [];
- var typeNodes = flatMap(extensions, function (node) {
- return node.types || [];
- });
- return new GraphQLUnionType(_objectSpread({}, config, {
- types: function types() {
- return [].concat(type.getTypes().map(replaceNamedType), typeNodes.map(function (node) {
- return astBuilder.getNamedType(node);
- }));
- },
- extensionASTNodes: config.extensionASTNodes.concat(extensions)
- }));
- }
- function extendField(field) {
- return _objectSpread({}, field, {
- type: replaceType(field.type),
- args: mapValue(field.args, extendArg)
- });
- }
- function extendArg(arg) {
- return _objectSpread({}, arg, {
- type: replaceType(arg.type)
- });
- }
- }
|