yazi/yazi-binding/src/elements/color.rs

use std::str::FromStr;

use mlua::{AnyUserData, ExternalError, ExternalResult, FromLua, IntoLua, Lua, MetaMethod, Table, UserData, Value};

#[derive(Clone, Copy, Default, UserData)]
pub struct Color(pub ratatui_core::style::Color);

impl Color {
	pub fn compose(lua: &Lua) -> mlua::Result<Value> {
		let new = lua.create_function(|_, (_, color): (Table, Self)| Ok(color))?;

		let color = lua.create_table()?;
		color.set_metatable(Some(lua.create_table_from([(MetaMethod::Call.name(), new)])?))?;

		color.into_lua(lua)
	}
}

impl TryFrom<&AnyUserData> for Color {
	type Error = mlua::Error;

	fn try_from(ud: &AnyUserData) -> Result<Self, Self::Error> { Ok(*ud.borrow()?) }
}

impl FromLua for Color {
	fn from_lua(value: Value, _: &Lua) -> mlua::Result<Self> {
		Ok(Self(match value {
			Value::Nil => ratatui_core::style::Color::Reset,
			Value::String(s) => ratatui_core::style::Color::from_str(&s.to_str()?).into_lua_err()?,
			Value::UserData(ud) => return Self::try_from(&ud),
			_ => Err("expected a Color".into_lua_err())?,
		}))
	}
}