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

use crate::ast::executable::{FragmentDefinition, OperationDefinition};
use crate::ast::{FromTokens, IsMatch, ParseError, Tokens};

#[derive(Debug)]
pub enum ExecutableDefinition<'a> {
    Operation(OperationDefinition<'a>),
    Fragment(FragmentDefinition<'a>),
}

impl<'a> FromTokens<'a> for ExecutableDefinition<'a> {
    fn from_tokens(tokens: &mut impl Tokens<'a>) -> Result<Self, ParseError> {
        if OperationDefinition::is_match(tokens) {
            OperationDefinition::from_tokens(tokens).map(Self::Operation)
        } else if FragmentDefinition::is_match(tokens) {
            FragmentDefinition::from_tokens(tokens).map(Self::Fragment)
        } else {
            Err(tokens.unexpected_token())
        }
    }
}

impl<'a> IsMatch<'a> for ExecutableDefinition<'a> {
    fn is_match(tokens: &mut impl Tokens<'a>) -> bool {
        OperationDefinition::is_match(tokens) || FragmentDefinition::is_match(tokens)
    }
}