Merge pull request #122 from h-michael/codegen

Enable generating graphql client code by cli

Author: tomhoule

graphql_client_cli/Cargo.toml

@@ -17,3 +17,4 @@ structopt = "0.2"
 serde = "1.0"
 serde_derive = "1.0"
 serde_json = "1.0"
+graphql_client_codegen = { path = "../graphql_client_codegen/", version = "0.4.0" }

graphql_client_cli/src/main.rs

@@ -4,12 +4,15 @@ extern crate reqwest;
 extern crate structopt;
 #[macro_use]
 extern crate graphql_client;
+extern crate graphql_client_codegen;
 #[macro_use]
 extern crate serde_derive;
 extern crate serde;
 extern crate serde_json;
 
 use reqwest::header::{HeaderMap, HeaderValue, CONTENT_TYPE, ACCEPT};
+use std::fs::File;
+use std::io::Write as IoWrite;
 use std::path::PathBuf;
 use structopt::StructOpt;
 
@@ -38,9 +41,22 @@ enum Cli {
     #[structopt(name = "generate")]
     Generate {
         // should be a glob
-        paths: String,
+        /// Path to graphql query file.
         #[structopt(parse(from_os_str))]
-        schema: PathBuf,
+        query_path: PathBuf,
+        /// Path to graphql schema file.
+        #[structopt(parse(from_os_str))]
+        schema_path: PathBuf,
+        /// Name of struct that is implementation target.
+        selected_operation: String,
+        /// Additional derives that will be added to the generated structs and enums for the response and the variables.
+        /// --additional-derives='Serialize,PartialEq'
+        #[structopt(short = "a", long = "additional-derives")]
+        additional_derives: Option<String>,
+        /// You can choose deprecation strategy from allow, deny, or warn.
+        /// Default value is warn.
+        #[structopt(short = "d", long = "deprecation-strategy",)]
+        deprecation_strategy: Option<String>,
         #[structopt(parse(from_os_str))]
         output: PathBuf,
     },
@@ -55,10 +71,20 @@ fn main() -> Result<(), failure::Error> {
             authorization,
         } => introspect_schema(schema_location, output, authorization),
         Cli::Generate {
-            paths: _,
-            schema: _,
-            output: _,
-        } => unimplemented!(),
+            query_path,
+            schema_path,
+            selected_operation,
+            additional_derives,
+            deprecation_strategy,
+            output,
+        } => generate_code(
+            query_path,
+            schema_path,
+            selected_operation,
+            additional_derives,
+            deprecation_strategy,
+            output,
+        ),
     }
 }
 
@@ -106,3 +132,34 @@ fn construct_headers() -> HeaderMap {
     headers.insert(ACCEPT, HeaderValue::from_static("application/json"));
     headers
 }
+
+fn generate_code(
+    query_path: PathBuf,
+    schema_path: PathBuf,
+    selected_operation: String,
+    additional_derives: Option<String>,
+    deprecation_strategy: Option<String>,
+    output: PathBuf,
+) -> Result<(), failure::Error> {
+    let deprecation_strategy = deprecation_strategy.as_ref().map(|s| s.as_str());
+    let deprecation_strategy = match deprecation_strategy {
+        Some("allow") => Some(graphql_client_codegen::deprecation::DeprecationStrategy::Allow),
+        Some("deny") => Some(graphql_client_codegen::deprecation::DeprecationStrategy::Deny),
+        Some("warn") => Some(graphql_client_codegen::deprecation::DeprecationStrategy::Warn),
+        _ => None,
+    };
+
+    let options = graphql_client_codegen::GraphQLClientDeriveOptions {
+        selected_operation,
+        additional_derives: additional_derives,
+        deprecation_strategy,
+    };
+    let gen = graphql_client_codegen::generate_module_token_stream(
+        query_path,
+        schema_path,
+        Some(options),
+    )?;
+    let mut file = File::create(output)?;
+    write!(file, "{}", gen.to_string());
+    Ok(())
+}

graphql_client_codegen/src/lib.rs

@@ -55,8 +55,10 @@ lazy_static! {
         CacheMap::default();
 }
 
-pub struct GraphQLClientDeriveOptions<'a> {
-    pub input: &'a syn::DeriveInput,
+pub struct GraphQLClientDeriveOptions {
+    pub selected_operation: String,
+    pub additional_derives: Option<String>,
+    pub deprecation_strategy: Option<deprecation::DeprecationStrategy>,
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -69,12 +71,13 @@ pub fn generate_module_token_stream(
     schema_path: std::path::PathBuf,
     options: Option<GraphQLClientDeriveOptions>,
 ) -> Result<TokenStream, failure::Error> {
-    let input = options.unwrap().input;
+    let options = options.unwrap();
 
-    let response_derives = attributes::extract_attr(input, "response_derives").ok();
+    let response_derives = options.additional_derives;
 
     // The user can determine what to do about deprecations.
-    let deprecation_strategy = deprecation::extract_deprecation_strategy(input)
+    let deprecation_strategy = options
+        .deprecation_strategy
         .unwrap_or(deprecation::DeprecationStrategy::Warn);
 
     // We need to qualify the query with the path to the crate it is part of
@@ -122,13 +125,16 @@ pub fn generate_module_token_stream(
         }
     };
 
-    let operation_string = input.ident.to_string();
-    let module_name = Ident::new(&operation_string.to_snake_case(), Span::call_site());
-    let struct_name = &input.ident;
+    let operation_string = options.selected_operation.clone();
+    let module_name = Ident::new(
+        options.selected_operation.to_snake_case().as_str(),
+        Span::call_site(),
+    );
+    let struct_name = Ident::new(options.selected_operation.as_str(), Span::call_site());
     let schema_output = codegen::response_for_query(
         schema,
         query,
-        input.ident.to_string(),
+        options.selected_operation.clone(),
         response_derives,
         deprecation_strategy,
     )?;

graphql_query_derive/src/lib.rs

@@ -11,8 +11,8 @@ pub fn graphql_query_derive(input: proc_macro::TokenStream) -> proc_macro::Token
     let input = TokenStream::from(input);
     let ast = syn::parse2(input).expect("Derive input is well formed");
     let (query_path, schema_path) = build_query_and_schema_path(&ast);
-    let option = GraphQLClientDeriveOptions { input: &ast };
-    let gen = generate_module_token_stream(query_path, schema_path, Some(option)).unwrap();
+    let options = build_graphql_client_derive_options(&ast);
+    let gen = generate_module_token_stream(query_path, schema_path, Some(options)).unwrap();
     gen.into()
 }
 
@@ -29,3 +29,16 @@ fn build_query_and_schema_path(
     let schema_path = ::std::path::Path::new(&cargo_manifest_dir).join(schema_path);
     (query_path, schema_path)
 }
+
+fn build_graphql_client_derive_options(input: &syn::DeriveInput) -> GraphQLClientDeriveOptions {
+    let response_derives = attributes::extract_attr(input, "response_derives").ok();
+    // The user can determine what to do about deprecations.
+    let deprecation_strategy = deprecation::extract_deprecation_strategy(input)
+        .unwrap_or(deprecation::DeprecationStrategy::Warn);
+
+    GraphQLClientDeriveOptions {
+        selected_operation: input.ident.to_string(),
+        additional_derives: response_derives,
+        deprecation_strategy: Some(deprecation_strategy),
+    }
+}