yazi/yazi-shared/src/event/actions.rs

use std::{ops::{Deref, DerefMut}, slice, vec};

use mlua::{ExternalError, FromLua, IntoLua, Lua, Value};
use serde::Deserializer;
use serde_with::{DeserializeAs, DisplayFromStr, OneOrMany};

use crate::{Source, event::Action};

#[derive(Clone, Debug, Default)]
pub struct Actions(pub Vec<Action>);

impl Deref for Actions {
	type Target = Vec<Action>;

	fn deref(&self) -> &Self::Target { &self.0 }
}

impl DerefMut for Actions {
	fn deref_mut(&mut self) -> &mut Self::Target { &mut self.0 }
}

impl From<Action> for Actions {
	fn from(value: Action) -> Self { Self(vec![value]) }
}

impl From<Vec<Action>> for Actions {
	fn from(value: Vec<Action>) -> Self { Self(value) }
}

impl Actions {
	pub fn set_source(&mut self, source: Source) {
		for action in &mut self.0 {
			action.source = source;
		}
	}
}

impl IntoIterator for Actions {
	type IntoIter = vec::IntoIter<Action>;
	type Item = Action;

	fn into_iter(self) -> Self::IntoIter { self.0.into_iter() }
}

impl<'a> IntoIterator for &'a Actions {
	type IntoIter = slice::Iter<'a, Action>;
	type Item = &'a Action;

	fn into_iter(self) -> Self::IntoIter { self.0.iter() }
}

impl FromLua for Actions {
	fn from_lua(value: Value, lua: &Lua) -> mlua::Result<Self> {
		Ok(match value {
			Value::Table(t) => t.sequence_values::<Action>().collect::<mlua::Result<Vec<_>>>()?.into(),
			v @ Value::String(_) => Action::from_lua(v, lua)?.into(),
			_ => Err("expected a string or a table of actions".into_lua_err())?,
		})
	}
}

impl IntoLua for Actions {
	fn into_lua(self, lua: &Lua) -> mlua::Result<Value> {
		lua.create_sequence_from(self.0)?.into_lua(lua)
	}
}

pub fn deserialize_actions<'de, D>(deserializer: D) -> Result<Actions, D::Error>
where
	D: Deserializer<'de>,
{
	let mut actions = Actions(OneOrMany::<DisplayFromStr>::deserialize_as(deserializer)?);
	actions.set_source(Source::Key);

	Ok(actions)
}