63f9c89429
* fix: add `balance_owner` attribute for ekubo_v2 components * refactor: rename `ekubo` into `ekubo-v2` * refactor: freeze `substreams-helper` version in ekubo V2 module * docs: improve integration test comment * docs: add TODO to remove `balance_owner` in favor of `AccountBalances` --------- 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(())
|
|
}
|