use std::{
  io::Write,
  env,
  path::Path,
  fs::{File, remove_file},
};

use dalek_ff_group::EdwardsPoint;

use monero_generators::bulletproofs_generators;

fn serialize(generators_string: &mut String, points: &[EdwardsPoint]) {
  for generator in points {
    generators_string.extend(
      format!(
        "
          dalek_ff_group::EdwardsPoint(
            curve25519_dalek::edwards::CompressedEdwardsY({:?}).decompress().unwrap()
          ),
        ",
        generator.compress().to_bytes()
      )
      .chars(),
    );
  }
}

fn generators(prefix: &'static str, path: &str) {
  let generators = bulletproofs_generators(prefix.as_bytes());
  #[allow(non_snake_case)]
  let mut G_str = String::new();
  serialize(&mut G_str, &generators.G);
  #[allow(non_snake_case)]
  let mut H_str = String::new();
  serialize(&mut H_str, &generators.H);

  let path = Path::new(&env::var("OUT_DIR").unwrap()).join(path);
  let _ = remove_file(&path);
  File::create(&path)
    .unwrap()
    .write_all(
      format!(
        "
          pub(crate) static GENERATORS_CELL: OnceLock<Generators> = OnceLock::new();
          pub fn GENERATORS() -> &'static Generators {{
            GENERATORS_CELL.get_or_init(|| Generators {{
              G: vec![
                {G_str}
              ],
              H: vec![
                {H_str}
              ],
            }})
          }}
        ",
      )
      .as_bytes(),
    )
    .unwrap();
}

fn main() {
  println!("cargo:rerun-if-changed=build.rs");

  generators("bulletproof", "generators.rs");
  generators("bulletproof_plus", "generators_plus.rs");
}