1use anyhow::Result;
11use chrono::{DateTime, Local};
12use crossterm::{
13 event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, MouseEventKind},
14 execute,
15 terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
16};
17use ratatui::{
18 Frame, Terminal,
19 backend::CrosstermBackend,
20 layout::{Constraint, Direction, Layout},
21 style::{Color, Modifier, Style},
22 text::{Line, Span},
23 widgets::Paragraph,
24};
25use std::io;
26use std::path::Path;
27
28use super::conversation::{ConversationHistory, ConversationManager};
29
30const WHEEL_STEP: usize = 3;
33
34const ROWS_PER_ENTRY: usize = 3;
37
38pub struct SessionEntry {
41 pub history: ConversationHistory,
42 pub size_bytes: u64,
43}
44
45pub fn select_conversation(
49 entries: Vec<SessionEntry>,
50 manager: &ConversationManager,
51 now: DateTime<Local>,
52) -> Result<Option<ConversationHistory>> {
53 if entries.is_empty() {
54 println!("No previous conversations found in this directory.");
55 return Ok(None);
56 }
57
58 enable_raw_mode()?;
59 let mut stdout = io::stdout();
60 execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
64 let backend = CrosstermBackend::new(stdout);
65 let mut terminal = Terminal::new(backend)?;
66
67 let mut state = SelectorState::new(entries);
68 let result = run_selector(&mut terminal, &mut state, manager, now);
69
70 disable_raw_mode()?;
71 execute!(
72 terminal.backend_mut(),
73 LeaveAlternateScreen,
74 DisableMouseCapture
75 )?;
76 terminal.show_cursor()?;
77
78 result
79}
80
81pub struct SelectorState {
84 entries: Vec<SessionEntry>,
85 query: String,
86 selected: usize,
88 scroll_offset: usize,
91 viewport_entries: usize,
94 pending_delete: Option<usize>,
96}
97
98impl SelectorState {
99 pub fn new(entries: Vec<SessionEntry>) -> Self {
100 Self {
101 entries,
102 query: String::new(),
103 selected: 0,
104 scroll_offset: 0,
105 viewport_entries: 0,
106 pending_delete: None,
107 }
108 }
109
110 fn filtered(&self) -> Vec<usize> {
115 if self.query.is_empty() {
116 return (0..self.entries.len()).collect();
117 }
118 let needle = self.query.to_lowercase();
119 self.entries
120 .iter()
121 .enumerate()
122 .filter(|(_, e)| entry_matches(&e.history, &needle))
123 .map(|(i, _)| i)
124 .collect()
125 }
126
127 fn current(&self) -> Option<&SessionEntry> {
129 self.filtered()
130 .get(self.selected)
131 .map(|&i| &self.entries[i])
132 }
133
134 fn move_down(&mut self) {
135 let n = self.filtered().len();
136 if n > 0 && self.selected + 1 < n {
137 self.selected += 1;
138 let visible = self.viewport_entries.max(1);
140 if self.selected >= self.scroll_offset + visible {
141 self.scroll_offset = self.selected + 1 - visible;
142 }
143 }
144 }
145
146 fn move_up(&mut self) {
147 if self.selected > 0 {
148 self.selected -= 1;
149 if self.selected < self.scroll_offset {
151 self.scroll_offset = self.selected;
152 }
153 }
154 }
155
156 fn push_query(&mut self, c: char) {
159 self.query.push(c);
160 self.selected = 0;
161 self.scroll_offset = 0;
162 }
163
164 fn pop_query(&mut self) {
165 self.query.pop();
166 self.selected = 0;
167 self.scroll_offset = 0;
168 }
169
170 fn scroll_viewport_down(&mut self) {
173 self.scroll_offset = self.scroll_offset.saturating_add(WHEEL_STEP);
174 }
175
176 fn scroll_viewport_up(&mut self) {
177 self.scroll_offset = self.scroll_offset.saturating_sub(WHEEL_STEP);
178 }
179
180 fn request_delete(&mut self) {
183 self.pending_delete = self.filtered().get(self.selected).copied();
184 }
185
186 fn cancel_delete(&mut self) {
187 self.pending_delete = None;
188 }
189
190 fn take_pending_delete(&mut self) -> Option<usize> {
191 self.pending_delete.take()
192 }
193
194 fn remove_entry(&mut self, entries_idx: usize) {
197 if entries_idx >= self.entries.len() {
198 return;
199 }
200 self.entries.remove(entries_idx);
201 let n = self.filtered().len();
202 if n == 0 {
203 self.selected = 0;
204 } else if self.selected >= n {
205 self.selected = n - 1;
206 }
207 }
208}
209
210fn entry_matches(history: &ConversationHistory, needle: &str) -> bool {
213 history.title.to_lowercase().contains(needle)
214 || history
215 .git_branch
216 .as_deref()
217 .is_some_and(|b| b.to_lowercase().contains(needle))
218}
219
220fn run_selector(
221 terminal: &mut Terminal<CrosstermBackend<io::Stdout>>,
222 state: &mut SelectorState,
223 manager: &ConversationManager,
224 now: DateTime<Local>,
225) -> Result<Option<ConversationHistory>> {
226 loop {
227 terminal.draw(|f| render(f, state, now))?;
228
229 match event::read()? {
230 Event::Key(key) => {
231 if state.pending_delete.is_some() {
235 if matches!(key.code, KeyCode::Char('y') | KeyCode::Char('Y')) {
236 if let Some(idx) = state.take_pending_delete() {
237 let id = state.entries[idx].history.id.clone();
238 if manager.delete_conversation(&id).is_ok() {
239 state.remove_entry(idx);
240 if state.entries.is_empty() {
241 return Ok(None);
242 }
243 }
244 }
245 } else {
246 state.cancel_delete();
247 }
248 continue;
249 }
250 match key.code {
251 KeyCode::Esc => return Ok(None),
252 KeyCode::Enter => return Ok(state.current().map(|e| e.history.clone())),
253 KeyCode::Down => state.move_down(),
254 KeyCode::Up => state.move_up(),
255 KeyCode::Delete => state.request_delete(),
258 KeyCode::Backspace => state.pop_query(),
259 KeyCode::Char(c) => state.push_query(c),
262 _ => {},
263 }
264 },
265 Event::Mouse(m) => match m.kind {
267 MouseEventKind::ScrollUp => state.scroll_viewport_up(),
268 MouseEventKind::ScrollDown => state.scroll_viewport_down(),
269 _ => {},
270 },
271 _ => {},
272 }
273 }
274}
275
276const ACCENT: Color = Color::Cyan;
279const META: Color = Color::Gray;
280const DIM: Color = Color::DarkGray;
281
282pub fn render(f: &mut Frame, state: &mut SelectorState, now: DateTime<Local>) {
289 let layout = Layout::default()
290 .direction(Direction::Vertical)
291 .constraints([
292 Constraint::Length(1), Constraint::Length(1), Constraint::Length(1), Constraint::Length(1), Constraint::Length(1), Constraint::Length(1), Constraint::Min(3), Constraint::Length(1), ]);
301 let [title, _s1, search, _s2, project, _s3, list, hints] = f.area().layout(&layout);
302
303 let filtered = state.filtered();
307 let visible = (list.height as usize / ROWS_PER_ENTRY).max(1);
308 state.viewport_entries = visible;
309 let max_offset = filtered.len().saturating_sub(visible);
310 if state.scroll_offset > max_offset {
311 state.scroll_offset = max_offset;
312 }
313 let scroll_offset = state.scroll_offset;
314 let selected = state.selected;
315 let pending_delete = state.pending_delete;
316
317 f.render_widget(
318 Paragraph::new(Line::from(Span::styled(
319 "Resume session",
320 Style::default().fg(ACCENT).add_modifier(Modifier::BOLD),
321 ))),
322 title,
323 );
324
325 let search_line = if state.query.is_empty() {
327 Line::from(Span::styled(" Search…", Style::default().fg(DIM)))
328 } else {
329 Line::from(vec![
330 Span::styled(" ", Style::default()),
331 Span::styled(state.query.clone(), Style::default().fg(Color::White)),
332 Span::styled("▏", Style::default().fg(ACCENT)),
333 ])
334 };
335 f.render_widget(Paragraph::new(search_line), search);
336
337 if let Some(name) = state
339 .entries
340 .first()
341 .map(|e| project_name(&e.history.project_path))
342 {
343 f.render_widget(
344 Paragraph::new(Line::from(Span::styled(
345 name,
346 Style::default().fg(META).add_modifier(Modifier::BOLD),
347 ))),
348 project,
349 );
350 }
351
352 let lines: Vec<Line> = if filtered.is_empty() {
353 vec![Line::from(Span::styled(
354 " No sessions match your search.",
355 Style::default().fg(DIM),
356 ))]
357 } else {
358 filtered
359 .iter()
360 .enumerate()
361 .skip(scroll_offset)
362 .take(visible)
363 .flat_map(|(row, &idx)| {
364 let entry = &state.entries[idx];
365 let is_selected = row == selected;
366 let (marker, title_style) = if is_selected {
367 (
368 "> ",
369 Style::default().fg(ACCENT).add_modifier(Modifier::BOLD),
370 )
371 } else {
372 (" ", Style::default().fg(Color::White))
373 };
374 let title_line = Line::from(vec![
375 Span::styled(marker, Style::default().fg(ACCENT)),
376 Span::styled(entry.history.title.clone(), title_style),
377 ]);
378 let meta_line = Line::from(vec![Span::styled(
379 format!(" {}", meta_label(entry, now)),
380 Style::default().fg(META),
381 )]);
382 [title_line, meta_line, Line::from("")]
383 })
384 .collect()
385 };
386 f.render_widget(Paragraph::new(lines), list);
387
388 let hints_line = if let Some(idx) = pending_delete {
390 let name: String = state
391 .entries
392 .get(idx)
393 .map(|e| e.history.title.chars().take(40).collect::<String>())
394 .unwrap_or_default();
395 Line::from(Span::styled(
396 format!("Delete \"{name}\"? y confirms · any other key cancels"),
397 Style::default().fg(Color::Yellow),
398 ))
399 } else {
400 Line::from(Span::styled(
401 "↑↓ select · type to search · del delete · enter resume · esc cancel",
402 Style::default().fg(DIM),
403 ))
404 };
405 f.render_widget(Paragraph::new(hints_line), hints);
406}
407
408fn meta_label(entry: &SessionEntry, now: DateTime<Local>) -> String {
411 let mut bits = vec![humanize_relative(now, entry.history.updated_at)];
412 if let Some(branch) = &entry.history.git_branch
413 && !branch.is_empty()
414 {
415 bits.push(branch.clone());
416 }
417 bits.push(humanize_size(entry.size_bytes));
418 bits.join(" · ")
419}
420
421fn project_name(project_path: &str) -> String {
423 Path::new(project_path)
424 .file_name()
425 .map(|n| n.to_string_lossy().into_owned())
426 .filter(|s| !s.is_empty())
427 .unwrap_or_else(|| project_path.to_string())
428}
429
430fn humanize_relative(now: DateTime<Local>, then: DateTime<Local>) -> String {
433 let secs = (now - then).num_seconds().max(0);
434 let (n, unit) = if secs < 45 {
435 return "just now".to_string();
436 } else if secs < 3600 {
437 (secs / 60, "minute")
438 } else if secs < 86_400 {
439 (secs / 3600, "hour")
440 } else if secs < 7 * 86_400 {
441 (secs / 86_400, "day")
442 } else if secs < 30 * 86_400 {
443 (secs / (7 * 86_400), "week")
444 } else {
445 (secs / (30 * 86_400), "month")
446 };
447 let n = n.max(1);
448 if n == 1 {
449 format!("1 {unit} ago")
450 } else {
451 format!("{n} {unit}s ago")
452 }
453}
454
455fn humanize_size(bytes: u64) -> String {
457 const KB: f64 = 1024.0;
458 const MB: f64 = 1024.0 * 1024.0;
459 let b = bytes as f64;
460 if b < KB {
461 format!("{bytes}B")
462 } else if b < MB {
463 format!("{:.1}KB", b / KB)
464 } else {
465 format!("{:.1}MB", b / MB)
466 }
467}
468
469#[cfg(test)]
470mod tests {
471 use super::*;
472 use chrono::TimeZone;
473 use ratatui::Terminal;
474 use ratatui::backend::TestBackend;
475 use std::collections::VecDeque;
476
477 fn at(y: i32, mo: u32, d: u32, h: u32, mi: u32) -> DateTime<Local> {
478 Local.with_ymd_and_hms(y, mo, d, h, mi, 0).unwrap()
479 }
480
481 fn history(title: &str, branch: Option<&str>, updated: DateTime<Local>) -> ConversationHistory {
482 ConversationHistory {
483 id: "20260101_000000_000".to_string(),
484 title: title.to_string(),
485 messages: Vec::new(),
486 model_name: "ollama/test".to_string(),
487 project_path: "/home/nsabaj/Development/source-clone".to_string(),
488 created_at: updated,
489 updated_at: updated,
490 total_tokens: None,
491 compactions: Vec::new(),
492 input_history: VecDeque::new(),
493 git_branch: branch.map(str::to_string),
494 safety_mode: None,
495 cumulative_tokens: 0,
496 last_token_usage: None,
497 cumulative_token_usage: crate::domain::TokenUsageTotals::default(),
498 context_usage: None,
499 }
500 }
501
502 fn entry(
503 title: &str,
504 branch: Option<&str>,
505 updated: DateTime<Local>,
506 size: u64,
507 ) -> SessionEntry {
508 SessionEntry {
509 history: history(title, branch, updated),
510 size_bytes: size,
511 }
512 }
513
514 #[test]
515 fn humanize_relative_scales_and_pluralizes() {
516 let now = at(2026, 7, 2, 12, 0);
517 assert_eq!(humanize_relative(now, now), "just now");
518 assert_eq!(
519 humanize_relative(now, at(2026, 7, 2, 11, 58)),
520 "2 minutes ago"
521 );
522 assert_eq!(humanize_relative(now, at(2026, 7, 2, 11, 0)), "1 hour ago");
523 assert_eq!(humanize_relative(now, at(2026, 7, 1, 12, 0)), "1 day ago");
524 assert_eq!(humanize_relative(now, at(2026, 6, 29, 12, 0)), "3 days ago");
525 assert_eq!(humanize_relative(now, at(2026, 6, 20, 12, 0)), "1 week ago");
526 assert_eq!(
527 humanize_relative(now, at(2026, 5, 1, 12, 0)),
528 "2 months ago"
529 );
530 assert_eq!(humanize_relative(now, at(2026, 7, 2, 12, 30)), "just now");
532 }
533
534 #[test]
535 fn humanize_size_picks_unit() {
536 assert_eq!(humanize_size(512), "512B");
537 assert_eq!(humanize_size(23_100), "22.6KB");
538 assert_eq!(humanize_size(1_367_426), "1.3MB");
539 }
540
541 #[test]
542 fn filtering_matches_title_and_branch_case_insensitively() {
543 let now = at(2026, 7, 2, 12, 0);
544 let mut state = SelectorState::new(vec![
545 entry("Examine the workspace", Some("main"), now, 100),
546 entry("Fix the parser", Some("feature/parser"), now, 200),
547 entry("Unrelated", Some("main"), now, 300),
548 ]);
549 assert_eq!(state.filtered().len(), 3);
551 state.push_query('E');
553 state.push_query('x');
554 assert_eq!(state.filtered(), vec![0]);
555 state.query.clear();
557 for c in "parser".chars() {
558 state.push_query(c);
559 }
560 assert_eq!(state.filtered(), vec![1]);
561 state.query.clear();
563 for c in "zzz".chars() {
564 state.push_query(c);
565 }
566 assert!(state.filtered().is_empty());
567 assert!(state.current().is_none());
568 }
569
570 #[test]
571 fn navigation_is_clamped_to_filtered_set() {
572 let now = at(2026, 7, 2, 12, 0);
573 let mut state =
574 SelectorState::new(vec![entry("A", None, now, 1), entry("B", None, now, 2)]);
575 state.move_up(); assert_eq!(state.selected, 0);
577 state.move_down();
578 assert_eq!(state.selected, 1);
579 state.move_down(); assert_eq!(state.selected, 1);
581 assert_eq!(state.current().unwrap().history.title, "B");
582 }
583
584 #[test]
585 fn render_shows_claude_code_style_rows() {
586 let now = at(2026, 7, 2, 12, 0);
587 let mut state = SelectorState::new(vec![
588 entry(
589 "Examine the workspace",
590 Some("main"),
591 at(2026, 7, 2, 10, 0),
592 1_367_426,
593 ),
594 entry(
595 "Older session",
596 Some("master"),
597 at(2026, 6, 29, 12, 0),
598 18_400_000,
599 ),
600 ]);
601 let backend = TestBackend::new(80, 24);
602 let mut terminal = Terminal::new(backend).unwrap();
603 terminal.draw(|f| render(f, &mut state, now)).unwrap();
604 let buf = terminal.backend().buffer();
605 let mut text = String::new();
606 for y in 0..buf.area.height {
607 for x in 0..buf.area.width {
608 text.push_str(buf[(x, y)].symbol());
609 }
610 text.push('\n');
611 }
612 assert!(text.contains("Resume session"), "header:\n{text}");
613 assert!(text.contains("Search"), "search placeholder:\n{text}");
614 assert!(text.contains("source-clone"), "project name:\n{text}");
615 assert!(text.contains("Examine the workspace"), "title:\n{text}");
616 assert!(text.contains("2 hours ago · main · 1.3MB"), "meta:\n{text}");
618 assert!(
619 text.contains("3 days ago · master · 17.5MB"),
620 "meta2:\n{text}"
621 );
622 assert!(text.contains("> Examine the workspace"), "marker:\n{text}");
624 assert!(text.contains("esc cancel"), "hints:\n{text}");
625 }
626
627 #[test]
628 fn render_empty_filter_shows_no_match_message() {
629 let now = at(2026, 7, 2, 12, 0);
630 let mut state = SelectorState::new(vec![entry("A", None, now, 1)]);
631 for c in "zzz".chars() {
632 state.push_query(c);
633 }
634 let backend = TestBackend::new(80, 24);
635 let mut terminal = Terminal::new(backend).unwrap();
636 terminal.draw(|f| render(f, &mut state, now)).unwrap();
637 let buf = terminal.backend().buffer();
638 let mut text = String::new();
639 for y in 0..buf.area.height {
640 for x in 0..buf.area.width {
641 text.push_str(buf[(x, y)].symbol());
642 }
643 }
644 assert!(text.contains("No sessions match"), "{text}");
645 }
646
647 fn dump(terminal: &Terminal<TestBackend>) -> String {
649 let buf = terminal.backend().buffer();
650 let mut text = String::new();
651 for y in 0..buf.area.height {
652 for x in 0..buf.area.width {
653 text.push_str(buf[(x, y)].symbol());
654 }
655 text.push('\n');
656 }
657 text
658 }
659
660 #[test]
661 fn arrowing_past_the_viewport_scrolls_to_follow_selection() {
662 let now = at(2026, 7, 2, 12, 0);
663 let mut state = SelectorState::new(
666 (0..8)
667 .map(|i| entry(&format!("Session {i}"), None, now, 100))
668 .collect(),
669 );
670 let backend = TestBackend::new(80, 14);
671 let mut terminal = Terminal::new(backend).unwrap();
672 terminal.draw(|f| render(f, &mut state, now)).unwrap();
674 for _ in 0..7 {
675 state.move_down();
676 }
677 terminal.draw(|f| render(f, &mut state, now)).unwrap();
678 let text = dump(&terminal);
679 assert!(
680 text.contains("> Session 7"),
681 "selection followed into view:\n{text}"
682 );
683 assert!(
684 !text.contains("Session 0"),
685 "top entries scrolled away:\n{text}"
686 );
687 }
688
689 #[test]
690 fn wheel_scrolls_viewport_without_moving_selection() {
691 let now = at(2026, 7, 2, 12, 0);
692 let mut state = SelectorState::new(
693 (0..8)
694 .map(|i| entry(&format!("Session {i}"), None, now, 100))
695 .collect(),
696 );
697 let backend = TestBackend::new(80, 14);
698 let mut terminal = Terminal::new(backend).unwrap();
699 terminal.draw(|f| render(f, &mut state, now)).unwrap();
700
701 state.scroll_viewport_down(); terminal.draw(|f| render(f, &mut state, now)).unwrap();
703 let text = dump(&terminal);
704 assert_eq!(state.selected, 0, "the wheel must not move the selection");
705 assert!(
706 !text.contains("Session 0"),
707 "viewport scrolled past the top:\n{text}"
708 );
709 }
710
711 #[test]
712 fn delete_flow_arms_confirm_then_drops_the_entry() {
713 let now = at(2026, 7, 2, 12, 0);
714 let mut state = SelectorState::new(vec![
715 entry("keep", None, now, 1),
716 entry("gone", None, now, 1),
717 ]);
718 state.move_down(); state.request_delete();
720 assert_eq!(
721 state.pending_delete,
722 Some(1),
723 "armed on the highlighted entry"
724 );
725 let idx = state.take_pending_delete().expect("pending");
728 state.remove_entry(idx);
729 assert_eq!(state.entries.len(), 1);
730 assert_eq!(
731 state.current().map(|e| e.history.title.as_str()),
732 Some("keep"),
733 "selection re-clamped onto the surviving entry"
734 );
735 }
736}