Merge pull request #18871 from Snuffleupagus/PDFDocument-more-ensure

Improve the promise-caching in the `PDFDocument.fieldObjects` getter
This commit is contained in:
Tim van der Meij 2024-10-08 20:11:46 +02:00 committed by GitHub
commit 1269c6a4f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1871,49 +1871,52 @@ class PDFDocument {
}
get fieldObjects() {
if (!this.formInfo.hasFields) {
return shadow(this, "fieldObjects", Promise.resolve(null));
}
const promise = this.pdfManager
.ensureDoc("formInfo")
.then(async formInfo => {
if (!formInfo.hasFields) {
return null;
}
const promise = Promise.all([
this.pdfManager.ensureDoc("annotationGlobals"),
this.pdfManager.ensureCatalog("acroForm"),
]).then(async ([annotationGlobals, acroForm]) => {
if (!annotationGlobals) {
return null;
}
const [annotationGlobals, acroForm] = await Promise.all([
this.pdfManager.ensureDoc("annotationGlobals"),
this.pdfManager.ensureCatalog("acroForm"),
]);
if (!annotationGlobals) {
return null;
}
const visitedRefs = new RefSet();
const allFields = Object.create(null);
const fieldPromises = new Map();
const orphanFields = new RefSetCache();
for (const fieldRef of await acroForm.getAsync("Fields")) {
await this.#collectFieldObjects(
"",
null,
fieldRef,
fieldPromises,
annotationGlobals,
visitedRefs,
orphanFields
);
}
const visitedRefs = new RefSet();
const allFields = Object.create(null);
const fieldPromises = new Map();
const orphanFields = new RefSetCache();
for (const fieldRef of await acroForm.getAsync("Fields")) {
await this.#collectFieldObjects(
"",
null,
fieldRef,
fieldPromises,
annotationGlobals,
visitedRefs,
orphanFields
);
}
const allPromises = [];
for (const [name, promises] of fieldPromises) {
allPromises.push(
Promise.all(promises).then(fields => {
fields = fields.filter(field => !!field);
if (fields.length > 0) {
allFields[name] = fields;
}
})
);
}
const allPromises = [];
for (const [name, promises] of fieldPromises) {
allPromises.push(
Promise.all(promises).then(fields => {
fields = fields.filter(field => !!field);
if (fields.length > 0) {
allFields[name] = fields;
}
})
);
}
await Promise.all(allPromises);
return { allFields, orphanFields };
});
await Promise.all(allPromises);
return { allFields, orphanFields };
});
return shadow(this, "fieldObjects", promise);
}
@ -1944,12 +1947,7 @@ class PDFDocument {
}
get calculationOrderIds() {
const acroForm = this.catalog.acroForm;
if (!acroForm?.has("CO")) {
return shadow(this, "calculationOrderIds", null);
}
const calculationOrder = acroForm.get("CO");
const calculationOrder = this.catalog.acroForm?.get("CO");
if (!Array.isArray(calculationOrder) || calculationOrder.length === 0) {
return shadow(this, "calculationOrderIds", null);
}
@ -1960,10 +1958,7 @@ class PDFDocument {
ids.push(id.toString());
}
}
if (ids.length === 0) {
return shadow(this, "calculationOrderIds", null);
}
return shadow(this, "calculationOrderIds", ids);
return shadow(this, "calculationOrderIds", ids.length ? ids : null);
}
get annotationGlobals() {