iso639 0.2.0

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

fn build_autonym_db() {
    let f = std::fs::File::open("./iso639-databases/iso639-autonyms.tsv").unwrap();
    let path = Path::new(&std::env::var("OUT_DIR").unwrap()).join("autonym_db.rs");
    let mut db = BufWriter::new(File::create(&path).unwrap());

    let mut names = std::collections::BTreeMap::new();

    let mut tsv_reader = csv::ReaderBuilder::new()
        .delimiter(b'\t')
        .has_headers(true)
        .from_reader(f);

    for result in tsv_reader.records() {
        let record = result.unwrap();

        let rec_name = format!("RECORD_{}", record[0].to_uppercase());
        let tag3 = &record[0];
        let tag1 = if record[1].trim() == "" {
            None
        } else {
            Some(&record[1])
        };
        let name = &record[2];
        let autonym = if record[3].trim() == "" {
            None
        } else {
            Some(&record[3])
        };
        let source = &record[4];

        writeln!(&mut db, "const {}: &'static Record = &Record {{", &rec_name).unwrap();
        names.insert(record[0].to_string(), rec_name.to_string());
        if record[1].trim() != "" {
            names.insert(record[1].to_string(), rec_name.to_string());
        }
        writeln!(
            &mut db,
            "    tag3: {:?}, tag1: {:?}, name: {:?}, autonym: {:?}, source: {:?}",
            tag3, tag1, name, autonym, source
        )
        .unwrap();
        writeln!(&mut db, "}};").unwrap();
    }

    write!(
        &mut db,
        "static AUTONYMS: phf::Map<&'static str, &'static Record> = "
    )
    .unwrap();

    let mut map = phf_codegen::Map::new();

    for (key, value) in names.iter() {
        map.entry(&**key, value);
    }

    writeln!(&mut db, "{}", map.build()).unwrap();
    write!(&mut db, ";\n").unwrap();
}

fn build_script_db() {
    let f = std::fs::File::open("./iso639-databases/iso639-default-script.tsv").unwrap();
    let path = Path::new(&std::env::var("OUT_DIR").unwrap()).join("script_db.rs");
    let mut db = BufWriter::new(File::create(&path).unwrap());

    let mut names = std::collections::BTreeMap::new();

    let mut tsv_reader = csv::ReaderBuilder::new()
        .delimiter(b'\t')
        .has_headers(true)
        .from_reader(f);

    for result in tsv_reader.records() {
        let record = result.unwrap();

        let rec_name = format!("RECORD_{}", record[0].to_uppercase());
        let tag3 = &record[0];
        let tag1 = if record[1].trim() == "" {
            None
        } else {
            Some(&record[1])
        };
        let script = &record[2];
        let source = &record[4];

        writeln!(&mut db, "const {}: &'static Record = &Record {{", &rec_name).unwrap();
        names.insert(record[0].to_string(), rec_name.to_string());
        if record[1].trim() != "" {
            names.insert(record[1].to_string(), rec_name.to_string());
        }
        writeln!(
            &mut db,
            "    tag3: {:?}, tag1: {:?}, script: {:?}, source: {:?}",
            tag3, tag1, script, source
        )
        .unwrap();
        writeln!(&mut db, "}};").unwrap();
    }

    write!(
        &mut db,
        "static SCRIPTS: phf::Map<&'static str, &'static Record> = "
    )
    .unwrap();

    let mut map = phf_codegen::Map::new();

    for (key, value) in names.iter() {
        map.entry(&**key, value);
    }

    writeln!(&mut db, "{}", map.build()).unwrap();
    write!(&mut db, ";\n").unwrap();
}

fn build_lcid_db() {
    let f = std::fs::File::open("./iso639-databases/iso639-lcids.tsv").unwrap();
    let path = Path::new(&std::env::var("OUT_DIR").unwrap()).join("lcid_db.rs");
    let mut db = BufWriter::new(File::create(&path).unwrap());

    let mut names = vec![];

    let mut tsv_reader = csv::ReaderBuilder::new()
        .delimiter(b'\t')
        .has_headers(true)
        .from_reader(f);

    for result in tsv_reader.records() {
        let record = result.unwrap();
        println!("{:?}", record);

        let mut name_chunks = vec![&record[0]];

        let tag3 = &record[0];
        let tag1 = if record[1].trim() == "" {
            None
        } else {
            Some(&record[1])
        };
        let script = if record[2].trim() == "" {
            None
        } else {
            name_chunks.push(&record[2]);
            Some(&record[2])
        };
        let region = if record[3].trim() == "" {
            None
        } else {
            name_chunks.push(&record[3]);
            Some(&record[3])
        };
        let lcid = record[4].parse::<u32>().unwrap();

        let rec_name = format!(
            "RECORD_{}",
            name_chunks.join("_").to_uppercase().replace("-", "_")
        );

        writeln!(&mut db, "const {}: &'static Record = &Record {{", &rec_name).unwrap();
        names.push(rec_name.to_string());
        writeln!(
            &mut db,
            "    tag3: {:?}, tag1: {:?}, script: {:?}, region: {:?}, lcid: {}",
            tag3, tag1, script, region, lcid,
        )
        .unwrap();
        writeln!(&mut db, "}};").unwrap();
    }

    write!(&mut db, "static LCIDS: &'static [&'static Record] = &[").unwrap();

    for name in names.iter() {
        writeln!(&mut db, "    {},", name).unwrap();
    }

    write!(&mut db, "];\n").unwrap();
}

fn main() {
    let mut process = std::process::Command::new("git")
        .args(&["submodule", "update", "--init"])
        .spawn()
        .unwrap();

    process.wait().unwrap();

    build_autonym_db();
    build_lcid_db();
    build_script_db();
}