mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-23 06:08:01 +00:00
Francisco Miguel García
107 lines
3.2 KiB
Rust
107 lines
3.2 KiB
Rust
// Copyright 2019-2022 Parity Technologies (UK) Ltd.
|
|
// This file is dual-licensed as Apache-2.0 or GPL-3.0.
|
|
// see LICENSE for license details.
|
|
|
|
use clap::Parser as ClapParser;
|
|
use color_eyre::eyre;
|
|
use jsonrpsee::client_transport::ws::Uri;
|
|
use std::{
|
|
fs,
|
|
io::Read,
|
|
path::PathBuf,
|
|
};
|
|
use subxt_codegen::DerivesRegistry;
|
|
|
|
/// Generate runtime API client code from metadata.
|
|
///
|
|
/// # Example (with code formatting)
|
|
///
|
|
/// `subxt codegen | rustfmt --edition=2018 --emit=stdout`
|
|
#[derive(Debug, ClapParser)]
|
|
pub struct Opts {
|
|
/// The url of the substrate node to query for metadata for codegen.
|
|
#[clap(name = "url", long, value_parser)]
|
|
url: Option<Uri>,
|
|
/// The path to the encoded metadata file.
|
|
#[clap(short, long, value_parser)]
|
|
file: Option<PathBuf>,
|
|
/// Additional derives
|
|
#[clap(long = "derive")]
|
|
derives: Vec<String>,
|
|
/// Additional derives for a given type.
|
|
///
|
|
/// Example `--derive-for-type my_module::my_type=serde::Serialize`.
|
|
#[clap(long = "derive-for-type", value_parser = derive_for_type_parser)]
|
|
derives_for_type: Vec<(String, String)>,
|
|
/// The `subxt` crate access path in the generated code.
|
|
/// Defaults to `::subxt`.
|
|
#[clap(long = "crate")]
|
|
crate_path: Option<String>,
|
|
}
|
|
|
|
fn derive_for_type_parser(src: &str) -> Result<(String, String), String> {
|
|
let (ty, derive) = src
|
|
.split_once('=')
|
|
.ok_or_else(|| String::from("Invalid pattern for `derive-for-type`. It should be `type=derive`, like `my_type=serde::Serialize`"))?;
|
|
|
|
Ok((ty.to_string(), derive.to_string()))
|
|
}
|
|
|
|
pub async fn run(opts: Opts) -> color_eyre::Result<()> {
|
|
let bytes = if let Some(file) = opts.file.as_ref() {
|
|
if opts.url.is_some() {
|
|
eyre::bail!("specify one of `--url` or `--file` but not both")
|
|
};
|
|
|
|
let mut file = fs::File::open(file)?;
|
|
let mut bytes = Vec::new();
|
|
file.read_to_end(&mut bytes)?;
|
|
bytes
|
|
} else {
|
|
let url = opts.url.unwrap_or_else(|| {
|
|
"http://localhost:9933"
|
|
.parse::<Uri>()
|
|
.expect("default url is valid")
|
|
});
|
|
subxt_codegen::utils::fetch_metadata_bytes(&url).await?
|
|
};
|
|
|
|
codegen(&bytes, opts.derives, opts.derives_for_type, opts.crate_path)?;
|
|
Ok(())
|
|
}
|
|
|
|
fn codegen(
|
|
metadata_bytes: &[u8],
|
|
raw_derives: Vec<String>,
|
|
derives_for_type: Vec<(String, String)>,
|
|
crate_path: Option<String>,
|
|
) -> color_eyre::Result<()> {
|
|
let item_mod = syn::parse_quote!(
|
|
pub mod api {}
|
|
);
|
|
|
|
let p = raw_derives
|
|
.iter()
|
|
.map(|raw| syn::parse_str(raw))
|
|
.collect::<Result<Vec<_>, _>>()?;
|
|
|
|
let crate_path = crate_path.map(Into::into).unwrap_or_default();
|
|
let mut derives = DerivesRegistry::new(&crate_path);
|
|
derives.extend_for_all(p.into_iter());
|
|
|
|
for (ty, derive) in derives_for_type.into_iter() {
|
|
let ty = syn::parse_str(&ty)?;
|
|
let derive = syn::parse_str(&derive)?;
|
|
derives.extend_for_type(ty, std::iter::once(derive), &crate_path)
|
|
}
|
|
|
|
let runtime_api = subxt_codegen::generate_runtime_api_from_bytes(
|
|
item_mod,
|
|
metadata_bytes,
|
|
derives,
|
|
crate_path,
|
|
);
|
|
println!("{}", runtime_api);
|
|
Ok(())
|
|
}
|