//! code 模式转译：language + code → argv + 临时脚本文件。
//!
//! code 模式（`/v1/exec` 的 `code`/`language` 字段）由本模块转译为 argv，复用现有进程执行路径。
//! 走脚本文件而非 `-c`，避免多行 / 模块导入 / shebang 问题；脚本经 `tempfile` 写入 OS 临时目录，
//! 返回的 `CodeTranslation.script_path`（`TempPath`）由调用方持有至执行结束——drop 时自动删除。
//! 流式执行场景须把 `CodeTranslation` move 进 SSE 闭包，确保子进程 open 脚本前文件不被删。

use std::io::Write;

use tempfile::NamedTempFile;

/// code 模式转译结果：argv 用于执行，script_path 须由调用方持有至执行结束。
pub struct CodeTranslation {
    pub argv: Vec<String>,
    /// 临时脚本路径，持有至执行结束（RAII：drop 时自动删除）
    #[allow(dead_code)]
    pub script_path: tempfile::TempPath,
}

/// 语言 → (interpreter, 脚本扩展名)。P0 仅支持 python；node/java 留 P2。
pub fn interpreter_for(language: &str) -> Result<(&'static str, &'static str), String> {
    match language {
        "python" => Ok(("python3", "py")),
        other => Err(format!("不支持的语言: {other}（P0 仅支持 python）")),
    }
}

/// 写 code 到临时脚本文件，返回 argv 与 TempPath（调用方持有至执行结束）。
pub async fn translate(language: &str, code: &str) -> Result<CodeTranslation, String> {
    let (interp, ext) = interpreter_for(language)?;
    // NamedTempFile 是同步 IO，文件小、写一次，无需 tokio 化。
    let mut tmp = NamedTempFile::with_suffix(format!(".{ext}"))
        .map_err(|e| format!("创建临时脚本失败: {e}"))?;
    tmp.write_all(code.as_bytes())
        .map_err(|e| format!("写入临时脚本失败: {e}"))?;
    // detach 成 TempPath：文件留在磁盘，TempPath drop 时删除。
    let script_path = tmp.into_temp_path();
    let argv = vec![
        interp.to_string(),
        script_path.to_string_lossy().into_owned(),
    ];
    Ok(CodeTranslation { argv, script_path })
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn interpreter_for_python_ok_others_rejected() {
        assert_eq!(interpreter_for("python").unwrap(), ("python3", "py"));
        assert!(interpreter_for("node").is_err());
        assert!(interpreter_for("").is_err());
    }

    #[tokio::test]
    async fn translate_python_writes_script_and_returns_argv() {
        let code = "print(1+1)\nimport sys\nprint(sys.argv)";
        let trans = translate("python", code).await.unwrap();
        assert_eq!(trans.argv[0], "python3");
        assert!(trans.argv[1].ends_with(".py"));
        let on_disk = std::fs::read_to_string(&trans.argv[1]).unwrap();
        assert_eq!(on_disk, code);
    }

    #[tokio::test]
    async fn translate_unsupported_language_errors() {
        assert!(translate("ruby", "puts 1").await.is_err());
    }

    #[tokio::test]
    async fn translate_temp_path_keeps_file_until_drop() {
        let trans = translate("python", "x = 1").await.unwrap();
        let path = trans.argv[1].clone();
        assert!(std::path::Path::new(&path).exists(), "持有 TempPath 时文件应存在");
        drop(trans);
        assert!(!std::path::Path::new(&path).exists(), "TempPath drop 后文件应删除");
    }
}