Add PDF archive store, raw-body IPC and export file helpers
The searchable PDF can be archived content-addressed under the local data dir with its sha256 and layout fingerprint; a different PDF for an archived invoice is refused and reads verify the hash. Export writes go through a raw-body command that remembers the last folder; reveal-in-folder and open fall back to xdg-open / explorer. Adds a Windows-safe export file name builder. Migration M4 adds the archive columns and last export dir.
This commit is contained in:
@@ -0,0 +1,211 @@
|
||||
//! User-visible files: exports written to a path from the save dialog, and the
|
||||
//! reveal/open helpers used afterwards.
|
||||
use super::raw::{decode_header_path, header_map, raw_body, required_header, write_atomic, Headers};
|
||||
use crate::AppState;
|
||||
use rusqlite::{params, Connection};
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process::Command;
|
||||
use tauri::ipc::Request;
|
||||
use tauri::State;
|
||||
use tauri_plugin_opener::OpenerExt;
|
||||
|
||||
pub fn write_export_file_impl(
|
||||
conn: &Connection,
|
||||
bytes: &[u8],
|
||||
headers: &Headers,
|
||||
) -> Result<String, String> {
|
||||
let raw = decode_header_path(required_header(headers, "x-path")?)?;
|
||||
if raw.trim().is_empty() {
|
||||
return Err("No file path given".to_string());
|
||||
}
|
||||
let path = PathBuf::from(&raw);
|
||||
if !path.is_absolute() {
|
||||
return Err("The export path must be absolute".to_string());
|
||||
}
|
||||
if bytes.is_empty() {
|
||||
return Err("Nothing to write: the file is empty".to_string());
|
||||
}
|
||||
let parent = path.parent().ok_or_else(|| "The export path has no folder".to_string())?;
|
||||
std::fs::create_dir_all(parent)
|
||||
.map_err(|e| format!("Could not create {}: {e}", parent.display()))?;
|
||||
write_atomic(&path, bytes)?;
|
||||
conn.execute(
|
||||
"UPDATE app_settings SET last_export_dir = ?1 WHERE id = 1",
|
||||
params![parent.to_string_lossy()],
|
||||
)
|
||||
.map_err(|e| e.to_string())?;
|
||||
Ok(raw)
|
||||
}
|
||||
|
||||
pub fn get_last_export_dir_impl(conn: &Connection) -> Result<String, String> {
|
||||
conn.query_row("SELECT last_export_dir FROM app_settings WHERE id = 1", [], |r| r.get(0))
|
||||
.map_err(|e| e.to_string())
|
||||
}
|
||||
|
||||
fn require_existing(path: &str) -> Result<PathBuf, String> {
|
||||
if path.trim().is_empty() {
|
||||
return Err("No file path given".to_string());
|
||||
}
|
||||
let path = PathBuf::from(path);
|
||||
if !path.exists() {
|
||||
return Err(format!("{} does not exist", path.display()));
|
||||
}
|
||||
Ok(path)
|
||||
}
|
||||
|
||||
/// Last resort when the opener plugin cannot reach a file manager (a minimal Linux
|
||||
/// install has no D-Bus file-manager service or portal). Arguments are passed
|
||||
/// individually; nothing goes through a shell.
|
||||
fn reveal_fallback_command(path: &Path) -> Command {
|
||||
#[cfg(target_os = "windows")]
|
||||
{
|
||||
let mut cmd = Command::new("explorer");
|
||||
cmd.arg(format!("/select,{}", path.display()));
|
||||
cmd
|
||||
}
|
||||
#[cfg(target_os = "macos")]
|
||||
{
|
||||
let mut cmd = Command::new("open");
|
||||
cmd.arg("-R").arg(path);
|
||||
cmd
|
||||
}
|
||||
#[cfg(not(any(target_os = "windows", target_os = "macos")))]
|
||||
{
|
||||
let folder = if path.is_dir() { path } else { path.parent().unwrap_or(path) };
|
||||
let mut cmd = Command::new("xdg-open");
|
||||
cmd.arg(folder);
|
||||
cmd
|
||||
}
|
||||
}
|
||||
|
||||
fn open_fallback_command(path: &Path) -> Command {
|
||||
#[cfg(target_os = "windows")]
|
||||
let mut cmd = Command::new("explorer");
|
||||
#[cfg(target_os = "macos")]
|
||||
let mut cmd = Command::new("open");
|
||||
#[cfg(not(any(target_os = "windows", target_os = "macos")))]
|
||||
let mut cmd = Command::new("xdg-open");
|
||||
cmd.arg(path);
|
||||
cmd
|
||||
}
|
||||
|
||||
fn spawn_detached(mut cmd: Command) -> Result<(), String> {
|
||||
let mut child = cmd.spawn().map_err(|e| format!("Could not start the file manager: {e}"))?;
|
||||
// Reap the child so it does not linger as a zombie; the result is irrelevant
|
||||
// (explorer.exe, for one, exits non-zero even on success).
|
||||
std::thread::spawn(move || {
|
||||
let _ = child.wait();
|
||||
});
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn write_export_file(request: Request<'_>, state: State<'_, AppState>) -> Result<String, String> {
|
||||
let bytes = raw_body(&request)?;
|
||||
let headers = header_map(&request);
|
||||
let conn = state.db.lock().map_err(|e| e.to_string())?;
|
||||
write_export_file_impl(&conn, bytes, &headers)
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub fn get_last_export_dir(state: State<AppState>) -> Result<String, String> {
|
||||
let conn = state.db.lock().map_err(|e| e.to_string())?;
|
||||
get_last_export_dir_impl(&conn)
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub fn reveal_in_folder(app: tauri::AppHandle, path: String) -> Result<(), String> {
|
||||
let path = require_existing(&path)?;
|
||||
if app.opener().reveal_item_in_dir(&path).is_ok() {
|
||||
return Ok(());
|
||||
}
|
||||
spawn_detached(reveal_fallback_command(&path))
|
||||
}
|
||||
|
||||
#[tauri::command]
|
||||
pub fn open_file(app: tauri::AppHandle, path: String) -> Result<(), String> {
|
||||
let path = require_existing(&path)?;
|
||||
if app.opener().open_path(path.to_string_lossy(), None::<&str>).is_ok() {
|
||||
return Ok(());
|
||||
}
|
||||
spawn_detached(open_fallback_command(&path))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use tempfile::tempdir;
|
||||
|
||||
fn headers(path: &str) -> Headers {
|
||||
Headers::from([("x-path".to_string(), path.to_string())])
|
||||
}
|
||||
|
||||
fn encode(path: &Path) -> String {
|
||||
percent_encoding::utf8_percent_encode(&path.to_string_lossy(), percent_encoding::NON_ALPHANUMERIC)
|
||||
.to_string()
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn writes_the_file_and_remembers_the_folder() {
|
||||
let dir = tempdir().unwrap();
|
||||
let conn = crate::db::open_in_memory().unwrap();
|
||||
assert_eq!(get_last_export_dir_impl(&conn).unwrap(), "");
|
||||
|
||||
// Non-ASCII, spaces and a folder that does not exist yet.
|
||||
let target = dir.path().join("New folder").join("\u{9ac}\u{9be}\u{982}\u{9b2}\u{9be} 1.pdf");
|
||||
let written = write_export_file_impl(&conn, b"%PDF-1.7", &headers(&encode(&target))).unwrap();
|
||||
assert_eq!(PathBuf::from(&written), target);
|
||||
assert_eq!(std::fs::read(&target).unwrap(), b"%PDF-1.7");
|
||||
assert_eq!(
|
||||
get_last_export_dir_impl(&conn).unwrap(),
|
||||
target.parent().unwrap().to_string_lossy()
|
||||
);
|
||||
|
||||
// Overwrites in place.
|
||||
write_export_file_impl(&conn, b"%PDF-2", &headers(&encode(&target))).unwrap();
|
||||
assert_eq!(std::fs::read(&target).unwrap(), b"%PDF-2");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn rejects_empty_relative_and_missing_paths() {
|
||||
let conn = crate::db::open_in_memory().unwrap();
|
||||
for bad in ["", " ", "out.pdf", "sub%2Fout.pdf", "..%2Fout.pdf"] {
|
||||
assert!(write_export_file_impl(&conn, b"x", &headers(bad)).is_err(), "{bad}");
|
||||
}
|
||||
assert!(write_export_file_impl(&conn, b"x", &Headers::new()).is_err());
|
||||
assert_eq!(get_last_export_dir_impl(&conn).unwrap(), "");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn rejects_an_empty_body_without_touching_the_folder_setting() {
|
||||
let dir = tempdir().unwrap();
|
||||
let conn = crate::db::open_in_memory().unwrap();
|
||||
let target = dir.path().join("a.pdf");
|
||||
assert!(write_export_file_impl(&conn, b"", &headers(&encode(&target))).is_err());
|
||||
assert!(!target.exists());
|
||||
assert_eq!(get_last_export_dir_impl(&conn).unwrap(), "");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn reveal_and_open_refuse_missing_paths() {
|
||||
let dir = tempdir().unwrap();
|
||||
assert!(require_existing("").is_err());
|
||||
let missing = dir.path().join("nope.pdf");
|
||||
assert!(require_existing(&missing.to_string_lossy()).unwrap_err().contains("does not exist"));
|
||||
assert!(require_existing(&dir.path().to_string_lossy()).is_ok());
|
||||
}
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
#[test]
|
||||
fn linux_fallback_opens_the_parent_folder_with_separate_args() {
|
||||
let dir = tempdir().unwrap();
|
||||
let file = dir.path().join("a b; rm -rf.pdf");
|
||||
std::fs::write(&file, b"x").unwrap();
|
||||
let cmd = reveal_fallback_command(&file);
|
||||
assert_eq!(cmd.get_program(), "xdg-open");
|
||||
let args: Vec<_> = cmd.get_args().collect();
|
||||
assert_eq!(args, [dir.path().as_os_str()]);
|
||||
let open = open_fallback_command(&file);
|
||||
assert_eq!(open.get_args().collect::<Vec<_>>(), [file.as_os_str()]);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user