use crate::picker::Picker;
use ansi_to_tui::IntoText;
use ratatui::{
layout::{Alignment, Constraint, Direction, Layout, Rect},
style::{Color, Modifier, Style},
text::{Line, Span, Text},
widgets::{Block, Borders, List, ListItem, Paragraph},
Frame,
};
use std::fmt::Display;
pub fn ui<T>(f: &mut Frame, app: &mut Picker<T>)
where
T: Sync + Send + Display,
{
match app.mode {
crate::picker::PickerMode::Help => {
render_help_screen(f, f.area(), app.help_scroll_offset);
}
_ => {
if app.has_preview() {
let main_chunks = Layout::default()
.direction(Direction::Horizontal)
.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
.split(f.area());
let left_chunks = Layout::default()
.direction(Direction::Vertical)
.constraints(
[
Constraint::Length(1),
Constraint::Length(3),
Constraint::Min(1),
]
.as_ref(),
)
.split(main_chunks[0]);
app.update_height(left_chunks[2].height - 3);
render_help(f, left_chunks[0], app);
render_search_input(f, app, left_chunks[1]);
render_items(f, app, left_chunks[2]);
render_preview(f, app, main_chunks[1]);
} else {
let chunks = Layout::default()
.direction(Direction::Vertical)
.constraints(
[
Constraint::Length(1),
Constraint::Length(3),
Constraint::Min(1),
]
.as_ref(),
)
.split(f.area());
app.update_height(chunks[2].height - 3);
render_help(f, chunks[0], app);
render_search_input(f, app, chunks[1]);
render_items(f, app, chunks[2]);
}
}
}
}
fn render_help<T>(f: &mut Frame, area: Rect, app: &Picker<T>)
where
T: Sync + Send + Display,
{
let left_text = vec![Line::from(vec![
Span::raw("Press "),
Span::styled("↑/↓", Style::default().add_modifier(Modifier::BOLD)),
Span::raw(" to navigate, "),
Span::styled("Tab", Style::default().add_modifier(Modifier::BOLD)),
Span::raw(" to select, "),
Span::styled("Enter", Style::default().add_modifier(Modifier::BOLD)),
Span::raw(" to confirm, "),
Span::styled("Esc", Style::default().add_modifier(Modifier::BOLD)),
Span::raw(" to quit"),
])];
let right_text = vec![Line::from(vec![
Span::styled(
app.running_threads().to_string(),
Style::default().add_modifier(Modifier::BOLD),
),
Span::raw(" indexers"),
])
.right_aligned()];
let spans = Layout::default()
.direction(Direction::Horizontal)
.constraints([Constraint::Min(40), Constraint::Length(12)])
.split(area);
let left_paragraph = Paragraph::new(left_text);
let right_paragraph = Paragraph::new(right_text);
f.render_widget(left_paragraph, spans[0]);
f.render_widget(right_paragraph, spans[1]);
}
fn render_search_input<T>(f: &mut Frame, app: &Picker<T>, area: Rect)
where
T: Sync + Send + Display,
{
let (text, cursor_index, title) = match app.mode {
crate::picker::PickerMode::Search => (&app.query, app.query_index, "Search"),
crate::picker::PickerMode::Editing => (&app.editing_text, app.editing_index, "Editing"),
crate::picker::PickerMode::Help => return,
};
let before_cursor = text.chars().take(cursor_index).collect::<String>();
let cursor_char = text.chars().nth(cursor_index).unwrap_or(' ');
let after_cursor = text.chars().skip(cursor_index + 1).collect::<String>();
let line = Line::from(vec![
Span::raw(before_cursor),
Span::styled(cursor_char.to_string(), Style::default().bg(Color::Blue)),
Span::raw(after_cursor),
]);
let input = Paragraph::new(line)
.style(Style::default())
.block(Block::default().borders(Borders::ALL).title(title));
let snapshot = app.snapshot();
let item_count_text = vec![Line::from(vec![Span::styled(
format!(
"{}/{}",
snapshot.matched_item_count(),
snapshot.item_count()
),
Style::default().add_modifier(Modifier::BOLD),
)])
.right_aligned()];
let item_count: Paragraph<'_> = Paragraph::new(item_count_text)
.style(Style::default())
.block(Block::default().borders(Borders::ALL).title("Status"));
let spans = Layout::default()
.direction(Direction::Horizontal)
.constraints([Constraint::Min(40), Constraint::Length(17)])
.split(area);
f.render_widget(input, spans[0]);
f.render_widget(item_count, spans[1]);
}
fn render_items<T>(f: &mut Frame, app: &mut Picker<T>, area: Rect)
where
T: Sync + Send + Display,
{
match app.mode {
crate::picker::PickerMode::Editing => {
render_autocomplete_suggestions(f, app, area);
}
crate::picker::PickerMode::Search => {
if app.matched_item_count() > 0 {
let items: Vec<ListItem> = app
.matched_items()
.iter()
.map(|item| {
let is_selected = item.is_selected();
let style = if is_selected {
Style::default()
.fg(Color::Yellow)
.add_modifier(Modifier::BOLD)
} else {
Style::default()
};
let prefix = if is_selected { "✓ " } else { " " };
let item_str = format!("{}", item);
let item_text = match item_str.as_str().into_text() {
Ok(text) => text,
Err(_) => Text::raw(item_str.clone()),
};
let mut lines: Vec<Line> = item_text.lines.into_iter().collect();
if let Some(first_line) = lines.first_mut() {
let mut new_spans = vec![Span::raw(prefix)];
new_spans.extend(first_line.spans.drain(..));
first_line.spans = new_spans;
} else {
lines.push(Line::from(prefix));
}
ListItem::new(Text::from(lines)).style(style)
})
.collect();
let items = List::new(items)
.block(Block::default().borders(Borders::ALL).title("Items"))
.highlight_style(
Style::default()
.bg(Color::Blue)
.add_modifier(Modifier::BOLD),
)
.highlight_symbol("> ");
f.render_stateful_widget(
items,
area,
&mut ratatui::widgets::ListState::default().with_selected(Some(
app.current_index
.saturating_sub(app.first_visible_item_index())
as usize,
)),
);
} else {
let chunks = Layout::default()
.direction(Direction::Vertical)
.constraints(
[
Constraint::Min(1),
Constraint::Length(1),
Constraint::Min(1),
]
.as_ref(),
)
.split(f.area());
let no_items_paragraph =
Paragraph::new("No items found").alignment(Alignment::Center);
f.render_widget(no_items_paragraph, chunks[1]);
}
}
crate::picker::PickerMode::Help => return,
}
}
fn render_autocomplete_suggestions<T>(f: &mut Frame, app: &Picker<T>, area: Rect)
where
T: Sync + Send + Display,
{
if !app.autocomplete_suggestions.is_empty() {
let items: Vec<ListItem> = app
.autocomplete_suggestions
.iter()
.map(|suggestion| {
let is_selected = suggestion.is_selected();
let style = if is_selected {
Style::default()
.fg(Color::Yellow)
.add_modifier(Modifier::BOLD)
} else {
Style::default()
};
let prefix = if is_selected { "✓ " } else { " " };
let content = format!("{}{}", prefix, suggestion);
ListItem::new(content).style(style)
})
.collect();
let items = List::new(items)
.block(Block::default().borders(Borders::ALL).title("Autocomplete"))
.highlight_style(
Style::default()
.bg(Color::Blue)
.add_modifier(Modifier::BOLD),
)
.highlight_symbol("> ");
f.render_stateful_widget(
items,
area,
&mut ratatui::widgets::ListState::default().with_selected(Some(app.autocomplete_index)),
);
} else {
let chunks = Layout::default()
.direction(Direction::Vertical)
.constraints(
[
Constraint::Min(1),
Constraint::Length(1),
Constraint::Min(1),
]
.as_ref(),
)
.split(f.area());
let no_suggestions_paragraph =
Paragraph::new("No autocomplete suggestions").alignment(Alignment::Center);
f.render_widget(no_suggestions_paragraph, chunks[1]);
}
}
fn render_preview<T>(f: &mut Frame, app: &Picker<T>, area: Rect)
where
T: Sync + Send + Display,
{
let lines = match app.preview_output().into_text() {
Ok(text) => text,
Err(e) => Text::from(format!("Could not render preview: {e}")),
};
let preview = Paragraph::new(lines)
.block(Block::default().borders(Borders::ALL).title("Preview"))
.wrap(ratatui::widgets::Wrap { trim: false });
f.render_widget(preview, area);
}
fn render_help_screen(f: &mut Frame, area: Rect, scroll_offset: u16) {
let help_text = vec![
Line::from(vec![Span::styled(
"Picleo Help",
Style::default()
.add_modifier(Modifier::BOLD)
.fg(Color::Yellow),
)]),
Line::from(""),
Line::from(vec![Span::styled(
"Search Mode:",
Style::default()
.add_modifier(Modifier::BOLD)
.fg(Color::Green),
)]),
Line::from(" ↑/↓, PgUp/PgDn, Home/End Navigate items"),
Line::from(" Tab Toggle item selection"),
Line::from(" Enter Confirm selection and exit"),
Line::from(" Esc Clear query or exit"),
Line::from(" Ctrl+c Exit program"),
Line::from(" Ctrl+u Clear query"),
Line::from(" Ctrl+a/e Move to start/end of query"),
Line::from(" Ctrl+k Delete to end of query"),
Line::from(" Ctrl+Backspace/Alt+Backspace Delete word backward"),
Line::from(" Ctrl+Delete/Alt+Delete Delete word forward"),
Line::from(" Ctrl+Left/Right Jump word backward/forward"),
Line::from(" Ctrl+d Edit current item (if editable)"),
Line::from(" Ctrl+n Create new item (if editable)"),
Line::from(" Ctrl+h Show this help"),
Line::from(""),
Line::from(vec![Span::styled(
"Editing Mode:",
Style::default()
.add_modifier(Modifier::BOLD)
.fg(Color::Green),
)]),
Line::from(" ↑/↓ Navigate autocomplete suggestions"),
Line::from(" Tab Toggle autocomplete selection"),
Line::from(" Enter Create items and return to search"),
Line::from(" Esc Cancel editing and return to search"),
Line::from(" Ctrl+c Exit program"),
Line::from(" Ctrl+u Clear editing text"),
Line::from(" Ctrl+a/e Move to start/end of text"),
Line::from(" Ctrl+k Delete to end of line"),
Line::from(" Ctrl+Backspace/Shift+Backspace Delete word backward"),
Line::from(" Ctrl+Delete/Shift+Delete Delete word forward"),
Line::from(" Ctrl+Left/Right Jump word backward/forward"),
Line::from(" Ctrl+h Show this help"),
Line::from(""),
Line::from(vec![Span::styled(
"Mouse Support:",
Style::default()
.add_modifier(Modifier::BOLD)
.fg(Color::Green),
)]),
Line::from(" Scroll Navigate items"),
Line::from(" Shift/Ctrl+Scroll Page navigation"),
Line::from(" Left Click Toggle item selection"),
Line::from(" Middle Click Toggle current item selection"),
Line::from(""),
Line::from(vec![
Span::styled("Press ", Style::default()),
Span::styled("Esc", Style::default().add_modifier(Modifier::BOLD)),
Span::styled(", ", Style::default()),
Span::styled("q", Style::default().add_modifier(Modifier::BOLD)),
Span::styled(", or ", Style::default()),
Span::styled("Ctrl+h", Style::default().add_modifier(Modifier::BOLD)),
Span::styled(" to close this help", Style::default()),
]),
];
let help_paragraph = Paragraph::new(help_text)
.block(
Block::default()
.borders(Borders::ALL)
.title("Help - Use ↑/↓ or j/k to scroll"),
)
.alignment(Alignment::Left)
.wrap(ratatui::widgets::Wrap { trim: false })
.scroll((scroll_offset, 0));
f.render_widget(help_paragraph, area);
}