5f319a6875
Previously, build files iterated through ABIs in an inconsistent order, causing unwanted changes when the build was rerun, as the iteration order would differ. This commit resolves the issue by sorting ABIs by name, ensuring consistent formatting and preventing unintended changes in future commits. Co-authored-by: zizou <111426680+flopell@users.noreply.github.com>
50 lines
1.3 KiB
Rust
50 lines
1.3 KiB
Rust
use anyhow::Result;
|
|
use std::{fs, io::Write};
|
|
use substreams_ethereum::Abigen;
|
|
|
|
fn main() -> Result<()> {
|
|
let abi_folder = "abi";
|
|
let output_folder = "src/abi";
|
|
|
|
let abis = fs::read_dir(abi_folder)?;
|
|
|
|
let mut files = abis.collect::<Result<Vec<_>, _>>()?;
|
|
|
|
// Sort the files by their name
|
|
files.sort_by_key(|a| a.file_name());
|
|
|
|
let mut mod_rs_content = String::new();
|
|
mod_rs_content.push_str("#![allow(clippy::all)]\n");
|
|
|
|
for file in files {
|
|
let file_name = file.file_name();
|
|
let file_name = file_name.to_string_lossy();
|
|
|
|
if !file_name.ends_with(".json") {
|
|
continue;
|
|
}
|
|
|
|
let contract_name = file_name.split('.').next().unwrap();
|
|
|
|
let input_path = format!("{}/{}", abi_folder, file_name);
|
|
let output_path = format!("{}/{}.rs", output_folder, contract_name);
|
|
|
|
mod_rs_content.push_str(&format!("pub mod {};\n", contract_name));
|
|
|
|
if std::path::Path::new(&output_path).exists() {
|
|
continue;
|
|
}
|
|
|
|
Abigen::new(contract_name, &input_path)?
|
|
.generate()?
|
|
.write_to_file(&output_path)?;
|
|
}
|
|
|
|
let mod_rs_path = format!("{}/mod.rs", output_folder);
|
|
let mut mod_rs_file = fs::File::create(mod_rs_path)?;
|
|
|
|
mod_rs_file.write_all(mod_rs_content.as_bytes())?;
|
|
|
|
Ok(())
|
|
}
|