feat: expose upsert file operations to Lua (#4266)

Author: sxyazi

Cargo.lock

@@ -71,7 +71,7 @@ regex                 = "1.13.1"
 regex-syntax          = "0.8.11"
 reqwest               = { version = "0.13.4", default-features = false, features = [ "rustls-no-provider" ] }
 rustls                = { version = "0.23.43", default-features = false, features = [ "ring", "std" ] }
-russh                 = { version = "0.62.6", default-features = false, features = [ "ring", "rsa" ] }
+russh                 = { version = "0.62.7", default-features = false, features = [ "ring", "rsa" ] }
 scopeguard            = "1.2.0"
 serde                 = { version = "1.0.229", features = [ "derive" ] }
 serde_json            = "1.0.151"

Cargo.toml

@@ -1,18 +1,19 @@
 use yazi_config::YAZI;
 use yazi_fs::Xdg;
+use yazi_macro::outln;
 
 use crate::cache::Cache;
 
 impl Cache {
 	pub(crate) fn clear() -> anyhow::Result<()> {
 		if YAZI.preview.cache_dir == *Xdg::temp_dir() {
-			println!("Clearing cache directory: \n{:?}", YAZI.preview.cache_dir);
+			outln!("Clearing cache directory: \n{:?}", YAZI.preview.cache_dir)?;
 			std::fs::remove_dir_all(&YAZI.preview.cache_dir)?;
 		} else {
-			println!(
+			outln!(
 				"You've changed the default cache directory, for your data's safety, please clear it manually: \n{:?}",
 				YAZI.preview.cache_dir
-			);
+			)?;
 		}
 
 		Ok(())

yazi-cli/src/cache/clear.rs

@@ -2,7 +2,7 @@ use anyhow::{Context, Result};
 use hashbrown::HashSet;
 use tokio::{io::AsyncWriteExt, time};
 use yazi_dds::{ClientReader, Payload, Stream, ember::EmberHi};
-use yazi_macro::try_format;
+use yazi_macro::{outln, try_format};
 
 use crate::dds::Dds;
 
@@ -26,7 +26,7 @@ impl Dds {
 				Some(s) => {
 					let kind = s.split(',').next();
 					if matches!(kind, Some(kind) if kinds.contains(kind)) {
-						println!("{s}");
+						outln!("{s}")?;
 					}
 				}
 				None => loop {

yazi-cli/src/dds/draw.rs

@@ -2,30 +2,29 @@ yazi_macro::mod_pub!(cache dds env package shared);
 
 yazi_macro::mod_flat!(args);
 
-use std::process::ExitCode;
-
 use clap::Parser;
 use yazi_macro::{errln, outln};
 use yazi_shared::LOCAL_SET;
 
 #[tokio::main]
-async fn main() -> ExitCode {
-	yazi_shim::init();
+async fn main() -> anyhow::Result<()> {
+	yazi_shim::init()?;
 	yazi_shared::init();
 	yazi_fs::init();
 
 	match LOCAL_SET.run_until(run()).await {
-		Ok(()) => ExitCode::SUCCESS,
+		Ok(()) => Ok(()),
 		Err(e) => {
 			for cause in e.chain() {
 				if let Some(ioerr) = cause.downcast_ref::<std::io::Error>()
 					&& ioerr.kind() == std::io::ErrorKind::BrokenPipe
 				{
-					return ExitCode::from(0);
+					return Ok(());
 				}
 			}
+
 			errln!("{e:#}").ok();
-			ExitCode::FAILURE
+			Err(e)
 		}
 	}
 }

yazi-cli/src/main.rs

@@ -141,6 +141,7 @@ fn op(lua: &Lua) -> mlua::Result<Function> {
 		b"part" => super::FilesOp::part(lua, t),
 		b"done" => super::FilesOp::done(lua, t),
 		b"size" => super::FilesOp::size(lua, t),
+		b"upsert" => super::FilesOp::upsert(lua, t),
 		_ => Err("Unknown operation".into_lua_err())?,
 	})
 }

yazi-plugin/src/fs/fs.rs

@@ -1,7 +1,8 @@
 use mlua::{Lua, Table, UserData};
 use yazi_codegen::FromLuaOwned;
+use yazi_fs::file::File;
 use yazi_macro::impl_data_any;
-use yazi_shared::url::UrlBuf;
+use yazi_shared::{path::PathBufDyn, url::UrlBuf};
 
 #[derive(Clone, FromLuaOwned, UserData)]
 pub(super) struct FilesOp(yazi_fs::FilesOp);
@@ -42,4 +43,12 @@ impl FilesOp {
 
 		Ok(Self(yazi_fs::FilesOp::Size(url, sizes.pairs().collect::<mlua::Result<_>>()?)))
 	}
+
+	pub(super) fn upsert(_: &Lua, t: Table) -> mlua::Result<Self> {
+		let url: UrlBuf = t.raw_get("url")?;
+		let files: Table = t.raw_get("files")?;
+		let files = files.pairs::<PathBufDyn, File>().collect::<mlua::Result<_>>()?;
+
+		Ok(Self(yazi_fs::FilesOp::Upserting(url, files)))
+	}
 }