platform/graphql_server/lib/introspection.dart

223 lines
5.3 KiB
Dart
Raw Normal View History

2018-08-03 17:26:11 +00:00
import 'package:graphql_parser/graphql_parser.dart';
2018-08-02 18:33:50 +00:00
import 'package:graphql_schema/graphql_schema.dart';
// TODO: How to handle custom types???
2018-08-02 19:22:16 +00:00
GraphQLSchema reflectSchema(GraphQLSchema schema, List<GraphQLType> allTypes) {
var objectTypes = fetchAllTypes(schema);
var typeType = _reflectSchemaTypes();
2018-08-03 17:26:11 +00:00
var directiveType = _reflectDirectiveType();
allTypes.addAll(objectTypes);
2018-08-02 18:33:50 +00:00
2018-08-02 19:22:16 +00:00
var schemaType = objectType('__Schema', fields: [
2018-08-02 18:33:50 +00:00
field(
'types',
type: listType(typeType),
resolve: (_, __) => objectTypes,
),
2018-08-02 19:22:16 +00:00
field(
'queryType',
type: typeType,
resolve: (_, __) => schema.query,
),
field(
'mutationType',
type: typeType,
resolve: (_, __) => schema.mutation,
),
2018-08-03 17:26:11 +00:00
field(
'directives',
type: listType(directiveType).nonNullable(),
resolve: (_, __) => [], // TODO: Actually fetch directives
),
2018-08-02 19:22:16 +00:00
]);
allTypes.addAll([
2018-08-03 17:26:11 +00:00
directiveType,
2018-08-02 19:22:16 +00:00
typeType,
schemaType,
_reflectFields(),
2018-08-02 18:33:50 +00:00
]);
var fields = <GraphQLField>[
field(
'__schema',
type: schemaType,
resolve: (_, __) => schemaType,
),
field(
'__type',
type: typeType,
arguments: [
new GraphQLFieldArgument('name', graphQLString.nonNullable())
],
resolve: (_, args) {
var name = args['name'] as String;
return objectTypes.firstWhere((t) => t.name == name,
orElse: () =>
throw new GraphQLException('No type named "$name" exists.'));
},
),
];
fields.addAll(schema.query.fields);
return new GraphQLSchema(
2018-08-02 19:22:16 +00:00
query: objectType(schema.query.name, fields: fields),
2018-08-02 18:33:50 +00:00
mutation: schema.mutation,
);
}
2018-08-02 19:22:16 +00:00
GraphQLObjectType _typeType;
GraphQLObjectType _reflectSchemaTypes() {
if (_typeType == null) {
_typeType = _createTypeType();
_typeType.fields.add(
field(
'ofType',
type: _reflectSchemaTypes(),
resolve: (type, _) {
if (type is GraphQLListType)
return type.innerType;
else if (type is GraphQLNonNullableType) return type.innerType;
return null;
},
),
);
var fieldType = _reflectFields();
var typeField = fieldType.fields
.firstWhere((f) => f.name == 'type', orElse: () => null);
if (typeField == null) {
fieldType.fields.add(
field(
'type',
type: _reflectSchemaTypes(),
resolve: (f, _) => (f as GraphQLField).type,
),
);
}
}
return _typeType;
}
GraphQLObjectType _createTypeType() {
2018-08-02 18:33:50 +00:00
var fieldType = _reflectFields();
2018-08-02 19:22:16 +00:00
return objectType('__Type', fields: [
2018-08-02 18:33:50 +00:00
field(
'name',
type: graphQLString,
2018-08-02 19:22:16 +00:00
resolve: (type, _) => (type as GraphQLType).name,
),
field(
'description',
type: graphQLString,
resolve: (type, _) => (type as GraphQLType).description,
2018-08-02 18:33:50 +00:00
),
field(
'kind',
type: graphQLString,
2018-08-02 19:22:16 +00:00
resolve: (type, _) {
var t = type as GraphQLType;
if (t is GraphQLScalarType)
return 'SCALAR';
else if (t is GraphQLObjectType)
return 'OBJECT';
else if (t is GraphQLListType)
return 'LIST';
else if (t is GraphQLNonNullableType)
return 'NON_NULL';
else
throw new UnsupportedError(
'Cannot get the kind of $t.'); // TODO: Interface + union
},
2018-08-02 18:33:50 +00:00
),
field(
'fields',
type: listType(fieldType),
2018-08-02 19:22:16 +00:00
resolve: (type, _) => type is GraphQLObjectType ? type.fields : [],
2018-08-02 18:33:50 +00:00
),
]);
}
2018-08-02 19:22:16 +00:00
GraphQLObjectType _fieldType;
2018-08-02 18:33:50 +00:00
GraphQLObjectType _reflectFields() {
2018-08-02 19:22:16 +00:00
if (_fieldType == null) {
_fieldType = _createFieldType();
}
return _fieldType;
}
GraphQLObjectType _createFieldType() {
return objectType('__Field', fields: [
field(
'name',
type: graphQLString,
resolve: (f, _) => (f as GraphQLField).name,
),
]);
2018-08-02 18:33:50 +00:00
}
2018-08-03 17:26:11 +00:00
GraphQLObjectType _reflectDirectiveType() {
return objectType('__Directive', fields: [
field(
'name',
type: graphQLString,
resolve: (obj, _) => (obj as DirectiveContext).NAME.span.text,
),
]);
}
2018-08-02 19:22:16 +00:00
List<GraphQLObjectType> fetchAllTypes(GraphQLSchema schema) {
2018-08-02 18:33:50 +00:00
var typess = <GraphQLType>[];
typess.addAll(_fetchAllTypesFromObject(schema.query));
if (schema.mutation != null) {
typess.addAll(_fetchAllTypesFromObject(schema.mutation)
.where((t) => t is GraphQLObjectType));
}
var types = <GraphQLObjectType>[];
for (var type in typess) {
if (type is GraphQLObjectType) types.add(type);
}
return types.toSet().toList();
}
List<GraphQLType> _fetchAllTypesFromObject(GraphQLObjectType objectType) {
var types = <GraphQLType>[objectType];
for (var field in objectType.fields) {
if (field.type is GraphQLObjectType) {
types.addAll(_fetchAllTypesFromObject(field.type as GraphQLObjectType));
} else {
types.addAll(_fetchAllTypesFromType(field.type));
}
}
return types;
}
Iterable<GraphQLType> _fetchAllTypesFromType(GraphQLType type) {
var types = <GraphQLType>[];
if (type is GraphQLNonNullableType) {
types.addAll(_fetchAllTypesFromType(type.innerType));
} else if (type is GraphQLListType) {
types.addAll(_fetchAllTypesFromType(type.innerType));
} else if (type is GraphQLObjectType) {
types.addAll(_fetchAllTypesFromObject(type));
}
// TODO: Enum, interface, union
return types;
}