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 = file;
        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!("{output_folder}/{contract_name}.rs");

        mod_rs_content.push_str(&format!("pub mod {contract_name};\n"));

        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!("{output_folder}/mod.rs");
    let mut mod_rs_file = fs::File::create(mod_rs_path)?;

    mod_rs_file.write_all(mod_rs_content.as_bytes())?;

    Ok(())
}