use ratatui::Frame;
use ratatui::layout::{Alignment, Constraint, Direction, Layout, Rect};
use ratatui::style::{Modifier, Style};
use ratatui::text::{Line, Span};
use ratatui::widgets::{Block, BorderType, Borders, Clear, List, ListItem, Paragraph};
use crate::ui::state::App;
pub fn render_search_popup(f: &mut Frame, app: &App, area: Rect) {
f.render_widget(Clear, area);
let popup_block = Block::default()
.borders(Borders::ALL)
.border_type(BorderType::Rounded)
.title("Search")
.border_style(Style::default().fg(app.theme.primary));
let inner_area = popup_block.inner(area);
f.render_widget(popup_block, area);
let popup_layout = Layout::default()
.direction(Direction::Vertical)
.constraints([Constraint::Length(3), Constraint::Min(3)])
.split(inner_area);
let display_text = if let Some(ref autocomplete) = app.search.autocomplete {
let base = &app.search.input;
let completion = &autocomplete[base.len()..];
Line::from(vec![
Span::styled(base.clone(), Style::default()),
Span::styled(completion, Style::default().fg(app.theme.secondary)),
])
} else {
Line::from(app.search.input.clone())
};
let input_block = Block::default()
.borders(Borders::ALL)
.border_type(BorderType::Rounded)
.title("Type to search (Tab to complete)")
.border_style(Style::default().fg(app.theme.primary));
let input = Paragraph::new(display_text)
.block(input_block)
.style(Style::default())
.alignment(Alignment::Left);
f.render_widget(input, popup_layout[0]);
if app.search.cursor_position <= app.search.input.len() {
f.set_cursor_position((
popup_layout[0].x + 1 + app.search.cursor_position as u16,
popup_layout[0].y + 1,
));
}
let mut result_items = Vec::new();
let max_display = area.height.saturating_sub(5) as usize;
let result_count = app.search.results.len();
let display_count = std::cmp::min(result_count, max_display);
let start_idx = if result_count <= max_display {
0
} else {
let half_display = max_display / 2;
if app.search.selected_idx < half_display {
0
} else if app.search.selected_idx + half_display >= result_count {
result_count - max_display
} else {
app.search.selected_idx - half_display
}
};
for i in 0..display_count {
let result_idx = start_idx + i;
if result_idx < app.search.results.len() {
let result = &app.search.results[result_idx];
let display_text = format!("{}/{}", result.item.category, result.item.name);
if app.search.input.is_empty() {
result_items.push(ListItem::new(Line::from(vec![Span::styled(
display_text,
Style::default().fg(app.theme.foreground),
)])));
} else {
let spans = highlight_match(&display_text, &result.indices, &app.theme);
result_items.push(ListItem::new(Line::from(spans)));
}
}
}
let result_count_text = format!("Found {} scripts", app.search.results.len());
let search_results = List::new(result_items)
.block(
Block::default()
.borders(Borders::ALL)
.border_type(BorderType::Rounded)
.title(result_count_text)
.border_style(Style::default().fg(app.theme.primary)),
)
.highlight_style(Style::default().bg(app.theme.secondary).fg(app.theme.background).bold())
.highlight_symbol("");
let mut search_list_state = ratatui::widgets::ListState::default();
search_list_state.select(Some(app.search.selected_idx.saturating_sub(start_idx)));
f.render_stateful_widget(search_results, popup_layout[1], &mut search_list_state);
}
fn highlight_match(
text: &str,
indices: &[usize],
theme: &crate::ui::theme::Theme,
) -> Vec<Span<'static>> {
if indices.is_empty() {
return vec![Span::styled(text.to_string(), Style::default().fg(theme.foreground))];
}
let mut spans = Vec::new();
let mut last_idx = 0usize;
for &idx in indices {
if idx >= text.len() {
break;
}
let Some((char_start, ch)) = text[idx..].char_indices().next() else {
break;
};
let char_byte_start = idx + char_start;
let char_end = char_byte_start + ch.len_utf8();
if char_byte_start > last_idx {
spans.push(Span::styled(
text[last_idx..char_byte_start].to_string(),
Style::default().fg(theme.foreground),
));
}
spans.push(Span::styled(
text[char_byte_start..char_end].to_string(),
Style::default().fg(theme.primary).add_modifier(Modifier::BOLD),
));
last_idx = char_end;
}
if last_idx < text.len() {
spans.push(Span::styled(
text[last_idx..].to_string(),
Style::default().fg(theme.foreground),
));
}
spans
}
#[cfg(test)]
mod tests {
use super::*;
use crate::ui::theme::Theme;
fn test_theme() -> Theme {
Theme::catppuccin_mocha()
}
#[test]
fn no_indices_returns_single_span() {
let spans = highlight_match("hello", &[], &test_theme());
assert_eq!(spans.len(), 1);
assert_eq!(spans[0].content, "hello");
}
#[test]
fn ascii_match_split() {
let spans = highlight_match("hello world", &[6], &test_theme());
assert_eq!(spans.len(), 3);
assert_eq!(spans[0].content, "hello ");
assert_eq!(spans[1].content, "w");
assert_eq!(spans[2].content, "orld");
}
#[test]
fn utf8_does_not_panic_on_multibyte_char() {
let text = "café";
let spans = highlight_match(text, &[3], &test_theme());
assert!(!spans.is_empty());
let matched: String = spans.iter().map(|s| s.content.as_ref()).collect();
assert_eq!(matched, text);
}
#[test]
fn utf8_emoji_does_not_split_codepoint() {
let text = "a😀b";
let spans = highlight_match(text, &[0], &test_theme());
let joined: String = spans.iter().map(|s| s.content.as_ref()).collect();
assert_eq!(joined, text);
}
#[test]
fn out_of_range_index_is_ignored() {
let spans = highlight_match("abc", &[100], &test_theme());
assert_eq!(spans.len(), 1);
assert_eq!(spans[0].content, "abc");
}
#[test]
fn first_char_match() {
let spans = highlight_match("abc", &[0], &test_theme());
assert_eq!(spans.len(), 2);
assert_eq!(spans[0].content, "a");
assert_eq!(spans[1].content, "bc");
}
}