extern crate cc;

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

fn main() {
	println!("cargo:rerun-if-env-changed=LOG");
	println!("cargo:rerun-if-env-changed=BOARD");

	let arch: String = std::env::var("ARCH").unwrap();
	let board: String = std::env::var("BOARD").unwrap();
	match arch.as_str() {
		"x86_64" => {
			gen_vector_asm().unwrap();
		}
		"riscv32" => {
			if let Ok(file_path) = gen_sfsimg_asm() {
				cc::Build::new()
					.file(&file_path)
					.flag("-march=rv32imac")
					.flag("-mabi=ilp32")
					.compile("sfsimg");
			}
		}
		"riscv64" => {
			if let Ok(file_path) = gen_sfsimg_asm() {
				cc::Build::new()
					.file(&file_path)
					.flag("-march=rv64imac")
					.flag("-mabi=lp64")
					.compile("sfsimg");
			}
			if board == "k210" {
				println!("cargo:rustc-link-search=native={}", "../tools/k210");
				println!("cargo:rustc-link-lib=static=kendryte");
			}
		}
		"aarch64" => {
			if let Ok(file_path) = gen_sfsimg_asm() {
				cc::Build::new().file(&file_path).compile("sfsimg");
			}
		}
		_ => panic!("Unknown arch {}", arch),
	}
}

fn gen_vector_asm() -> Result<()> {
	let mut f = File::create("src/arch/x86_64/interrupt/vector.asm").unwrap();

	writeln!(f, "# generated by build.rs - do not edit")?;
	writeln!(f, ".section .text")?;
	writeln!(f, ".intel_syntax noprefix")?;
	for i in 0..256 {
		writeln!(f, "vector{}:", i)?;
		if !(i == 8 || (i >= 10 && i <= 14) || i == 17) {
			writeln!(f, "\tpush 0")?;
		}
		writeln!(f, "\tpush {}", i)?;
		writeln!(f, "\tjmp __alltraps")?;
	}

	writeln!(f, "\n.section .rodata")?;
	writeln!(f, ".global __vectors")?;
	writeln!(f, "__vectors:")?;
	for i in 0..256 {
		writeln!(f, "\t.quad vector{}", i)?;
	}
	Ok(())
}

fn gen_sfsimg_asm() -> Result<std::path::PathBuf> {
	let out_dir = std::env::var("OUT_DIR").unwrap();
	let sfsimg = std::env::var("SFSIMG").unwrap();

	let file_path = Path::new(&out_dir).join("sfsimg.S");
	let mut f = File::create(&file_path).unwrap();

	write!(f, "# generated by build.rs - do not edit")?;
	write!(f, r#"
	.section .rodata
	.align 12
	.global _user_img_start
	.global _user_img_end
_user_img_start:
    .incbin "{}"
_user_img_end:
"#, sfsimg)?;

	println!("cargo:rerun-if-changed={}", sfsimg);
	println!("cargo:rerun-if-env-changed=SFSIMG");

	Ok(file_path)
}
