use super::gen_c::create_c_test_program;
use super::gen_c::setup_c_file_paths;
use super::gen_rust::{create_rust_test_program, setup_rust_file_paths};
use super::intrinsic::IntrinsicDefinition;
use super::intrinsic_helpers::IntrinsicTypeDefinition;
use std::fs::File;
use std::io::Write;

pub fn write_file(filename: &String, code: String) {
    let mut file = File::create(&filename).unwrap();
    file.write_all(code.into_bytes().as_slice()).unwrap();
}

pub fn write_c_testfiles<T: IntrinsicTypeDefinition + Sized>(
    intrinsics: &Vec<&dyn IntrinsicDefinition<T>>,
    target: &str,
    c_target: &str,
    headers: &[&str],
    notice: &str,
    arch_specific_definitions: &[&str],
) -> Vec<String> {
    let intrinsics_name_list = intrinsics
        .iter()
        .map(|i| i.name().clone())
        .collect::<Vec<_>>();
    let filename_mapping = setup_c_file_paths(&intrinsics_name_list);

    intrinsics.iter().for_each(|&i| {
        let c_code = create_c_test_program(
            i,
            headers,
            target,
            c_target,
            notice,
            arch_specific_definitions,
        );
        match filename_mapping.get(&i.name()) {
            Some(filename) => write_file(filename, c_code),
            None => {}
        };
    });

    intrinsics_name_list
}

pub fn write_rust_testfiles<T: IntrinsicTypeDefinition>(
    intrinsics: Vec<&dyn IntrinsicDefinition<T>>,
    rust_target: &str,
    notice: &str,
    definitions: &str,
    cfg: &str,
) -> Vec<String> {
    let intrinsics_name_list = intrinsics
        .iter()
        .map(|i| i.name().clone())
        .collect::<Vec<_>>();
    let filename_mapping = setup_rust_file_paths(&intrinsics_name_list);

    intrinsics.iter().for_each(|&i| {
        let rust_code = create_rust_test_program(i, rust_target, notice, definitions, cfg);
        match filename_mapping.get(&i.name()) {
            Some(filename) => write_file(filename, rust_code),
            None => {}
        }
    });

    intrinsics_name_list
}