use crate::primitives::display_width::str_width;
use crate::view::file_tree::{FileExplorerDecorationCache, FileTreeView, NodeId};
use crate::view::theme::Theme;
use ratatui::{
layout::Rect,
style::{Color, Modifier, Style},
text::{Line, Span},
widgets::{Block, Borders, List, ListItem, ListState},
Frame,
};
use std::collections::HashSet;
use std::path::PathBuf;
pub struct FileExplorerRenderer;
impl FileExplorerRenderer {
fn folder_has_modified_files(
folder_path: &PathBuf,
files_with_unsaved_changes: &HashSet<PathBuf>,
) -> bool {
for modified_file in files_with_unsaved_changes {
if modified_file.starts_with(folder_path) {
return true;
}
}
false
}
#[allow(clippy::too_many_arguments)]
pub fn render(
view: &mut FileTreeView,
frame: &mut Frame,
area: Rect,
is_focused: bool,
files_with_unsaved_changes: &HashSet<PathBuf>,
decorations: &FileExplorerDecorationCache,
keybinding_resolver: &crate::input::keybindings::KeybindingResolver,
current_context: crate::input::keybindings::KeyContext,
theme: &Theme,
close_button_hovered: bool,
) {
let viewport_height = area.height.saturating_sub(2) as usize;
view.set_viewport_height(viewport_height);
let display_nodes = view.get_display_nodes();
let scroll_offset = view.get_scroll_offset();
let selected_index = view.get_selected_index();
let scroll_offset = scroll_offset.min(display_nodes.len());
let visible_end = (scroll_offset + viewport_height).min(display_nodes.len());
let visible_items = &display_nodes[scroll_offset..visible_end];
let content_width = area.width.saturating_sub(3) as usize;
let items: Vec<ListItem> = visible_items
.iter()
.enumerate()
.map(|(viewport_idx, &(node_id, indent))| {
let actual_idx = scroll_offset + viewport_idx;
let is_selected = selected_index == Some(actual_idx);
Self::render_node(
view,
node_id,
indent,
is_selected,
is_focused,
files_with_unsaved_changes,
decorations,
theme,
content_width,
)
})
.collect();
let title = if let Some(keybinding) = keybinding_resolver.get_keybinding_for_action(
&crate::input::keybindings::Action::FocusFileExplorer,
current_context,
) {
format!(" File Explorer ({}) ", keybinding)
} else {
" File Explorer ".to_string()
};
let (title_style, border_style) = if is_focused {
(
Style::default()
.fg(theme.editor_bg)
.bg(theme.editor_fg)
.add_modifier(Modifier::BOLD),
Style::default().fg(theme.cursor),
)
} else {
(
Style::default().fg(theme.line_number_fg),
Style::default().fg(theme.split_separator_fg),
)
};
let list = List::new(items)
.block(
Block::default()
.borders(Borders::ALL)
.title(title)
.title_style(title_style)
.border_style(border_style)
.style(Style::default().bg(theme.editor_bg)),
)
.highlight_style(if is_focused {
Style::default().bg(theme.selection_bg).fg(theme.editor_fg)
} else {
Style::default().bg(theme.current_line_bg)
});
let mut list_state = ListState::default();
if let Some(selected) = selected_index {
if selected >= scroll_offset && selected < scroll_offset + viewport_height {
list_state.select(Some(selected - scroll_offset));
}
}
frame.render_stateful_widget(list, area, &mut list_state);
let close_button_x = area.x + area.width.saturating_sub(3);
let close_fg = if close_button_hovered {
theme.tab_close_hover_fg
} else {
theme.line_number_fg
};
let close_button =
ratatui::widgets::Paragraph::new("×").style(Style::default().fg(close_fg));
let close_area = Rect::new(close_button_x, area.y, 1, 1);
frame.render_widget(close_button, close_area);
if is_focused {
if let Some(selected) = selected_index {
if selected >= scroll_offset && selected < scroll_offset + viewport_height {
let cursor_x = area.x + 1;
let cursor_y = area.y + 1 + (selected - scroll_offset) as u16;
let cursor_indicator = ratatui::widgets::Paragraph::new("▌")
.style(Style::default().fg(theme.cursor));
let cursor_area = ratatui::layout::Rect::new(cursor_x, cursor_y, 1, 1);
frame.render_widget(cursor_indicator, cursor_area);
frame.set_cursor_position((cursor_x, cursor_y));
}
}
}
}
#[allow(clippy::too_many_arguments)]
fn render_node(
view: &FileTreeView,
node_id: NodeId,
indent: usize,
is_selected: bool,
is_focused: bool,
files_with_unsaved_changes: &HashSet<PathBuf>,
decorations: &FileExplorerDecorationCache,
theme: &Theme,
content_width: usize,
) -> ListItem<'static> {
let node = view.tree().get_node(node_id).expect("Node should exist");
let mut spans = Vec::new();
let indent_width = indent * 2;
let indicator_width = if node.is_dir() { 2 } else { 2 }; let name_width = str_width(&node.entry.name);
let left_side_width = indent_width + indicator_width + name_width;
if indent > 0 {
spans.push(Span::raw(" ".repeat(indent)));
}
if node.is_dir() {
let indicator = if node.is_expanded() {
"▼ "
} else if node.is_collapsed() {
"> "
} else if node.is_loading() {
"⟳ "
} else {
"! "
};
spans.push(Span::styled(
indicator,
Style::default().fg(theme.diagnostic_warning_fg),
));
} else {
spans.push(Span::raw(" "));
}
let name_style = if is_selected && is_focused {
Style::default().fg(theme.editor_fg)
} else if node
.entry
.metadata
.as_ref()
.map(|m| m.is_hidden)
.unwrap_or(false)
{
Style::default().fg(theme.line_number_fg)
} else if node.entry.is_symlink() {
Style::default().fg(theme.syntax_type)
} else if node.is_dir() {
Style::default().fg(theme.syntax_keyword)
} else {
Style::default().fg(theme.editor_fg)
};
spans.push(Span::styled(node.entry.name.clone(), name_style));
let has_unsaved = if node.is_dir() {
Self::folder_has_modified_files(&node.entry.path, files_with_unsaved_changes)
} else {
files_with_unsaved_changes.contains(&node.entry.path)
};
let direct_decoration = decorations.direct_for_path(&node.entry.path);
let bubbled_decoration = if node.is_dir() {
decorations
.bubbled_for_path(&node.entry.path)
.filter(|_| direct_decoration.is_none())
} else {
None
};
let right_indicator: Option<(String, Color)> = if has_unsaved {
Some(("●".to_string(), theme.diagnostic_warning_fg))
} else if let Some(decoration) = direct_decoration {
let symbol = Self::decoration_symbol(&decoration.symbol);
Some((symbol, Self::decoration_color(decoration)))
} else {
bubbled_decoration
.map(|decoration| ("●".to_string(), Self::decoration_color(decoration)))
};
let right_indicator_width = right_indicator
.as_ref()
.map(|(s, _)| str_width(s))
.unwrap_or(0);
let error_text = if node.is_error() { " [Error]" } else { "" };
let error_width = str_width(error_text);
let total_right_width = right_indicator_width + error_width;
let min_gap = 1;
let padding = if left_side_width + min_gap + total_right_width < content_width {
content_width - left_side_width - total_right_width
} else {
min_gap
};
spans.push(Span::raw(" ".repeat(padding)));
if let Some((symbol, color)) = right_indicator {
spans.push(Span::styled(symbol, Style::default().fg(color)));
}
if node.is_error() {
spans.push(Span::styled(
error_text,
Style::default().fg(theme.diagnostic_error_fg),
));
}
ListItem::new(Line::from(spans)).style(Style::default().bg(theme.editor_bg))
}
fn decoration_symbol(symbol: &str) -> String {
symbol
.chars()
.next()
.map(|c| c.to_string())
.unwrap_or_else(|| " ".to_string())
}
fn decoration_color(decoration: &crate::view::file_tree::FileExplorerDecoration) -> Color {
let [r, g, b] = decoration.color;
Color::Rgb(r, g, b)
}
}