Custom derive for cli

cli/src/main.rs

@@ -33,6 +33,12 @@ use std::{
     path::PathBuf,
 };
 use structopt::StructOpt;
+use subxt_codegen::GeneratedTypeDerives;
+use syn::{
+    punctuated::Punctuated,
+    Ident,
+    __private::Span,
+};
 
 /// Utilities for working with substrate metadata for subxt.
 #[derive(Debug, StructOpt)]
@@ -70,6 +76,9 @@ enum Command {
         /// the path to the encoded metadata file.
         #[structopt(short, long, parse(from_os_str))]
         file: Option<PathBuf>,
+        /// Additional derives
+        #[structopt(long = "derive")]
+        derives: Vec<String>,
     },
 }
 
@@ -102,7 +111,7 @@ fn main() -> color_eyre::Result<()> {
                 }
             }
         }
-        Command::Codegen { url, file } => {
+        Command::Codegen { url, file, derives } => {
             if let Some(file) = file.as_ref() {
                 if url.is_some() {
                     eyre::bail!("specify one of `--url` or `--file` but not both")
@@ -111,15 +120,15 @@ fn main() -> color_eyre::Result<()> {
                 let mut file = fs::File::open(file)?;
                 let mut bytes = Vec::new();
                 file.read_to_end(&mut bytes)?;
-                codegen(&mut &bytes[..])?;
+                codegen(&mut &bytes[..], derives)?;
                 return Ok(())
             }
 
             let url = url.unwrap_or_else(|| {
                 url::Url::parse("http://localhost:9933").expect("default url is valid")
             });
             let (_, bytes) = fetch_metadata(&url)?;
-            codegen(&mut &bytes[..])?;
+            codegen(&mut &bytes[..], derives)?;
             Ok(())
         }
     }
@@ -145,13 +154,24 @@ fn fetch_metadata(url: &url::Url) -> color_eyre::Result<(String, Vec<u8>)> {
     Ok((hex_data, bytes))
 }
 
-fn codegen<I: Input>(encoded: &mut I) -> color_eyre::Result<()> {
+fn codegen<I: Input>(
+    encoded: &mut I,
+    raw_derives: Vec<String>,
+) -> color_eyre::Result<()> {
     let metadata = <RuntimeMetadataPrefixed as Decode>::decode(encoded)?;
     let generator = subxt_codegen::RuntimeGenerator::new(metadata);
     let item_mod = syn::parse_quote!(
         pub mod api {}
     );
-    let runtime_api = generator.generate_runtime(item_mod, Default::default());
+
+    let mut p: Punctuated<syn::Path, syn::Token![,]> = Punctuated::new();
+    for raw in raw_derives {
+        p.push(Ident::new(&raw, Span::call_site()).into());
+    }
+    let mut derives = GeneratedTypeDerives::default();
+    derives.append(p.iter().cloned());
+
+    let runtime_api = generator.generate_runtime(item_mod, derives);
     println!("{}", runtime_api);
     Ok(())
 }