shed/src/prompt/readline/vicmd.rs

use super::register::{append_register, read_register, write_register};

#[derive(Clone,Copy,Debug)]
pub struct RegisterName {
	name: Option<char>,
	count: usize,
	append: bool
}

impl RegisterName {
	pub fn new(name: Option<char>, count: Option<usize>) -> Self {
		let Some(ch) = name else {
			return Self::default()
		};

		let append = ch.is_uppercase();
		let name = ch.to_ascii_lowercase();
		Self {
			name: Some(name),
			count: count.unwrap_or(1),
			append
		}
	}
	pub fn name(&self) -> Option<char> {
		self.name
	}
	pub fn is_append(&self) -> bool {
		self.append
	}
	pub fn count(&self) -> usize {
		self.count
	}
	pub fn write_to_register(&self, buf: String) {
		if self.append {
			append_register(self.name, buf);
		} else {
			write_register(self.name, buf);
		}
	}
	pub fn read_from_register(&self) -> Option<String> {
		read_register(self.name)
	}
}

impl Default for RegisterName {
	fn default() -> Self {
		Self {
			name: None,
			count: 1,
			append: false
		}
	}
}

#[derive(Clone,Default,Debug)]
pub struct ViCmd {
	pub register: RegisterName,
	pub verb: Option<VerbCmd>,
	pub motion: Option<MotionCmd>,
	pub raw_seq: String, 
}

impl ViCmd {
	pub fn new() -> Self {
		Self::default()
	}
	pub fn set_motion(&mut self, motion: MotionCmd) {
		self.motion = Some(motion)
	}
	pub fn set_verb(&mut self, verb: VerbCmd) {
		self.verb = Some(verb)
	}
	pub fn verb(&self) -> Option<&VerbCmd> {
		self.verb.as_ref()
	}
	pub fn motion(&self) -> Option<&MotionCmd> {
		self.motion.as_ref()
	}
	pub fn verb_count(&self) -> usize {
		self.verb.as_ref().map(|v| v.0).unwrap_or(1)
	}
	pub fn motion_count(&self) -> usize {
		self.motion.as_ref().map(|m| m.0).unwrap_or(1)
	}
	pub fn is_repeatable(&self) -> bool {
		self.verb.as_ref().is_some_and(|v| v.1.is_repeatable())
	}
	pub fn is_cmd_repeat(&self) -> bool {
		self.verb.as_ref().is_some_and(|v| matches!(v.1,Verb::RepeatLast))
	}
	pub fn is_motion_repeat(&self) -> bool {
		self.motion.as_ref().is_some_and(|m| matches!(m.1,Motion::RepeatMotion | Motion::RepeatMotionRev))
	}
	pub fn is_char_search(&self) -> bool {
		self.motion.as_ref().is_some_and(|m| matches!(m.1, Motion::CharSearch(..)))
	}
	pub fn should_submit(&self) -> bool {
		self.verb.as_ref().is_some_and(|v| matches!(v.1, Verb::AcceptLine))
	}
	pub fn is_undo_op(&self) -> bool {
		self.verb.as_ref().is_some_and(|v| matches!(v.1, Verb::Undo | Verb::Redo))
	}
	pub fn is_line_motion(&self) -> bool {
		self.motion.as_ref().is_some_and(|m| matches!(m.1, Motion::LineUp | Motion::LineDown))
	}
	pub fn is_mode_transition(&self) -> bool {
		self.verb.as_ref().is_some_and(|v| {
			matches!(v.1, 
				Verb::Change |
				Verb::InsertMode |
				Verb::InsertModeLineBreak(_) |
				Verb::NormalMode |
				Verb::VisualMode |
				Verb::ReplaceMode
			)
		})
	}
}

#[derive(Clone,Debug)]
pub struct VerbCmd(pub usize,pub Verb);
#[derive(Clone,Debug)]
pub struct MotionCmd(pub usize,pub Motion);

impl MotionCmd {
	pub fn invert_char_motion(self) -> Self {
		let MotionCmd(count,Motion::CharSearch(dir, dest, ch)) = self else {
			unreachable!()
		};
		let new_dir = match dir {
			Direction::Forward => Direction::Backward,
			Direction::Backward => Direction::Forward,
		};
		MotionCmd(count,Motion::CharSearch(new_dir, dest, ch))
	}
}

#[derive(Debug, Clone, Eq, PartialEq)]
#[non_exhaustive]
pub enum Verb {
	Delete,
	DeleteChar(Anchor),
	Change,
	Yank,
	ReplaceChar(char),
	Substitute,
	ToggleCase,
	Complete,
	CompleteBackward,
	Undo,
	Redo,
	RepeatLast,
	Put(Anchor),
	ReplaceMode,
	InsertMode,
	InsertModeLineBreak(Anchor),
	NormalMode,
	VisualMode,
	JoinLines,
	InsertChar(char),
	Insert(String),
	Breakline(Anchor),
	Indent,
	Dedent,
	Equalize,
	AcceptLine,
	Builder(VerbBuilder),
	EndOfFile
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum VerbBuilder {
}

impl Verb {
	pub fn needs_motion(&self) -> bool {
		matches!(self, 
			Self::Indent |
			Self::Dedent |
			Self::Delete |
			Self::Change |
			Self::Yank
		)
	}
	pub fn is_repeatable(&self) -> bool {
		matches!(self,
			Self::Delete |
			Self::DeleteChar(_) |
			Self::Change |
			Self::ReplaceChar(_) |
			Self::Substitute |
			Self::ToggleCase |
			Self::Put(_) |
			Self::ReplaceMode |
			Self::InsertModeLineBreak(_) |
			Self::JoinLines |
			Self::InsertChar(_) |
			Self::Insert(_) |
			Self::Breakline(_) |
			Self::Indent |
			Self::Dedent |
			Self::Equalize
		)
	}
	pub fn is_edit(&self) -> bool {
		matches!(self,
			Self::Delete |
			Self::DeleteChar(_) |
			Self::Change |
			Self::ReplaceChar(_) |
			Self::Substitute |
			Self::ToggleCase |
			Self::RepeatLast |
			Self::Put(_) |
			Self::ReplaceMode |
			Self::InsertModeLineBreak(_) |
			Self::JoinLines |
			Self::InsertChar(_) |
			Self::Insert(_) |
			Self::Breakline(_) |
			Self::EndOfFile
		)
	}
	pub fn is_char_insert(&self) -> bool {
		matches!(self, 
			Self::Change |
			Self::InsertChar(_) |
			Self::ReplaceChar(_)
		)
	}
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum Motion {
	/// Whole current line (not really a movement but a range)
	WholeLine,
	TextObj(TextObj, Bound),
	BeginningOfFirstWord,
	/// beginning-of-line
	BeginningOfLine,
	/// end-of-line
	EndOfLine,
	/// backward-word, vi-prev-word
	BackwardWord(To, Word), // Backward until start of word
																	 /// forward-word, vi-end-word, vi-next-word
	ForwardWord(To, Word), // Forward until start/end of word
																			/// character-search, character-search-backward, vi-char-search
	CharSearch(Direction,Dest,char),
	/// backward-char
	BackwardChar,
	/// forward-char
	ForwardChar,
	/// move to the same column on the previous line
	LineUp,
	/// move to the same column on the previous visual line
	ScreenLineUp,
	/// move to the same column on the next line
	LineDown,
	/// move to the same column on the next visual line
	ScreenLineDown, 
	/// Whole user input (not really a movement but a range)
	WholeBuffer,
	/// beginning-of-register
	BeginningOfBuffer,
	/// end-of-register
	EndOfBuffer,
	ToColumn(usize),
	Range(usize,usize),
	Builder(MotionBuilder),
	RepeatMotion,
	RepeatMotionRev,
	Null
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum MotionBuilder {
	CharSearch(Option<Direction>,Option<Dest>,Option<char>),
	TextObj(Option<TextObj>,Option<Bound>)
}

impl Motion {
	pub fn needs_verb(&self) -> bool {
		matches!(self, Self::TextObj(_, _))
	}
}

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum Anchor {
	After,
	Before
}
#[derive(Debug, Clone, Eq, PartialEq)]
pub enum TextObj {
	/// `iw`, `aw` — inner word, around word
	Word(Word),

	/// for stuff like 'dd'
	Line,

	/// `is`, `as` — inner sentence, around sentence
	Sentence,

	/// `ip`, `ap` — inner paragraph, around paragraph
	Paragraph,

	/// `i"`, `a"` — inner/around double quotes
	DoubleQuote,
	/// `i'`, `a'`
	SingleQuote,
	/// `i\``, `a\``
	BacktickQuote,

	/// `i)`, `a)` — round parens
	Paren,
	/// `i]`, `a]`
	Bracket,
	/// `i}`, `a}`
	Brace,
	/// `i<`, `a<`
	Angle,

	/// `it`, `at` — HTML/XML tags (if you support it)
	Tag,

	/// Custom user-defined objects maybe?
	Custom(char),
}

#[derive(Debug, Copy, Clone, Eq, PartialEq)]
pub enum Word {
	Big,
	Normal
}
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum Bound {
	Inside,
	Around
}

#[derive(Default, Debug, Clone, Copy, Eq, PartialEq)]
pub enum Direction {
	#[default]
	Forward,
	Backward
}

#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum Dest {
	On,
	Before,
	After
}

#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum To {
	Start,
	End
}