mirror of https://github.com/TaKO8Ki/gobang
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
272 lines
8.3 KiB
Rust
272 lines
8.3 KiB
Rust
use super::{is_nonalphanumeric, is_whitespace};
|
|
use crate::components::compute_character_width;
|
|
use crate::event::Key;
|
|
use unicode_width::UnicodeWidthStr;
|
|
|
|
pub struct Input {
|
|
pub value: Vec<char>,
|
|
pub cursor_position: u16,
|
|
pub cursor_index: usize,
|
|
}
|
|
|
|
impl Input {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
value: Vec::new(),
|
|
cursor_index: 0,
|
|
cursor_position: 0,
|
|
}
|
|
}
|
|
|
|
pub fn value_str(&self) -> String {
|
|
self.value.iter().collect()
|
|
}
|
|
|
|
pub fn value_width(&self) -> u16 {
|
|
self.value_str().width() as u16
|
|
}
|
|
|
|
pub fn reset(&mut self) {
|
|
self.value = Vec::new();
|
|
self.cursor_index = 0;
|
|
self.cursor_position = 0;
|
|
}
|
|
|
|
fn cannot_go_left(&self) -> bool {
|
|
self.value.is_empty() || self.cursor_index == 0 || self.value_width() == 0
|
|
}
|
|
|
|
fn cannot_go_right(&self) -> bool {
|
|
self.cursor_index == self.value.len()
|
|
}
|
|
|
|
fn cursor_index_backwards_until(&self, char_fun: &dyn Fn(char) -> bool) -> usize {
|
|
let mut result = 0;
|
|
|
|
for i in (0..self.cursor_index - 1).rev() {
|
|
if char_fun(self.value[i]) {
|
|
result = i + 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
fn delete_left_until(&mut self, new_cursor_index: usize) {
|
|
let mut tail = self.value.to_vec().drain(self.cursor_index..).collect();
|
|
|
|
self.cursor_index = new_cursor_index;
|
|
self.value.truncate(new_cursor_index);
|
|
self.cursor_position = self.value_width();
|
|
self.value.append(&mut tail);
|
|
}
|
|
|
|
pub fn handle_key(&mut self, key: Key) -> (Option<Key>, bool) {
|
|
match key {
|
|
Key::Char(c) => {
|
|
self.value.insert(self.cursor_index, c);
|
|
self.cursor_index += 1;
|
|
self.cursor_position += compute_character_width(c);
|
|
|
|
return (Some(key), true);
|
|
}
|
|
Key::Delete | Key::Backspace => {
|
|
if self.cannot_go_left() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
let last_c = self.value.remove(self.cursor_index - 1);
|
|
self.cursor_index -= 1;
|
|
self.cursor_position -= compute_character_width(last_c);
|
|
return (Some(key), true);
|
|
}
|
|
Key::Right => {
|
|
if self.cannot_go_right() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
let next_c = self.value[self.cursor_index];
|
|
self.cursor_index += 1;
|
|
self.cursor_position += compute_character_width(next_c);
|
|
return (Some(key), true);
|
|
}
|
|
Key::Ctrl('e') => {
|
|
if self.cannot_go_right() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
self.cursor_index = self.value.len();
|
|
self.cursor_position = self.value_width();
|
|
return (Some(key), true);
|
|
}
|
|
Key::Left => {
|
|
if self.cannot_go_left() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
self.cursor_index -= 1;
|
|
self.cursor_position = self
|
|
.cursor_position
|
|
.saturating_sub(compute_character_width(self.value[self.cursor_index]));
|
|
return (Some(key), true);
|
|
}
|
|
Key::Ctrl('a') => {
|
|
if self.cannot_go_left() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
self.cursor_index = 0;
|
|
self.cursor_position = 0;
|
|
return (Some(key), true);
|
|
}
|
|
Key::Ctrl('w') => {
|
|
if self.cannot_go_left() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
let new_cursor_index = self.cursor_index_backwards_until(&is_whitespace);
|
|
self.delete_left_until(new_cursor_index);
|
|
|
|
return (Some(key), true);
|
|
}
|
|
Key::AltBackspace => {
|
|
if self.cannot_go_left() {
|
|
return (Some(key), false);
|
|
}
|
|
|
|
let new_cursor_index = self.cursor_index_backwards_until(&is_nonalphanumeric);
|
|
self.delete_left_until(new_cursor_index);
|
|
|
|
return (Some(key), true);
|
|
}
|
|
_ => (None, false),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
use super::Input;
|
|
use crate::components::compute_character_width;
|
|
use crate::event::Key;
|
|
|
|
#[test]
|
|
fn test_adds_new_chars_for_char_key() {
|
|
let mut input = Input::new();
|
|
input.handle_key(Key::Char('a'));
|
|
|
|
assert_eq!(input.value, vec!['a']);
|
|
assert_eq!(input.cursor_index, 1);
|
|
assert_eq!(input.cursor_position, compute_character_width('a'));
|
|
}
|
|
|
|
#[test]
|
|
fn test_deletes_chars_for_backspace_and_delete_key() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a', 'b'];
|
|
input.cursor_index = 2;
|
|
input.cursor_position = input.value_width();
|
|
|
|
input.handle_key(Key::Delete);
|
|
input.handle_key(Key::Backspace);
|
|
|
|
assert_eq!(input.value, Vec::<char>::new());
|
|
assert_eq!(input.cursor_index, 0);
|
|
assert_eq!(input.cursor_position, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_moves_cursor_left_for_left_key() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a'];
|
|
input.cursor_index = 1;
|
|
input.cursor_position = compute_character_width('a');
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::Left);
|
|
|
|
assert_eq!(matched_key, Some(Key::Left));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a']);
|
|
assert_eq!(input.cursor_index, 0);
|
|
assert_eq!(input.cursor_position, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_moves_cursor_right_for_right_key() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a'];
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::Right);
|
|
|
|
assert_eq!(matched_key, Some(Key::Right));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a']);
|
|
assert_eq!(input.cursor_index, 1);
|
|
assert_eq!(input.cursor_position, compute_character_width('a'));
|
|
}
|
|
|
|
#[test]
|
|
fn test_jumps_to_beginning_for_ctrl_a() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a', 'b', 'c'];
|
|
input.cursor_index = 3;
|
|
input.cursor_position = input.value_width();
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::Ctrl('a'));
|
|
|
|
assert_eq!(matched_key, Some(Key::Ctrl('a')));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a', 'b', 'c']);
|
|
assert_eq!(input.cursor_index, 0);
|
|
assert_eq!(input.cursor_position, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_jumps_to_end_for_ctrl_e() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a', 'b', 'c'];
|
|
input.cursor_index = 0;
|
|
input.cursor_position = 0;
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::Ctrl('e'));
|
|
|
|
assert_eq!(matched_key, Some(Key::Ctrl('e')));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a', 'b', 'c']);
|
|
assert_eq!(input.cursor_index, 3);
|
|
assert_eq!(input.cursor_position, input.value_width());
|
|
}
|
|
|
|
#[test]
|
|
fn test_deletes_word_for_ctrl_w() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a', ' ', 'c', 'd'];
|
|
input.cursor_index = 3;
|
|
input.cursor_position = input.value_width();
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::Ctrl('w'));
|
|
|
|
assert_eq!(matched_key, Some(Key::Ctrl('w')));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a', ' ', 'd']);
|
|
assert_eq!(input.cursor_index, 2);
|
|
}
|
|
|
|
#[test]
|
|
fn test_deletes_til_nonalphanumeric_for_alt_backspace() {
|
|
let mut input = Input::new();
|
|
input.value = vec!['a', '-', 'c', 'd'];
|
|
input.cursor_index = 3;
|
|
input.cursor_position = input.value_width();
|
|
|
|
let (matched_key, input_changed) = input.handle_key(Key::AltBackspace);
|
|
|
|
assert_eq!(matched_key, Some(Key::AltBackspace));
|
|
assert_eq!(input_changed, true);
|
|
assert_eq!(input.value, vec!['a', '-', 'd']);
|
|
assert_eq!(input.cursor_index, 2);
|
|
}
|
|
}
|