yazi/yazi-plugin/src/fs/op.rs

use mlua::{Lua, Table, UserData};
use yazi_codegen::FromLuaOwned;
use yazi_macro::impl_data_any;
use yazi_shared::url::UrlBuf;

#[derive(Clone, FromLuaOwned, UserData)]
pub(super) struct FilesOp(yazi_fs::FilesOp);

impl_data_any!(FilesOp => yazi_fs::FilesOp; from_into_lua = inherit);

impl From<FilesOp> for yazi_fs::FilesOp {
	fn from(op: FilesOp) -> Self { op.0 }
}

impl AsRef<yazi_fs::FilesOp> for FilesOp {
	fn as_ref(&self) -> &yazi_fs::FilesOp { &self.0 }
}

impl FilesOp {
	pub(super) fn part(_: &Lua, t: Table) -> mlua::Result<Self> {
		let id = t.raw_get("id")?;
		let url = t.raw_get("url")?;
		let files: Table = t.raw_get("files")?;

		Ok(Self(yazi_fs::FilesOp::Part(
			url,
			files.sequence_values().collect::<mlua::Result<Vec<_>>>()?,
			id,
		)))
	}

	pub(super) fn done(_: &Lua, t: Table) -> mlua::Result<Self> {
		let id = t.raw_get("id")?;
		let file = t.raw_get("file")?;

		Ok(Self(yazi_fs::FilesOp::Done(file, id)))
	}

	pub(super) fn size(_: &Lua, t: Table) -> mlua::Result<Self> {
		let url: UrlBuf = t.raw_get("url")?;
		let sizes: Table = t.raw_get("sizes")?;

		Ok(Self(yazi_fs::FilesOp::Size(url, sizes.pairs().collect::<mlua::Result<_>>()?)))
	}
}