hakorune/src/runner/modes/common_util/pyvm.rs

// no extra imports needed

/// Run PyVM harness over a MIR module, returning the exit code
#[allow(dead_code)]
pub fn run_pyvm_harness(module: &crate::mir::MirModule, tag: &str) -> Result<i32, String> {
    let py3 = which::which("python3").map_err(|e| format!("python3 not found: {}", e))?;
    // Resolve runner path relative to CWD or NYASH_ROOT fallback
    let mut runner_buf = std::path::PathBuf::from("tools/pyvm_runner.py");
    if !runner_buf.exists() {
        if let Ok(root) = std::env::var("NYASH_ROOT") {
            let alt = std::path::Path::new(&root).join("tools/pyvm_runner.py");
            if alt.exists() { runner_buf = alt; }
        }
    }
    if !runner_buf.exists() {
        return Err(format!("PyVM runner not found: tools/pyvm_runner.py (cwd) or $NYASH_ROOT/tools/pyvm_runner.py"));
    }
    let tmp_dir = std::path::Path::new("tmp");
    let _ = std::fs::create_dir_all(tmp_dir);
    let mir_json_path = tmp_dir.join("nyash_pyvm_mir.json");
    crate::runner::mir_json_emit::emit_mir_json_for_harness_bin(module, &mir_json_path)
        .map_err(|e| format!("PyVM MIR JSON emit error: {}", e))?;
    crate::cli_v!("[ny-compiler] using PyVM ({} ) → {}", tag, mir_json_path.display());
    // Determine entry function (prefer Main.main; top-level main only if allowed)
    let allow_top = crate::config::env::entry_allow_toplevel_main();
    let entry = if module.functions.contains_key("Main.main") {
        "Main.main"
    } else if allow_top && module.functions.contains_key("main") {
        "main"
    } else if module.functions.contains_key("main") {
        eprintln!("[entry] Warning: using top-level 'main' without explicit allow; set NYASH_ENTRY_ALLOW_TOPLEVEL_MAIN=1 to silence.");
        "main"
    } else {
        "Main.main"
    };
    // Optional: Mini‑VM stdin loader — when enabled, read entire stdin and pass as argv[0]
    let mut cmd = std::process::Command::new(py3);
    crate::runner::child_env::apply_core_wrapper_env(&mut cmd);
    if std::env::var("NYASH_MINIVM_READ_STDIN").ok().as_deref() == Some("1") {
        use std::io::Read;
        let mut buf = String::new();
        let _ = std::io::stdin().read_to_string(&mut buf);
        // Wrap as argv JSON array [string]
        let arg_json = serde_json::json!([buf]).to_string();
        cmd.env("NYASH_SCRIPT_ARGS_JSON", arg_json);
    }
    let status = cmd
        .args([
            runner_buf.to_string_lossy().as_ref(),
            "--in",
            &mir_json_path.display().to_string(),
            "--entry",
            entry,
            "--args-env",
            "NYASH_SCRIPT_ARGS_JSON",
        ])
        .status()
        .map_err(|e| format!("spawn pyvm: {}", e))?;
    let code = status.code().unwrap_or(1);
    if !status.success() {
        crate::cli_v!("❌ PyVM ({}) failed (status={})", tag, code);
    }
    Ok(code)
}

/// Run PyVM harness over a nyash_rust (lib) MIR module, returning the exit code
#[allow(dead_code)]
pub fn run_pyvm_harness_lib(module: &nyash_rust::mir::MirModule, tag: &str) -> Result<i32, String> {
    let py3 = which::which("python3").map_err(|e| format!("python3 not found: {}", e))?;
    let mut runner_buf = std::path::PathBuf::from("tools/pyvm_runner.py");
    if !runner_buf.exists() {
        if let Ok(root) = std::env::var("NYASH_ROOT") {
            let alt = std::path::Path::new(&root).join("tools/pyvm_runner.py");
            if alt.exists() { runner_buf = alt; }
        }
    }
    if !runner_buf.exists() {
        return Err(format!("PyVM runner not found: tools/pyvm_runner.py (cwd) or $NYASH_ROOT/tools/pyvm_runner.py"));
    }
    let tmp_dir = std::path::Path::new("tmp");
    let _ = std::fs::create_dir_all(tmp_dir);
    let mir_json_path = tmp_dir.join("nyash_pyvm_mir.json");
    crate::runner::mir_json_emit::emit_mir_json_for_harness(module, &mir_json_path)
        .map_err(|e| format!("PyVM MIR JSON emit error: {}", e))?;
    crate::cli_v!("[Runner] using PyVM ({} ) → {}", tag, mir_json_path.display());
    // Determine entry function (prefer Main.main; top-level main only if allowed)
    let allow_top = crate::config::env::entry_allow_toplevel_main();
    let entry = if module.functions.contains_key("Main.main") {
        "Main.main"
    } else if allow_top && module.functions.contains_key("main") {
        "main"
    } else if module.functions.contains_key("main") {
        eprintln!("[entry] Warning: using top-level 'main' without explicit allow; set NYASH_ENTRY_ALLOW_TOPLEVEL_MAIN=1 to silence.");
        "main"
    } else {
        "Main.main"
    };
    let mut cmd = std::process::Command::new(py3);
    crate::runner::child_env::apply_core_wrapper_env(&mut cmd);
    if std::env::var("NYASH_MINIVM_READ_STDIN").ok().as_deref() == Some("1") {
        use std::io::Read;
        let mut buf = String::new();
        let _ = std::io::stdin().read_to_string(&mut buf);
        let arg_json = serde_json::json!([buf]).to_string();
        cmd.env("NYASH_SCRIPT_ARGS_JSON", arg_json);
    }
    let status = cmd
        .args([
            runner_buf.to_string_lossy().as_ref(),
            "--in",
            &mir_json_path.display().to_string(),
            "--entry",
            entry,
            "--args-env",
            "NYASH_SCRIPT_ARGS_JSON",
        ])
        .status()
        .map_err(|e| format!("spawn pyvm: {}", e))?;
    let code = status.code().unwrap_or(1);
    if !status.success() {
        crate::cli_v!("❌ PyVM ({}) failed (status={})", tag, code);
    }
    Ok(code)
}