more work on vi line editing

This commit is contained in:
2025-05-20 02:55:41 -04:00
parent b30de3232b
commit 25534d4cd9
4 changed files with 519 additions and 152 deletions

View File

@@ -1,6 +1,6 @@
use std::ops::Range; use std::ops::Range;
use crate::{libsh::error::ShResult, prompt::readline::linecmd::Anchor}; use crate::{libsh::{error::ShResult, term::{Style, Styled}}, prompt::readline::linecmd::Anchor};
use super::linecmd::{At, CharSearch, MoveCmd, Movement, Verb, VerbCmd, Word}; use super::linecmd::{At, CharSearch, MoveCmd, Movement, Verb, VerbCmd, Word};
@@ -19,16 +19,91 @@ impl LineBuf {
self.buffer = init.to_string().chars().collect(); self.buffer = init.to_string().chars().collect();
self self
} }
pub fn display_lines(&self) -> Vec<String> {
let line_bullet = "".styled(Style::Dim);
self.split_lines()
.into_iter()
.enumerate()
.map(|(i, line)| {
if i == 0 {
line.to_string()
} else {
format!("{line_bullet}{line}")
}
})
.collect()
}
pub fn repos_cursor(&mut self) {
if self.cursor >= self.len() {
self.cursor = self.len_minus_one();
}
}
pub fn split_lines(&self) -> Vec<String> {
let line = self.prepare_line();
let mut lines = vec![];
let mut cur_line = String::new();
for ch in line.chars() {
match ch {
'\n' => lines.push(std::mem::take(&mut cur_line)),
_ => cur_line.push(ch)
}
}
lines.push(cur_line);
lines
}
pub fn count_lines(&self) -> usize { pub fn count_lines(&self) -> usize {
self.buffer.iter().filter(|&&c| c == '\n').count() self.buffer.iter().filter(|&&c| c == '\n').count()
} }
pub fn cursor(&self) -> usize { pub fn cursor(&self) -> usize {
self.cursor self.cursor
} }
pub fn prepare_line(&self) -> String {
self.buffer
.iter()
.filter(|&&c| c != '\r')
.collect::<String>()
}
pub fn clear(&mut self) { pub fn clear(&mut self) {
self.buffer.clear(); self.buffer.clear();
self.cursor = 0; self.cursor = 0;
} }
pub fn cursor_display_coords(&self) -> (usize, usize) {
let mut x = 0;
let mut y = 0;
for i in 0..self.cursor() {
let ch = self.get_char(i);
match ch {
'\n' => {
y += 1;
x = 0;
}
'\r' => continue,
_ => {
x += 1;
}
}
}
(x, y)
}
pub fn cursor_real_coords(&self) -> (usize,usize) {
let mut x = 0;
let mut y = 0;
for i in 0..self.cursor() {
let ch = self.get_char(i);
match ch {
'\n' => {
y += 1;
x = 0;
}
_ => {
x += 1;
}
}
}
(x, y)
}
pub fn backspace(&mut self) { pub fn backspace(&mut self) {
if self.cursor() == 0 { if self.cursor() == 0 {
return return
@@ -46,14 +121,17 @@ impl LineBuf {
if pos < self.cursor() { if pos < self.cursor() {
self.cursor = self.cursor.saturating_sub(1) self.cursor = self.cursor.saturating_sub(1)
} }
if self.cursor() >= self.buffer.len() { }
self.cursor = self.buffer.len().saturating_sub(1) pub fn insert_at_pos(&mut self, pos: usize, ch: char) {
} self.buffer.insert(pos, ch)
} }
pub fn insert_at_cursor(&mut self, ch: char) { pub fn insert_at_cursor(&mut self, ch: char) {
self.buffer.insert(self.cursor, ch); self.buffer.insert(self.cursor, ch);
self.move_cursor_right(); self.move_cursor_right();
} }
pub fn insert_after_cursor(&mut self, ch: char) {
self.buffer.insert(self.cursor, ch);
}
pub fn backspace_at_cursor(&mut self) { pub fn backspace_at_cursor(&mut self) {
assert!(self.cursor <= self.buffer.len()); assert!(self.cursor <= self.buffer.len());
if self.buffer.is_empty() { if self.buffer.is_empty() {
@@ -108,11 +186,55 @@ impl LineBuf {
pub fn len(&self) -> usize { pub fn len(&self) -> usize {
self.buffer.len() self.buffer.len()
} }
pub fn len_minus_one(&self) -> usize {
self.buffer.len().saturating_sub(1)
}
pub fn is_empty(&self) -> bool { pub fn is_empty(&self) -> bool {
self.buffer.is_empty() self.buffer.is_empty()
} }
pub fn cursor_char(&self) -> char { pub fn cursor_char(&self) -> Option<&char> {
self.buffer[self.cursor] self.buffer.get(self.cursor())
}
pub fn get_char(&self, pos: usize) -> char {
assert!((0..self.len()).contains(&pos));
self.buffer[pos]
}
pub fn prev_char(&self) -> Option<char> {
if self.cursor() == 0 {
None
} else {
Some(self.get_char(self.cursor() - 1))
}
}
pub fn next_char(&self) -> Option<char> {
if self.cursor() == self.len_minus_one() {
None
} else {
Some(self.get_char(self.cursor() + 1))
}
}
pub fn on_word_bound_left(&self) -> bool {
if self.cursor() == 0 {
return false
}
let Some(ch) = self.cursor_char() else {
return false
};
let cur_char_class = CharClass::from(*ch);
let prev_char_pos = self.cursor().saturating_sub(1).max(0);
cur_char_class.is_opposite(self.get_char(prev_char_pos))
}
pub fn on_word_bound_right(&self) -> bool {
if self.cursor() >= self.len_minus_one() {
return false
}
let Some(ch) = self.cursor_char() else {
return false
};
let cur_char_class = CharClass::from(*ch);
let next_char_pos = self.cursor().saturating_add(1).min(self.len());
cur_char_class.is_opposite(self.get_char(next_char_pos))
} }
fn backward_until<F: Fn(usize) -> bool>(&self, mut start: usize, cond: F) -> usize { fn backward_until<F: Fn(usize) -> bool>(&self, mut start: usize, cond: F) -> usize {
while start > 0 && !cond(start) { while start > 0 && !cond(start) {
@@ -126,7 +248,7 @@ impl LineBuf {
} }
start start
} }
pub fn calc_range(&self, movement: &Movement) -> Range<usize> { pub fn calc_range(&mut self, movement: &Movement) -> Range<usize> {
let mut start = self.cursor(); let mut start = self.cursor();
let mut end = self.cursor(); let mut end = self.cursor();
@@ -149,28 +271,73 @@ impl LineBuf {
end = self.forward_until(end, |pos| self.buffer[pos] == '\n'); end = self.forward_until(end, |pos| self.buffer[pos] == '\n');
} }
Movement::BackwardWord(word) => { Movement::BackwardWord(word) => {
let cur_char = self.cursor_char();
match word { match word {
Word::Big => { Word::Big => {
if self.cursor_char().is_none() {
self.cursor = self.cursor.saturating_sub(1);
start = start.saturating_sub(1)
}
// Skip whitespace
let Some(cur_char) = self.cursor_char() else {
return start..end
};
if cur_char.is_whitespace() { if cur_char.is_whitespace() {
start = self.backward_until(start, |pos| !self.buffer[pos].is_whitespace()) start = self.backward_until(start, |pos| !self.buffer[pos].is_whitespace())
} }
let ch_class = CharClass::from(self.get_char(start));
let should_step = self.prev_char().is_some_and(|ch| ch_class.is_opposite(ch));
// If we are on a word boundary, move forward one character
// If we are now on whitespace, skip it
if should_step {
start = start.saturating_sub(1).max(0);
if self.get_char(start).is_whitespace() {
start = self.backward_until(start, |pos| !self.buffer[pos].is_whitespace())
}
}
start = self.backward_until(start, |pos| self.buffer[pos].is_whitespace()); start = self.backward_until(start, |pos| self.buffer[pos].is_whitespace());
start += 1; if self.get_char(start).is_whitespace() {
start += 1;
}
} }
Word::Normal => { Word::Normal => {
if cur_char.is_alphanumeric() || cur_char == '_' { if self.cursor_char().is_none() {
start = self.backward_until(start, |pos| !(self.buffer[pos].is_alphanumeric() || self.buffer[pos] == '_')); self.cursor = self.cursor.saturating_sub(1);
start += 1; start = start.saturating_sub(1)
} else { }
start = self.backward_until(start, |pos| (self.buffer[pos].is_alphanumeric() || self.buffer[pos] == '_')); let Some(cur_char) = self.cursor_char() else {
return start..end
};
// Skip whitespace
if cur_char.is_whitespace() {
start = self.backward_until(start, |pos| !self.get_char(pos).is_whitespace())
}
let ch_class = CharClass::from(self.get_char(start));
let should_step = self.prev_char().is_some_and(|ch| ch_class.is_opposite(ch));
// If we are on a word boundary, move forward one character
// If we are now on whitespace, skip it
if should_step {
start = start.saturating_sub(1).max(0);
if self.get_char(start).is_whitespace() {
start = self.backward_until(start, |pos| !self.get_char(pos).is_whitespace())
}
}
// Find an alternate charclass to stop at
let cur_char = self.get_char(start);
let cur_char_class = CharClass::from(cur_char);
start = self.backward_until(start, |pos| cur_char_class.is_opposite(self.get_char(pos)));
if cur_char_class.is_opposite(self.get_char(start)) {
start += 1; start += 1;
} }
} }
} }
} }
Movement::ForwardWord(at, word) => { Movement::ForwardWord(at, word) => {
let cur_char = self.cursor_char(); let Some(cur_char) = self.cursor_char() else {
return start..end
};
let is_ws = |pos: usize| self.buffer[pos].is_whitespace(); let is_ws = |pos: usize| self.buffer[pos].is_whitespace();
let not_ws = |pos: usize| !self.buffer[pos].is_whitespace(); let not_ws = |pos: usize| !self.buffer[pos].is_whitespace();
@@ -178,38 +345,68 @@ impl LineBuf {
Word::Big => { Word::Big => {
if cur_char.is_whitespace() { if cur_char.is_whitespace() {
end = self.forward_until(end, not_ws); end = self.forward_until(end, not_ws);
} else { }
end = self.forward_until(end, is_ws);
end = self.forward_until(end, not_ws); let ch_class = CharClass::from(self.buffer[end]);
let should_step = self.next_char().is_some_and(|ch| ch_class.is_opposite(ch));
if should_step {
end = end.saturating_add(1).min(self.len_minus_one());
if self.get_char(end).is_whitespace() {
end = self.forward_until(end, |pos| !self.get_char(pos).is_whitespace())
}
} }
match at { match at {
At::Start => {/* Done */} At::Start => {
if !should_step {
end = self.forward_until(end, is_ws);
end = self.forward_until(end, not_ws);
}
}
At::AfterEnd => { At::AfterEnd => {
end = self.forward_until(end, is_ws); end = self.forward_until(end, is_ws);
} }
At::BeforeEnd => { At::BeforeEnd => {
end = self.forward_until(end, is_ws); end = self.forward_until(end, is_ws);
end = end.saturating_sub(1); if self.buffer.get(end).is_some_and(|ch| ch.is_whitespace()) {
end = end.saturating_sub(1);
}
} }
} }
} }
Word::Normal => { Word::Normal => {
let ch_class = CharClass::from(self.buffer[end]);
if cur_char.is_whitespace() { if cur_char.is_whitespace() {
end = self.forward_until(end, not_ws); end = self.forward_until(end, not_ws);
} else { }
end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos]))
let ch_class = CharClass::from(self.buffer[end]);
let should_step = self.next_char().is_some_and(|ch| ch_class.is_opposite(ch));
if should_step {
end = end.saturating_add(1).min(self.len_minus_one());
if self.get_char(end).is_whitespace() {
end = self.forward_until(end, |pos| !self.get_char(pos).is_whitespace())
}
} }
match at { match at {
At::Start => {/* Done */ } At::Start => {
if !should_step {
end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos]));
if self.get_char(end).is_whitespace() {
end = self.forward_until(end, |pos| !self.get_char(pos).is_whitespace())
}
}
}
At::AfterEnd => { At::AfterEnd => {
end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos])); end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos]));
} }
At::BeforeEnd => { At::BeforeEnd => {
end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos])); end = self.forward_until(end, |pos| ch_class.is_opposite(self.buffer[pos]));
end = end.saturating_sub(1); if self.buffer.get(end).is_some_and(|ch| ch.is_whitespace()) {
end = end.saturating_sub(1);
}
} }
} }
} }
@@ -225,51 +422,58 @@ impl LineBuf {
Movement::CharSearch(char_search) => { Movement::CharSearch(char_search) => {
match char_search { match char_search {
CharSearch::FindFwd(ch) => { CharSearch::FindFwd(ch) => {
let search = self.forward_until(end, |pos| self.buffer[pos] == *ch); let ch = ch.unwrap();
end = end.saturating_add(1).min(self.len_minus_one());
let search = self.forward_until(end, |pos| self.buffer[pos] == ch);
// we check anyway because it may have reached the end without finding anything // we check anyway because it may have reached the end without finding anything
if self.buffer[search] == *ch { if self.buffer.get(search).is_some_and(|&s_ch| s_ch == ch) {
end = search; end = search;
} }
} }
CharSearch::FwdTo(ch) => { CharSearch::FwdTo(ch) => {
let search = self.forward_until(end, |pos| self.buffer[pos] == *ch); let ch = ch.unwrap();
end = end.saturating_add(1).min(self.len_minus_one());
let search = self.forward_until(end, |pos| self.buffer[pos] == ch);
// we check anyway because it may have reached the end without finding anything // we check anyway because it may have reached the end without finding anything
if self.buffer[search] == *ch { if self.buffer.get(search).is_some_and(|&s_ch| s_ch == ch) {
end = search.saturating_sub(1); end = search.saturating_sub(1);
} }
} }
CharSearch::FindBkwd(ch) => { CharSearch::FindBkwd(ch) => {
let search = self.forward_until(start, |pos| self.buffer[pos] == *ch); let ch = ch.unwrap();
start = start.saturating_sub(1);
let search = self.backward_until(start, |pos| self.buffer[pos] == ch);
// we check anyway because it may have reached the end without finding anything // we check anyway because it may have reached the end without finding anything
if self.buffer[search] == *ch { if self.buffer.get(search).is_some_and(|&s_ch| s_ch == ch) {
start = search; start = search;
} }
} }
CharSearch::BkwdTo(ch) => { CharSearch::BkwdTo(ch) => {
let search = self.forward_until(start, |pos| self.buffer[pos] == *ch); let ch = ch.unwrap();
start = start.saturating_sub(1);
let search = self.backward_until(start, |pos| self.buffer[pos] == ch);
// we check anyway because it may have reached the end without finding anything // we check anyway because it may have reached the end without finding anything
if self.buffer[search] == *ch { if self.buffer.get(search).is_some_and(|&s_ch| s_ch == ch) {
start = search.saturating_add(1); start = search.saturating_add(1);
} }
} }
} }
} }
Movement::ViFirstPrint => todo!(),
Movement::LineUp => todo!(), Movement::LineUp => todo!(),
Movement::LineDown => todo!(), Movement::LineDown => todo!(),
Movement::WholeBuffer => { Movement::WholeBuffer => {
start = 0; start = 0;
end = self.len().saturating_sub(1); end = self.len_minus_one();
} }
Movement::BeginningOfBuffer => { Movement::BeginningOfBuffer => {
start = 0; start = 0;
} }
Movement::EndOfBuffer => { Movement::EndOfBuffer => {
end = self.len().saturating_sub(1); end = self.len_minus_one();
} }
Movement::Null => {/* nothing */} Movement::Null => {/* nothing */}
} }
@@ -299,6 +503,22 @@ impl LineBuf {
} }
} }
} }
Verb::Breakline(anchor) => {
match anchor {
Anchor::Before => {
let last_newline = self.backward_until(self.cursor(), |pos| self.get_char(pos) == '\n');
self.cursor = last_newline;
self.insert_at_cursor('\n');
self.insert_at_cursor('\r');
}
Anchor::After => {
let next_newline = self.forward_until(self.cursor(), |pos| self.get_char(pos) == '\n');
self.cursor = next_newline;
self.insert_at_cursor('\n');
self.insert_at_cursor('\r');
}
}
}
Verb::InsertChar(ch) => self.insert_at_cursor(ch), Verb::InsertChar(ch) => self.insert_at_cursor(ch),
Verb::InsertMode => todo!(), Verb::InsertMode => todo!(),
Verb::JoinLines => todo!(), Verb::JoinLines => todo!(),
@@ -308,8 +528,22 @@ impl LineBuf {
Verb::Put(_) => todo!(), Verb::Put(_) => todo!(),
Verb::Undo => todo!(), Verb::Undo => todo!(),
Verb::RepeatLast => todo!(), Verb::RepeatLast => todo!(),
Verb::Dedent => todo!(), Verb::Dedent => {
Verb::Indent => todo!(), let mut start_pos = self.backward_until(self.cursor(), |pos| self.get_char(pos) == '\n');
if self.get_char(start_pos) == '\n' {
start_pos += 1;
}
if self.get_char(start_pos) == '\t' {
self.delete_pos(start_pos);
}
}
Verb::Indent => {
let mut line_start = self.backward_until(self.cursor(), |pos| self.get_char(pos) == '\n');
if self.get_char(line_start) == '\n' {
line_start += 1;
}
self.insert_at_pos(line_start, '\t');
}
Verb::ReplaceChar(_) => todo!(), Verb::ReplaceChar(_) => todo!(),
_ => unreachable!() _ => unreachable!()
} }
@@ -333,10 +567,21 @@ impl LineBuf {
Verb::Delete => { Verb::Delete => {
(0..move_count).for_each(|_| { (0..move_count).for_each(|_| {
let range = self.calc_range(&movement); let range = self.calc_range(&movement);
let range = range.start..(range.end + 1).min(self.len());
self.buffer.drain(range);
self.repos_cursor();
});
}
Verb::Change => {
(0..move_count).for_each(|_| {
let range = self.calc_range(&movement);
let range = range.start..(range.end + 1).min(self.len());
self.buffer.drain(range);
self.repos_cursor();
}); });
} }
Verb::DeleteOne(anchor) => todo!(), Verb::DeleteOne(anchor) => todo!(),
Verb::Change => todo!(), Verb::Breakline(anchor) => todo!(),
Verb::Yank => todo!(), Verb::Yank => todo!(),
Verb::ReplaceChar(_) => todo!(), Verb::ReplaceChar(_) => todo!(),
Verb::Substitute => todo!(), Verb::Substitute => todo!(),
@@ -358,13 +603,13 @@ impl LineBuf {
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)] #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub enum CharClass { pub enum CharClass {
AlphaNum, AlphaNum,
Symbol Symbol,
} }
impl CharClass { impl CharClass {
pub fn is_opposite(&self, ch: char) -> bool { pub fn is_opposite(&self, other: char) -> bool {
let opp_class = CharClass::from(ch); let other_class = CharClass::from(other);
opp_class != *self other_class != *self
} }
} }
@@ -379,7 +624,7 @@ impl From<char> for CharClass {
} }
pub fn strip_ansi_codes(s: &str) -> String { pub fn strip_ansi_codes_and_escapes(s: &str) -> String {
let mut out = String::with_capacity(s.len()); let mut out = String::with_capacity(s.len());
let mut chars = s.chars().peekable(); let mut chars = s.chars().peekable();
@@ -395,7 +640,11 @@ pub fn strip_ansi_codes(s: &str) -> String {
chars.next(); // consume intermediate characters chars.next(); // consume intermediate characters
} }
} else { } else {
out.push(c); match c {
'\n' |
'\r' => { /* Continue */ }
_ => out.push(c)
}
} }
} }
out out

View File

@@ -34,6 +34,18 @@ impl ViCmdBuilder {
let Self { verb_count, verb, move_count, movement: _ } = self; let Self { verb_count, verb, move_count, movement: _ } = self;
Self { verb_count, verb, move_count, movement: Some(movement) } Self { verb_count, verb, move_count, movement: Some(movement) }
} }
pub fn verb_count(&self) -> Option<u16> {
self.verb_count
}
pub fn move_count(&self) -> Option<u16> {
self.move_count
}
pub fn movement(&self) -> Option<&Movement> {
self.movement.as_ref()
}
pub fn verb(&self) -> Option<&Verb> {
self.verb.as_ref()
}
pub fn append_digit(&mut self, digit: char) { pub fn append_digit(&mut self, digit: char) {
// Convert char digit to a number (assuming ASCII '0'..'9') // Convert char digit to a number (assuming ASCII '0'..'9')
let digit_val = digit.to_digit(10).expect("digit must be 0-9") as u16; let digit_val = digit.to_digit(10).expect("digit must be 0-9") as u16;
@@ -128,6 +140,7 @@ pub enum Verb {
/// `J` — join lines /// `J` — join lines
JoinLines, JoinLines,
InsertChar(char), InsertChar(char),
Breakline(Anchor),
Indent, Indent,
Dedent Dedent
} }
@@ -147,6 +160,7 @@ impl Verb {
Verb::Dedent | Verb::Dedent |
Verb::Indent | Verb::Indent |
Verb::InsertChar(_) | Verb::InsertChar(_) |
Verb::Breakline(_) |
Verb::ReplaceChar(_) => false, Verb::ReplaceChar(_) => false,
Verb::Delete | Verb::Delete |
Verb::Change | Verb::Change |
@@ -171,8 +185,6 @@ pub enum Movement {
ForwardWord(At, Word), // Forward until start/end of word ForwardWord(At, Word), // Forward until start/end of word
/// character-search, character-search-backward, vi-char-search /// character-search, character-search-backward, vi-char-search
CharSearch(CharSearch), CharSearch(CharSearch),
/// vi-first-print
ViFirstPrint,
/// backward-char /// backward-char
BackwardChar, BackwardChar,
/// forward-char /// forward-char
@@ -200,7 +212,6 @@ impl Movement {
Self::BackwardWord(_) | Self::BackwardWord(_) |
Self::ForwardWord(_, _) | Self::ForwardWord(_, _) |
Self::CharSearch(_) | Self::CharSearch(_) |
Self::ViFirstPrint |
Self::BackwardChar | Self::BackwardChar |
Self::ForwardChar | Self::ForwardChar |
Self::LineUp | Self::LineUp |
@@ -334,10 +345,10 @@ pub enum Anchor {
#[derive(Debug, Clone, Eq, PartialEq, Copy)] #[derive(Debug, Clone, Eq, PartialEq, Copy)]
pub enum CharSearch { pub enum CharSearch {
FindFwd(char), FindFwd(Option<char>),
FwdTo(char), FwdTo(Option<char>),
FindBkwd(char), FindBkwd(Option<char>),
BkwdTo(char) BkwdTo(Option<char>),
} }
#[derive(Debug, Clone, Eq, PartialEq, Copy)] #[derive(Debug, Clone, Eq, PartialEq, Copy)]
@@ -347,13 +358,6 @@ pub enum Word {
} }
const fn repeat_count(previous: RepeatCount, new: Option<RepeatCount>) -> RepeatCount {
match new {
Some(n) => n,
None => previous,
}
}
#[derive(Default,Debug,Clone,Copy,PartialEq,Eq,PartialOrd,Ord)] #[derive(Default,Debug,Clone,Copy,PartialEq,Eq,PartialOrd,Ord)]
pub enum InputMode { pub enum InputMode {
Normal, Normal,

View File

@@ -1,8 +1,8 @@
use std::{arch::asm, os::fd::BorrowedFd}; use std::{arch::asm, os::fd::BorrowedFd};
use keys::KeyEvent; use keys::KeyEvent;
use line::{strip_ansi_codes, LineBuf}; use line::{strip_ansi_codes_and_escapes, LineBuf};
use linecmd::{Anchor, At, InputMode, LineCmd, MoveCmd, Movement, Verb, VerbCmd, ViCmd, ViCmdBuilder, Word}; use linecmd::{Anchor, At, CharSearch, InputMode, LineCmd, MoveCmd, Movement, Verb, VerbCmd, ViCmd, ViCmdBuilder, Word};
use nix::{libc::STDIN_FILENO, sys::termios::{self, Termios}, unistd::read}; use nix::{libc::STDIN_FILENO, sys::termios::{self, Termios}, unistd::read};
use term::Terminal; use term::Terminal;
use unicode_width::UnicodeWidthStr; use unicode_width::UnicodeWidthStr;
@@ -13,6 +13,33 @@ pub mod line;
pub mod keys; pub mod keys;
pub mod linecmd; pub mod linecmd;
/// Add a verb to a specified ViCmdBuilder, then build it
///
/// Returns the built value as a LineCmd::ViCmd
macro_rules! build_verb {
($cmd:expr,$verb:expr) => {{
$cmd.with_verb($verb).build().map(|cmd| LineCmd::ViCmd(cmd))
}}
}
/// Add a movement to a specified ViCmdBuilder, then build it
///
/// Returns the built value as a LineCmd::ViCmd
macro_rules! build_movement {
($cmd:expr,$move:expr) => {{
$cmd.with_movement($move).build().map(|cmd| LineCmd::ViCmd(cmd))
}}
}
/// Add both a movement and a verb to a specified ViCmdBuilder, then build it
///
/// Returns the built value as a LineCmd::ViCmd
macro_rules! build_moveverb {
($cmd:expr,$verb:expr,$move:expr) => {{
$cmd.with_movement($move).with_verb($verb).build().map(|cmd| LineCmd::ViCmd(cmd))
}}
}
#[derive(Default,Debug)] #[derive(Default,Debug)]
pub struct FernReader { pub struct FernReader {
pub term: Terminal, pub term: Terminal,
@@ -70,19 +97,50 @@ impl FernReader {
} }
let mut prompt_lines = self.prompt.lines().peekable(); let mut prompt_lines = self.prompt.lines().peekable();
let mut last_line_len = 0; let mut last_line_len = 0;
let lines = self.line.display_lines();
while let Some(line) = prompt_lines.next() { while let Some(line) = prompt_lines.next() {
if prompt_lines.peek().is_none() { if prompt_lines.peek().is_none() {
last_line_len = strip_ansi_codes(line).width(); last_line_len = strip_ansi_codes_and_escapes(line).width();
self.term.write(line); self.term.write(line);
} else { } else {
self.term.writeln(line); self.term.writeln(line);
} }
} }
let line = self.pack_line(); let num_lines = lines.len();
self.term.write(&line); let mut lines_iter = lines.into_iter().peekable();
let cursor_offset = self.line.cursor() + last_line_len; while let Some(line) = lines_iter.next() {
self.term.write(&format!("\r\x1b[{}C", cursor_offset)); if lines_iter.peek().is_some() {
self.term.writeln(&line);
} else {
self.term.write(&line);
}
}
if num_lines == 1 {
let cursor_offset = self.line.cursor() + last_line_len;
self.term.write(&format!("\r\x1b[{}C", cursor_offset));
} else {
let (x, y) = self.line.cursor_display_coords();
// Y-axis movements are 1-indexed and must move up from the bottom
// Therefore, add 1 to Y and subtract that number from the number of lines
// to find the number of times we have to push the cursor upward
let y = num_lines.saturating_sub(y+1);
if y > 0 {
self.term.write(&format!("\r\x1b[{}A", y))
}
self.term.write(&format!("\r\x1b[{}C", x+2)); // Factor in the line bullet thing
}
match self.edit_mode {
InputMode::Replace |
InputMode::Insert => {
self.term.write("\x1b[6 q")
}
InputMode::Normal |
InputMode::Visual => {
self.term.write("\x1b[2 q")
}
}
} }
pub fn next_cmd(&mut self) -> ShResult<LineCmd> { pub fn next_cmd(&mut self) -> ShResult<LineCmd> {
let vi_cmd = ViCmdBuilder::new(); let vi_cmd = ViCmdBuilder::new();
@@ -97,12 +155,7 @@ impl FernReader {
use keys::{KeyEvent as E, KeyCode as K, ModKeys as M}; use keys::{KeyEvent as E, KeyCode as K, ModKeys as M};
let key = self.term.read_key(); let key = self.term.read_key();
let cmd = match key { let cmd = match key {
E(K::Char(ch), M::NONE) => { E(K::Char(ch), M::NONE) => build_verb!(pending_cmd, Verb::InsertChar(ch))?,
let cmd = pending_cmd
.with_verb(Verb::InsertChar(ch))
.build()?;
LineCmd::ViCmd(cmd)
}
E(K::Char('H'), M::CTRL) | E(K::Char('H'), M::CTRL) |
E(K::Backspace, M::NONE) => LineCmd::backspace(), E(K::Backspace, M::NONE) => LineCmd::backspace(),
@@ -114,10 +167,7 @@ impl FernReader {
E(K::Esc, M::NONE) => { E(K::Esc, M::NONE) => {
self.edit_mode = InputMode::Normal; self.edit_mode = InputMode::Normal;
let cmd = pending_cmd build_movement!(pending_cmd, Movement::BackwardChar)?
.with_movement(Movement::BackwardChar)
.build()?;
LineCmd::ViCmd(cmd)
} }
E(K::Char('D'), M::CTRL) => LineCmd::EndOfFile, E(K::Char('D'), M::CTRL) => LineCmd::EndOfFile,
_ => { _ => {
@@ -131,6 +181,31 @@ impl FernReader {
pub fn get_normal_cmd(&mut self, mut pending_cmd: ViCmdBuilder) -> ShResult<LineCmd> { pub fn get_normal_cmd(&mut self, mut pending_cmd: ViCmdBuilder) -> ShResult<LineCmd> {
use keys::{KeyEvent as E, KeyCode as K, ModKeys as M}; use keys::{KeyEvent as E, KeyCode as K, ModKeys as M};
let key = self.term.read_key(); let key = self.term.read_key();
if let E(K::Char(ch), M::NONE) = key {
if pending_cmd.movement().is_some_and(|m| matches!(m, Movement::CharSearch(_))) {
let Movement::CharSearch(charsearch) = pending_cmd.movement().unwrap() else {unreachable!()};
match charsearch {
CharSearch::FindFwd(_) => {
let finalized = CharSearch::FindFwd(Some(ch));
return build_movement!(pending_cmd, Movement::CharSearch(finalized))
}
CharSearch::FwdTo(_) => {
let finalized = CharSearch::FwdTo(Some(ch));
return build_movement!(pending_cmd, Movement::CharSearch(finalized))
}
CharSearch::FindBkwd(_) => {
let finalized = CharSearch::FindBkwd(Some(ch));
return build_movement!(pending_cmd, Movement::CharSearch(finalized))
}
CharSearch::BkwdTo(_) => {
let finalized = CharSearch::BkwdTo(Some(ch));
return build_movement!(pending_cmd, Movement::CharSearch(finalized))
}
}
}
}
if let E(K::Char(digit @ '0'..='9'), M::NONE) = key { if let E(K::Char(digit @ '0'..='9'), M::NONE) = key {
pending_cmd.append_digit(digit); pending_cmd.append_digit(digit);
return self.get_normal_cmd(pending_cmd); return self.get_normal_cmd(pending_cmd);
@@ -144,55 +219,91 @@ impl FernReader {
} }
E(K::Char('j'), M::NONE) => LineCmd::LineDownOrNextHistory, E(K::Char('j'), M::NONE) => LineCmd::LineDownOrNextHistory,
E(K::Char('k'), M::NONE) => LineCmd::LineUpOrPreviousHistory, E(K::Char('k'), M::NONE) => LineCmd::LineUpOrPreviousHistory,
E(K::Char('l'), M::NONE) => { E(K::Char('D'), M::NONE) => build_moveverb!(pending_cmd,Verb::Delete,Movement::EndOfLine)?,
let cmd = pending_cmd E(K::Char('C'), M::NONE) => build_moveverb!(pending_cmd,Verb::Change,Movement::EndOfLine)?,
.with_movement(Movement::ForwardChar) E(K::Char('Y'), M::NONE) => build_moveverb!(pending_cmd,Verb::Yank,Movement::EndOfLine)?,
.build()?; E(K::Char('l'), M::NONE) => build_movement!(pending_cmd,Movement::ForwardChar)?,
LineCmd::ViCmd(cmd) E(K::Char('w'), M::NONE) => build_movement!(pending_cmd,Movement::ForwardWord(At::Start, Word::Normal))?,
E(K::Char('W'), M::NONE) => build_movement!(pending_cmd,Movement::ForwardWord(At::Start, Word::Big))?,
E(K::Char('b'), M::NONE) => build_movement!(pending_cmd,Movement::BackwardWord(Word::Normal))?,
E(K::Char('B'), M::NONE) => build_movement!(pending_cmd,Movement::BackwardWord(Word::Big))?,
E(K::Char('e'), M::NONE) => build_movement!(pending_cmd,Movement::ForwardWord(At::BeforeEnd, Word::Normal))?,
E(K::Char('E'), M::NONE) => build_movement!(pending_cmd,Movement::ForwardWord(At::BeforeEnd, Word::Big))?,
E(K::Char('^'), M::NONE) => build_movement!(pending_cmd,Movement::BeginningOfFirstWord)?,
E(K::Char('0'), M::NONE) => build_movement!(pending_cmd,Movement::BeginningOfLine)?,
E(K::Char('$'), M::NONE) => build_movement!(pending_cmd,Movement::EndOfLine)?,
E(K::Char('x'), M::NONE) => build_verb!(pending_cmd,Verb::DeleteOne(Anchor::After))?,
E(K::Char('o'), M::NONE) => {
self.edit_mode = InputMode::Insert;
build_verb!(pending_cmd,Verb::Breakline(Anchor::After))?
} }
E(K::Char('w'), M::NONE) => { E(K::Char('O'), M::NONE) => {
let cmd = pending_cmd self.edit_mode = InputMode::Insert;
.with_movement(Movement::ForwardWord(At::Start, Word::Normal)) build_verb!(pending_cmd,Verb::Breakline(Anchor::Before))?
.build()?;
LineCmd::ViCmd(cmd)
}
E(K::Char('W'), M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::ForwardWord(At::Start, Word::Big))
.build()?;
LineCmd::ViCmd(cmd)
}
E(K::Char('b'), M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::BackwardWord(Word::Normal))
.build()?;
LineCmd::ViCmd(cmd)
}
E(K::Char('B'), M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::BackwardWord(Word::Big))
.build()?;
LineCmd::ViCmd(cmd)
}
E(K::Char('x'), M::NONE) => {
let cmd = pending_cmd
.with_verb(Verb::DeleteOne(Anchor::After))
.build()?;
LineCmd::ViCmd(cmd)
} }
E(K::Char('i'), M::NONE) => { E(K::Char('i'), M::NONE) => {
self.edit_mode = InputMode::Insert; self.edit_mode = InputMode::Insert;
let cmd = pending_cmd LineCmd::Null
.with_movement(Movement::BackwardChar)
.build()?;
LineCmd::ViCmd(cmd)
} }
E(K::Char('I'), M::NONE) => { E(K::Char('I'), M::NONE) => {
self.edit_mode = InputMode::Insert; self.edit_mode = InputMode::Insert;
let cmd = pending_cmd build_movement!(pending_cmd,Movement::BeginningOfFirstWord)?
.with_movement(Movement::BeginningOfFirstWord) }
.build()?; E(K::Char('a'), M::NONE) => {
LineCmd::ViCmd(cmd) self.edit_mode = InputMode::Insert;
build_movement!(pending_cmd,Movement::ForwardChar)?
}
E(K::Char('A'), M::NONE) => {
self.edit_mode = InputMode::Insert;
build_movement!(pending_cmd,Movement::EndOfLine)?
}
E(K::Char('c'), M::NONE) => {
if pending_cmd.verb() == Some(&Verb::Change) {
build_moveverb!(pending_cmd,Verb::Change,Movement::WholeLine)?
} else {
pending_cmd = pending_cmd.with_verb(Verb::Change);
self.get_normal_cmd(pending_cmd)?
}
}
E(K::Char('>'), M::NONE) => {
if pending_cmd.verb() == Some(&Verb::Indent) {
build_verb!(pending_cmd,Verb::Indent)?
} else {
pending_cmd = pending_cmd.with_verb(Verb::Indent);
self.get_normal_cmd(pending_cmd)?
}
}
E(K::Char('<'), M::NONE) => {
if pending_cmd.verb() == Some(&Verb::Dedent) {
build_verb!(pending_cmd,Verb::Dedent)?
} else {
pending_cmd = pending_cmd.with_verb(Verb::Dedent);
self.get_normal_cmd(pending_cmd)?
}
}
E(K::Char('d'), M::NONE) => {
if pending_cmd.verb() == Some(&Verb::Delete) {
LineCmd::ViCmd(pending_cmd.with_movement(Movement::WholeLine).build()?)
} else {
pending_cmd = pending_cmd.with_verb(Verb::Delete);
self.get_normal_cmd(pending_cmd)?
}
}
E(K::Char('f'), M::NONE) => {
pending_cmd = pending_cmd.with_movement(Movement::CharSearch(CharSearch::FindFwd(None)));
self.get_normal_cmd(pending_cmd)?
}
E(K::Char('F'), M::NONE) => {
pending_cmd = pending_cmd.with_movement(Movement::CharSearch(CharSearch::FindBkwd(None)));
self.get_normal_cmd(pending_cmd)?
}
E(K::Char('t'), M::NONE) => {
pending_cmd = pending_cmd.with_movement(Movement::CharSearch(CharSearch::FwdTo(None)));
self.get_normal_cmd(pending_cmd)?
}
E(K::Char('T'), M::NONE) => {
pending_cmd = pending_cmd.with_movement(Movement::CharSearch(CharSearch::BkwdTo(None)));
self.get_normal_cmd(pending_cmd)?
} }
_ => { _ => {
flog!(INFO, "unhandled key in get_normal_cmd, trying common_cmd..."); flog!(INFO, "unhandled key in get_normal_cmd, trying common_cmd...");
@@ -205,44 +316,18 @@ impl FernReader {
pub fn common_cmd(&mut self, key: KeyEvent, pending_cmd: ViCmdBuilder) -> ShResult<LineCmd> { pub fn common_cmd(&mut self, key: KeyEvent, pending_cmd: ViCmdBuilder) -> ShResult<LineCmd> {
use keys::{KeyEvent as E, KeyCode as K, ModKeys as M}; use keys::{KeyEvent as E, KeyCode as K, ModKeys as M};
match key { match key {
E(K::Home, M::NONE) => { E(K::Home, M::NONE) => build_movement!(pending_cmd,Movement::BeginningOfLine),
let cmd = pending_cmd E(K::End, M::NONE) => build_movement!(pending_cmd,Movement::EndOfLine),
.with_movement(Movement::BeginningOfLine) E(K::Left, M::NONE) => build_movement!(pending_cmd,Movement::BackwardChar),
.build()?; E(K::Right, M::NONE) => build_movement!(pending_cmd,Movement::ForwardChar),
Ok(LineCmd::ViCmd(cmd)) E(K::Delete, M::NONE) => build_moveverb!(pending_cmd,Verb::Delete,Movement::ForwardChar),
} E(K::Up, M::NONE) => Ok(LineCmd::LineUpOrPreviousHistory),
E(K::End, M::NONE) => { E(K::Down, M::NONE) => Ok(LineCmd::LineDownOrNextHistory),
let cmd = pending_cmd E(K::Enter, M::NONE) => Ok(LineCmd::AcceptLine),
.with_movement(Movement::EndOfLine)
.build()?;
Ok(LineCmd::ViCmd(cmd))
}
E(K::Left, M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::BackwardChar)
.build()?;
Ok(LineCmd::ViCmd(cmd))
}
E(K::Right, M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::ForwardChar)
.build()?;
Ok(LineCmd::ViCmd(cmd))
}
E(K::Delete, M::NONE) => {
let cmd = pending_cmd
.with_movement(Movement::ForwardChar)
.with_verb(Verb::Delete)
.build()?;
Ok(LineCmd::ViCmd(cmd))
}
E(K::Backspace, M::NONE) | E(K::Backspace, M::NONE) |
E(K::Char('h'), M::CTRL) => { E(K::Char('h'), M::CTRL) => {
Ok(LineCmd::backspace()) Ok(LineCmd::backspace())
} }
E(K::Up, M::NONE) => Ok(LineCmd::LineUpOrPreviousHistory),
E(K::Down, M::NONE) => Ok(LineCmd::LineDownOrNextHistory),
E(K::Enter, M::NONE) => Ok(LineCmd::AcceptLine),
_ => Err(ShErr::simple(ShErrKind::ReadlineErr,format!("Unhandled common key event: {key:?}"))) _ => Err(ShErr::simple(ShErrKind::ReadlineErr,format!("Unhandled common key event: {key:?}")))
} }
} }
@@ -251,10 +336,14 @@ impl FernReader {
ViCmd::MoveVerb(verb_cmd, move_cmd) => { ViCmd::MoveVerb(verb_cmd, move_cmd) => {
self.last_effect = Some(verb_cmd.clone()); self.last_effect = Some(verb_cmd.clone());
self.last_movement = Some(move_cmd.clone()); self.last_movement = Some(move_cmd.clone());
let VerbCmd { verb_count, verb } = verb_cmd; let VerbCmd { verb_count, verb } = verb_cmd;
for _ in 0..verb_count { for _ in 0..verb_count {
self.line.exec_vi_cmd(Some(verb.clone()), Some(move_cmd.clone()))?; self.line.exec_vi_cmd(Some(verb.clone()), Some(move_cmd.clone()))?;
} }
if verb == Verb::Change {
self.edit_mode = InputMode::Insert
}
} }
ViCmd::Verb(verb_cmd) => { ViCmd::Verb(verb_cmd) => {
self.last_effect = Some(verb_cmd.clone()); self.last_effect = Some(verb_cmd.clone());
@@ -318,3 +407,9 @@ impl FernReader {
} }
} }
impl Drop for FernReader {
fn drop(&mut self) {
self.term.write("\x1b[2 q");
}
}

View File

@@ -92,6 +92,25 @@ impl Terminal {
} }
KeyEvent(KeyCode::Null, ModKeys::empty()) KeyEvent(KeyCode::Null, ModKeys::empty())
} }
pub fn cursor_pos(&self) -> (usize, usize) {
self.write("\x1b[6n");
let mut buf = [0u8;32];
let n = self.read_byte(&mut buf);
let response = std::str::from_utf8(&buf[..n]).unwrap_or("");
let mut row = 0;
let mut col = 0;
if let Some(caps) = response.strip_prefix("\x1b[").and_then(|s| s.strip_suffix("R")) {
let mut parts = caps.split(';');
if let (Some(rowstr), Some(colstr)) = (parts.next(), parts.next()) {
row = rowstr.parse().unwrap_or(1);
col = colstr.parse().unwrap_or(1);
}
}
(row,col)
}
} }
impl Default for Terminal { impl Default for Terminal {