Sakurai/pdf.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #13283 from Snuffleupagus/NameOrNumberTree-getAll-map

Browse Source 
Change `NameOrNumberTree.getAll` to return a `Map` rather than an Object
This commit is contained in:
Tim van der Meij 2021-04-23 20:53:52 +02:00 committed by GitHub
commit a6e3ad4c72
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/core/catalog.js
View File

@ -514,9 +514,8 @@ class Catalog {
const obj = this._readDests(), const obj = this._readDests(),
dests = Object.create(null); dests = Object.create(null);
if (obj instanceof NameTree) { if (obj instanceof NameTree) {
const names = obj.getAll(); for (const [key, value] of obj.getAll()) {
for (const name in names) { dests[key] = fetchDestination(value);
dests[name] = fetchDestination(names[name]);
} }
} else if (obj instanceof Dict) { } else if (obj instanceof Dict) {
obj.forEach(function (key, value) { obj.forEach(function (key, value) {
@ -582,8 +581,9 @@ class Catalog {
currentIndex = 1; currentIndex = 1;
for (let i = 0, ii = this.numPages; i < ii; i++) { for (let i = 0, ii = this.numPages; i < ii; i++) {
if (i in nums) { const labelDict = nums.get(i);
const labelDict = nums[i];
if (labelDict !== undefined) {
if (!isDict(labelDict)) { if (!isDict(labelDict)) {
throw new FormatError("PageLabel is not a dictionary."); throw new FormatError("PageLabel is not a dictionary.");
} }
@ -879,15 +879,14 @@ class Catalog {
const obj = this._catDict.get("Names"); const obj = this._catDict.get("Names");
let attachments = null; let attachments = null;
if (obj && obj.has("EmbeddedFiles")) { if (obj instanceof Dict && obj.has("EmbeddedFiles")) {
const nameTree = new NameTree(obj.getRaw("EmbeddedFiles"), this.xref); const nameTree = new NameTree(obj.getRaw("EmbeddedFiles"), this.xref);
const names = nameTree.getAll(); for (const [key, value] of nameTree.getAll()) {
for (const name in names) { const fs = new FileSpec(value, this.xref);
const fs = new FileSpec(names[name], this.xref);
if (!attachments) { if (!attachments) {
attachments = Object.create(null); attachments = Object.create(null);
} }
attachments[stringToPDFString(name)] = fs.serializable; attachments[stringToPDFString(key)] = fs.serializable;
} }
} }
return shadow(this, "attachments", attachments); return shadow(this, "attachments", attachments);
@ -895,18 +894,20 @@ class Catalog {
_collectJavaScript() { _collectJavaScript() {
const obj = this._catDict.get("Names"); const obj = this._catDict.get("Names");
let javaScript = null; let javaScript = null;
function appendIfJavaScriptDict(name, jsDict) { function appendIfJavaScriptDict(name, jsDict) {
const type = jsDict.get("S"); if (!(jsDict instanceof Dict)) {
if (!isName(type, "JavaScript")) { return;
}
if (!isName(jsDict.get("S"), "JavaScript")) {
return; return;
} }
let js = jsDict.get("JS"); let js = jsDict.get("JS");
if (isStream(js)) { if (isStream(js)) {
js = bytesToString(js.getBytes()); js = bytesToString(js.getBytes());
} else if (!isString(js)) { } else if (typeof js !== "string") {
return; return;
} }
@ -916,22 +917,15 @@ class Catalog {
javaScript.set(name, stringToPDFString(js)); javaScript.set(name, stringToPDFString(js));
} }
if (obj && obj.has("JavaScript")) { if (obj instanceof Dict && obj.has("JavaScript")) {
const nameTree = new NameTree(obj.getRaw("JavaScript"), this.xref); const nameTree = new NameTree(obj.getRaw("JavaScript"), this.xref);
const names = nameTree.getAll(); for (const [key, value] of nameTree.getAll()) {
for (const name in names) { appendIfJavaScriptDict(key, value);
// We don't use most JavaScript in PDF documents. This code is
// defensive so we don't cause errors on document load.
const jsDict = names[name];
if (isDict(jsDict)) {
appendIfJavaScriptDict(name, jsDict);
}
} }
} }
// Append OpenAction "JavaScript" actions, if any, to the JavaScript map.
// Append OpenAction "JavaScript" actions to the JavaScript array.
const openAction = this._catDict.get("OpenAction"); const openAction = this._catDict.get("OpenAction");
if (isDict(openAction) && isName(openAction.get("S"), "JavaScript")) { if (openAction) {
appendIfJavaScriptDict("OpenAction", openAction); appendIfJavaScriptDict("OpenAction", openAction);
} }

15
src/core/name_number_tree.js
View File

@ -32,9 +32,9 @@ class NameOrNumberTree {
} }
getAll() { getAll() {
const dict = Object.create(null); const map = new Map();
if (!this.root) { if (!this.root) {
return dict; return map;
} }
const xref = this.xref; const xref = this.xref;
// Reading Name/Number tree. // Reading Name/Number tree.
@ -59,13 +59,14 @@ class NameOrNumberTree {
continue; continue;
} }
const entries = obj.get(this._type); const entries = obj.get(this._type);
if (Array.isArray(entries)) { if (!Array.isArray(entries)) {
for (let i = 0, ii = entries.length; i < ii; i += 2) { continue;
dict[xref.fetchIfRef(entries[i])] = xref.fetchIfRef(entries[i + 1]); }
} for (let i = 0, ii = entries.length; i < ii; i += 2) {
map.set(xref.fetchIfRef(entries[i]), xref.fetchIfRef(entries[i + 1]));
} }
} }
return dict; return map;
} }
get(key) { get(key) {