Add visibility option

Author: h-michael

graphql_client_cli/src/generate.rs

@@ -3,6 +3,7 @@ use graphql_client_codegen::*;
 use std::fs::File;
 use std::io::Write as IoWrite;
 use std::path::PathBuf;
+use syn;
 
 pub fn generate_code(
     query_path: PathBuf,
@@ -11,6 +12,7 @@ pub fn generate_code(
     additional_derives: Option<String>,
     deprecation_strategy: Option<String>,
     no_formatting: bool,
+    module_visibility: Option<String>,
     output: PathBuf,
 ) -> Result<(), failure::Error> {
     let deprecation_strategy = deprecation_strategy.as_ref().map(|s| s.as_str());
@@ -21,10 +23,22 @@ pub fn generate_code(
         _ => None,
     };
 
+    let module_visibility = module_visibility.as_ref().map(|s| s.as_str());
+    let module_visibility = match module_visibility {
+        Some("pub") => syn::VisPublic {
+            pub_token: <Token![pub]>::default(),
+        }.into(),
+        Some("private") => syn::Visibility::Inherited {},
+        _ => syn::VisPublic {
+            pub_token: <Token![pub]>::default(),
+        }.into(),
+    };
+
     let options = GraphQLClientDeriveOptions {
         struct_name: selected_operation,
         additional_derives,
         deprecation_strategy,
+        module_visibility,
     };
 
     let gen = generate_module_token_stream(query_path, schema_path, Some(options))?;

graphql_client_cli/src/main.rs

@@ -10,6 +10,8 @@ extern crate graphql_client_codegen;
 extern crate serde_derive;
 extern crate serde;
 extern crate serde_json;
+#[macro_use]
+extern crate syn;
 
 #[cfg(feature = "rustfmt")]
 extern crate rustfmt_nightly as rustfmt;
@@ -57,6 +59,10 @@ enum Cli {
         /// Formating feature is disabled as default installation.
         #[structopt(long = "no-formatting")]
         no_formatting: bool,
+        /// You can choose module and target struct visibility from pub and private.
+        /// Default value is pub.
+        #[structopt(short = "m", long = "module_visibility")]
+        module_visibility: Option<String>,
         #[structopt(parse(from_os_str))]
         output: PathBuf,
     },
@@ -77,6 +83,7 @@ fn main() -> Result<(), failure::Error> {
             additional_derives,
             deprecation_strategy,
             no_formatting,
+            module_visibility,
             output,
         } => generate::generate_code(
             query_path,
@@ -85,6 +92,7 @@ fn main() -> Result<(), failure::Error> {
             additional_derives,
             deprecation_strategy,
             no_formatting,
+            module_visibility,
             output,
         ),
     }

graphql_client_codegen/src/lib.rs

@@ -23,6 +23,7 @@ extern crate syn;
 extern crate quote;
 
 use proc_macro2::TokenStream;
+use syn::Visibility;
 
 /// Derive-related code. This will be moved into graphql_query_derive.
 pub mod attributes;
@@ -71,6 +72,8 @@ pub struct GraphQLClientDeriveOptions {
     pub additional_derives: Option<String>,
     /// The deprecation strategy to adopt.
     pub deprecation_strategy: Option<deprecation::DeprecationStrategy>,
+    /// target struct visibility.
+    pub module_visibility: Visibility,
 }
 
 /// Generates the code for a Rust module given a query, a schema and options.
@@ -81,6 +84,7 @@ pub fn generate_module_token_stream(
 ) -> Result<TokenStream, failure::Error> {
     let options = options.unwrap();
 
+    let module_visibility = options.module_visibility;
     let response_derives = options.additional_derives;
 
     // The user can determine what to do about deprecations.
@@ -157,7 +161,7 @@ pub fn generate_module_token_stream(
     )?;
 
     let result = quote!(
-        pub mod #module_name {
+        #module_visibility mod #module_name {
             #![allow(non_camel_case_types)]
             #![allow(non_snake_case)]
             #![allow(dead_code)]

graphql_query_derive/src/lib.rs

@@ -44,8 +44,9 @@ fn build_graphql_client_derive_options(input: &syn::DeriveInput) -> GraphQLClien
         .unwrap_or(deprecation::DeprecationStrategy::default());
 
     GraphQLClientDeriveOptions {
-        struct_name: input.ident.to_string(),
+        struct_name: input.clone().ident.to_string(),
         additional_derives: response_derives,
         deprecation_strategy: Some(deprecation_strategy),
+        module_visibility: input.clone().vis,
     }
 }