rona 2.25.0

//! Git Status Operations
//!
//! Git status processing functionality using the git CLI for handling different
//! file states and contexts.

use std::{collections::HashSet, process::Command};

use crate::errors::{GitError, Result, RonaError};

/// Unquotes a git path.
///
/// When a path contains special characters (spaces, non-ASCII bytes, etc.),
/// git wraps it in double quotes and uses C-style escape sequences. This
/// function strips the surrounding quotes and unescapes the content.
fn unquote_git_path(path: &str) -> String {
    if path.starts_with('"') && path.ends_with('"') && path.len() >= 2 {
        let inner = &path[1..path.len() - 1];
        // Collect raw bytes so that multi-byte UTF-8 octal sequences (e.g. \303\242 -> â)
        // are decoded correctly at the end rather than being misinterpreted as Latin-1.
        let mut result: Vec<u8> = Vec::with_capacity(inner.len());
        let mut chars = inner.chars().peekable();
        while let Some(ch) = chars.next() {
            if ch != '\\' {
                let mut buf = [0u8; 4];
                result.extend_from_slice(ch.encode_utf8(&mut buf).as_bytes());
                continue;
            }
            match chars.next() {
                Some('\\') | None => result.push(b'\\'),
                Some('"') => result.push(b'"'),
                Some('n') => result.push(b'\n'),
                Some('t') => result.push(b'\t'),
                Some('r') => result.push(b'\r'),
                Some(c @ '0'..='7') => {
                    // Octal escape: up to 3 digits
                    let mut octal = String::from(c);
                    for _ in 0..2 {
                        match chars.peek() {
                            Some(&d) if d.is_ascii_digit() && d <= '7' => {
                                octal.push(d);
                                chars.next();
                            }
                            _ => break,
                        }
                    }
                    if let Ok(byte) = u8::from_str_radix(&octal, 8) {
                        result.push(byte);
                    }
                }
                Some(c) => {
                    result.push(b'\\');
                    let mut buf = [0u8; 4];
                    result.extend_from_slice(c.encode_utf8(&mut buf).as_bytes());
                }
            }
        }
        return String::from_utf8_lossy(&result).into_owned();
    }
    path.to_string()
}

/// Runs `git status --porcelain=v1` and returns the output lines.
///
/// Each line has the format `XY PATH` where X is the index status and Y is the
/// working-tree status. For renamed files, the path may include ` -> ` separating
/// the old and new names.
///
/// # Errors
/// * If the git command fails or we are not in a git repository
fn run_git_status() -> Result<Vec<String>> {
    let output = Command::new("git")
        .args(["status", "--porcelain=v1"])
        .output()
        .map_err(RonaError::Io)?;

    if output.status.success() {
        let stdout = String::from_utf8_lossy(&output.stdout);
        return Ok(stdout.lines().map(String::from).collect());
    }

    let stderr = String::from_utf8_lossy(&output.stderr);
    if stderr.to_lowercase().contains("not a git repository") {
        return Err(RonaError::Git(GitError::RepositoryNotFound));
    }

    Err(RonaError::Git(GitError::CommandFailed {
        command: "git status".to_string(),
        output: stderr.trim().to_string(),
    }))
}

/// Returns the new paths of all staged renamed files.
///
/// Uses `git diff --cached --name-status --diff-filter=R` which outputs lines like:
/// `R100\told_name\tnew_name`
///
/// # Errors
/// * If the git command fails
fn get_renamed_new_paths() -> Result<Vec<String>> {
    let output = Command::new("git")
        .args(["diff", "--cached", "--name-status", "--diff-filter=R"])
        .output()
        .map_err(RonaError::Io)?;

    if !output.status.success() {
        return Ok(Vec::new());
    }

    let stdout = String::from_utf8_lossy(&output.stdout);
    let paths = stdout
        .lines()
        .filter_map(|line| {
            let parts: Vec<&str> = line.splitn(3, '\t').collect();
            if parts.len() >= 3 {
                Some(parts[2].to_string())
            } else {
                None
            }
        })
        .collect();

    Ok(paths)
}

/// Returns a list of all files that appear in git status
/// (modified, untracked, staged - but not deleted)
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Vec<String>` - List of files from git status
pub fn get_status_files() -> Result<Vec<String>> {
    let lines = run_git_status()?;
    let mut files: HashSet<String> = HashSet::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let mut chars = line.chars();
        let index_char = chars.next().unwrap_or(' ');
        let wt_char = chars.next().unwrap_or(' ');
        let path = unquote_git_path(&line[3..]);

        // Skip index-deleted entries unless the working tree has modifications
        if index_char == 'D' && wt_char != 'M' && wt_char != '?' {
            continue;
        }

        // Skip working-tree-deleted files
        if wt_char == 'D' {
            continue;
        }

        // For renames, collect new paths separately below
        if index_char == 'R' {
            continue;
        }

        files.insert(path);
    }

    // Add new paths for renamed files
    for path in get_renamed_new_paths()? {
        files.insert(path);
    }

    Ok(files.into_iter().collect())
}

/// A single entry from `git status` that has unstaged changes and can be staged.
///
/// Used by the interactive add mode (`rona -a -i`) to present a `MultiSelect` of
/// changed files. The [`Display`] implementation renders a human-readable status
/// label followed by the path, e.g. `modified    src/main.rs`.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StatusEntry {
    /// Path to the file, relative to the repository root.
    pub path: String,
    /// Short, human-readable status label (e.g. "modified", "untracked").
    pub status: &'static str,
}

impl std::fmt::Display for StatusEntry {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{:<11} {}", self.status, self.path)
    }
}

/// Returns the files that currently have unstaged changes and can be staged.
///
/// This includes untracked files and files with working-tree modifications,
/// deletions or type changes. Fully staged files with no remaining working-tree
/// changes are omitted, since there is nothing left to stage for them.
///
/// The list is sorted by path for stable, predictable ordering in the selector.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Vec<StatusEntry>` - The stageable files with their status labels
pub fn get_stageable_files() -> Result<Vec<StatusEntry>> {
    let lines = run_git_status()?;
    let mut entries = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let mut chars = line.chars();
        let index_char = chars.next().unwrap_or(' ');
        let wt_char = chars.next().unwrap_or(' ');

        let is_untracked = index_char == '?' && wt_char == '?';

        // Skip files whose working tree matches the index (nothing left to stage).
        if wt_char == ' ' && !is_untracked {
            continue;
        }

        // For renamed-and-modified entries the path is "old -> new"; stage the new path.
        let raw_path = &line[3..];
        let path_part = raw_path.rsplit(" -> ").next().unwrap_or(raw_path);
        let path = unquote_git_path(path_part);

        let status = match wt_char {
            'D' => "deleted",
            'T' => "type change",
            '?' => "untracked",
            _ => "modified",
        };

        entries.push(StatusEntry { path, status });
    }

    entries.sort_by(|a, b| a.path.cmp(&b.path));
    Ok(entries)
}

/// Returns the files that currently have staged changes in the index.
///
/// Used by the interactive reset mode (`rona reset -i`) to present a `MultiSelect`
/// of staged files that can be unstaged. The status label reflects the staged
/// (index-side) change, e.g. "new file", "modified" or "deleted".
///
/// The list is sorted by path for stable, predictable ordering in the selector.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Vec<StatusEntry>` - The staged files with their status labels
pub fn get_staged_files() -> Result<Vec<StatusEntry>> {
    let lines = run_git_status()?;
    let mut entries = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let index_char = line.chars().next().unwrap_or(' ');

        // Skip entries with no staged change (unmodified or untracked index side).
        if index_char == ' ' || index_char == '?' {
            continue;
        }

        // For renamed/copied entries the path is "old -> new"; unstage the new path.
        let raw_path = &line[3..];
        let path_part = raw_path.rsplit(" -> ").next().unwrap_or(raw_path);
        let path = unquote_git_path(path_part);

        let status = match index_char {
            'A' => "new file",
            'D' => "deleted",
            'R' => "renamed",
            'C' => "copied",
            'T' => "type change",
            _ => "modified",
        };

        entries.push(StatusEntry { path, status });
    }

    entries.sort_by(|a, b| a.path.cmp(&b.path));
    Ok(entries)
}

/// Returns the tracked files that have working-tree changes which can be discarded.
///
/// Used by the interactive restore mode (`rona restore -i`) to present a
/// `MultiSelect` of files whose working-tree modifications, deletions or type
/// changes can be reverted with `git restore`. Untracked files are excluded
/// because `git restore` cannot act on them.
///
/// The list is sorted by path for stable, predictable ordering in the selector.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Vec<StatusEntry>` - The restorable files with their status labels
pub fn get_restorable_files() -> Result<Vec<StatusEntry>> {
    let lines = run_git_status()?;
    let mut entries = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let mut chars = line.chars();
        let index_char = chars.next().unwrap_or(' ');
        let wt_char = chars.next().unwrap_or(' ');

        // Only tracked files with working-tree changes can be restored.
        if index_char == '?' || !matches!(wt_char, 'M' | 'D' | 'T') {
            continue;
        }

        // For renamed-and-modified entries the path is "old -> new"; restore the new path.
        let raw_path = &line[3..];
        let path_part = raw_path.rsplit(" -> ").next().unwrap_or(raw_path);
        let path = unquote_git_path(path_part);

        let status = match wt_char {
            'D' => "deleted",
            'T' => "type change",
            _ => "modified",
        };

        entries.push(StatusEntry { path, status });
    }

    entries.sort_by(|a, b| a.path.cmp(&b.path));
    Ok(entries)
}

/// Processes deleted files that need to be staged for deletion.
/// Only returns files that are deleted in the working directory but not yet staged.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Result<Vec<String>>` - Files that need to be staged for deletion
pub fn process_deleted_files_for_staging() -> Result<Vec<String>> {
    let lines = run_git_status()?;
    let mut deleted_files = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let mut chars = line.chars();
        let index_char = chars.next().unwrap_or(' ');
        let wt_char = chars.next().unwrap_or(' ');
        let path = unquote_git_path(&line[3..]);

        // Working-tree deleted but NOT staged for deletion (index char != 'D')
        if wt_char == 'D' && index_char != 'D' {
            deleted_files.push(path);
        }
    }

    Ok(deleted_files)
}

/// Processes deleted files for commit message generation.
/// Returns all deleted files that are staged for deletion.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Result<Vec<String>>` - All deleted files for the commit message
pub fn process_deleted_files_for_commit_message() -> Result<Vec<String>> {
    let lines = run_git_status()?;
    let mut deleted_files = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let index_char = line.chars().next().unwrap_or(' ');
        let path = unquote_git_path(&line[3..]);

        // Index-deleted (staged deletion)
        if index_char == 'D' {
            deleted_files.push(path);
        }
    }

    Ok(deleted_files)
}

/// Processes the git status.
/// Returns the modified/added/renamed/type-changed files in the index,
/// to prepare the git commit message.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Result<Vec<String>>` - The modified/added files
pub fn process_git_status() -> Result<Vec<String>> {
    let lines = run_git_status()?;
    let mut files = Vec::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let index_char = line.chars().next().unwrap_or(' ');
        let path = unquote_git_path(&line[3..]);

        match index_char {
            'M' | 'A' | 'T' => files.push(path),
            _ => {} // 'R' (renamed) files are collected separately below; skip all others
        }
    }

    // Add new paths for renamed files
    files.extend(get_renamed_new_paths()?);

    Ok(files)
}

/// Returns all file paths currently staged in the index.
///
/// Used after `git add -A` to discover which staged files should be unstaged
/// based on exclude patterns.
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Result<Vec<String>>` - All staged file paths
pub fn get_all_staged_file_paths() -> Result<Vec<String>> {
    let lines = run_git_status()?;
    let mut files: HashSet<String> = HashSet::new();

    for line in &lines {
        if line.len() < 4 {
            continue;
        }

        let mut chars = line.chars();
        let index_char = chars.next().unwrap_or(' ');

        // Skip untracked and purely unstaged entries
        if index_char == ' ' || index_char == '?' {
            continue;
        }

        // Renames are handled separately to get the new path
        if index_char == 'R' {
            continue;
        }

        let path = unquote_git_path(&line[3..]);
        files.insert(path);
    }

    // Add new paths for renamed files
    for path in get_renamed_new_paths()? {
        files.insert(path);
    }

    Ok(files.into_iter().collect())
}

/// Counts the number of renamed files in the git status.
///
/// This function helps with accurate file counting since renamed files appear
/// as 2 lines in `git diff --cached --numstat` (one deletion, one addition).
///
/// # Errors
/// * If reading git status fails
///
/// # Returns
/// * `Result<usize>` - The count of renamed files
pub fn count_renamed_files() -> Result<usize> {
    let lines = run_git_status()?;
    let count = lines
        .iter()
        .filter(|line| !line.is_empty() && line.starts_with('R'))
        .count();
    Ok(count)
}

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

    #[test]
    fn test_unquote_plain_path() {
        assert_eq!(unquote_git_path("src/main.rs"), "src/main.rs");
    }

    #[test]
    fn test_unquote_quoted_path_with_spaces() {
        assert_eq!(
            unquote_git_path("\"assets/foo bar/file.txt\""),
            "assets/foo bar/file.txt"
        );
    }

    #[test]
    fn test_unquote_escape_sequences() {
        assert_eq!(unquote_git_path("\"a\\\\b\""), "a\\b");
        assert_eq!(unquote_git_path("\"a\\\"b\""), "a\"b");
        assert_eq!(unquote_git_path("\"a\\nb\""), "a\nb");
    }

    #[test]
    fn test_unquote_octal_escape() {
        // Space is octal 040
        assert_eq!(unquote_git_path("\"a\\040b\""), "a b");
    }

    #[test]
    fn test_unquote_multibyte_utf8_octal() {
        // â is U+00E2, encoded in UTF-8 as 0xC3 0xA2 (octal \303\242)
        // git quotes filenames like "Marags\303\242-Display.otf"
        assert_eq!(
            unquote_git_path("\"Marags\\303\\242-Display.otf\""),
            "Maragsâ-Display.otf"
        );
    }
}