1 use std::{io, sync::MutexGuard}; 2 3 use crate::utils::{ 4 terminal::TermManager, 5 ui::{ 6 event::{KeyEventCallback, WarpUiCallBackType}, 7 uicore::{UiCore, CONTENT_WINSIZE}, 8 }, 9 }; 10 11 pub trait CommonOp: KeyEventCallback { 12 fn remove_line(&self, ui: &mut MutexGuard<UiCore>) -> io::Result<()> { 13 TermManager::clear_current_line()?; 14 TermManager::clear_under_cursor()?; 15 let y = ui.cursor.y() as usize; 16 let old_line_count = ui.buffer.line_count(); 17 let old_offset = ui.buffer.offset(); 18 19 let count = old_line_count - y as usize; 20 ui.buffer.delete_line(y + ui.buffer.offset() as usize); 21 ui.render_content(y as u16, count.max(1))?; 22 23 if y + old_offset == old_line_count - 1 { 24 self.up(ui)?; 25 } 26 27 if old_line_count == 1 { 28 ui.cursor.move_to_columu(0)?; 29 ui.buffer.insert_char('\n' as u8, 0, 0); 30 ui.render_content(0, 1)?; 31 } 32 33 Ok(()) 34 } 35 36 fn remove_n_line(&self, ui: &mut MutexGuard<UiCore>, n: u16) -> io::Result<()> { 37 let linecount = ui.buffer.line_count() as u16; 38 let y = ui.cursor.y(); 39 40 // 实际能删除的行数 41 let to_delete = n.min(linecount - y); 42 for _ in 0..to_delete { 43 self.remove_line(ui)?; 44 } 45 Ok(()) 46 } 47 fn remove_word(&self, ui: &mut MutexGuard<UiCore>) -> io::Result<()> { 48 let x = ui.cursor.x(); 49 let y = ui.cursor.y(); 50 let next_word_pos = ui.buffer.search_nextw_begin(x, y); 51 let linesize = ui.buffer.get_linesize(y); 52 53 // 如果下一个单词在当前行，则删除当前单词 54 if next_word_pos < linesize.into() { 55 ui.buffer.remove_str(x, y, next_word_pos - x as usize); 56 } else { 57 // 如果下一个单词在下一行，则删除当前行剩余部分 58 self.left(ui)?; 59 ui.buffer.delete_line(y.into()); 60 self.down(ui)?; 61 } 62 ui.render_content(y, 1)?; 63 return Ok(()); 64 } 65 fn jump_to_next_word(&self, ui: &mut MutexGuard<UiCore>) -> io::Result<WarpUiCallBackType> { 66 let x = ui.cursor.x(); 67 let y = ui.cursor.y(); 68 let pos = ui.buffer.search_nextw_begin(x, y); 69 let linesize = ui.buffer.get_linesize(y); 70 let abs_y = y + ui.buffer.offset() as u16; 71 72 if pos < linesize as usize { 73 // 如果下一个单词在当前行，则移动光标到该单词的起始位置 74 ui.cursor.move_to_columu(pos as u16)?; 75 } else if y as usize + ui.buffer.offset() < ui.buffer.line_count() - 1 { 76 // 如果当前行不是最后一行，则移动到下一行的单词起始位置 77 let next_word_pos = ui.buffer.search_nextw_begin(0, y + 1) as u16; 78 let next_linesize = ui.buffer.get_linesize_abs(abs_y + 1); 79 self.down(ui)?; 80 ui.cursor 81 .move_to_columu(next_word_pos.min(next_linesize - 1))?; 82 ui.cursor.highlight(Some(y))?; 83 } else { 84 // 如果当前行是最后一行，则移动到当前行的末尾 85 ui.cursor.move_to_columu(linesize as u16 - 1)?; 86 } 87 return Ok(WarpUiCallBackType::None); 88 } 89 fn move_to_line(&self, ui: &mut MutexGuard<UiCore>, line: u16) -> io::Result<()> { 90 let x = ui.cursor.x(); 91 let y = ui.cursor.y(); 92 let new_y = ui.buffer.goto_line(line as usize); 93 let new_x = x.min(ui.buffer.get_linesize(new_y)) as u16; 94 ui.cursor.move_to(new_x, new_y)?; 95 ui.render_content(0, CONTENT_WINSIZE.read().unwrap().rows as usize)?; 96 ui.cursor.highlight(Some(y))?; 97 return Ok(()); 98 } 99 100 fn locate_prevw_begin(&self, ui: &mut MutexGuard<UiCore>, x: u16, abs_y: u16) -> (u16, u16) { 101 // 如果光标已在行首，则尝试移动到上一行的单词首字母 102 if x == 0 { 103 if abs_y == 0 { 104 return (0, 0); 105 } 106 let last_y = abs_y - 1; 107 let end_of_prev_line = ui.buffer.get_linesize_abs(last_y) - 1; 108 let prev_word_pos = ui.buffer.search_prevw_begin_abs(end_of_prev_line, last_y); 109 return (prev_word_pos as u16, last_y); 110 } 111 112 let prev_word_pos = ui.buffer.search_prevw_begin_abs(x, abs_y); 113 114 return (prev_word_pos as u16, abs_y); 115 } 116 fn locate_nextw_ending(&self, ui: &mut MutexGuard<UiCore>, x: u16, y: u16) -> (u16, u16) { 117 let linesize = ui.buffer.get_linesize(y) as usize; 118 119 // y的绝对位置 120 let abs_y = ui.buffer.offset() as u16 + y; 121 // 如果光标已经在当前行的末尾或最后一个字符(x + 2)，则尝试移动到下一行的末尾或单词末尾 122 if x as usize + 2 >= linesize { 123 if abs_y < ui.buffer.line_count() as u16 - 1 { 124 let next_end_pos = ui.buffer.search_nextw_end(0, y + 1) as u16; 125 return (next_end_pos, abs_y + 1); 126 } else { 127 // 如果已经是最后一行，则保持光标在当前行的末尾 128 let x = if linesize > 0 { linesize - 1 } else { 0 }; 129 return (x as u16, abs_y); 130 } 131 } 132 133 let next_end_pos = ui.buffer.search_nextw_end(x, y) as u16; 134 // 如果下一个单词的末尾在当前行，则移动光标到该单词的末尾 135 return (next_end_pos.min(linesize as u16 - 1), abs_y); 136 } 137 } 138