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

use crate::parser::query::OperationDefinition;
use crate::validation::visitor::{Visitor, VisitorContext};
use crate::Positioned;

#[derive(Default)]
pub struct UploadFile;

impl<'a> Visitor<'a> for UploadFile {
    fn enter_operation_definition(
        &mut self,
        ctx: &mut VisitorContext<'a>,
        operation_definition: &'a Positioned<OperationDefinition>,
    ) {
        if let OperationDefinition::Query(query) = &operation_definition.node {
            for var in &query.variable_definitions {
                if let Some(ty) = ctx.registry.concrete_type_by_parsed_type(&var.var_type) {
                    if ty.name() == "Upload" {
                        ctx.report_error(
                            vec![var.position()],
                            "The Upload type is only allowed to be defined on a mutation",
                        );
                    }
                }
            }
        } else if let OperationDefinition::Subscription(subscription) = &operation_definition.node {
            for var in &subscription.variable_definitions {
                if let Some(ty) = ctx.registry.concrete_type_by_parsed_type(&var.var_type) {
                    if ty.name() == "Upload" {
                        ctx.report_error(
                            vec![var.position()],
                            "The Upload type is only allowed to be defined on a mutation",
                        );
                    }
                }
            }
        }
    }
}