ftui_widgets/

file_picker.rs

#![forbid(unsafe_code)]

//! File picker widget for browsing and selecting files.
//!
//! Provides a TUI file browser with keyboard navigation. The widget
//! renders a directory listing with cursor selection and supports
//! entering subdirectories and navigating back to parents.
//!
//! # Architecture
//!
//! - [`FilePicker`] — stateless configuration and rendering
//! - [`FilePickerState`] — mutable navigation state (cursor, directory, entries)
//! - [`DirEntry`] — a single file/directory entry
//!
//! The widget uses [`StatefulWidget`] so the application owns the state
//! and can read the selected path.

use crate::{StatefulWidget, clear_text_area, clear_text_row, draw_text_span};
use ftui_core::geometry::Rect;
use ftui_render::frame::Frame;
use ftui_style::Style;
use std::{
    io,
    path::{Path, PathBuf},
};

/// A single entry in a directory listing.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct DirEntry {
    /// Display name.
    pub name: String,
    /// Full path.
    pub path: PathBuf,
    /// Whether this is a directory.
    pub is_dir: bool,
}

impl DirEntry {
    /// Create a directory entry.
    pub fn dir(name: impl Into<String>, path: impl Into<PathBuf>) -> Self {
        Self {
            name: name.into(),
            path: path.into(),
            is_dir: true,
        }
    }

    /// Create a file entry.
    pub fn file(name: impl Into<String>, path: impl Into<PathBuf>) -> Self {
        Self {
            name: name.into(),
            path: path.into(),
            is_dir: false,
        }
    }
}

/// Mutable state for the file picker.
#[derive(Debug, Clone)]
pub struct FilePickerState {
    /// Current directory being displayed.
    pub current_dir: PathBuf,
    /// Root directory for confinement (if set, cannot navigate above this).
    pub root: Option<PathBuf>,
    /// Directory entries (sorted: dirs first, then files).
    pub entries: Vec<DirEntry>,
    /// Currently highlighted index.
    pub cursor: usize,
    /// Scroll offset (first visible row).
    pub offset: usize,
    /// The selected/confirmed path (set when user presses enter on a file).
    pub selected: Option<PathBuf>,
    /// Navigation history for going back.
    history: Vec<(PathBuf, usize)>,
}

impl FilePickerState {
    /// Create a new state with the given directory and entries.
    pub fn new(current_dir: PathBuf, entries: Vec<DirEntry>) -> Self {
        Self {
            current_dir,
            root: None,
            entries,
            cursor: 0,
            offset: 0,
            selected: None,
            history: Vec::new(),
        }
    }

    /// Set a root directory to confine navigation.
    ///
    /// When set, the user cannot navigate to a parent directory above this root.
    #[must_use]
    pub fn with_root(mut self, root: impl Into<PathBuf>) -> Self {
        self.root = Some(root.into());
        self
    }

    /// Create state from a directory path by reading the filesystem.
    ///
    /// Sorts entries: directories first (alphabetical), then files (alphabetical).
    /// Returns an error if the directory cannot be read.
    pub fn from_path(path: impl AsRef<Path>) -> std::io::Result<Self> {
        let path = path.as_ref().to_path_buf();
        let entries = read_directory(&path)?;
        Ok(Self::new(path, entries))
    }

    /// Move cursor up.
    pub fn cursor_up(&mut self) {
        if self.cursor > 0 {
            self.cursor -= 1;
        }
    }

    /// Move cursor down.
    pub fn cursor_down(&mut self) {
        if !self.entries.is_empty() && self.cursor < self.entries.len() - 1 {
            self.cursor += 1;
        }
    }

    /// Move cursor to the first entry.
    pub fn cursor_home(&mut self) {
        self.cursor = 0;
    }

    /// Move cursor to the last entry.
    pub fn cursor_end(&mut self) {
        if !self.entries.is_empty() {
            self.cursor = self.entries.len() - 1;
        }
    }

    /// Page up by `page_size` rows.
    pub fn page_up(&mut self, page_size: usize) {
        self.cursor = self.cursor.saturating_sub(page_size);
    }

    /// Page down by `page_size` rows.
    pub fn page_down(&mut self, page_size: usize) {
        if !self.entries.is_empty() {
            self.cursor = (self.cursor + page_size).min(self.entries.len() - 1);
        }
    }

    /// Enter the selected directory (if cursor is on a directory).
    ///
    /// Returns `Ok(true)` if navigation succeeded, `Ok(false)` if cursor is on a file,
    /// or an error if the directory cannot be read.
    pub fn enter(&mut self) -> std::io::Result<bool> {
        let Some(entry) = self.entries.get(self.cursor) else {
            return Ok(false);
        };

        if !entry.is_dir {
            if let Some(root) = &self.root {
                ensure_path_within_root(
                    &entry.path,
                    root,
                    "Cannot select a file outside root directory",
                )?;
            }
            self.selected = Some(entry.path.clone());
            return Ok(false);
        }

        let new_dir = entry.path.clone();

        if let Some(root) = &self.root {
            ensure_path_within_root(
                &new_dir,
                root,
                "Cannot traverse outside root directory via symlink",
            )?;
        }

        let new_entries = read_directory(&new_dir)?;

        self.history.push((self.current_dir.clone(), self.cursor));
        self.current_dir = new_dir;
        self.entries = new_entries;
        self.cursor = 0;
        self.offset = 0;
        Ok(true)
    }

    /// Go back to the parent directory.
    ///
    /// Returns `Ok(true)` if navigation succeeded.
    pub fn go_back(&mut self) -> std::io::Result<bool> {
        // If root is set, prevent going above it using canonicalized paths
        if let Some(root) = &self.root {
            let (resolved_curr, resolved_root) = canonicalize_candidate_and_root(
                &self.current_dir,
                root,
                "current file picker directory",
            )?;
            if resolved_curr == resolved_root || !resolved_curr.starts_with(&resolved_root) {
                return Ok(false);
            }
        }

        if let Some((prev_dir, prev_cursor)) = self.history.last().cloned() {
            if let Some(root) = &self.root {
                ensure_path_within_root(
                    &prev_dir,
                    root,
                    "Cannot restore directory outside root directory",
                )?;
            }
            self.history.pop();
            let entries = read_directory(&prev_dir)?;
            self.current_dir = prev_dir;
            self.entries = entries;
            self.cursor = prev_cursor.min(self.entries.len().saturating_sub(1));
            self.offset = 0;
            return Ok(true);
        }

        // No history — try parent directory
        if let Some(parent) = self.current_dir.parent().map(|p| p.to_path_buf()) {
            if let Some(root) = &self.root
                && !path_is_within_root(&parent, root)?
            {
                return Ok(false); // Block parent traversal outside root
            }

            let entries = read_directory(&parent)?;
            self.current_dir = parent;
            self.entries = entries;
            self.cursor = 0;
            self.offset = 0;
            return Ok(true);
        }

        Ok(false)
    }

    /// Ensure scroll offset keeps cursor visible for the given viewport height.
    fn adjust_scroll(&mut self, visible_rows: usize) {
        if visible_rows == 0 {
            return;
        }
        if self.cursor < self.offset {
            self.offset = self.cursor;
        }
        if self.cursor >= self.offset + visible_rows {
            self.offset = self.cursor + 1 - visible_rows;
        }
    }
}

fn canonicalize_for_confinement(path: &Path, label: &str) -> io::Result<PathBuf> {
    std::fs::canonicalize(path).map_err(|error| {
        io::Error::new(
            error.kind(),
            format!("Cannot resolve {label} {}: {error}", path.display()),
        )
    })
}

fn canonicalize_candidate_and_root(
    candidate: &Path,
    root: &Path,
    label: &str,
) -> io::Result<(PathBuf, PathBuf)> {
    let resolved_candidate = canonicalize_for_confinement(candidate, label)?;
    let resolved_root = canonicalize_for_confinement(root, "file picker root")?;
    Ok((resolved_candidate, resolved_root))
}

fn path_is_within_root(candidate: &Path, root: &Path) -> io::Result<bool> {
    let (resolved_candidate, resolved_root) =
        canonicalize_candidate_and_root(candidate, root, "file picker path")?;
    Ok(resolved_candidate.starts_with(resolved_root))
}

fn ensure_path_within_root(candidate: &Path, root: &Path, message: &str) -> io::Result<()> {
    if path_is_within_root(candidate, root)? {
        return Ok(());
    }

    Err(io::Error::new(io::ErrorKind::PermissionDenied, message))
}

/// Read a directory and return sorted entries (dirs first, then files).
fn read_directory(path: &Path) -> std::io::Result<Vec<DirEntry>> {
    let mut dirs = Vec::new();
    let mut files = Vec::new();

    for entry in std::fs::read_dir(path)? {
        let entry = entry?;
        let name = entry.file_name().to_string_lossy().to_string();
        let mut file_type = entry.file_type()?;
        let full_path = entry.path();

        // If it's a symlink, check what it points to
        if file_type.is_symlink()
            && let Ok(metadata) = std::fs::metadata(&full_path)
        {
            file_type = metadata.file_type();
        }

        if file_type.is_dir() {
            dirs.push(DirEntry::dir(name, full_path));
        } else {
            files.push(DirEntry::file(name, full_path));
        }
    }

    dirs.sort_by_key(|a| a.name.to_lowercase());
    files.sort_by_key(|a| a.name.to_lowercase());

    dirs.extend(files);
    Ok(dirs)
}

/// Configuration and rendering for the file picker widget.
///
/// # Example
///
/// ```ignore
/// let picker = FilePicker::new()
///     .dir_style(Style::new().fg(PackedRgba::rgb(100, 100, 255)))
///     .cursor_style(Style::new().bold());
///
/// let mut state = FilePickerState::from_path(".").unwrap();
/// picker.render(area, &mut frame, &mut state);
/// ```
#[derive(Debug, Clone)]
pub struct FilePicker {
    /// Style for directory entries.
    pub dir_style: Style,
    /// Style for file entries.
    pub file_style: Style,
    /// Style for the cursor row.
    pub cursor_style: Style,
    /// Style for the header (current directory).
    pub header_style: Style,
    /// Whether to show the current directory path as a header.
    pub show_header: bool,
    /// Prefix for directory entries.
    pub dir_prefix: &'static str,
    /// Prefix for file entries.
    pub file_prefix: &'static str,
}

impl Default for FilePicker {
    fn default() -> Self {
        Self {
            dir_style: Style::default(),
            file_style: Style::default(),
            cursor_style: Style::default(),
            header_style: Style::default(),
            show_header: true,
            dir_prefix: "📁 ",
            file_prefix: "  ",
        }
    }
}

impl FilePicker {
    /// Create a new file picker with default styles.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the directory entry style.
    #[must_use]
    pub fn dir_style(mut self, style: Style) -> Self {
        self.dir_style = style;
        self
    }

    /// Set the file entry style.
    #[must_use]
    pub fn file_style(mut self, style: Style) -> Self {
        self.file_style = style;
        self
    }

    /// Set the cursor (highlight) style.
    #[must_use]
    pub fn cursor_style(mut self, style: Style) -> Self {
        self.cursor_style = style;
        self
    }

    /// Set the header style.
    #[must_use]
    pub fn header_style(mut self, style: Style) -> Self {
        self.header_style = style;
        self
    }

    /// Toggle header display.
    #[must_use]
    pub fn show_header(mut self, show: bool) -> Self {
        self.show_header = show;
        self
    }
}

impl StatefulWidget for FilePicker {
    type State = FilePickerState;

    fn render(&self, area: Rect, frame: &mut Frame, state: &mut Self::State) {
        if area.is_empty() {
            return;
        }

        let deg = frame.buffer.degradation;
        if !deg.render_content() {
            return;
        }

        clear_text_area(frame, area, Style::default());

        let header_style = if deg.apply_styling() {
            self.header_style
        } else {
            Style::default()
        };
        let dir_style = if deg.apply_styling() {
            self.dir_style
        } else {
            Style::default()
        };
        let file_style = if deg.apply_styling() {
            self.file_style
        } else {
            Style::default()
        };
        let cursor_style = if deg.apply_styling() {
            self.cursor_style
        } else {
            Style::default()
        };

        let mut y = area.y;
        let max_y = area.bottom();

        // Header: current directory path
        if self.show_header && y < max_y {
            clear_text_row(frame, Rect::new(area.x, y, area.width, 1), header_style);
            let header = state.current_dir.to_string_lossy();
            draw_text_span(frame, area.x, y, &header, header_style, area.right());
            y += 1;
        }

        if y >= max_y {
            return;
        }

        let visible_rows = (max_y - y) as usize;
        state.adjust_scroll(visible_rows);

        if state.entries.is_empty() {
            clear_text_row(frame, Rect::new(area.x, y, area.width, 1), file_style);
            draw_text_span(
                frame,
                area.x,
                y,
                "(empty directory)",
                file_style,
                area.right(),
            );
            return;
        }

        let end_idx = (state.offset + visible_rows).min(state.entries.len());
        for (i, entry) in state.entries[state.offset..end_idx].iter().enumerate() {
            if y >= max_y {
                break;
            }

            let actual_idx = state.offset + i;
            let is_cursor = actual_idx == state.cursor;

            let prefix = if entry.is_dir {
                self.dir_prefix
            } else {
                self.file_prefix
            };

            let base_style = if entry.is_dir { dir_style } else { file_style };

            let style = if is_cursor {
                cursor_style.merge(&base_style)
            } else {
                base_style
            };

            clear_text_row(frame, Rect::new(area.x, y, area.width, 1), style);

            // Draw cursor indicator
            let mut x = area.x;
            if is_cursor {
                draw_text_span(frame, x, y, "> ", cursor_style, area.right());
                x = x.saturating_add(2);
            } else {
                x = x.saturating_add(2);
            }

            // Draw prefix + name
            x = draw_text_span(frame, x, y, prefix, style, area.right());
            draw_text_span(frame, x, y, &entry.name, style, area.right());

            y += 1;
        }
    }
}

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

    fn buf_to_lines(buf: &ftui_render::buffer::Buffer) -> Vec<String> {
        let mut lines = Vec::new();
        for y in 0..buf.height() {
            let mut row = String::with_capacity(buf.width() as usize);
            for x in 0..buf.width() {
                let ch = buf
                    .get(x, y)
                    .and_then(|c| c.content.as_char())
                    .unwrap_or(' ');
                row.push(ch);
            }
            lines.push(row);
        }
        lines
    }

    fn make_entries() -> Vec<DirEntry> {
        vec![
            DirEntry::dir("docs", "/tmp/docs"),
            DirEntry::dir("src", "/tmp/src"),
            DirEntry::file("README.md", "/tmp/README.md"),
            DirEntry::file("main.rs", "/tmp/main.rs"),
        ]
    }

    fn make_state() -> FilePickerState {
        FilePickerState::new(PathBuf::from("/tmp"), make_entries())
    }

    #[test]
    fn dir_entry_constructors() {
        let d = DirEntry::dir("src", "/src");
        assert!(d.is_dir);
        assert_eq!(d.name, "src");

        let f = DirEntry::file("main.rs", "/main.rs");
        assert!(!f.is_dir);
        assert_eq!(f.name, "main.rs");
    }

    #[test]
    fn state_cursor_movement() {
        let mut state = make_state();
        assert_eq!(state.cursor, 0);

        state.cursor_down();
        assert_eq!(state.cursor, 1);

        state.cursor_down();
        state.cursor_down();
        assert_eq!(state.cursor, 3);

        // Can't go past end
        state.cursor_down();
        assert_eq!(state.cursor, 3);

        state.cursor_up();
        assert_eq!(state.cursor, 2);

        state.cursor_home();
        assert_eq!(state.cursor, 0);

        // Can't go before start
        state.cursor_up();
        assert_eq!(state.cursor, 0);

        state.cursor_end();
        assert_eq!(state.cursor, 3);
    }

    #[test]
    fn state_page_navigation() {
        let entries: Vec<DirEntry> = (0..20)
            .map(|i| DirEntry::file(format!("file{i}.txt"), format!("/tmp/file{i}.txt")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/tmp"), entries);

        state.page_down(5);
        assert_eq!(state.cursor, 5);

        state.page_down(5);
        assert_eq!(state.cursor, 10);

        state.page_up(3);
        assert_eq!(state.cursor, 7);

        state.page_up(100);
        assert_eq!(state.cursor, 0);

        state.page_down(100);
        assert_eq!(state.cursor, 19);
    }

    #[test]
    fn state_empty_entries() {
        let mut state = FilePickerState::new(PathBuf::from("/tmp"), vec![]);
        state.cursor_down(); // should not panic
        state.cursor_up();
        state.cursor_end();
        state.cursor_home();
        state.page_down(10);
        state.page_up(10);
        assert_eq!(state.cursor, 0);
    }

    #[test]
    fn adjust_scroll_keeps_cursor_visible() {
        let entries: Vec<DirEntry> = (0..20)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);

        state.cursor = 15;
        state.adjust_scroll(5);
        // cursor=15 should be visible in a 5-row window
        assert!(state.offset <= 15);
        assert!(state.offset + 5 > 15);

        state.cursor = 0;
        state.adjust_scroll(5);
        assert_eq!(state.offset, 0);
    }

    #[test]
    fn render_basic() {
        let picker = FilePicker::new().show_header(false);
        let mut state = make_state();

        let area = Rect::new(0, 0, 30, 5);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 5, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);

        // First entry should have cursor indicator "> "
        assert!(lines[0].starts_with("> "));
        // Should contain directory and file names
        let all_text = lines.join("\n");
        assert!(all_text.contains("docs"));
        assert!(all_text.contains("src"));
        assert!(all_text.contains("README.md"));
        assert!(all_text.contains("main.rs"));
    }

    #[test]
    fn render_with_header() {
        let picker = FilePicker::new().show_header(true);
        let mut state = make_state();

        let area = Rect::new(0, 0, 30, 6);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 6, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);

        // First line should be the directory path
        assert!(lines[0].starts_with("/tmp"));
    }

    #[test]
    fn render_empty_directory() {
        let picker = FilePicker::new().show_header(false);
        let mut state = FilePickerState::new(PathBuf::from("/empty"), vec![]);

        let area = Rect::new(0, 0, 30, 3);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 3, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);

        assert!(lines[0].contains("empty directory"));
    }

    #[test]
    fn render_scrolling() {
        let entries: Vec<DirEntry> = (0..20)
            .map(|i| DirEntry::file(format!("file{i:02}.txt"), format!("/tmp/file{i:02}.txt")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/tmp"), entries);
        let picker = FilePicker::new().show_header(false);

        // Move cursor to item 15, viewport is 5 rows
        state.cursor = 15;
        let area = Rect::new(0, 0, 30, 5);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 5, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);

        // file15 should be visible (with cursor)
        let all_text = lines.join("\n");
        assert!(all_text.contains("file15"));
    }

    #[test]
    fn cursor_style_applied_to_selected_row() {
        use ftui_render::cell::PackedRgba;

        let picker = FilePicker::new()
            .show_header(false)
            .cursor_style(Style::new().fg(PackedRgba::rgb(255, 0, 0)));
        let mut state = make_state();
        state.cursor = 1; // "src"

        let area = Rect::new(0, 0, 30, 4);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 4, &mut pool);

        picker.render(area, &mut frame, &mut state);

        // The cursor row (y=1) should have the cursor indicator
        let lines = buf_to_lines(&frame.buffer);
        assert!(lines[1].starts_with("> "));
        // Non-cursor rows should not
        assert!(!lines[0].starts_with("> "));
    }

    #[test]
    fn selected_set_on_file_entry() {
        let mut state = make_state();
        state.cursor = 2; // README.md (a file)

        // enter() on a file should set selected
        let result = state.enter();
        assert!(result.is_ok());
        assert_eq!(state.selected, Some(PathBuf::from("/tmp/README.md")));
    }

    #[test]
    fn enter_on_file_rejects_canonical_path_outside_root() {
        let root = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
        let repo = root
            .parent()
            .and_then(Path::parent)
            .expect("crate should be under workspace crates directory");
        let outside_file = repo.join("Cargo.toml");
        let mut state = FilePickerState::new(
            root.clone(),
            vec![DirEntry::file("Cargo.toml", outside_file)],
        )
        .with_root(root);

        let error = state
            .enter()
            .expect_err("root confinement should reject outside file selections");

        assert_eq!(error.kind(), std::io::ErrorKind::PermissionDenied);
        assert!(state.selected.is_none());
    }

    #[test]
    fn enter_on_directory_with_unresolvable_root_fails_closed() {
        let current_dir = std::env::current_dir().expect("test should run inside the workspace");
        let missing_root =
            current_dir.join(format!(".missing-file-picker-root-{}", std::process::id()));
        let target_dir = std::env::temp_dir();
        let mut state = FilePickerState::new(
            current_dir.clone(),
            vec![DirEntry::dir("tmp", target_dir.clone())],
        )
        .with_root(missing_root);

        let error = state
            .enter()
            .expect_err("unresolvable confinement root should fail closed");

        assert_eq!(error.kind(), std::io::ErrorKind::NotFound);
        assert_eq!(state.current_dir, current_dir);
        assert_eq!(state.entries[0].path, target_dir);
    }

    // ── DirEntry edge cases ───────────────────────────────────────

    #[test]
    fn dir_entry_equality() {
        let a = DirEntry::dir("src", "/src");
        let b = DirEntry::dir("src", "/src");
        assert_eq!(a, b);

        let c = DirEntry::file("src", "/src");
        assert_ne!(a, c, "dir vs file should differ");
    }

    #[test]
    fn dir_entry_clone() {
        let orig = DirEntry::file("main.rs", "/main.rs");
        let cloned = orig.clone();
        assert_eq!(orig, cloned);
    }

    #[test]
    fn dir_entry_debug_format() {
        let e = DirEntry::dir("test", "/test");
        let dbg = format!("{e:?}");
        assert!(dbg.contains("test"));
        assert!(dbg.contains("is_dir: true"));
    }

    // ── FilePickerState construction ──────────────────────────────

    #[test]
    fn state_new_defaults() {
        let state = FilePickerState::new(PathBuf::from("/home"), vec![]);
        assert_eq!(state.current_dir, PathBuf::from("/home"));
        assert_eq!(state.cursor, 0);
        assert_eq!(state.offset, 0);
        assert!(state.selected.is_none());
        assert!(state.root.is_none());
        assert!(state.entries.is_empty());
    }

    #[test]
    fn state_with_root_sets_root() {
        let state = FilePickerState::new(PathBuf::from("/home/user"), vec![]).with_root("/home");
        assert_eq!(state.root, Some(PathBuf::from("/home")));
    }

    // ── Cursor on single entry ────────────────────────────────────

    #[test]
    fn cursor_movement_single_entry() {
        let entries = vec![DirEntry::file("only.txt", "/only.txt")];
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);

        assert_eq!(state.cursor, 0);
        state.cursor_down();
        assert_eq!(state.cursor, 0, "can't go past single entry");
        state.cursor_up();
        assert_eq!(state.cursor, 0);
        state.cursor_end();
        assert_eq!(state.cursor, 0);
        state.cursor_home();
        assert_eq!(state.cursor, 0);
    }

    #[test]
    fn page_down_clamps_to_last() {
        let entries: Vec<DirEntry> = (0..5)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);

        state.page_down(100);
        assert_eq!(state.cursor, 4);
    }

    #[test]
    fn page_up_clamps_to_zero() {
        let entries: Vec<DirEntry> = (0..5)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);
        state.cursor = 3;

        state.page_up(100);
        assert_eq!(state.cursor, 0);
    }

    #[test]
    fn page_operations_on_empty_entries() {
        let mut state = FilePickerState::new(PathBuf::from("/"), vec![]);
        state.page_down(10);
        assert_eq!(state.cursor, 0);
        state.page_up(10);
        assert_eq!(state.cursor, 0);
    }

    // ── enter() edge cases ────────────────────────────────────────

    #[test]
    fn enter_on_empty_entries_returns_false() {
        let mut state = FilePickerState::new(PathBuf::from("/"), vec![]);
        let result = state.enter();
        assert!(result.is_ok());
        assert!(!result.unwrap());
        assert!(state.selected.is_none());
    }

    #[test]
    fn enter_on_file_sets_selected_without_navigation() {
        let entries = vec![
            DirEntry::dir("sub", "/sub"),
            DirEntry::file("readme.txt", "/readme.txt"),
        ];
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);
        state.cursor = 1;

        let result = state.enter().unwrap();
        assert!(!result, "enter on file returns false (no navigation)");
        assert_eq!(state.selected, Some(PathBuf::from("/readme.txt")));
        // Current directory unchanged.
        assert_eq!(state.current_dir, PathBuf::from("/"));
    }

    // ── go_back() edge cases ──────────────────────────────────────

    #[test]
    fn go_back_blocked_at_root() {
        let root = std::env::temp_dir();
        let mut state = FilePickerState::new(root.clone(), vec![]).with_root(root);

        let changed = state.go_back().unwrap();
        assert!(!changed, "go_back should be blocked when already at root");
    }

    #[test]
    fn go_back_without_history_uses_parent_directory() {
        let current = std::env::temp_dir();
        let parent = current
            .parent()
            .expect("temp_dir should have a parent")
            .to_path_buf();

        let mut state = FilePickerState::new(current.clone(), vec![]);
        let changed = state.go_back().unwrap();

        assert!(
            changed,
            "go_back should navigate to parent when history is empty"
        );
        assert_eq!(state.current_dir, parent);
        assert_eq!(state.cursor, 0, "parent navigation resets cursor to home");
    }

    #[test]
    fn go_back_restores_history_cursor_with_clamp() {
        let child = std::env::temp_dir();
        let parent = child
            .parent()
            .expect("temp_dir should have a parent")
            .to_path_buf();

        let mut state = FilePickerState::new(
            parent.clone(),
            vec![
                DirEntry::file("placeholder.txt", parent.join("placeholder.txt")),
                DirEntry::dir("child", child.clone()),
            ],
        );
        state.cursor = 1;

        let entered = state.enter().unwrap();
        assert!(entered, "enter should navigate into selected directory");

        let went_back = state.go_back().unwrap();
        assert!(
            went_back,
            "go_back should restore previous directory from history"
        );
        assert_eq!(state.current_dir, parent);

        let expected_cursor = 1.min(state.entries.len().saturating_sub(1));
        assert_eq!(state.cursor, expected_cursor);
    }

    // ── adjust_scroll edge cases ──────────────────────────────────

    #[test]
    fn adjust_scroll_zero_visible_rows_is_noop() {
        let entries: Vec<DirEntry> = (0..10)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);
        state.cursor = 5;
        state.offset = 0;

        state.adjust_scroll(0);
        assert_eq!(
            state.offset, 0,
            "zero visible rows should not change offset"
        );
    }

    #[test]
    fn adjust_scroll_cursor_above_viewport() {
        let entries: Vec<DirEntry> = (0..20)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);
        state.offset = 10;
        state.cursor = 5;

        state.adjust_scroll(5);
        assert_eq!(state.offset, 5, "offset should snap to cursor");
    }

    #[test]
    fn adjust_scroll_cursor_below_viewport() {
        let entries: Vec<DirEntry> = (0..20)
            .map(|i| DirEntry::file(format!("f{i}"), format!("/f{i}")))
            .collect();
        let mut state = FilePickerState::new(PathBuf::from("/"), entries);
        state.offset = 0;
        state.cursor = 10;

        state.adjust_scroll(5);
        // cursor=10 should be the last visible row: offset + 5 > 10 → offset = 6
        assert_eq!(state.offset, 6);
    }

    // ── FilePicker builder ────────────────────────────────────────

    #[test]
    fn file_picker_default_values() {
        let picker = FilePicker::default();
        assert!(picker.show_header);
        assert_eq!(picker.dir_prefix, "📁 ");
        assert_eq!(picker.file_prefix, "  ");
    }

    #[test]
    fn file_picker_builder_chain() {
        let picker = FilePicker::new()
            .dir_style(Style::default())
            .file_style(Style::default())
            .cursor_style(Style::default())
            .header_style(Style::default())
            .show_header(false);
        assert!(!picker.show_header);
    }

    #[test]
    fn file_picker_debug_format() {
        let picker = FilePicker::new();
        let dbg = format!("{picker:?}");
        assert!(dbg.contains("FilePicker"));
    }

    // ── Render edge cases ─────────────────────────────────────────

    #[test]
    fn render_zero_area_is_noop() {
        let picker = FilePicker::new();
        let mut state = make_state();

        let area = Rect::new(0, 0, 0, 0);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 5, &mut pool);

        picker.render(area, &mut frame, &mut state);
        // No crash, buffer untouched.
        let lines = buf_to_lines(&frame.buffer);
        assert!(lines[0].trim().is_empty());
    }

    #[test]
    fn render_height_one_shows_only_header() {
        let picker = FilePicker::new().show_header(true);
        let mut state = make_state();

        let area = Rect::new(0, 0, 30, 1);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 5, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);
        // Only the header row should have content.
        assert!(lines[0].starts_with("/tmp"));
        // Row 1 should be empty (no room for entries).
        assert!(lines[1].trim().is_empty());
    }

    #[test]
    fn render_no_header_uses_full_area_for_entries() {
        let picker = FilePicker::new().show_header(false);
        let mut state = make_state();

        let area = Rect::new(0, 0, 30, 4);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 4, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);
        // First line should be an entry (cursor on first entry), not a header.
        assert!(lines[0].starts_with("> "));
    }

    #[test]
    fn render_cursor_on_last_entry() {
        let picker = FilePicker::new().show_header(false);
        let mut state = make_state();
        state.cursor = 3; // last entry: main.rs

        let area = Rect::new(0, 0, 30, 5);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 5, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);
        // The cursor row should contain "main.rs".
        let cursor_line = lines.iter().find(|l| l.starts_with("> ")).unwrap();
        assert!(cursor_line.contains("main.rs"));
    }

    #[test]
    fn render_area_offset() {
        // Render into a sub-area of a larger buffer.
        let picker = FilePicker::new().show_header(false);
        let mut state = make_state();

        let area = Rect::new(5, 2, 20, 3);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(30, 10, &mut pool);

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);
        // Rows 0 and 1 should be empty (area starts at y=2).
        assert!(lines[0].trim().is_empty());
        assert!(lines[1].trim().is_empty());
        // Row 2 should have content starting at x=5.
        assert!(lines[2].len() >= 7);
    }

    #[test]
    fn render_shorter_header_and_fewer_entries_clear_stale_content() {
        let picker = FilePicker::new().show_header(true);
        let mut state = FilePickerState::new(PathBuf::from("/tmp/very/long/path"), make_entries());

        let area = Rect::new(0, 0, 24, 4);
        let mut pool = GraphemePool::new();
        let mut frame = Frame::new(24, 4, &mut pool);

        picker.render(area, &mut frame, &mut state);

        state.current_dir = PathBuf::from("/x");
        state.entries = vec![DirEntry::file("a", "/x/a")];
        state.cursor = 0;
        state.offset = 0;

        picker.render(area, &mut frame, &mut state);
        let lines = buf_to_lines(&frame.buffer);

        assert_eq!(lines[0], format!("{:<24}", "/x"));
        assert!(lines[1].starts_with(">   a"));
        assert_eq!(lines[2], " ".repeat(24));
        assert_eq!(lines[3], " ".repeat(24));
    }
}