cynic-codegen 0.14.1

Codegen for cynic - a GraphQL query builder & data mapper for Rust
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

use proc_macro2::TokenStream;

use crate::{schema, Ident};

/// We generate a UnionStruct for each union in the schema.
///
/// These are output as empty structs that can be used as the TypeLock
/// in a SelectorStruct.
#[derive(Debug)]
pub struct UnionStruct {
    pub name: Ident,
    pub subtypes: Vec<Ident>,
}

impl UnionStruct {
    pub fn from_union(union: &schema::UnionType) -> Self {
        UnionStruct {
            name: Ident::for_type(&union.name),
            subtypes: union.types.iter().map(Ident::for_type).collect(),
        }
    }
}

impl quote::ToTokens for UnionStruct {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        use quote::{quote, TokenStreamExt};

        let name = &self.name;
        let subtypes = &self.subtypes;

        tokens.append_all(quote! {
            #[allow(dead_code)]
            pub struct #name {}

            #(
                impl ::cynic::selection_set::HasSubtype<#subtypes> for #name {}
            )*
        });
    }
}