armybox 0.3.0

//! pwd - print name of current/working directory
//!
//! POSIX.1-2017 compliant implementation.
//! Reference: https://pubs.opengroup.org/onlinepubs/9699919799/utilities/pwd.html

use crate::io;
use crate::sys;
use crate::applets::get_arg;

/// pwd - print name of current/working directory
///
/// # Synopsis
/// ```text
/// pwd [-L|-P]
/// ```
///
/// # Description
/// Prints the absolute pathname of the current working directory.
///
/// # Options
/// - `-L`: Use PWD from environment (logical path)
/// - `-P`: Print physical path without symlinks (default)
///
/// # Exit Status
/// - 0: Success
/// - >0: An error occurred
pub fn pwd(argc: i32, argv: *const *const u8) -> i32 {
    let mut logical = false;

    // Parse options - last one wins per POSIX
    for i in 1..argc {
        if let Some(arg) = unsafe { get_arg(argv, i) } {
            if arg == b"-L" {
                logical = true;
            } else if arg == b"-P" {
                logical = false;
            }
        }
    }

    if logical {
        // Try PWD environment variable first
        if let Some(pwd) = io::getenv(b"PWD") {
            // Verify it starts with / and refers to the same directory
            if !pwd.is_empty() && pwd[0] == b'/' {
                // Verify PWD points to the same inode as "."
                let mut st_pwd: libc::stat = unsafe { core::mem::zeroed() };
                let mut st_dot: libc::stat = unsafe { core::mem::zeroed() };
                if io::stat(pwd, &mut st_pwd) == 0
                    && io::stat(b".", &mut st_dot) == 0
                    && st_pwd.st_dev == st_dot.st_dev
                    && st_pwd.st_ino == st_dot.st_ino
                {
                    io::write_all(1, pwd);
                    io::write_str(1, b"\n");
                    return 0;
                }
            }
        }
        // Fall through to physical if PWD is unset or invalid
    }

    let mut buf = [0u8; 4096];
    let ret = unsafe { libc::getcwd(buf.as_mut_ptr() as *mut i8, buf.len()) };
    if !ret.is_null() {
        io::write_all(1, &buf[..io::strlen_arr(&buf)]);
        io::write_str(1, b"\n");
        0
    } else {
        sys::perror(b"getcwd");
        1
    }
}

#[cfg(test)]
mod tests {
    extern crate std;
    use std::sync::atomic::{AtomicUsize, Ordering};

    static TEST_COUNTER: AtomicUsize = AtomicUsize::new(0);
    use std::process::Command;
    use std::fs;
    use std::path::PathBuf;

    fn get_armybox_path() -> PathBuf {
        if let Ok(path) = std::env::var("ARMYBOX_PATH") {
            return PathBuf::from(path);
        }
        // Use absolute paths since tests may change current_dir
        let manifest_dir = std::env::var("CARGO_MANIFEST_DIR")
            .map(PathBuf::from)
            .unwrap_or_else(|_| std::env::current_dir().unwrap());
        let release = manifest_dir.join("target/release/armybox");
        if release.exists() { return release; }
        manifest_dir.join("target/debug/armybox")
    }

    fn setup() -> PathBuf {
        let counter = TEST_COUNTER.fetch_add(1, Ordering::SeqCst);
        let dir = std::env::temp_dir().join(format!("armybox_pwd_test_{}_{}", std::process::id(), counter));
        let _ = fs::remove_dir_all(&dir);
        fs::create_dir_all(&dir).unwrap();
        dir
    }

    fn cleanup(dir: &std::path::Path) {
        let _ = fs::remove_dir_all(dir);
    }

    #[test]
    fn test_pwd_basic() {
        let armybox = get_armybox_path();
        if !armybox.exists() { return; }

        let dir = setup();

        let output = Command::new(&armybox)
            .args(["pwd"])
            .current_dir(&dir)
            .output()
            .unwrap();

        assert_eq!(output.status.code(), Some(0));
        let stdout = std::string::String::from_utf8_lossy(&output.stdout);
        // Should contain the temp dir path
        assert!(stdout.trim().len() > 0);
        cleanup(&dir);
    }

    #[test]
    fn test_pwd_output_ends_with_newline() {
        let armybox = get_armybox_path();
        if !armybox.exists() { return; }

        let output = Command::new(&armybox)
            .args(["pwd"])
            .output()
            .unwrap();

        assert_eq!(output.status.code(), Some(0));
        let stdout = std::string::String::from_utf8_lossy(&output.stdout);
        assert!(stdout.ends_with('\n'));
    }

    #[test]
    fn test_pwd_absolute_path() {
        let armybox = get_armybox_path();
        if !armybox.exists() { return; }

        let output = Command::new(&armybox)
            .args(["pwd"])
            .output()
            .unwrap();

        assert_eq!(output.status.code(), Some(0));
        let stdout = std::string::String::from_utf8_lossy(&output.stdout);
        // Absolute path should start with /
        assert!(stdout.trim().starts_with('/'));
    }

    #[test]
    fn test_pwd_physical_flag() {
        let armybox = get_armybox_path();
        if !armybox.exists() { return; }

        let output = Command::new(&armybox)
            .args(["pwd", "-P"])
            .output()
            .unwrap();

        assert_eq!(output.status.code(), Some(0));
        let stdout = std::string::String::from_utf8_lossy(&output.stdout);
        assert!(stdout.trim().starts_with('/'));
    }

    #[test]
    fn test_pwd_logical_flag() {
        let armybox = get_armybox_path();
        if !armybox.exists() { return; }

        let dir = setup();

        // Set PWD to the test dir and run with -L
        let output = Command::new(&armybox)
            .args(["pwd", "-L"])
            .current_dir(&dir)
            .env("PWD", dir.to_str().unwrap())
            .output()
            .unwrap();

        assert_eq!(output.status.code(), Some(0));
        let stdout = std::string::String::from_utf8_lossy(&output.stdout);
        assert!(stdout.trim().len() > 0);
        cleanup(&dir);
    }
}