yazi/yazi-core/src/which/which.rs

use yazi_config::keymap::{ChordCow, Key};
use yazi_macro::{emit, render_and};

#[derive(Default)]
pub struct Which {
	pub times: usize,
	pub cands: Vec<ChordCow>,

	// Active state
	pub active: bool,
	pub silent: bool,
}

impl Which {
	pub fn r#type(&mut self, key: Key) -> bool {
		self.cands.retain(|c| c.on.len() > self.times && c.on[self.times] == key);
		self.times += 1;

		if self.cands.is_empty() {
			self.reset();
		} else if self.cands.len() == 1 {
			emit!(Seq(self.cands.remove(0).into_seq()));
			self.reset();
		} else if let Some(i) = self.cands.iter().position(|c| c.on.len() == self.times) {
			emit!(Seq(self.cands.remove(i).into_seq()));
			self.reset();
		}

		render_and!(true)
	}

	fn reset(&mut self) {
		self.times = 0;
		self.cands.clear();

		self.active = false;
		self.silent = false;
	}
}