graphql-composition 0.12.2

An implementation of GraphQL federated schema composition
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use super::*;
use crate::federated_graph::DirectiveLocations;

pub(super) fn ingest_directive_definitions(ctx: &mut Context<'_>) {
    for definition in ctx.document.definitions() {
        match definition {
            ast::Definition::Directive(directive_definition) => {
                let definition_name_id = ctx.subgraphs.strings.intern(directive_definition.name());
                ingest_directive_definition(ctx, directive_definition, definition_name_id);
            }
            ast::Definition::Schema(_)
            | ast::Definition::SchemaExtension(_)
            | ast::Definition::TypeExtension(_)
            | ast::Definition::Type(_) => {}
        }
    }
}

fn ingest_directive_definition(
    ctx: &mut Context<'_>,
    directive_definition: ast::DirectiveDefinition<'_>,
    name: subgraphs::StringId,
) {
    let mut locations = DirectiveLocations::default();
    let subgraphs = &mut ctx.subgraphs;
    let subgraph_id = ctx.subgraph_id;

    for location in directive_definition.locations() {
        let location = match location {
            ast::DirectiveLocation::Query => DirectiveLocations::QUERY,
            ast::DirectiveLocation::Mutation => DirectiveLocations::MUTATION,
            ast::DirectiveLocation::Subscription => DirectiveLocations::SUBSCRIPTION,
            ast::DirectiveLocation::Field => DirectiveLocations::FIELD,
            ast::DirectiveLocation::FragmentDefinition => DirectiveLocations::FRAGMENT_DEFINITION,
            ast::DirectiveLocation::FragmentSpread => DirectiveLocations::FRAGMENT_SPREAD,
            ast::DirectiveLocation::InlineFragment => DirectiveLocations::INLINE_FRAGMENT,
            ast::DirectiveLocation::VariableDefinition => DirectiveLocations::VARIABLE_DEFINITION,
            ast::DirectiveLocation::Schema => DirectiveLocations::SCHEMA,
            ast::DirectiveLocation::Scalar => DirectiveLocations::SCALAR,
            ast::DirectiveLocation::Object => DirectiveLocations::OBJECT,
            ast::DirectiveLocation::FieldDefinition => DirectiveLocations::FIELD_DEFINITION,
            ast::DirectiveLocation::ArgumentDefinition => DirectiveLocations::ARGUMENT_DEFINITION,
            ast::DirectiveLocation::Interface => DirectiveLocations::INTERFACE,
            ast::DirectiveLocation::Union => DirectiveLocations::UNION,
            ast::DirectiveLocation::Enum => DirectiveLocations::ENUM,
            ast::DirectiveLocation::EnumValue => DirectiveLocations::ENUM_VALUE,
            ast::DirectiveLocation::InputObject => DirectiveLocations::INPUT_OBJECT,
            ast::DirectiveLocation::InputFieldDefinition => DirectiveLocations::INPUT_FIELD_DEFINITION,
        };

        locations |= location;
    }

    let mut arguments = Vec::with_capacity(directive_definition.arguments().len());

    for argument in directive_definition.arguments() {
        let argument_name = subgraphs.strings.intern(argument.name());
        let r#type = subgraphs.intern_field_type(argument.ty());
        let default_value = argument
            .default_value()
            .map(|default| crate::ast_value_to_subgraph_value(default, subgraphs));

        let directives = argument
            .directives()
            .map(|directive| subgraphs::InputValueDefinitionDirective {
                name: subgraphs.strings.intern(directive.name()),
                arguments: directive
                    .arguments()
                    .map(|arg| {
                        (
                            subgraphs.strings.intern(arg.name()),
                            crate::ast_value_to_subgraph_value(arg.value(), subgraphs),
                        )
                    })
                    .collect(),
            })
            .collect();

        arguments.push(subgraphs::InputValueDefinition {
            name: argument_name,
            r#type,
            default_value,
            directives,
        });
    }

    subgraphs.push_directive_definition(subgraphs::DirectiveDefinition {
        subgraph_id,
        name,
        locations,
        arguments,
        repeatable: directive_definition.is_repeatable(),
    });
}