76 lines
2.6 KiB
Rust
76 lines
2.6 KiB
Rust
|
use std::fs;
|
|||
|
|
use std::io::Write;
|
|||
|
|
use std::path::{Path, PathBuf};
|
|||
|
|
use std::process::Command;
|
|||
|
|
|
|||
|
|
pub struct Opts {
|
|||
|
|
pub out: Option<PathBuf>,
|
|||
|
|
pub nyrt: Option<PathBuf>,
|
|||
|
|
pub opt_level: Option<String>,
|
|||
|
|
pub timeout_ms: Option<u64>,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn resolve_ny_llvmc() -> PathBuf {
|
|||
|
|
if let Ok(s) = std::env::var("NYASH_NY_LLVM_COMPILER") {
|
|||
|
|
if !s.is_empty() { return PathBuf::from(s); }
|
|||
|
|
}
|
|||
|
|
if let Ok(p) = which::which("ny-llvmc") { return p; }
|
|||
|
|
PathBuf::from("target/release/ny-llvmc")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// Compile MIR(JSON v0) to an object file (.o) using ny-llvmc. Returns the output path.
|
|||
|
|
/// Fail‑Fast: prints stable tags and returns Err with the same message.
|
|||
|
|
pub fn mir_json_to_object(mir_json: &str, opts: Opts) -> Result<PathBuf, String> {
|
|||
|
|
// Basic shape check for MIR(JSON v0)
|
|||
|
|
if !mir_json.contains("\"functions\"") || !mir_json.contains("\"blocks\"") {
|
|||
|
|
let tag = "[llvmemit/input/invalid] missing functions/blocks keys";
|
|||
|
|
eprintln!("{}", tag);
|
|||
|
|
return Err(tag.into());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let ny_llvmc = resolve_ny_llvmc();
|
|||
|
|
if !ny_llvmc.exists() {
|
|||
|
|
let tag = format!("[llvmemit/ny-llvmc/not-found] path={}", ny_llvmc.display());
|
|||
|
|
eprintln!("{}", tag);
|
|||
|
|
return Err(tag);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Write MIR JSON to temp
|
|||
|
|
let tmp_dir = std::env::temp_dir();
|
|||
|
|
let in_path = tmp_dir.join("hako_llvm_in.json");
|
|||
|
|
{
|
|||
|
|
let mut f = fs::File::create(&in_path).map_err(|e| format!("[llvmemit/tmp/write-failed] {}", e))?;
|
|||
|
|
f.write_all(mir_json.as_bytes()).map_err(|e| format!("[llvmemit/tmp/write-failed] {}", e))?;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Output path
|
|||
|
|
let out_path = if let Some(p) = opts.out.clone() { p } else { tmp_dir.join("hako_llvm_out.o") };
|
|||
|
|
if let Some(parent) = out_path.parent() { let _ = fs::create_dir_all(parent); }
|
|||
|
|
|
|||
|
|
// Build command: ny-llvmc --in <json> --emit obj --out <out>
|
|||
|
|
let mut cmd = Command::new(&ny_llvmc);
|
|||
|
|
cmd.arg("--in").arg(&in_path)
|
|||
|
|
.arg("--emit").arg("obj")
|
|||
|
|
.arg("--out").arg(&out_path);
|
|||
|
|
if let Some(nyrt) = opts.nyrt.as_ref() { cmd.arg("--nyrt").arg(nyrt); }
|
|||
|
|
if let Some(level) = opts.opt_level.as_ref() {
|
|||
|
|
cmd.env("HAKO_LLVM_OPT_LEVEL", level);
|
|||
|
|
cmd.env("NYASH_LLVM_OPT_LEVEL", level);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let status = cmd.status().map_err(|e| format!("[llvmemit/spawn/error] {}", e))?;
|
|||
|
|
if !status.success() {
|
|||
|
|
let code = status.code().unwrap_or(1);
|
|||
|
|
let tag = format!("[llvmemit/ny-llvmc/failed status={}]", code);
|
|||
|
|
eprintln!("{}", tag);
|
|||
|
|
return Err(tag);
|
|||
|
|
}
|
|||
|
|
if !out_path.exists() {
|
|||
|
|
let tag = format!("[llvmemit/output/missing] {}", out_path.display());
|
|||
|
|
eprintln!("{}", tag);
|
|||
|
|
return Err(tag);
|
|||
|
|
}
|
|||
|
|
Ok(out_path)
|
|||
|
|
}
|
|||
|
|
|