use ratatui::style::{Modifier, Style};
use ratatui::text::Span;
use crate::ui::theme::Theme;
pub fn text_cursor_spans(
content: &str,
cursor: usize,
viewport_offset: usize,
visible_width: usize,
) -> Vec<Span<'static>> {
if visible_width == 0 {
return vec![];
}
let chars: Vec<char> = content.chars().collect();
let total = chars.len();
let vp = viewport_offset.min(total);
let view_end = vp.saturating_add(visible_width).min(total);
let visible: Vec<char> = chars[vp..view_end].to_vec();
let cursor_in_view = cursor.saturating_sub(vp);
let cursor_style = Style::default()
.bg(Theme::CURSOR_FG)
.fg(Theme::SELECTION_BG)
.add_modifier(Modifier::BOLD);
if cursor >= total {
let text: String = visible.iter().collect();
vec![
Span::raw(text),
Span::styled("\u{2588}", Style::default().fg(Theme::CURSOR_FG)),
]
} else if cursor_in_view < visible.len() {
let before: String = visible[..cursor_in_view].iter().collect();
let cursor_char: String = visible[cursor_in_view].to_string();
let after: String = visible[cursor_in_view + 1..].iter().collect();
vec![
Span::raw(before),
Span::styled(cursor_char, cursor_style),
Span::raw(after),
]
} else {
let text: String = visible.iter().collect();
vec![Span::raw(text)]
}
}
#[cfg(test)]
mod tests {
use super::*;
fn spans_to_strings(spans: &[Span<'_>]) -> Vec<String> {
spans.iter().map(|s| s.content.to_string()).collect()
}
#[test]
fn cursor_at_beginning() {
let spans = text_cursor_spans("abc", 0, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["", "a", "bc"]);
}
#[test]
fn cursor_at_middle() {
let spans = text_cursor_spans("abc", 1, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["a", "b", "c"]);
}
#[test]
fn cursor_at_end() {
let spans = text_cursor_spans("abc", 3, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["abc", "\u{2588}"]);
}
#[test]
fn empty_string() {
let spans = text_cursor_spans("", 0, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["", "\u{2588}"]);
}
#[test]
fn multibyte_characters() {
let spans = text_cursor_spans("あいう", 1, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["あ", "い", "う"]);
}
#[test]
fn viewport_offset_positive() {
let spans = text_cursor_spans("abcdef", 3, 2, 3);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["c", "d", "e"]);
}
#[test]
fn viewport_offset_beyond_text_length() {
let spans = text_cursor_spans("abc", 3, 10, 5);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["", "\u{2588}"]);
}
#[test]
fn visible_width_one() {
let spans = text_cursor_spans("abc", 1, 1, 1);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["", "b", ""]);
}
#[test]
fn visible_width_zero() {
let spans = text_cursor_spans("abc", 1, 0, 0);
assert!(spans.is_empty());
}
#[test]
fn visible_width_usize_max_sentinel() {
let spans = text_cursor_spans("hello", 2, 0, usize::MAX);
let texts = spans_to_strings(&spans);
assert_eq!(texts, vec!["he", "l", "lo"]);
}
}