@graphql-tools/utils
Version:
Common package containing utils and types for GraphQL tools
100 lines (99 loc) • 3.78 kB
JavaScript
import { asArray } from './helpers.js';
import { MapperKind } from './Interfaces.js';
import { mapSchema } from './mapSchema.js';
function handleDirectiveExtensions(extensions, removeDirectives) {
extensions = extensions || {};
const { directives: existingDirectives, ...rest } = extensions;
const finalExtensions = {
...rest,
};
if (!removeDirectives) {
if (existingDirectives != null) {
const directives = {};
for (const directiveName in existingDirectives) {
directives[directiveName] = [...asArray(existingDirectives[directiveName])];
}
finalExtensions.directives = directives;
}
}
return finalExtensions;
}
export function extractExtensionsFromSchema(schema, removeDirectives = false) {
const result = {
schemaExtensions: handleDirectiveExtensions(schema.extensions, removeDirectives),
types: {},
};
mapSchema(schema, {
[MapperKind.OBJECT_TYPE]: type => {
result.types[type.name] = {
fields: {},
type: 'object',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.INTERFACE_TYPE]: type => {
result.types[type.name] = {
fields: {},
type: 'interface',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.FIELD]: (field, fieldName, typeName) => {
result.types[typeName].fields[fieldName] = {
arguments: {},
extensions: handleDirectiveExtensions(field.extensions, removeDirectives),
};
const args = field.args;
if (args != null) {
for (const argName in args) {
result.types[typeName].fields[fieldName].arguments[argName] =
handleDirectiveExtensions(args[argName].extensions, removeDirectives);
}
}
return field;
},
[MapperKind.ENUM_TYPE]: type => {
result.types[type.name] = {
values: {},
type: 'enum',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.ENUM_VALUE]: (value, typeName, _schema, valueName) => {
result.types[typeName].values[valueName] = handleDirectiveExtensions(value.extensions, removeDirectives);
return value;
},
[MapperKind.SCALAR_TYPE]: type => {
result.types[type.name] = {
type: 'scalar',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.UNION_TYPE]: type => {
result.types[type.name] = {
type: 'union',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.INPUT_OBJECT_TYPE]: type => {
result.types[type.name] = {
fields: {},
type: 'input',
extensions: handleDirectiveExtensions(type.extensions, removeDirectives),
};
return type;
},
[MapperKind.INPUT_OBJECT_FIELD]: (field, fieldName, typeName) => {
result.types[typeName].fields[fieldName] = {
extensions: handleDirectiveExtensions(field.extensions, removeDirectives),
};
return field;
},
});
return result;
}