feat(fs): use IPC streaming for read and write file (#1251)

* feat(fs): use IPC streaming for read and write file

* fix build

* clippy
pull/1314/head
Lucas Fernandes Nogueira 1 year ago committed by GitHub
parent 5661cce795
commit 1d8662e71a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

File diff suppressed because one or more lines are too long

@ -698,12 +698,14 @@ async function readFile(
throw new TypeError("Must be a file URL."); throw new TypeError("Must be a file URL.");
} }
const arr = await invoke<number[]>("plugin:fs|read_file", { const arr = await invoke<ArrayBuffer | number[]>("plugin:fs|read_file", {
path: path instanceof URL ? path.toString() : path, path: path instanceof URL ? path.toString() : path,
options, options,
}); });
return Uint8Array.from(arr); return arr instanceof ArrayBuffer
? new Uint8Array(arr)
: Uint8Array.from(arr);
} }
/** /**
@ -1007,10 +1009,11 @@ async function writeFile(
throw new TypeError("Must be a file URL."); throw new TypeError("Must be a file URL.");
} }
await invoke("plugin:fs|write_file", { await invoke("plugin:fs|write_file", data, {
path: path instanceof URL ? path.toString() : path, headers: {
data: Array.from(data), path: path instanceof URL ? path.toString() : path,
options, options: JSON.stringify(options),
},
}); });
} }

@ -336,7 +336,7 @@ pub fn read_file<R: Runtime>(
command_scope: CommandScope<Entry>, command_scope: CommandScope<Entry>,
path: SafePathBuf, path: SafePathBuf,
options: Option<BaseOptions>, options: Option<BaseOptions>,
) -> CommandResult<Vec<u8>> { ) -> CommandResult<tauri::ipc::Response> {
let resolved_path = resolve_path( let resolved_path = resolve_path(
&webview, &webview,
&global_scope, &global_scope,
@ -345,6 +345,7 @@ pub fn read_file<R: Runtime>(
options.as_ref().and_then(|o| o.base_dir), options.as_ref().and_then(|o| o.base_dir),
)?; )?;
std::fs::read(&resolved_path) std::fs::read(&resolved_path)
.map(tauri::ipc::Response::new)
.map_err(|e| { .map_err(|e| {
format!( format!(
"failed to read file at path: {} with error: {e}", "failed to read file at path: {} with error: {e}",
@ -756,11 +757,27 @@ pub fn write_file<R: Runtime>(
webview: Webview<R>, webview: Webview<R>,
global_scope: GlobalScope<Entry>, global_scope: GlobalScope<Entry>,
command_scope: CommandScope<Entry>, command_scope: CommandScope<Entry>,
path: SafePathBuf, request: tauri::ipc::Request<'_>,
data: Vec<u8>,
options: Option<WriteFileOptions>,
) -> CommandResult<()> { ) -> CommandResult<()> {
write_file_inner(webview, &global_scope, &command_scope, path, &data, options) if let tauri::ipc::InvokeBody::Raw(data) = request.body() {
let path = request
.headers()
.get("path")
.ok_or_else(|| anyhow::anyhow!("missing file path").into())
.and_then(|p| {
p.to_str()
.map_err(|e| anyhow::anyhow!("invalid path: {e}").into())
})
.and_then(|p| SafePathBuf::new(p.into()).map_err(CommandError::from))?;
let options = request
.headers()
.get("options")
.and_then(|p| p.to_str().ok())
.and_then(|opts| serde_json::from_str(opts).ok());
write_file_inner(webview, &global_scope, &command_scope, path, data, options)
} else {
Err(anyhow::anyhow!("unexpected invoke body").into())
}
} }
#[tauri::command] #[tauri::command]

@ -77,7 +77,7 @@ fn socket_cleanup(socket: &PathBuf) {
} }
fn notify_singleton(socket: &PathBuf) -> Result<(), Error> { fn notify_singleton(socket: &PathBuf) -> Result<(), Error> {
let stream = UnixStream::connect(&socket)?; let stream = UnixStream::connect(socket)?;
let mut bf = BufWriter::new(&stream); let mut bf = BufWriter::new(&stream);
let args_joined = std::env::args().collect::<Vec<String>>().join("\0"); let args_joined = std::env::args().collect::<Vec<String>>().join("\0");
bf.write_all(args_joined.as_bytes())?; bf.write_all(args_joined.as_bytes())?;
@ -91,7 +91,7 @@ fn listen_for_other_instances<A: Runtime>(
app: AppHandle<A>, app: AppHandle<A>,
mut cb: Box<SingleInstanceCallback<A>>, mut cb: Box<SingleInstanceCallback<A>>,
) { ) {
match UnixListener::bind(&socket) { match UnixListener::bind(socket) {
Ok(listener) => { Ok(listener) => {
let cwd = std::env::current_dir() let cwd = std::env::current_dir()
.unwrap_or_default() .unwrap_or_default()
@ -108,7 +108,7 @@ fn listen_for_other_instances<A: Runtime>(
Ok(_) => { Ok(_) => {
let args: Vec<String> = let args: Vec<String> =
s.split('\0').map(String::from).collect(); s.split('\0').map(String::from).collect();
cb(&app.clone().app_handle(), args, cwd.clone()); cb(app.app_handle(), args, cwd.clone());
} }
Err(e) => log::debug!("single_instance failed to be notified: {e}"), Err(e) => log::debug!("single_instance failed to be notified: {e}"),
} }

Loading…
Cancel
Save