use crate::keymap::{At, CharSearch, Movement, RepeatCount, Word};
use std::cell::RefCell;
use std::cmp::min;
use std::fmt;
use std::iter;
use std::ops::{Deref, Index, Range};
use std::rc::Rc;
use std::string::Drain;
use std::sync::{Arc, Mutex};
use unicode_segmentation::UnicodeSegmentation;
pub(crate) const MAX_LINE: usize = 4096;
pub(crate) const INDENT: &str = " ";
#[derive(Clone, Copy)]
pub enum WordAction {
Capitalize,
Lowercase,
Uppercase,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub(crate) enum Direction {
Forward,
Backward,
}
impl Default for Direction {
fn default() -> Self {
Direction::Forward
}
}
pub(crate) trait DeleteListener {
fn start_killing(&mut self);
fn delete(&mut self, idx: usize, string: &str, dir: Direction);
fn stop_killing(&mut self);
}
pub(crate) trait ChangeListener: DeleteListener {
fn insert_char(&mut self, idx: usize, c: char);
fn insert_str(&mut self, idx: usize, string: &str);
fn replace(&mut self, idx: usize, old: &str, new: &str);
}
pub struct LineBuffer {
buf: String, pos: usize, can_growth: bool, dl: Option<Arc<Mutex<dyn DeleteListener>>>,
cl: Option<Rc<RefCell<dyn ChangeListener>>>,
}
impl fmt::Debug for LineBuffer {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("LineBuffer")
.field("buf", &self.buf)
.field("pos", &self.pos)
.finish()
}
}
impl LineBuffer {
#[must_use]
pub fn with_capacity(capacity: usize) -> Self {
Self {
buf: String::with_capacity(capacity),
pos: 0,
can_growth: false,
dl: None,
cl: None,
}
}
pub(crate) fn can_growth(mut self, can_growth: bool) -> Self {
self.can_growth = can_growth;
self
}
fn must_truncate(&self, new_len: usize) -> bool {
!self.can_growth && new_len > self.buf.capacity()
}
#[cfg(test)]
pub(crate) fn init(
line: &str,
pos: usize,
cl: Option<Rc<RefCell<dyn ChangeListener>>>,
) -> Self {
let mut lb = Self::with_capacity(MAX_LINE);
assert!(lb.insert_str(0, line));
lb.set_pos(pos);
lb.cl = cl;
lb
}
pub(crate) fn set_delete_listener(&mut self, dl: Arc<Mutex<dyn DeleteListener>>) {
self.dl = Some(dl);
}
pub(crate) fn set_change_listener(&mut self, dl: Rc<RefCell<dyn ChangeListener>>) {
self.cl = Some(dl);
}
#[must_use]
pub fn as_str(&self) -> &str {
&self.buf
}
#[must_use]
pub fn into_string(self) -> String {
self.buf
}
#[must_use]
pub fn pos(&self) -> usize {
self.pos
}
pub fn set_pos(&mut self, pos: usize) {
assert!(pos <= self.buf.len());
self.pos = pos;
}
#[must_use]
pub fn len(&self) -> usize {
self.buf.len()
}
#[must_use]
pub fn is_empty(&self) -> bool {
self.buf.is_empty()
}
pub fn update(&mut self, buf: &str, pos: usize) {
assert!(pos <= buf.len());
let end = self.len();
self.drain(0..end, Direction::default());
let max = self.buf.capacity();
if self.must_truncate(buf.len()) {
self.insert_str(0, &buf[..max]);
if pos > max {
self.pos = max;
} else {
self.pos = pos;
}
} else {
self.insert_str(0, buf);
self.pos = pos;
}
}
fn end_of_line(&self) -> usize {
if let Some(n) = self.buf[self.pos..].find('\n') {
n + self.pos
} else {
self.buf.len()
}
}
fn start_of_line(&self) -> usize {
if let Some(i) = self.buf[..self.pos].rfind('\n') {
i + 1
} else {
0
}
}
pub(crate) fn grapheme_at_cursor(&self) -> Option<&str> {
if self.pos == self.buf.len() {
None
} else {
self.buf[self.pos..].graphemes(true).next()
}
}
#[must_use]
pub fn next_pos(&self, n: RepeatCount) -> Option<usize> {
if self.pos == self.buf.len() {
return None;
}
self.buf[self.pos..]
.grapheme_indices(true)
.take(n)
.last()
.map(|(i, s)| i + self.pos + s.len())
}
fn prev_pos(&self, n: RepeatCount) -> Option<usize> {
if self.pos == 0 {
return None;
}
self.buf[..self.pos]
.grapheme_indices(true)
.rev()
.take(n)
.last()
.map(|(i, _)| i)
}
pub fn insert(&mut self, ch: char, n: RepeatCount) -> Option<bool> {
let shift = ch.len_utf8() * n;
if self.must_truncate(self.buf.len() + shift) {
return None;
}
let push = self.pos == self.buf.len();
if n == 1 {
self.buf.insert(self.pos, ch);
for cl in &self.cl {
if let Ok(mut cl) = cl.try_borrow_mut() {
cl.insert_char(self.pos, ch);
} }
} else {
let text = iter::repeat(ch).take(n).collect::<String>();
let pos = self.pos;
self.insert_str(pos, &text);
}
self.pos += shift;
Some(push)
}
pub fn yank(&mut self, text: &str, n: RepeatCount) -> Option<bool> {
let shift = text.len() * n;
if text.is_empty() || self.must_truncate(self.buf.len() + shift) {
return None;
}
let push = self.pos == self.buf.len();
let pos = self.pos;
if n == 1 {
self.insert_str(pos, text);
} else {
let text = text.repeat(n);
self.insert_str(pos, &text);
}
self.pos += shift;
Some(push)
}
pub fn yank_pop(&mut self, yank_size: usize, text: &str) -> Option<bool> {
let end = self.pos;
let start = end - yank_size;
self.drain(start..end, Direction::default());
self.pos -= yank_size;
self.yank(text, 1)
}
pub fn move_backward(&mut self, n: RepeatCount) -> bool {
match self.prev_pos(n) {
Some(pos) => {
self.pos = pos;
true
}
None => false,
}
}
pub fn move_forward(&mut self, n: RepeatCount) -> bool {
match self.next_pos(n) {
Some(pos) => {
self.pos = pos;
true
}
None => false,
}
}
pub fn move_buffer_start(&mut self) -> bool {
if self.pos > 0 {
self.pos = 0;
true
} else {
false
}
}
pub fn move_buffer_end(&mut self) -> bool {
if self.pos == self.buf.len() {
false
} else {
self.pos = self.buf.len();
true
}
}
pub fn move_home(&mut self) -> bool {
let start = self.start_of_line();
if self.pos > start {
self.pos = start;
true
} else {
false
}
}
pub fn move_end(&mut self) -> bool {
let end = self.end_of_line();
if self.pos == end {
false
} else {
self.pos = end;
true
}
}
#[must_use]
pub fn is_end_of_input(&self) -> bool {
self.pos >= self.buf.trim_end().len()
}
pub fn delete(&mut self, n: RepeatCount) -> Option<String> {
match self.next_pos(n) {
Some(pos) => {
let start = self.pos;
let chars = self
.drain(start..pos, Direction::Forward)
.collect::<String>();
Some(chars)
}
None => None,
}
}
pub fn backspace(&mut self, n: RepeatCount) -> bool {
match self.prev_pos(n) {
Some(pos) => {
let end = self.pos;
self.drain(pos..end, Direction::Backward);
self.pos = pos;
true
}
None => false,
}
}
pub fn kill_line(&mut self) -> bool {
if !self.buf.is_empty() && self.pos < self.buf.len() {
let start = self.pos;
let end = self.end_of_line();
if start == end {
self.delete(1);
} else {
self.drain(start..end, Direction::Forward);
}
true
} else {
false
}
}
pub fn kill_buffer(&mut self) -> bool {
if !self.buf.is_empty() && self.pos < self.buf.len() {
let start = self.pos;
let end = self.buf.len();
self.drain(start..end, Direction::Forward);
true
} else {
false
}
}
pub fn discard_line(&mut self) -> bool {
if self.pos > 0 && !self.buf.is_empty() {
let start = self.start_of_line();
let end = self.pos;
if end == start {
self.backspace(1)
} else {
self.drain(start..end, Direction::Backward);
self.pos = start;
true
}
} else {
false
}
}
pub fn discard_buffer(&mut self) -> bool {
if self.pos > 0 && !self.buf.is_empty() {
let end = self.pos;
self.drain(0..end, Direction::Backward);
self.pos = 0;
true
} else {
false
}
}
pub fn transpose_chars(&mut self) -> bool {
if self.pos == 0 || self.buf.graphemes(true).count() < 2 {
return false;
}
if self.pos == self.buf.len() {
self.move_backward(1);
}
let chars = self.delete(1).unwrap();
self.move_backward(1);
self.yank(&chars, 1);
self.move_forward(1);
true
}
fn prev_word_pos(&self, pos: usize, word_def: Word, n: RepeatCount) -> Option<usize> {
if pos == 0 {
return None;
}
let mut sow = 0;
let mut gis = self.buf[..pos].grapheme_indices(true).rev();
'outer: for _ in 0..n {
sow = 0;
let mut gj = gis.next();
'inner: loop {
if let Some((j, y)) = gj {
let gi = gis.next();
if let Some((_, x)) = gi {
if is_start_of_word(word_def, x, y) {
sow = j;
break 'inner;
}
gj = gi;
} else {
break 'outer;
}
} else {
break 'outer;
}
}
}
Some(sow)
}
pub fn move_to_prev_word(&mut self, word_def: Word, n: RepeatCount) -> bool {
if let Some(pos) = self.prev_word_pos(self.pos, word_def, n) {
self.pos = pos;
true
} else {
false
}
}
pub fn delete_prev_word(&mut self, word_def: Word, n: RepeatCount) -> bool {
if let Some(pos) = self.prev_word_pos(self.pos, word_def, n) {
let end = self.pos;
self.drain(pos..end, Direction::Backward);
self.pos = pos;
true
} else {
false
}
}
fn next_word_pos(&self, pos: usize, at: At, word_def: Word, n: RepeatCount) -> Option<usize> {
if pos == self.buf.len() {
return None;
}
let mut wp = 0;
let mut gis = self.buf[pos..].grapheme_indices(true);
let mut gi = if at == At::BeforeEnd {
gis.next()
} else {
None
};
'outer: for _ in 0..n {
wp = 0;
gi = gis.next();
'inner: loop {
if let Some((i, x)) = gi {
let gj = gis.next();
if let Some((j, y)) = gj {
if at == At::Start && is_start_of_word(word_def, x, y) {
wp = j;
break 'inner;
} else if at != At::Start && is_end_of_word(word_def, x, y) {
if word_def == Word::Emacs || at == At::AfterEnd {
wp = j;
} else {
wp = i;
}
break 'inner;
}
gi = gj;
} else {
break 'outer;
}
} else {
break 'outer;
}
}
}
if wp == 0 {
if word_def == Word::Emacs || at == At::AfterEnd {
Some(self.buf.len())
} else {
match gi {
Some((i, _)) if i != 0 => Some(i + pos),
_ => None,
}
}
} else {
Some(wp + pos)
}
}
pub fn move_to_next_word(&mut self, at: At, word_def: Word, n: RepeatCount) -> bool {
if let Some(pos) = self.next_word_pos(self.pos, at, word_def, n) {
self.pos = pos;
true
} else {
false
}
}
pub fn move_to_line_up(&mut self, n: RepeatCount) -> bool {
match self.buf[..self.pos].rfind('\n') {
Some(off) => {
let column = self.buf[off + 1..self.pos].graphemes(true).count();
let mut dest_start = self.buf[..off].rfind('\n').map_or(0, |n| n + 1);
let mut dest_end = off;
for _ in 1..n {
if dest_start == 0 {
break;
}
dest_end = dest_start - 1;
dest_start = self.buf[..dest_end].rfind('\n').map_or(0, |n| n + 1);
}
let gidx = self.buf[dest_start..dest_end]
.grapheme_indices(true)
.nth(column);
self.pos = gidx.map_or(off, |(idx, _)| dest_start + idx); true
}
None => false,
}
}
fn n_lines_up(&self, n: RepeatCount) -> Option<(usize, usize)> {
let mut start = if let Some(off) = self.buf[..self.pos].rfind('\n') {
off + 1
} else {
return None;
};
let end = self.buf[self.pos..]
.find('\n')
.map_or_else(|| self.buf.len(), |x| self.pos + x + 1);
for _ in 0..n {
if let Some(off) = self.buf[..start - 1].rfind('\n') {
start = off + 1;
} else {
start = 0;
break;
}
}
Some((start, end))
}
fn n_lines_down(&self, n: RepeatCount) -> Option<(usize, usize)> {
let mut end = if let Some(off) = self.buf[self.pos..].find('\n') {
self.pos + off + 1
} else {
return None;
};
let start = self.buf[..self.pos].rfind('\n').unwrap_or(0);
for _ in 0..n {
if let Some(off) = self.buf[end..].find('\n') {
end = end + off + 1;
} else {
end = self.buf.len();
break;
};
}
Some((start, end))
}
pub fn move_to_line_down(&mut self, n: RepeatCount) -> bool {
match self.buf[self.pos..].find('\n') {
Some(off) => {
let line_start = self.buf[..self.pos].rfind('\n').map_or(0, |n| n + 1);
let column = self.buf[line_start..self.pos].graphemes(true).count();
let mut dest_start = self.pos + off + 1;
let mut dest_end = self.buf[dest_start..]
.find('\n')
.map_or_else(|| self.buf.len(), |v| dest_start + v);
for _ in 1..n {
if dest_end == self.buf.len() {
break;
}
dest_start = dest_end + 1;
dest_end = self.buf[dest_start..]
.find('\n')
.map_or_else(|| self.buf.len(), |v| dest_start + v);
}
self.pos = self.buf[dest_start..dest_end]
.grapheme_indices(true)
.nth(column)
.map_or(dest_end, |(idx, _)| dest_start + idx); debug_assert!(self.pos <= self.buf.len());
true
}
None => false,
}
}
fn search_char_pos(&self, cs: CharSearch, n: RepeatCount) -> Option<usize> {
let mut shift = 0;
let search_result = match cs {
CharSearch::Backward(c) | CharSearch::BackwardAfter(c) => self.buf[..self.pos]
.char_indices()
.rev()
.filter(|&(_, ch)| ch == c)
.take(n)
.last()
.map(|(i, _)| i),
CharSearch::Forward(c) | CharSearch::ForwardBefore(c) => {
if let Some(cc) = self.grapheme_at_cursor() {
shift = self.pos + cc.len();
if shift < self.buf.len() {
self.buf[shift..]
.char_indices()
.filter(|&(_, ch)| ch == c)
.take(n)
.last()
.map(|(i, _)| i)
} else {
None
}
} else {
None
}
}
};
search_result.map(|pos| match cs {
CharSearch::Backward(_) => pos,
CharSearch::BackwardAfter(c) => pos + c.len_utf8(),
CharSearch::Forward(_) => shift + pos,
CharSearch::ForwardBefore(_) => {
shift + pos
- self.buf[..shift + pos]
.chars()
.next_back()
.unwrap()
.len_utf8()
}
})
}
pub fn move_to(&mut self, cs: CharSearch, n: RepeatCount) -> bool {
if let Some(pos) = self.search_char_pos(cs, n) {
self.pos = pos;
true
} else {
false
}
}
pub fn delete_word(&mut self, at: At, word_def: Word, n: RepeatCount) -> bool {
if let Some(pos) = self.next_word_pos(self.pos, at, word_def, n) {
let start = self.pos;
self.drain(start..pos, Direction::Forward);
true
} else {
false
}
}
pub fn delete_to(&mut self, cs: CharSearch, n: RepeatCount) -> bool {
let search_result = match cs {
CharSearch::ForwardBefore(c) => self.search_char_pos(CharSearch::Forward(c), n),
_ => self.search_char_pos(cs, n),
};
if let Some(pos) = search_result {
match cs {
CharSearch::Backward(_) | CharSearch::BackwardAfter(_) => {
let end = self.pos;
self.pos = pos;
self.drain(pos..end, Direction::Backward);
}
CharSearch::ForwardBefore(_) => {
let start = self.pos;
self.drain(start..pos, Direction::Forward);
}
CharSearch::Forward(c) => {
let start = self.pos;
self.drain(start..pos + c.len_utf8(), Direction::Forward);
}
};
true
} else {
false
}
}
fn skip_whitespace(&self) -> Option<usize> {
if self.pos == self.buf.len() {
return None;
}
self.buf[self.pos..]
.grapheme_indices(true)
.find_map(|(i, ch)| {
if ch.chars().all(char::is_alphanumeric) {
Some(i)
} else {
None
}
})
.map(|i| i + self.pos)
}
pub fn edit_word(&mut self, a: WordAction) -> bool {
if let Some(start) = self.skip_whitespace() {
if let Some(end) = self.next_word_pos(start, At::AfterEnd, Word::Emacs, 1) {
if start == end {
return false;
}
let word = self
.drain(start..end, Direction::default())
.collect::<String>();
let result = match a {
WordAction::Capitalize => {
let ch = word.graphemes(true).next().unwrap();
let cap = ch.to_uppercase();
cap + &word[ch.len()..].to_lowercase()
}
WordAction::Lowercase => word.to_lowercase(),
WordAction::Uppercase => word.to_uppercase(),
};
self.insert_str(start, &result);
self.pos = start + result.len();
return true;
}
}
false
}
pub fn transpose_words(&mut self, n: RepeatCount) -> bool {
let word_def = Word::Emacs;
self.move_to_next_word(At::AfterEnd, word_def, n);
let w2_end = self.pos;
self.move_to_prev_word(word_def, 1);
let w2_beg = self.pos;
self.move_to_prev_word(word_def, n);
let w1_beg = self.pos;
self.move_to_next_word(At::AfterEnd, word_def, 1);
let w1_end = self.pos;
if w1_beg == w2_beg || w2_beg < w1_end {
return false;
}
let w1 = self.buf[w1_beg..w1_end].to_owned();
let w2 = self
.drain(w2_beg..w2_end, Direction::default())
.collect::<String>();
self.insert_str(w2_beg, &w1);
self.drain(w1_beg..w1_end, Direction::default());
self.insert_str(w1_beg, &w2);
self.pos = w2_end;
true
}
pub fn replace(&mut self, range: Range<usize>, text: &str) {
let start = range.start;
for cl in &self.cl {
if let Ok(mut cl) = cl.try_borrow_mut() {
cl.replace(start, self.buf.index(range.clone()), text);
} }
self.buf.drain(range);
if start == self.buf.len() {
self.buf.push_str(text);
} else {
self.buf.insert_str(start, text);
}
self.pos = start + text.len();
}
pub fn insert_str(&mut self, idx: usize, s: &str) -> bool {
for cl in &self.cl {
if let Ok(mut cl) = cl.try_borrow_mut() {
cl.insert_str(idx, s);
} }
if idx == self.buf.len() {
self.buf.push_str(s);
true
} else {
self.buf.insert_str(idx, s);
false
}
}
pub fn delete_range(&mut self, range: Range<usize>) {
self.set_pos(range.start);
self.drain(range, Direction::default());
}
fn drain(&mut self, range: Range<usize>, dir: Direction) -> Drain<'_> {
for dl in &self.dl {
let lock = dl.try_lock();
if let Ok(mut dl) = lock {
dl.delete(range.start, &self.buf[range.start..range.end], dir);
}
}
for cl in &self.cl {
if let Ok(mut cl) = cl.try_borrow_mut() {
cl.delete(range.start, &self.buf[range.start..range.end], dir);
} }
self.buf.drain(range)
}
#[must_use]
pub fn copy(&self, mvt: &Movement) -> Option<String> {
if self.is_empty() {
return None;
}
match *mvt {
Movement::WholeLine => {
let start = self.start_of_line();
let end = self.end_of_line();
if start == end {
None
} else {
Some(self.buf[start..self.pos].to_owned())
}
}
Movement::BeginningOfLine => {
let start = self.start_of_line();
if self.pos == start {
None
} else {
Some(self.buf[start..self.pos].to_owned())
}
}
Movement::ViFirstPrint => {
if self.pos == 0 {
None
} else {
self.next_word_pos(0, At::Start, Word::Big, 1)
.map(|pos| self.buf[pos..self.pos].to_owned())
}
}
Movement::EndOfLine => {
let end = self.end_of_line();
if self.pos == end {
None
} else {
Some(self.buf[self.pos..end].to_owned())
}
}
Movement::EndOfBuffer => {
if self.pos == self.buf.len() {
None
} else {
Some(self.buf[self.pos..].to_owned())
}
}
Movement::WholeBuffer => {
if self.buf.is_empty() {
None
} else {
Some(self.buf.clone())
}
}
Movement::BeginningOfBuffer => {
if self.pos == 0 {
None
} else {
Some(self.buf[..self.pos].to_owned())
}
}
Movement::BackwardWord(n, word_def) => self
.prev_word_pos(self.pos, word_def, n)
.map(|pos| self.buf[pos..self.pos].to_owned()),
Movement::ForwardWord(n, at, word_def) => self
.next_word_pos(self.pos, at, word_def, n)
.map(|pos| self.buf[self.pos..pos].to_owned()),
Movement::ViCharSearch(n, cs) => {
let search_result = match cs {
CharSearch::ForwardBefore(c) => self.search_char_pos(CharSearch::Forward(c), n),
_ => self.search_char_pos(cs, n),
};
search_result.map(|pos| match cs {
CharSearch::Backward(_) | CharSearch::BackwardAfter(_) => {
self.buf[pos..self.pos].to_owned()
}
CharSearch::ForwardBefore(_) => self.buf[self.pos..pos].to_owned(),
CharSearch::Forward(c) => self.buf[self.pos..pos + c.len_utf8()].to_owned(),
})
}
Movement::BackwardChar(n) => self
.prev_pos(n)
.map(|pos| self.buf[pos..self.pos].to_owned()),
Movement::ForwardChar(n) => self
.next_pos(n)
.map(|pos| self.buf[self.pos..pos].to_owned()),
Movement::LineUp(n) => {
if let Some((start, end)) = self.n_lines_up(n) {
Some(self.buf[start..end].to_owned())
} else {
None
}
}
Movement::LineDown(n) => {
if let Some((start, end)) = self.n_lines_down(n) {
Some(self.buf[start..end].to_owned())
} else {
None
}
}
}
}
pub fn kill(&mut self, mvt: &Movement) -> bool {
let notify = !matches!(*mvt, Movement::ForwardChar(_) | Movement::BackwardChar(_));
if notify {
if let Some(dl) = self.dl.as_ref() {
let mut dl = dl.lock().unwrap();
dl.start_killing();
}
}
let killed = match *mvt {
Movement::ForwardChar(n) => {
self.delete(n).is_some()
}
Movement::BackwardChar(n) => {
self.backspace(n)
}
Movement::EndOfLine => {
self.kill_line()
}
Movement::WholeLine => {
self.move_home();
self.kill_line()
}
Movement::BeginningOfLine => {
self.discard_line()
}
Movement::BackwardWord(n, word_def) => {
self.delete_prev_word(word_def, n)
}
Movement::ForwardWord(n, at, word_def) => {
self.delete_word(at, word_def, n)
}
Movement::ViCharSearch(n, cs) => self.delete_to(cs, n),
Movement::LineUp(n) => {
if let Some((start, end)) = self.n_lines_up(n) {
self.delete_range(start..end);
true
} else {
false
}
}
Movement::LineDown(n) => {
if let Some((start, end)) = self.n_lines_down(n) {
self.delete_range(start..end);
true
} else {
false
}
}
Movement::ViFirstPrint => {
false }
Movement::EndOfBuffer => {
self.kill_buffer()
}
Movement::BeginningOfBuffer => {
self.discard_buffer()
}
Movement::WholeBuffer => {
self.move_buffer_start();
self.kill_buffer()
}
};
if notify {
if let Some(dl) = self.dl.as_ref() {
let mut dl = dl.lock().unwrap();
dl.stop_killing();
}
}
killed
}
pub fn indent(&mut self, mvt: &Movement, amount: usize, dedent: bool) -> bool {
let pair = match *mvt {
Movement::WholeLine
| Movement::BeginningOfLine
| Movement::ViFirstPrint
| Movement::EndOfLine
| Movement::BackwardChar(..)
| Movement::ForwardChar(..)
| Movement::ViCharSearch(..) => Some((self.pos, self.pos)),
Movement::EndOfBuffer => Some((self.pos, self.buf.len())),
Movement::WholeBuffer => Some((0, self.buf.len())),
Movement::BeginningOfBuffer => Some((0, self.pos)),
Movement::BackwardWord(n, word_def) => self
.prev_word_pos(self.pos, word_def, n)
.map(|pos| (pos, self.pos)),
Movement::ForwardWord(n, at, word_def) => self
.next_word_pos(self.pos, at, word_def, n)
.map(|pos| (self.pos, pos)),
Movement::LineUp(n) => self.n_lines_up(n),
Movement::LineDown(n) => self.n_lines_down(n),
};
let (start, end) = pair.unwrap_or((self.pos, self.pos));
let start = self.buf[..start].rfind('\n').map_or(0, |pos| pos + 1);
let end = self.buf[end..]
.rfind('\n')
.map_or_else(|| self.buf.len(), |pos| end + pos);
let mut index = start;
if dedent {
for line in self.buf[start..end].to_string().split('\n') {
let max = line.len() - line.trim_start().len();
let deleting = min(max, amount);
self.drain(index..index + deleting, Direction::default());
if self.pos >= index {
if self.pos.saturating_sub(index) < deleting {
self.pos = index;
} else {
self.pos -= deleting;
}
}
index += line.len() + 1 - deleting;
}
} else {
for line in self.buf[start..end].to_string().split('\n') {
for off in (0..amount).step_by(INDENT.len()) {
self.insert_str(index, &INDENT[..min(amount - off, INDENT.len())]);
}
if self.pos >= index {
self.pos += amount;
}
index += amount + line.len() + 1;
}
}
true
}
}
impl Deref for LineBuffer {
type Target = str;
fn deref(&self) -> &str {
self.as_str()
}
}
fn is_start_of_word(word_def: Word, previous: &str, grapheme: &str) -> bool {
(!is_word_char(word_def, previous) && is_word_char(word_def, grapheme))
|| (word_def == Word::Vi && !is_other_char(previous) && is_other_char(grapheme))
}
fn is_end_of_word(word_def: Word, grapheme: &str, next: &str) -> bool {
(!is_word_char(word_def, next) && is_word_char(word_def, grapheme))
|| (word_def == Word::Vi && !is_other_char(next) && is_other_char(grapheme))
}
fn is_word_char(word_def: Word, grapheme: &str) -> bool {
match word_def {
Word::Emacs => grapheme.chars().all(char::is_alphanumeric),
Word::Vi => is_vi_word_char(grapheme),
Word::Big => !grapheme.chars().any(char::is_whitespace),
}
}
fn is_vi_word_char(grapheme: &str) -> bool {
grapheme.chars().all(char::is_alphanumeric) || grapheme == "_"
}
fn is_other_char(grapheme: &str) -> bool {
!(grapheme.chars().any(char::is_whitespace) || is_vi_word_char(grapheme))
}
#[cfg(test)]
mod test {
use super::{ChangeListener, DeleteListener, Direction, LineBuffer, WordAction, MAX_LINE};
use crate::keymap::{At, CharSearch, Word};
use std::cell::RefCell;
use std::rc::Rc;
struct Listener {
deleted_str: Option<String>,
}
impl Listener {
fn new() -> Rc<RefCell<Listener>> {
let l = Listener { deleted_str: None };
Rc::new(RefCell::new(l))
}
fn assert_deleted_str_eq(&self, expected: &str) {
let actual = self.deleted_str.as_ref().expect("no deleted string");
assert_eq!(expected, actual)
}
}
impl DeleteListener for Listener {
fn start_killing(&mut self) {}
fn delete(&mut self, _: usize, string: &str, _: Direction) {
self.deleted_str = Some(string.to_owned());
}
fn stop_killing(&mut self) {}
}
impl ChangeListener for Listener {
fn insert_char(&mut self, _: usize, _: char) {}
fn insert_str(&mut self, _: usize, _: &str) {}
fn replace(&mut self, _: usize, _: &str, _: &str) {}
}
#[test]
fn next_pos() {
let s = LineBuffer::init("ö̲g̈", 0, None);
assert_eq!(7, s.len());
let pos = s.next_pos(1);
assert_eq!(Some(4), pos);
let s = LineBuffer::init("ö̲g̈", 4, None);
let pos = s.next_pos(1);
assert_eq!(Some(7), pos);
}
#[test]
fn prev_pos() {
let s = LineBuffer::init("ö̲g̈", 4, None);
assert_eq!(7, s.len());
let pos = s.prev_pos(1);
assert_eq!(Some(0), pos);
let s = LineBuffer::init("ö̲g̈", 7, None);
let pos = s.prev_pos(1);
assert_eq!(Some(4), pos);
}
#[test]
fn insert() {
let mut s = LineBuffer::with_capacity(MAX_LINE);
let push = s.insert('α', 1).unwrap();
assert_eq!("α", s.buf);
assert_eq!(2, s.pos);
assert!(push);
let push = s.insert('ß', 1).unwrap();
assert_eq!("αß", s.buf);
assert_eq!(4, s.pos);
assert!(push);
s.pos = 0;
let push = s.insert('γ', 1).unwrap();
assert_eq!("γαß", s.buf);
assert_eq!(2, s.pos);
assert!(!push);
}
#[test]
fn yank_after() {
let mut s = LineBuffer::init("αß", 2, None);
s.move_forward(1);
let ok = s.yank("γδε", 1);
assert_eq!(Some(true), ok);
assert_eq!("αßγδε", s.buf);
assert_eq!(10, s.pos);
}
#[test]
fn yank_before() {
let mut s = LineBuffer::init("αε", 2, None);
let ok = s.yank("ßγδ", 1);
assert_eq!(Some(false), ok);
assert_eq!("αßγδε", s.buf);
assert_eq!(8, s.pos);
}
#[test]
fn moves() {
let mut s = LineBuffer::init("αß", 4, None);
let ok = s.move_backward(1);
assert_eq!("αß", s.buf);
assert_eq!(2, s.pos);
assert!(ok);
let ok = s.move_forward(1);
assert_eq!("αß", s.buf);
assert_eq!(4, s.pos);
assert!(ok);
let ok = s.move_home();
assert_eq!("αß", s.buf);
assert_eq!(0, s.pos);
assert!(ok);
let ok = s.move_end();
assert_eq!("αß", s.buf);
assert_eq!(4, s.pos);
assert!(ok);
}
#[test]
fn move_home_end_multiline() {
let text = "αa\nsdf ßc\nasdf";
let mut s = LineBuffer::init(text, 7, None);
let ok = s.move_home();
assert_eq!(text, s.buf);
assert_eq!(4, s.pos);
assert!(ok);
let ok = s.move_home();
assert_eq!(text, s.buf);
assert_eq!(4, s.pos);
assert!(!ok);
let ok = s.move_end();
assert_eq!(text, s.buf);
assert_eq!(11, s.pos);
assert!(ok);
let ok = s.move_end();
assert_eq!(text, s.buf);
assert_eq!(11, s.pos);
assert!(!ok);
}
#[test]
fn move_buffer_multiline() {
let text = "αa\nsdf ßc\nasdf";
let mut s = LineBuffer::init(text, 7, None);
let ok = s.move_buffer_start();
assert_eq!(text, s.buf);
assert_eq!(0, s.pos);
assert!(ok);
let ok = s.move_buffer_start();
assert_eq!(text, s.buf);
assert_eq!(0, s.pos);
assert!(!ok);
let ok = s.move_buffer_end();
assert_eq!(text, s.buf);
assert_eq!(text.len(), s.pos);
assert!(ok);
let ok = s.move_buffer_end();
assert_eq!(text, s.buf);
assert_eq!(text.len(), s.pos);
assert!(!ok);
}
#[test]
fn move_grapheme() {
let mut s = LineBuffer::init("ag̈", 4, None);
assert_eq!(4, s.len());
let ok = s.move_backward(1);
assert!(ok);
assert_eq!(1, s.pos);
let ok = s.move_forward(1);
assert!(ok);
assert_eq!(4, s.pos);
}
#[test]
fn delete() {
let cl = Listener::new();
let mut s = LineBuffer::init("αß", 2, Some(cl.clone()));
let chars = s.delete(1);
assert_eq!("α", s.buf);
assert_eq!(2, s.pos);
assert_eq!(Some("ß".to_owned()), chars);
let ok = s.backspace(1);
assert_eq!("", s.buf);
assert_eq!(0, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("α");
}
#[test]
fn kill() {
let cl = Listener::new();
let mut s = LineBuffer::init("αßγδε", 6, Some(cl.clone()));
let ok = s.kill_line();
assert_eq!("αßγ", s.buf);
assert_eq!(6, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("δε");
s.pos = 4;
let ok = s.discard_line();
assert_eq!("γ", s.buf);
assert_eq!(0, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("αß");
}
#[test]
fn kill_multiline() {
let cl = Listener::new();
let mut s = LineBuffer::init("αß\nγδ 12\nε f4", 7, Some(cl.clone()));
let ok = s.kill_line();
assert_eq!("αß\nγ\nε f4", s.buf);
assert_eq!(7, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("δ 12");
let ok = s.kill_line();
assert_eq!("αß\nγε f4", s.buf);
assert_eq!(7, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("\n");
let ok = s.kill_line();
assert_eq!("αß\nγ", s.buf);
assert_eq!(7, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ε f4");
let ok = s.kill_line();
assert_eq!(7, s.pos);
assert!(!ok);
}
#[test]
fn discard_multiline() {
let cl = Listener::new();
let mut s = LineBuffer::init("αß\nc γδε", 9, Some(cl.clone()));
let ok = s.discard_line();
assert_eq!("αß\nδε", s.buf);
assert_eq!(5, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("c γ");
let ok = s.discard_line();
assert_eq!("αßδε", s.buf);
assert_eq!(4, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("\n");
let ok = s.discard_line();
assert_eq!("δε", s.buf);
assert_eq!(0, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("αß");
let ok = s.discard_line();
assert_eq!(0, s.pos);
assert!(!ok);
}
#[test]
fn transpose() {
let mut s = LineBuffer::init("aßc", 1, None);
let ok = s.transpose_chars();
assert_eq!("ßac", s.buf);
assert_eq!(3, s.pos);
assert!(ok);
s.buf = String::from("aßc");
s.pos = 3;
let ok = s.transpose_chars();
assert_eq!("acß", s.buf);
assert_eq!(4, s.pos);
assert!(ok);
s.buf = String::from("aßc");
s.pos = 4;
let ok = s.transpose_chars();
assert_eq!("acß", s.buf);
assert_eq!(4, s.pos);
assert!(ok);
}
#[test]
fn move_to_prev_word() {
let mut s = LineBuffer::init("a ß c", 6, None); let ok = s.move_to_prev_word(Word::Emacs, 1);
assert_eq!("a ß c", s.buf);
assert_eq!(2, s.pos); assert!(ok);
assert!(s.move_end()); assert_eq!(7, s.pos);
let ok = s.move_to_prev_word(Word::Emacs, 1);
assert!(ok);
assert_eq!(6, s.pos); let ok = s.move_to_prev_word(Word::Emacs, 2);
assert!(ok);
assert_eq!(0, s.pos);
}
#[test]
fn move_to_prev_vi_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 19, None);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(17, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(15, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(12, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(11, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(7, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(6, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(ok);
assert_eq!(0, s.pos);
let ok = s.move_to_prev_word(Word::Vi, 1);
assert!(!ok);
}
#[test]
fn move_to_prev_big_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 19, None);
let ok = s.move_to_prev_word(Word::Big, 1);
assert!(ok);
assert_eq!(17, s.pos);
let ok = s.move_to_prev_word(Word::Big, 1);
assert!(ok);
assert_eq!(6, s.pos);
let ok = s.move_to_prev_word(Word::Big, 1);
assert!(ok);
assert_eq!(0, s.pos);
let ok = s.move_to_prev_word(Word::Big, 1);
assert!(!ok);
}
#[test]
fn move_to_forward() {
let mut s = LineBuffer::init("αßγδε", 2, None);
let ok = s.move_to(CharSearch::ForwardBefore('ε'), 1);
assert!(ok);
assert_eq!(6, s.pos);
let mut s = LineBuffer::init("αßγδε", 2, None);
let ok = s.move_to(CharSearch::Forward('ε'), 1);
assert!(ok);
assert_eq!(8, s.pos);
let mut s = LineBuffer::init("αßγδε", 2, None);
let ok = s.move_to(CharSearch::Forward('ε'), 10);
assert!(ok);
assert_eq!(8, s.pos);
}
#[test]
fn move_to_backward() {
let mut s = LineBuffer::init("αßγδε", 8, None);
let ok = s.move_to(CharSearch::BackwardAfter('ß'), 1);
assert!(ok);
assert_eq!(4, s.pos);
let mut s = LineBuffer::init("αßγδε", 8, None);
let ok = s.move_to(CharSearch::Backward('ß'), 1);
assert!(ok);
assert_eq!(2, s.pos);
}
#[test]
fn delete_prev_word() {
let cl = Listener::new();
let mut s = LineBuffer::init("a ß c", 6, Some(cl.clone()));
let ok = s.delete_prev_word(Word::Big, 1);
assert_eq!("a c", s.buf);
assert_eq!(2, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ß ");
}
#[test]
fn move_to_next_word() {
let mut s = LineBuffer::init("a ß c", 1, None); let ok = s.move_to_next_word(At::AfterEnd, Word::Emacs, 1);
assert_eq!("a ß c", s.buf);
assert!(ok);
assert_eq!(4, s.pos); let ok = s.move_to_next_word(At::AfterEnd, Word::Emacs, 1);
assert!(ok);
assert_eq!(7, s.pos); s.move_home();
let ok = s.move_to_next_word(At::AfterEnd, Word::Emacs, 1);
assert!(ok);
assert_eq!(1, s.pos); let ok = s.move_to_next_word(At::AfterEnd, Word::Emacs, 2);
assert!(ok);
assert_eq!(7, s.pos); }
#[test]
fn move_to_end_of_word() {
let mut s = LineBuffer::init("a ßeta c", 1, None);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert_eq!("a ßeta c", s.buf);
assert_eq!(6, s.pos);
assert!(ok);
}
#[test]
fn move_to_end_of_vi_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 0, None);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(4, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(6, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(10, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(11, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(14, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(15, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(ok);
assert_eq!(18, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Vi, 1);
assert!(!ok);
}
#[test]
fn move_to_end_of_big_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 0, None);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Big, 1);
assert!(ok);
assert_eq!(4, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Big, 1);
assert!(ok);
assert_eq!(15, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Big, 1);
assert!(ok);
assert_eq!(18, s.pos);
let ok = s.move_to_next_word(At::BeforeEnd, Word::Big, 1);
assert!(!ok);
}
#[test]
fn move_to_start_of_word() {
let mut s = LineBuffer::init("a ß c", 2, None);
let ok = s.move_to_next_word(At::Start, Word::Emacs, 1);
assert_eq!("a ß c", s.buf);
assert_eq!(6, s.pos);
assert!(ok);
}
#[test]
fn move_to_start_of_vi_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 0, None);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(6, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(7, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(11, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(12, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(15, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(17, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(ok);
assert_eq!(18, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Vi, 1);
assert!(!ok);
}
#[test]
fn move_to_start_of_big_word() {
let mut s = LineBuffer::init("alpha ,beta/rho; mu", 0, None);
let ok = s.move_to_next_word(At::Start, Word::Big, 1);
assert!(ok);
assert_eq!(6, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Big, 1);
assert!(ok);
assert_eq!(17, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Big, 1);
assert!(ok);
assert_eq!(18, s.pos);
let ok = s.move_to_next_word(At::Start, Word::Big, 1);
assert!(!ok);
}
#[test]
fn delete_word() {
let cl = Listener::new();
let mut s = LineBuffer::init("a ß c", 1, Some(cl.clone()));
let ok = s.delete_word(At::AfterEnd, Word::Emacs, 1);
assert_eq!("a c", s.buf);
assert_eq!(1, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq(" ß");
let mut s = LineBuffer::init("test", 0, Some(cl.clone()));
let ok = s.delete_word(At::AfterEnd, Word::Vi, 1);
assert_eq!("", s.buf);
assert_eq!(0, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("test");
}
#[test]
fn delete_til_start_of_word() {
let cl = Listener::new();
let mut s = LineBuffer::init("a ß c", 2, Some(cl.clone()));
let ok = s.delete_word(At::Start, Word::Emacs, 1);
assert_eq!("a c", s.buf);
assert_eq!(2, s.pos);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ß ");
}
#[test]
fn delete_to_forward() {
let cl = Listener::new();
let mut s = LineBuffer::init("αßγδε", 2, Some(cl.clone()));
let ok = s.delete_to(CharSearch::ForwardBefore('ε'), 1);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ßγδ");
assert_eq!("αε", s.buf);
assert_eq!(2, s.pos);
let mut s = LineBuffer::init("αßγδε", 2, Some(cl.clone()));
let ok = s.delete_to(CharSearch::Forward('ε'), 1);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ßγδε");
assert_eq!("α", s.buf);
assert_eq!(2, s.pos);
}
#[test]
fn delete_to_backward() {
let cl = Listener::new();
let mut s = LineBuffer::init("αßγδε", 8, Some(cl.clone()));
let ok = s.delete_to(CharSearch::BackwardAfter('α'), 1);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ßγδ");
assert_eq!("αε", s.buf);
assert_eq!(2, s.pos);
let mut s = LineBuffer::init("αßγδε", 8, Some(cl.clone()));
let ok = s.delete_to(CharSearch::Backward('ß'), 1);
assert!(ok);
cl.borrow().assert_deleted_str_eq("ßγδ");
assert_eq!("αε", s.buf);
assert_eq!(2, s.pos);
}
#[test]
fn edit_word() {
let mut s = LineBuffer::init("a ßeta c", 1, None);
assert!(s.edit_word(WordAction::Uppercase));
assert_eq!("a SSETA c", s.buf);
assert_eq!(7, s.pos);
let mut s = LineBuffer::init("a ßetA c", 1, None);
assert!(s.edit_word(WordAction::Lowercase));
assert_eq!("a ßeta c", s.buf);
assert_eq!(7, s.pos);
let mut s = LineBuffer::init("a ßETA c", 1, None);
assert!(s.edit_word(WordAction::Capitalize));
assert_eq!("a SSeta c", s.buf);
assert_eq!(7, s.pos);
let mut s = LineBuffer::init("test", 1, None);
assert!(s.edit_word(WordAction::Capitalize));
assert_eq!("tEst", s.buf);
assert_eq!(4, s.pos);
}
#[test]
fn transpose_words() {
let mut s = LineBuffer::init("ßeta / δelta__", 15, None);
assert!(s.transpose_words(1));
assert_eq!("δelta__ / ßeta", s.buf);
assert_eq!(16, s.pos);
let mut s = LineBuffer::init("ßeta / δelta", 14, None);
assert!(s.transpose_words(1));
assert_eq!("δelta / ßeta", s.buf);
assert_eq!(14, s.pos);
let mut s = LineBuffer::init(" / δelta", 8, None);
assert!(!s.transpose_words(1));
let mut s = LineBuffer::init("ßeta / __", 9, None);
assert!(!s.transpose_words(1));
}
#[test]
fn move_by_line() {
let text = "aa123\nsdf bc\nasdf";
let mut s = LineBuffer::init(text, 14, None);
let ok = s.move_to_line_up(1);
assert_eq!(7, s.pos);
assert!(ok);
let ok = s.move_to_line_up(1);
assert_eq!(1, s.pos);
assert!(ok);
let ok = s.move_to_line_up(1);
assert_eq!(1, s.pos);
assert!(!ok);
let ok = s.move_to_line_down(1);
assert_eq!(7, s.pos);
assert!(ok);
let ok = s.move_to_line_down(1);
assert_eq!(14, s.pos);
assert!(ok);
let ok = s.move_to_line_down(1);
assert_eq!(14, s.pos);
assert!(!ok);
let ok = s.move_to_line_up(2);
assert_eq!(1, s.pos);
assert!(ok);
let ok = s.move_to_line_down(2);
assert_eq!(14, s.pos);
assert!(ok);
}
}