Enable generating graphql clent code by cli

Author: h-michael

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,21 @@ 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
+        selected_operation: String,
+        /// Additional derives
+        /// --additional-derives='Serialize,PartialEq'
+        #[structopt(short = "a", long = "additional-derives")]
+        additional_derives: Option<String>,
+        /// allow, deny, or warn
+        #[structopt(short = "d", long = "deprecation-strategy",)]
+        deprecation_strategy: Option<String>,
         #[structopt(parse(from_os_str))]
         output: PathBuf,
     },
@@ -55,10 +70,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 +131,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(())
+}