Derive definitions for introspected schema (initial draft)

Author: tomhoule

graphql_query_derive/Cargo.toml

@@ -11,5 +11,7 @@ failure = "*"
 quote = "^0.6"
 syn = "*"
 proc-macro2 = { version = "*", features = ["nightly"] }
+serde = "1.0"
+serde_derive = "1.0"
 heck = "*"
 graphql-parser = "*"

graphql_query_derive/src/enums.rs

@@ -1,6 +1,6 @@
 use proc_macro2::{Ident, Span, TokenStream};
 
-pub const ENUMS_PREFIX: &'static str = "RustEnum_";
+pub const ENUMS_PREFIX: &'static str = "";
 
 #[derive(Debug, PartialEq)]
 pub struct GqlEnum {

graphql_query_derive/src/field_type.rs

@@ -1,8 +1,8 @@
+use enums::ENUMS_PREFIX;
 use graphql_parser::schema;
 use proc_macro2::{Ident, Span, TokenStream};
 use query::QueryContext;
 use schema::DEFAULT_SCALARS;
-use enums::ENUMS_PREFIX;
 
 #[derive(Debug, PartialEq)]
 pub enum FieldType {
@@ -19,10 +19,18 @@ impl FieldType {
                 let just_the_prefix = Ident::new(prefix, Span::call_site());
                 let name_string = name.to_string();
 
-                let name = if context.schema.scalars.contains(&name_string) || DEFAULT_SCALARS.iter().find(|elem| elem == &&name_string).is_some()  {
+                let name = if context.schema.scalars.contains(&name_string)
+                    || DEFAULT_SCALARS
+                        .iter()
+                        .find(|elem| elem == &&name_string)
+                        .is_some()
+                {
                     name.clone()
                 } else if context.schema.enums.contains_key(&name_string) {
-                    Ident::new(&format!("{}{}", ENUMS_PREFIX, &name_string), Span::call_site())
+                    Ident::new(
+                        &format!("{}{}", ENUMS_PREFIX, &name_string),
+                        Span::call_site(),
+                    )
                 } else {
                     just_the_prefix
                 };

graphql_query_derive/src/fragments.rs

@@ -1,6 +1,6 @@
+use graphql_parser::query::SelectionSet;
 use proc_macro2::{Ident, Span, TokenStream};
 use query::QueryContext;
-use graphql_parser::query::SelectionSet;
 
 #[derive(Debug, PartialEq)]
 pub struct GqlFragment {
@@ -12,10 +12,19 @@ pub struct GqlFragment {
 impl GqlFragment {
     pub fn to_rust(&self, context: &QueryContext) -> TokenStream {
         let name_ident = Ident::new(&self.name, Span::call_site());
+        let object = context.schema.objects.get(&self.on).expect("oh, noes");
+        let field_impls = object
+            .field_impls_for_selection(context, &self.selection, &self.name)
+            .unwrap();
+        let fields = object.response_fields_for_selection(context, &self.selection, &self.name);
 
         quote!{
             #[derive(Debug, Deserialize)]
-            pub struct #name_ident;
+            pub struct #name_ident {
+                #(#fields,)*
+            }
+
+            #(#field_impls)*
         }
     }
 }