double_o/

commands_patterns.rs

//! Pattern listing functionality for `oo patterns` commands.

use std::io::{self, Write};
use std::path::Path;

use crate::{init, learn, pattern};

/// Line printed under the `User (<dir>):` header when no user pattern is listable.
const EMPTY_USER_PATTERNS_LINE: &str = "  no learned patterns yet";

/// Write a single pattern entry with flags.
///
/// Helper function to reduce duplication in pattern listing code.
/// Writes the command_match with optional success/failure flags.
fn write_pattern_entry<W: Write>(
    out: &mut W,
    cmd_match: &str,
    has_success: bool,
    has_failure: bool,
) -> io::Result<()> {
    if has_success || has_failure {
        let mut flags = Vec::new();
        if has_success {
            flags.push("success");
        }
        if has_failure {
            flags.push("failure");
        }
        writeln!(out, "  {cmd_match}  [{}]", flags.join("] ["))
    } else {
        writeln!(out, "  {cmd_match}")
    }
}

/// Prefix an I/O error with the path that failed, as `oo: <path>: <error>` reads.
fn with_path(path: &Path, e: &io::Error) -> io::Error {
    io::Error::new(e.kind(), format!("{}: {e}", path.display()))
}

/// Write pattern entries from a single directory, returning true if any were found.
///
/// Each line is written with a two-space indent so callers can add section headers.
/// A missing directory, or one with no parseable `.toml` files, writes nothing and
/// returns false. Any other read failure (permission denied, I/O error) is returned
/// as an error naming the path, so an unreadable directory is never reported as empty.
fn write_patterns_in<W: Write>(out: &mut W, dir: &Path) -> io::Result<bool> {
    let entries = match std::fs::read_dir(dir) {
        Ok(e) => e,
        Err(e) if e.kind() == io::ErrorKind::NotFound => return Ok(false),
        Err(e) => return Err(with_path(dir, &e)),
    };

    let mut found = false;
    for entry in entries {
        let path = entry.map_err(|e| with_path(dir, &e))?.path();
        if path.extension().and_then(|e| e.to_str()) != Some("toml") {
            continue;
        }
        // Only regular files are pattern files. Reading a directory fails and a FIFO would
        // block forever, so both are skipped. Metadata follows symlinks, so a dangling one is
        // NotFound and skipped; any other stat failure (e.g. EACCES on an unsearchable dir)
        // must surface rather than look like an empty directory.
        let is_file = match pattern::is_pattern_file(&path) {
            Ok(b) => b,
            Err(e) => return Err(with_path(&path, &e)),
        };
        if !is_file {
            continue;
        }
        let text = match std::fs::read_to_string(&path) {
            Ok(text) => text,
            // Not UTF-8 means it is not a pattern file; a file removed since listing is gone.
            Err(e)
                if matches!(
                    e.kind(),
                    io::ErrorKind::InvalidData | io::ErrorKind::NotFound
                ) =>
            {
                continue;
            }
            Err(e) => return Err(with_path(&path, &e)),
        };
        let Ok(parsed) = toml::from_str::<toml::Value>(&text) else {
            continue;
        };
        found = true;

        let cmd_match = parsed
            .get("command_match")
            .and_then(|v| v.as_str())
            .unwrap_or("(unknown)");
        let has_success = parsed.get("success").is_some();
        let has_failure = parsed.get("failure").is_some();

        write_pattern_entry(out, cmd_match, has_success, has_failure)?;
    }
    Ok(found)
}

/// Print pattern entries from a single directory, returning true if any were found.
///
/// Each line is printed with a two-space indent so callers can add section headers.
/// A read or write failure is printed to stderr and returns false, so callers cannot tell it
/// apart from an empty directory.
pub fn list_patterns_in(dir: &Path) -> bool {
    match write_patterns_in(&mut io::stdout().lock(), dir) {
        Ok(found) => found,
        Err(e) => {
            eprintln!("oo: {e}");
            false
        }
    }
}

/// List learned pattern files from a single directory (legacy test helper).
///
/// Uses the same rendering as the `User` section of `oo patterns`: the empty-state
/// line when nothing is listed. Returns 0 on success, and 1 with `oo: <path>: <error>`
/// on stderr when the directory cannot be read.
pub fn cmd_patterns_in(dir: &Path) -> i32 {
    let result = write_patterns_in(&mut io::stdout().lock(), dir);
    match result {
        Ok(true) => 0,
        Ok(false) => {
            println!("{EMPTY_USER_PATTERNS_LINE}");
            0
        }
        Err(e) => {
            eprintln!("oo: {e}");
            1
        }
    }
}

/// Write the `User (<dir>):` section, with the empty-state line when nothing is listed.
fn write_user_section<W: Write>(out: &mut W, dir: &Path) -> io::Result<()> {
    writeln!(out, "User ({}):", dir.display())?;
    if !write_patterns_in(out, dir)? {
        writeln!(out, "{EMPTY_USER_PATTERNS_LINE}")?;
    }
    Ok(())
}

/// Write the built-in patterns.
fn write_builtins<W: Write>(out: &mut W) -> io::Result<()> {
    for pat in pattern::builtins() {
        let cmd_match = pat.command_match.as_str();
        let has_success = pat.success.is_some();
        let has_failure = pat.failure.is_some();

        write_pattern_entry(out, cmd_match, has_success, has_failure)?;
    }
    Ok(())
}

/// Write the full `oo patterns` listing.
fn write_all_patterns<W: Write>(out: &mut W) -> io::Result<()> {
    let project_dir = std::env::current_dir()
        .map(|cwd| init::project_patterns_dir(&cwd))
        .ok();
    let user_dir = learn::patterns_dir();

    // Show built-in patterns first
    writeln!(out, "Built-in ({} patterns):", pattern::builtins().len())?;
    write_builtins(out)?;
    writeln!(out)?;

    if let Some(ref pdir) = project_dir {
        if pdir.exists() {
            writeln!(out, "Project ({}):", pdir.display())?;
            write_patterns_in(out, pdir)?;
            writeln!(out)?;
        }
    }

    write_user_section(out, &user_dir)
}

/// List patterns from both project-local and user config directories.
pub fn cmd_patterns() -> i32 {
    match write_all_patterns(&mut io::stdout().lock()) {
        Ok(()) => 0,
        Err(e) => {
            eprintln!("oo: {e}");
            1
        }
    }
}

#[cfg(test)]
#[path = "patterns_empty_tests.rs"]
mod tests;