yazi/yazi-actor/src/lives/yanked.rs

use std::ops::Deref;

use mlua::{AnyUserData, MetaMethod, MultiValue, ObjectLike, UserData, UserDataFields, UserDataMethods};
use yazi_binding::Iter;
use yazi_fs::file::File;
use yazi_shim::mlua::get_metatable;

use super::{Lives, PtrCell};

pub(super) struct Yanked {
	inner: PtrCell<yazi_core::mgr::Yanked>,
	iter:  AnyUserData,
}

impl Deref for Yanked {
	type Target = yazi_core::mgr::Yanked;

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

impl Yanked {
	pub(super) fn make(inner: &yazi_core::mgr::Yanked) -> mlua::Result<AnyUserData> {
		let inner = PtrCell::from(inner);

		Lives::scoped_userdata(Self {
			inner,
			iter: Lives::scoped_userdata(Iter::new(
				inner.as_static().iter().map(File::from),
				Some(inner.len()),
			))?,
		})
	}
}

impl UserData for Yanked {
	fn add_fields<F: UserDataFields<Self>>(fields: &mut F) {
		fields.add_field_method_get("is_cut", |_, me| Ok(me.cut));
	}

	fn add_methods<M: UserDataMethods<Self>>(methods: &mut M) {
		methods.add_meta_method(MetaMethod::Len, |_, me, ()| Ok(me.len()));

		methods.add_meta_method(MetaMethod::Pairs, |lua, me, ()| {
			get_metatable(lua, &me.iter)?
				.call_function::<MultiValue>(MetaMethod::Pairs.name(), me.iter.clone())
		});
	}
}