ZacharyZhang-NY
b7fca0146d
ElyDoc, ElyTable, ElyShow (LibreOfficeKit), ElyPDF (MuPDF), ElyTree (Git), ElyMark (Milkdown and CodeMirror in CEF). ElyMail and ElyAgent follow.
165 lines
8.5 KiB
Rust
165 lines
8.5 KiB
Rust
//! Exports and the repaired copy (C11.8).
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use ely_office::{Saving as _, Tabs as _};
|
|
use ely_pdf_engine::Optimizing;
|
|
use gpui::TestAppContext;
|
|
|
|
use super::{file, settle, shown, windowed};
|
|
use crate::output::{Kind, image_names};
|
|
|
|
#[test]
|
|
fn image_names_pad_alike() {
|
|
assert_eq!(image_names("doc", &[0, 9], 12), ["doc-01.png", "doc-10.png"]);
|
|
}
|
|
|
|
#[test]
|
|
fn pages_export_as_images_and_text() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let path = file(dir.path(), "doc.pdf", &[(612.0, 792.0, &["First"]), (612.0, 792.0, &["Second"])]);
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path], true);
|
|
shown(cx, &viewer);
|
|
let id = cx.update(|_, app| viewer.read(app).core.tabs[0].id);
|
|
let targets: Vec<PathBuf> = ["doc-1.png", "doc-2.png"].iter().map(|name| dir.path().join(name)).collect();
|
|
cx.update(|_, app| viewer.update(app, |viewer, cx| viewer.write_images(id, (vec![0, 1], 0), 72.0, targets.clone(), cx)));
|
|
settle(cx, |_| targets.iter().all(|target| std::fs::metadata(target).is_ok_and(|meta| meta.len() > 100)));
|
|
let txt = dir.path().join("doc.txt");
|
|
let how = Optimizing { compress: true, from: 300, to: 0 };
|
|
cx.update(|_, app| viewer.update(app, |viewer, cx| viewer.write_export(id, Kind::Text, how, txt.clone(), cx)));
|
|
settle(cx, |_| txt.is_file());
|
|
let text = std::fs::read_to_string(&txt).expect("UTF-8");
|
|
assert!(text.contains("First") && text.contains("Second"), "{text}");
|
|
}
|
|
|
|
#[test]
|
|
fn a_damaged_file_opens_repaired_and_saves_as_a_new_file() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let path = file(dir.path(), "broken.pdf", &[(612.0, 792.0, &["Hello"])]);
|
|
let mut bytes = std::fs::read(&path).expect("reads");
|
|
let at = bytes.windows(9).rposition(|w| w == b"startxref").expect("the trailer");
|
|
bytes.truncate(at);
|
|
bytes.extend_from_slice(b"startxref\n9\n%%EOF\n");
|
|
std::fs::write(&path, &bytes).expect("writes");
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path.clone()], true);
|
|
let view = shown(cx, &viewer);
|
|
assert_eq!(cx.update(|_, app| viewer.read(app).active_tab().and_then(|tab| tab.risk)), Some(ely_office::Risk::Repaired));
|
|
assert!(cx.update(|_, app| view.read(app).repair.as_ref().is_some_and(|report| !report.is_empty())));
|
|
cx.update(|_, app| viewer.update(app, |viewer, cx| viewer.show_repair(true, cx)));
|
|
super::frames(cx);
|
|
let id = cx.update(|_, app| viewer.read(app).core.tabs[0].id);
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| viewer.write_as(id, path.clone(), None, window, cx)));
|
|
super::frames(cx);
|
|
assert_eq!(std::fs::read(&path).expect("reads"), bytes, "Save As refuses the damaged original");
|
|
let fixed = dir.path().join("fixed.pdf");
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| viewer.write_as(id, fixed.clone(), None, window, cx)));
|
|
settle(cx, |app| fixed.is_file() && viewer.read(app).active_tab().is_some_and(|tab| tab.risk.is_none() && tab.view != view));
|
|
}
|
|
|
|
#[test]
|
|
fn an_edit_made_while_saving_stays_and_the_tab_reopens_after_the_next_save() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let path = file(dir.path(), "broken.pdf", &[(612.0, 792.0, &["Hello"])]);
|
|
let mut bytes = std::fs::read(&path).expect("reads");
|
|
let at = bytes.windows(9).rposition(|w| w == b"startxref").expect("the trailer");
|
|
bytes.truncate(at);
|
|
bytes.extend_from_slice(b"startxref\n9\n%%EOF\n");
|
|
std::fs::write(&path, &bytes).expect("writes");
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path], true);
|
|
let view = shown(cx, &viewer);
|
|
let (id, fixed) = (cx.update(|_, app| viewer.read(app).core.tabs[0].id), dir.path().join("fixed.pdf"));
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| {
|
|
viewer.write_as(id, fixed.clone(), None, window, cx);
|
|
// Lands after the save took its snapshot.
|
|
view.update(cx, |view, cx| view.edit("rotating", |w, doc| w.rotate_pages(doc, &[0], 90), cx));
|
|
}));
|
|
settle(cx, |app| fixed.is_file() && view.read(app).edited == 1 && viewer.read(app).active_tab().is_some_and(|tab| !tab.exporting));
|
|
for _ in 0..10 {
|
|
super::frames(cx);
|
|
}
|
|
let tab = cx.update(|_, app| viewer.read(app).active_tab().map(|tab| (tab.view == view, tab.dirty(app))));
|
|
assert_eq!(tab, Some((true, true)), "the edit stays, unsaved");
|
|
assert_eq!(cx.update(|_, app| view.read(app).sizes[0]), (792.0, 612.0));
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| viewer.save(id, None, window, cx)));
|
|
settle(cx, |app| viewer.read(app).active_tab().is_some_and(|tab| tab.view != view && !tab.reopen));
|
|
let reopened = shown(cx, &viewer);
|
|
assert_eq!(cx.update(|_, app| reopened.read(app).sizes[0]), (792.0, 612.0), "the reopened file holds the edit");
|
|
}
|
|
|
|
#[test]
|
|
fn picked_pages_from_before_an_edit_export_nothing() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let path = file(dir.path(), "doc.pdf", &[(100.0, 100.0, &["One"]), (200.0, 100.0, &["Two"]), (300.0, 100.0, &["Three"])]);
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path], true);
|
|
let view = shown(cx, &viewer);
|
|
let id = cx.update(|_, app| viewer.read(app).core.tabs[0].id);
|
|
let target = dir.path().join("doc-2.png");
|
|
cx.update(|_, app| viewer.update(app, |viewer, cx| {
|
|
let version = view.read(cx).version;
|
|
view.update(cx, |view, cx| view.edit("deleting", |w, doc| w.delete_pages(doc, &[0]), cx));
|
|
viewer.write_images(id, (vec![1], version), 72.0, vec![target.clone()], cx);
|
|
}));
|
|
settle(cx, |app| view.read(app).edited == 1 && view.read(app).settled());
|
|
for _ in 0..10 {
|
|
super::frames(cx);
|
|
}
|
|
assert!(!target.exists(), "page 2 is another page now");
|
|
}
|
|
|
|
/// A page drawing `/Im`, a picture that is no zlib.
|
|
fn broken_picture(dir: &std::path::Path, height: f32, y: f32) -> (PathBuf, Vec<u8>) {
|
|
let picture = "<< /Type /XObject /Subtype /Image /Width 4 /Height 4 /ColorSpace /DeviceRGB /BitsPerComponent 8 /Filter /FlateDecode /Length 12 >>\nstream\nnot a zlib!!\nendstream".to_string();
|
|
let page = format!(" /MediaBox [0 0 612 {height}] /Resources << /XObject << /Im 5 0 R >> >>");
|
|
let bytes = ely_pdf_engine::sample::one_page("", &page, &format!("q 100 0 0 100 72 {y} cm /Im Do Q"), &[picture]);
|
|
let path = dir.join("broken.pdf");
|
|
std::fs::write(&path, &bytes).expect("writes");
|
|
(path, bytes)
|
|
}
|
|
|
|
#[test]
|
|
fn a_picture_that_breaks_as_it_draws_stops_an_overwrite() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let (path, bytes) = broken_picture(dir.path(), 792.0, 600.0);
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path.clone()], true);
|
|
shown(cx, &viewer);
|
|
settle(cx, |app| viewer.read(app).active_tab().and_then(|tab| tab.risk) == Some(ely_office::Risk::Repaired));
|
|
let id = cx.update(|_, app| viewer.read(app).core.tabs[0].id);
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| viewer.save(id, None, window, cx)));
|
|
for _ in 0..10 {
|
|
super::frames(cx);
|
|
}
|
|
assert_eq!(std::fs::read(&path).expect("reads"), bytes);
|
|
}
|
|
|
|
#[test]
|
|
fn a_picture_never_drawn_is_checked_before_an_overwrite() {
|
|
let dir = tempfile::tempdir().expect("a temp dir");
|
|
let mut cx = TestAppContext::single();
|
|
let (one, _) = broken_picture(dir.path(), 792.0, 600.0);
|
|
// Thirty blank pages first: the picture's page is never drawn.
|
|
let mut worker = ely_pdf_engine::Worker::spawn().expect("the worker");
|
|
let mut doc = worker.open(&one).expect("opens").doc;
|
|
for _ in 0..30 {
|
|
let mark = worker.insert_blank(doc, 0, (612.0, 792.0)).expect("inserts");
|
|
doc = doc.at(&mark).expect("the new state");
|
|
}
|
|
let path = dir.path().join("long.pdf");
|
|
worker.save(doc, &path).expect("saves");
|
|
let bytes = std::fs::read(&path).expect("reads");
|
|
let (viewer, cx) = windowed(&mut cx, dir.path(), &[path.clone()], true);
|
|
shown(cx, &viewer);
|
|
for _ in 0..10 {
|
|
super::frames(cx);
|
|
}
|
|
let id = cx.update(|_, app| viewer.read(app).core.tabs[0].id);
|
|
assert_eq!(cx.update(|_, app| viewer.read(app).active_tab().and_then(|tab| tab.risk)), None, "nothing drawn yet");
|
|
cx.update(|window, app| viewer.update(app, |viewer, cx| viewer.save(id, None, window, cx)));
|
|
settle(cx, |app| viewer.read(app).active_tab().and_then(|tab| tab.risk) == Some(ely_office::Risk::Repaired));
|
|
assert_eq!(std::fs::read(&path).expect("reads"), bytes, "Save turned into Save As");
|
|
}
|