Add caller-thread progress for native sheet processing

Co-authored-by: TKIPisalegacycipher <git@chir.us>
This commit is contained in:
skylartaylor
2026-10-13 01:20:21 +02:00
co-authored by TKIPisalegacycipher
parent f03ec27ab2
commit ebc4d8cfbd
4 changed files with 150 additions and 4 deletions
+8
View File
@@ -54,6 +54,14 @@ is closed and requests cancellation. Calls into a driver and interactive output-
are not covered by this notification timeout. Windows answers when the job is in
the queue ("spooled 3 sheet(s)"): a paused queue or an offline printer keeps it there. CUPS jobs
never use a PDF temporary file. Printing is synchronous; UI callers should submit on a worker.
`submit_with_progress` reports `(processed, sheets)` before the first Windows sheet and after
each sheet is successfully added to the native print job. These are source PDF sheets, excluding
copies and duplex; the events describe sheet processing, and do not confirm physical output.
The callback runs on the thread calling submission, can capture non-`Send` state, and should
return promptly. It is never retained. Keep waiting for the final `Result` even at full progress:
the job can still fail. The API does not provide cancellation. CUPS reports no progress.
UI callers must forward events to their own progress state; adopting the API alone does not
change an application's UI.
Receipts are backend messages, not portable job IDs. `Job::duplex = None` preserves the queue's
duplex default; `Some(Duplex::Off)` forces simplex. `Job::scaling` defaults to `Scaling::Exact`;
callers exporting arbitrary document sizes can choose `Scaling::ShrinkToFit`. Exact mode reports
+17 -2
View File
@@ -8,6 +8,8 @@
#![deny(clippy::unwrap_used, clippy::expect_used, clippy::panic, clippy::unimplemented, clippy::todo, clippy::unreachable)]
#[cfg(any(test, windows))]
mod progress;
#[cfg(any(test, windows))]
mod sheets;
#[cfg(test)]
@@ -437,17 +439,30 @@ pub fn parse_windows_printers(out: &str) -> Vec<Printer> {
/// Send a print-ready PDF to the spooler. Returns the spooler's message: CUPS includes a job id;
/// Windows confirms submission only ("spooled 3 sheet(s)").
pub fn submit(pdf: &[u8], job: &Job) -> Result<String, Error> {
submit_with_progress(pdf, job, &mut |_, _| {})
}
/// [`submit`] with Windows sheet-processing progress: `(processed, sheets)` before the first
/// sheet and after each sheet is successfully added to the native print job. Counts refer to
/// source PDF sheets, regardless of copies or duplex. CUPS reports no progress.
///
/// The callback runs synchronously on the thread calling this function, need not be `Send`,
/// and should return promptly. It is never retained. All sheets processed is not terminal
/// success: only the returned result confirms submission, and a later failure may follow the
/// last progress event. This does not report physical printing or provide cancellation.
pub fn submit_with_progress(pdf: &[u8], job: &Job, progress: &mut dyn FnMut(usize, usize)) -> Result<String, Error> {
#[cfg(all(unix, not(target_arch = "wasm32")))]
{
let _ = progress;
submit_via(std::process::Command::new("lp"), pdf, job)
}
#[cfg(windows)]
{
windows::submit(pdf, job)
windows::submit_with_progress(pdf, job, progress)
}
#[cfg(not(any(windows, all(unix, not(target_arch = "wasm32")))))]
{
let _ = (pdf, job);
let _ = (pdf, job, progress);
Err(Error::Spool("printing to a printer isn't available on this platform yet; save the print-ready PDF instead".into()))
}
}
+110
View File
@@ -0,0 +1,110 @@
//! Bridge native worker events to callbacks on the submission thread.
use std::sync::mpsc::{SyncSender, sync_channel};
use crate::Error;
pub(crate) fn run<T: Send>(
progress: &mut dyn FnMut(usize, usize),
work: impl FnOnce(SyncSender<(usize, usize)>) -> Result<T, Error> + Send,
) -> Result<T, Error> {
// A slow UI callback must not accumulate one queued event per PDF sheet. The sender must
// ignore disconnection: dropping a progress receiver does not cancel an already-started job.
let (sender, receiver) = sync_channel(1);
std::thread::scope(move |scope| {
let worker = std::thread::Builder::new()
.name("print job".into())
.spawn_scoped(scope, move || work(sender))
.map_err(|e| Error::Spool(format!("the print job could not be started: {e}")))?;
// Own the receiver here so unwinding a callback closes it before scope joins the
// worker; otherwise a bounded sender could block forever during that join.
for (processed, sheets) in receiver {
progress(processed, sheets);
}
worker.join().map_err(|_| Error::Spool("the print job stopped unexpectedly; check the printer's queue before printing again".into()))?
})
}
#[cfg(test)]
mod tests {
use super::*;
use std::cell::RefCell;
use std::rc::Rc;
#[test]
fn progress_arrives_on_the_calling_thread_before_submission_finishes() {
let caller = std::thread::current().id();
let seen = Rc::new(RefCell::new(Vec::new()));
let observations = Rc::clone(&seen);
let (observed, waiting) = sync_channel(0);
let result = run(
&mut move |processed, sheets| {
assert_eq!(std::thread::current().id(), caller);
observations.borrow_mut().push((processed, sheets));
observed.send(()).unwrap();
},
move |events| {
assert_ne!(std::thread::current().id(), caller);
for completed in 0..=3 {
events.send((completed, 3)).unwrap();
// No terminal result until the caller processes this intermediate event.
waiting.recv().unwrap();
}
Ok("accepted")
},
);
assert_eq!(result, Ok("accepted"));
assert_eq!(*seen.borrow(), [(0, 3), (1, 3), (2, 3), (3, 3)]);
}
#[test]
fn complete_progress_does_not_hide_a_terminal_failure() {
let mut seen = Vec::new();
let error = Error::Spool("the native print package failed".into());
let result = run(&mut |done, total| seen.push((done, total)), |events| {
events.send((0, 1)).unwrap();
events.send((1, 1)).unwrap();
Err::<(), _>(error.clone())
});
assert_eq!(seen, [(0, 1), (1, 1)]);
assert_eq!(result, Err(error));
}
#[test]
fn a_partial_job_keeps_its_failure_and_last_processed_sheet() {
let mut seen = Vec::new();
let result = run(&mut |done, total| seen.push((done, total)), |events| {
events.send((0, 3)).unwrap();
events.send((1, 3)).unwrap();
Err::<(), _>(Error::Spool("the second sheet could not be rendered".into()))
});
assert_eq!(seen, [(0, 3), (1, 3)]);
assert_eq!(result, Err(Error::Spool("the second sheet could not be rendered".into())));
}
#[test]
fn worker_failure_is_reported_without_inventing_progress() {
let mut seen = Vec::new();
let result = run(&mut |done, total| seen.push((done, total)), |_| -> Result<(), Error> {
panic!("synthetic native worker failure");
});
assert!(seen.is_empty());
assert!(result.is_err_and(|e| e.to_string().contains("check the printer's queue")));
}
#[test]
fn callback_unwind_closes_progress_and_reaps_the_worker() {
let finished = std::sync::atomic::AtomicBool::new(false);
let caught = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
run(&mut |_, _| panic!("synthetic caller callback failure"), |events| {
for processed in 0..=20 {
let _ = events.send((processed, 20));
}
finished.store(true, std::sync::atomic::Ordering::SeqCst);
Ok(())
})
}));
assert!(caught.is_err());
assert!(finished.load(std::sync::atomic::Ordering::SeqCst));
}
}
+15 -2
View File
@@ -368,11 +368,22 @@ fn windows_jobs_reach_the_spooler() {
let listed = spool::printers().into_iter().find(|p| p.name == name);
let mixed = test_pdf(&[LETTER, LETTER_LANDSCAPE, LETTER], true);
let job = Job { printer: Some(name.into()), copies: 2, duplex: Some(Duplex::LongEdge), title: "mixed.pdf".into(), ..Job::default() };
let sent = spool::submit(&mixed, &job);
let caller = std::thread::current().id();
// Rc proves callers are not required to make their callback Send just because native
// printing uses a COM worker. Counts describe PDF sheets, despite copies and duplex.
let processed = std::rc::Rc::new(std::cell::RefCell::new(Vec::new()));
let observed = std::rc::Rc::clone(&processed);
let sent = spool::submit_with_progress(&mixed, &job, &mut move |done, total| {
assert_eq!(std::thread::current().id(), caller);
observed.borrow_mut().push((done, total));
});
let tabloid = test_pdf(&[TABLOID, LETTER], false);
let shrink = Job { printer: Some(name.into()), scaling: Scaling::ShrinkToFit, title: "shrink".into(), ..Job::default() };
let shrunk = spool::submit(&tabloid, &shrink);
let exact_refused = spool::submit(&tabloid, &Job { printer: Some(name.into()), ..Job::default() });
let mut refused_progress = Vec::new();
let exact_refused = spool::submit_with_progress(&tabloid, &Job { printer: Some(name.into()), ..Job::default() }, &mut |done, total| {
refused_progress.push((done, total));
});
let queued = windows_powershell(
"@(Get-PrintJob -PrinterName $name | Sort-Object Id | ForEach-Object { $_.DocumentName + ' ' + $_.TotalPages }) -join '|'",
name,
@@ -385,8 +396,10 @@ fn windows_jobs_reach_the_spooler() {
);
assert!(listed.as_ref().is_some_and(|p| p.two_sided.is_some()), "the new printer is listed, its duplex support known: {listed:?}");
assert_eq!(sent.as_deref(), Ok("spooled 3 sheet(s)"), "{sent:?}");
assert_eq!(*processed.borrow(), [(0, 3), (1, 3), (2, 3), (3, 3)]);
assert_eq!(shrunk.as_deref(), Ok("spooled 2 sheet(s)"), "{shrunk:?}");
assert!(matches!(exact_refused, Err(Error::Spool(ref m)) if m.contains("one paper size")), "{exact_refused:?}");
assert!(refused_progress.is_empty(), "rejection before a job starts has no progress: {refused_progress:?}");
assert_eq!(queued.as_deref().map(str::trim), Ok("mixed.pdf 3|shrink.pdf 2"), "one job each, all sheets, in the paused queue: {queued:?}");
assert!(cleaned.is_ok(), "the test printer is removed again: {cleaned:?}");
// Asking for a printer that isn't there is an error, never a job on another one.