dcr/src/core/builder/nasm.rs

use crate::core::builder::BuildContext;
use crate::platform;
use std::fs;
use std::path::{Path, PathBuf};
use std::process::Command;
use std::time::Instant;

pub fn build(ctx: &BuildContext) -> Result<f64, String> {
    if ctx.language.to_lowercase().as_str() != "asm" {
        return Err("NASM backend requires build.language = \"asm\"".to_string());
    }
    let assembler = if ctx.compiler.is_empty() {
        "nasm"
    } else {
        ctx.compiler
    };
    let start_time = Instant::now();
    let sources = collect_sources(ctx.language, ctx.exclude_dirs)?;
    let obj_dir = Path::new("./target").join(ctx.profile).join("obj");
    let objects = build_objects(assembler, &sources, &obj_dir, ctx, "o")?;

    if ctx.kind == "staticlib" {
        let lib_path = platform::lib_path(ctx.profile, ctx.project_name, ctx.target_dir);
        let archiver = ctx.archiver.unwrap_or(if cfg!(target_os = "windows") {
            "lib"
        } else {
            "ar"
        });
        let mut cmd = Command::new(archiver);
        if cfg!(target_os = "windows") && archiver == "lib" {
            cmd.arg("/nologo").arg(format!("/OUT:{lib_path}"));
        } else {
            cmd.arg("rcs").arg(&lib_path);
        }
        for obj in &objects {
            cmd.arg(obj);
        }
        match cmd.status() {
            Ok(status) if status.success() => {
                let elapsed = ((start_time.elapsed().as_secs_f64() * 100.0).trunc()) / 100.0;
                return Ok(elapsed);
            }
            Ok(_) => return Err("Build failed".to_string()),
            Err(err) => return Err(format!("Build failed: {err}")),
        }
    }

    let mut cmd = Command::new(ctx.linker.unwrap_or("cc"));
    if ctx.kind == "sharedlib" {
        if cfg!(target_os = "macos") {
            cmd.arg("-dynamiclib");
        } else {
            cmd.arg("-shared");
        }
    }
    for obj in &objects {
        cmd.arg(obj);
    }
    for dir in ctx.lib_dirs {
        cmd.arg(format!("-L{dir}"));
    }
    for lib in ctx.libs {
        cmd.arg(format!("-l{lib}"));
    }
    for flag in ctx.ldflags {
        cmd.arg(flag);
    }
    let out_path = if ctx.kind == "sharedlib" {
        platform::shared_lib_path(ctx.profile, ctx.project_name, ctx.target_dir)
    } else {
        platform::bin_path(ctx.profile, ctx.project_name, ctx.target_dir)
    };
    cmd.arg("-o").arg(out_path);

    match cmd.status() {
        Ok(status) if status.success() => {
            let elapsed = ((start_time.elapsed().as_secs_f64() * 100.0).trunc()) / 100.0;
            Ok(elapsed)
        }
        Ok(_) => Err("Build failed".to_string()),
        Err(err) => Err(format!("Build failed: {err}")),
    }
}

pub(crate) fn collect_sources(
    language: &str,
    exclude_dirs: &[PathBuf],
) -> Result<Vec<String>, String> {
    let lang = language.to_lowercase();
    let mut sources = Vec::new();
    collect_sources_rec("./src", &lang, &mut sources, exclude_dirs)?;
    sources.sort();
    if sources.is_empty() {
        return Err("No source files found in ./src".to_string());
    }
    Ok(sources)
}

fn collect_sources_rec(
    dir: &str,
    lang: &str,
    out: &mut Vec<String>,
    exclude_dirs: &[PathBuf],
) -> Result<(), String> {
    let dir_path = Path::new(dir);
    let full_dir = if dir_path.is_absolute() {
        dir_path.to_path_buf()
    } else {
        std::env::current_dir()
            .map_err(|err| format!("src dir error: {err}"))?
            .join(dir_path)
    };
    if is_excluded(&full_dir, exclude_dirs) {
        return Ok(());
    }
    let entries = fs::read_dir(&full_dir).map_err(|err| format!("src dir error: {err}"))?;
    for entry in entries {
        let entry = entry.map_err(|err| format!("src dir error: {err}"))?;
        let path = entry.path();
        if path.is_dir() {
            if is_excluded(&path, exclude_dirs) {
                continue;
            }
            collect_sources_rec(&path.to_string_lossy(), lang, out, exclude_dirs)?;
            continue;
        }
        if !path.is_file() {
            continue;
        }
        let ext_raw = path.extension().and_then(|v| v.to_str()).unwrap_or("");
        let ext = ext_raw.to_lowercase();
        let file = normalize_source_path(&path);
        let allowed = lang == "asm" && (ext == "asm" || (ext == "s" && ext_raw == "s"));
        if allowed {
            out.push(file);
        }
    }
    Ok(())
}

fn is_excluded(path: &Path, exclude_dirs: &[PathBuf]) -> bool {
    exclude_dirs.iter().any(|dir| path.starts_with(dir))
}

fn normalize_source_path(path: &Path) -> String {
    if !path.is_absolute() {
        return path.to_string_lossy().to_string();
    }
    if let Ok(base) = std::env::current_dir()
        && let Ok(rel) = path.strip_prefix(&base)
    {
        return format!("./{}", rel.to_string_lossy());
    }
    path.to_string_lossy().to_string()
}

fn build_objects(
    assembler: &str,
    sources: &[String],
    obj_dir: &Path,
    ctx: &BuildContext,
    obj_ext: &str,
) -> Result<Vec<String>, String> {
    let mut objects = Vec::new();
    let format = nasm_format(ctx.platform);
    for source in sources {
        let obj_path = object_path(obj_dir, source, obj_ext);
        if let Some(parent) = Path::new(&obj_path).parent() {
            fs::create_dir_all(parent).map_err(|err| format!("obj dir error: {err}"))?;
        }
        if needs_rebuild(source, &obj_path) {
            let mut cmd = Command::new(assembler);
            cmd.arg("-f")
                .arg(format)
                .arg(source)
                .arg("-o")
                .arg(&obj_path);
            for flag in ctx.cflags {
                cmd.arg(flag);
            }
            match cmd.status() {
                Ok(status) if status.success() => {}
                Ok(_) => return Err("Build failed".to_string()),
                Err(err) => return Err(format!("Build failed: {err}")),
            }
        }
        objects.push(obj_path);
    }
    Ok(objects)
}

fn nasm_format(platform: Option<&str>) -> &'static str {
    if let Some(p) = platform {
        let p = p.to_lowercase().replace('-', "_");
        if p == "x86" || (p.starts_with('i') && p.ends_with("86") && p.len() == 4) {
            #[cfg(target_os = "macos")]
            {
                return "macho32";
            }
            #[cfg(target_os = "windows")]
            {
                return "win32";
            }
            #[cfg(not(any(target_os = "macos", target_os = "windows")))]
            {
                return "elf32";
            }
        }
    }
    #[cfg(target_os = "linux")]
    {
        "elf64"
    }
    #[cfg(target_os = "macos")]
    {
        "macho64"
    }
    #[cfg(target_os = "windows")]
    {
        "win64"
    }
    #[cfg(not(any(target_os = "linux", target_os = "macos", target_os = "windows")))]
    {
        "elf64"
    }
}

fn object_path(obj_dir: &Path, source: &str, obj_ext: &str) -> String {
    let src_path = Path::new(source);
    let rel = src_path
        .strip_prefix("./src")
        .or_else(|_| src_path.strip_prefix("src"))
        .unwrap_or(src_path);
    let mut out = obj_dir.join(rel);
    out.set_extension(obj_ext.trim_start_matches('.'));
    out.to_string_lossy().to_string()
}

fn needs_rebuild(source: &str, object: &str) -> bool {
    let src_time = fs::metadata(source).and_then(|m| m.modified());
    let obj_time = fs::metadata(object).and_then(|m| m.modified());
    match (src_time, obj_time) {
        (Ok(s), Ok(o)) => s > o,
        (Ok(_), Err(_)) => true,
        _ => true,
    }
}