@adnsistemas/pdf-lib 1.18.0 → 2.4.3

package/src/core/streams/AsciiHexStream.ts

@@ -6,8 +6,8 @@
  * under the Apache 2.0 open source license.
  */
 
-import DecodeStream from 'src/core/streams/DecodeStream';
-import { StreamType } from 'src/core/streams/Stream';
+import DecodeStream from './DecodeStream';
+import { StreamType } from './Stream';
 
 class AsciiHexStream extends DecodeStream {
   private stream: StreamType;

package/src/core/streams/DecodeStream.ts

@@ -1,5 +1,5 @@
-import { MethodNotImplementedError } from 'src/core/errors';
-import Stream, { StreamType } from 'src/core/streams/Stream';
+import { MethodNotImplementedError } from '../errors';
+import Stream, { StreamType } from './Stream';
 
 /*
  * Copyright 2012 Mozilla Foundation

package/src/core/streams/DecryptStream.ts

@@ -0,0 +1,56 @@
+import DecodeStream from './DecodeStream';
+import { StreamType } from './Stream';
+
+const chunkSize = 512;
+
+type DecryptFnType = (
+  arg1: Uint8Array | Uint8ClampedArray,
+  arg2: boolean,
+) => Uint8Array;
+
+class DecryptStream extends DecodeStream {
+  private stream: StreamType;
+  private initialized: boolean;
+  private nextChunk: Uint8Array | Uint8ClampedArray | null;
+  private decrypt: DecryptFnType;
+
+  constructor(
+    stream: StreamType,
+    decrypt: DecryptFnType,
+    maybeLength?: number,
+  ) {
+    super(maybeLength);
+
+    this.stream = stream;
+    this.decrypt = decrypt;
+    this.nextChunk = null;
+    this.initialized = false;
+  }
+
+  readBlock() {
+    let chunk;
+    if (this.initialized) {
+      chunk = this.nextChunk;
+    } else {
+      chunk = this.stream.getBytes(chunkSize);
+      this.initialized = true;
+    }
+    if (!chunk || chunk.length === 0) {
+      this.eof = true;
+      return;
+    }
+    this.nextChunk = this.stream.getBytes(chunkSize);
+    const hasMoreData = this.nextChunk && this.nextChunk.length > 0;
+
+    const decrypt = this.decrypt;
+    chunk = decrypt(chunk, !hasMoreData);
+
+    const bufferLength = this.bufferLength;
+    const newLength = bufferLength + chunk.length;
+    const buffer = this.ensureBuffer(newLength);
+    buffer.set(chunk, bufferLength);
+    this.bufferLength = newLength;
+  }
+}
+
+export default DecryptStream;

package/src/core/streams/FlateStream.ts

@@ -15,8 +15,8 @@
  */
 
 /* tslint:disable  no-conditional-assignment */
-import DecodeStream from 'src/core/streams/DecodeStream';
-import { StreamType } from 'src/core/streams/Stream';
+import DecodeStream from '../../core/streams/DecodeStream';
+import { StreamType } from '../../core/streams/Stream';
 
 // prettier-ignore
 const codeLenCodeMap = new Int32Array([
@@ -272,6 +272,7 @@ class FlateStream extends DecodeStream {
     buffer = this.buffer;
     let limit = buffer ? buffer.length : 0;
     let pos = this.bufferLength;
+    // eslint-disable-next-line no-constant-condition
     while (true) {
       let code1 = this.getCode(litCodeTable);
       if (code1 < 256) {

package/src/core/streams/LZWStream.ts

@@ -6,14 +6,14 @@
  * under the Apache 2.0 open source license.
  */
 
-import DecodeStream from 'src/core/streams/DecodeStream';
-import { StreamType } from 'src/core/streams/Stream';
+import DecodeStream from './DecodeStream';
+import { StreamType } from './Stream';
 
 class LZWStream extends DecodeStream {
   private stream: StreamType;
   private cachedData: number;
   private bitsCached: number;
-  private lzwState: {
+  private lzwState?: {
     earlyChange: 0 | 1;
     codeLength: number;
     nextCode: number;

package/src/core/streams/RunLengthStream.ts

@@ -6,8 +6,8 @@
  * under the Apache 2.0 open source license.
  */
 
-import DecodeStream from 'src/core/streams/DecodeStream';
-import { StreamType } from 'src/core/streams/Stream';
+import DecodeStream from './DecodeStream';
+import { StreamType } from './Stream';
 
 class RunLengthStream extends DecodeStream {
   private stream: StreamType;

package/src/core/streams/decode.ts

@@ -1,19 +1,16 @@
-import {
-  UnexpectedObjectTypeError,
-  UnsupportedEncodingError,
-} from 'src/core/errors';
-import PDFArray from 'src/core/objects/PDFArray';
-import PDFDict from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFNull from 'src/core/objects/PDFNull';
-import PDFNumber from 'src/core/objects/PDFNumber';
-import PDFRawStream from 'src/core/objects/PDFRawStream';
-import Ascii85Stream from 'src/core/streams/Ascii85Stream';
-import AsciiHexStream from 'src/core/streams/AsciiHexStream';
-import FlateStream from 'src/core/streams/FlateStream';
-import LZWStream from 'src/core/streams/LZWStream';
-import RunLengthStream from 'src/core/streams/RunLengthStream';
-import Stream, { StreamType } from 'src/core/streams/Stream';
+import { UnexpectedObjectTypeError, UnsupportedEncodingError } from '../errors';
+import PDFArray from '../objects/PDFArray';
+import PDFDict from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFNull from '../objects/PDFNull';
+import PDFNumber from '../objects/PDFNumber';
+import PDFRawStream from '../objects/PDFRawStream';
+import Ascii85Stream from './Ascii85Stream';
+import AsciiHexStream from './AsciiHexStream';
+import FlateStream from './FlateStream';
+import LZWStream from './LZWStream';
+import RunLengthStream from './RunLengthStream';
+import Stream, { StreamType } from './Stream';
 
 const decodeStream = (
   stream: StreamType,
@@ -45,9 +42,17 @@ const decodeStream = (
   throw new UnsupportedEncodingError(encoding.asString());
 };
 
-export const decodePDFRawStream = ({ dict, contents }: PDFRawStream) => {
+export const decodePDFRawStream = ({
+  dict,
+  contents,
+  transform,
+}: PDFRawStream) => {
   let stream: StreamType = new Stream(contents);
 
+  if (transform) {
+    stream = transform.createStream(stream, contents.length);
+  }
+
   const Filter = dict.lookup(PDFName.of('Filter'));
   const DecodeParms = dict.lookup(PDFName.of('DecodeParms'));
 
@@ -65,7 +70,7 @@ export const decodePDFRawStream = ({ dict, contents }: PDFRawStream) => {
         DecodeParms && (DecodeParms as PDFArray).lookupMaybe(idx, PDFDict),
       );
     }
-  } else if (!!Filter) {
+  } else if (Filter) {
     throw new UnexpectedObjectTypeError([PDFName, PDFArray], Filter);
   }
 

package/src/core/structures/PDFCatalog.ts

@@ -1,9 +1,9 @@
-import PDFDict, { DictMap } from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFRef from 'src/core/objects/PDFRef';
-import PDFContext from 'src/core/PDFContext';
-import PDFPageTree from 'src/core/structures/PDFPageTree';
-import { PDFAcroForm } from 'src/core/acroform';
+import PDFDict, { DictMap } from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFRef from '../objects/PDFRef';
+import PDFContext from '../PDFContext';
+import PDFPageTree from './PDFPageTree';
+import { PDFAcroForm } from '../acroform';
 import ViewerPreferences from '../interactive/ViewerPreferences';
 
 class PDFCatalog extends PDFDict {

package/src/core/structures/PDFContentStream.ts

@@ -1,8 +1,8 @@
-import PDFDict from 'src/core/objects/PDFDict';
-import PDFOperator from 'src/core/operators/PDFOperator';
-import PDFContext from 'src/core/PDFContext';
-import PDFFlateStream from 'src/core/structures/PDFFlateStream';
-import CharCodes from 'src/core/syntax/CharCodes';
+import PDFDict from '../objects/PDFDict';
+import PDFOperator from '../operators/PDFOperator';
+import PDFContext from '../PDFContext';
+import PDFFlateStream from './PDFFlateStream';
+import CharCodes from '../syntax/CharCodes';
 
 class PDFContentStream extends PDFFlateStream {
   static of = (dict: PDFDict, operators: PDFOperator[], encode = true) =>

package/src/core/structures/PDFCrossRefStream.ts

@@ -1,9 +1,9 @@
-import PDFDict from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFRef from 'src/core/objects/PDFRef';
-import PDFContext from 'src/core/PDFContext';
-import PDFFlateStream from 'src/core/structures/PDFFlateStream';
-import { bytesFor, Cache, reverseArray, sizeInBytes, sum } from 'src/utils';
+import PDFDict from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFRef from '../objects/PDFRef';
+import PDFContext from '../PDFContext';
+import PDFFlateStream from './PDFFlateStream';
+import { bytesFor, Cache, reverseArray, sizeInBytes, sum } from '../../utils';
 
 export enum EntryType {
   Deleted = 0,

package/src/core/structures/PDFFlateStream.ts

@@ -1,13 +1,13 @@
 import pako from 'pako';
 
-import { MethodNotImplementedError } from 'src/core/errors';
-import PDFDict from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFStream from 'src/core/objects/PDFStream';
-import { Cache } from 'src/utils';
+import { MethodNotImplementedError } from '../errors';
+import PDFDict from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFStream from '../objects/PDFStream';
+import { Cache } from '../../utils';
 
 class PDFFlateStream extends PDFStream {
-  protected readonly contentsCache: Cache<Uint8Array>;
+  protected contentsCache: Cache<Uint8Array>;
   protected readonly encode: boolean;
 
   constructor(dict: PDFDict, encode: boolean) {
@@ -38,6 +38,10 @@ class PDFFlateStream extends PDFStream {
       'getUnencodedContents',
     );
   }
+
+  updateContents(contents: Uint8Array): void {
+    this.contentsCache = Cache.populatedBy(() => contents);
+  }
 }
 
 export default PDFFlateStream;

package/src/core/structures/PDFObjectStream.ts

@@ -1,11 +1,11 @@
-import PDFName from 'src/core/objects/PDFName';
-import PDFNumber from 'src/core/objects/PDFNumber';
-import PDFObject from 'src/core/objects/PDFObject';
-import PDFRef from 'src/core/objects/PDFRef';
-import PDFContext from 'src/core/PDFContext';
-import PDFFlateStream from 'src/core/structures/PDFFlateStream';
-import CharCodes from 'src/core/syntax/CharCodes';
-import { copyStringIntoBuffer, last } from 'src/utils';
+import PDFName from '../objects/PDFName';
+import PDFNumber from '../objects/PDFNumber';
+import PDFObject from '../objects/PDFObject';
+import PDFRef from '../objects/PDFRef';
+import PDFContext from '../PDFContext';
+import PDFFlateStream from './PDFFlateStream';
+import CharCodes from '../syntax/CharCodes';
+import { copyStringIntoBuffer, last } from '../../utils';
 
 export type IndirectObject = [PDFRef, PDFObject];
 

package/src/core/structures/PDFPageLeaf.ts

@@ -1,12 +1,12 @@
-import PDFArray from 'src/core/objects/PDFArray';
-import PDFDict, { DictMap } from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFNumber from 'src/core/objects/PDFNumber';
-import PDFObject from 'src/core/objects/PDFObject';
-import PDFRef from 'src/core/objects/PDFRef';
-import PDFStream from 'src/core/objects/PDFStream';
-import PDFContext from 'src/core/PDFContext';
-import PDFPageTree from 'src/core/structures/PDFPageTree';
+import PDFArray from '../objects/PDFArray';
+import PDFDict, { DictMap } from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFNumber from '../objects/PDFNumber';
+import PDFObject from '../objects/PDFObject';
+import PDFRef from '../objects/PDFRef';
+import PDFStream from '../objects/PDFStream';
+import PDFContext from '../PDFContext';
+import PDFPageTree from './PDFPageTree';
 
 class PDFPageLeaf extends PDFDict {
   static readonly InheritableEntries = [

package/src/core/structures/PDFPageTree.ts

@@ -1,11 +1,11 @@
-import PDFArray from 'src/core/objects/PDFArray';
-import PDFDict, { DictMap } from 'src/core/objects/PDFDict';
-import PDFName from 'src/core/objects/PDFName';
-import PDFNumber from 'src/core/objects/PDFNumber';
-import PDFRef from 'src/core/objects/PDFRef';
-import PDFContext from 'src/core/PDFContext';
-import PDFPageLeaf from 'src/core/structures/PDFPageLeaf';
-import { InvalidTargetIndexError, CorruptPageTreeError } from 'src/core/errors';
+import PDFArray from '../objects/PDFArray';
+import PDFDict, { DictMap } from '../objects/PDFDict';
+import PDFName from '../objects/PDFName';
+import PDFNumber from '../objects/PDFNumber';
+import PDFRef from '../objects/PDFRef';
+import PDFContext from '../PDFContext';
+import PDFPageLeaf from './PDFPageLeaf';
+import { InvalidTargetIndexError, CorruptPageTreeError } from '../errors';
 
 export type TreeNode = PDFPageTree | PDFPageLeaf;
 

package/src/core/syntax/Delimiters.ts

@@ -1,4 +1,4 @@
-import CharCodes from 'src/core/syntax/CharCodes';
+import CharCodes from './CharCodes';
 
 export const IsDelimiter = new Uint8Array(256);
 

package/src/core/syntax/Irregular.ts

@@ -1,6 +1,6 @@
-import CharCodes from 'src/core/syntax/CharCodes';
-import { IsDelimiter } from 'src/core/syntax/Delimiters';
-import { IsWhitespace } from 'src/core/syntax/Whitespace';
+import CharCodes from './CharCodes';
+import { IsDelimiter } from './Delimiters';
+import { IsWhitespace } from './Whitespace';
 
 export const IsIrregular = new Uint8Array(256);
 

package/src/core/syntax/Keywords.ts

@@ -1,4 +1,4 @@
-import CharCodes from 'src/core/syntax/CharCodes';
+import CharCodes from './CharCodes';
 
 const { Space, CarriageReturn, Newline } = CharCodes;
 

package/src/core/syntax/Numeric.ts

@@ -1,4 +1,4 @@
-import CharCodes from 'src/core/syntax/CharCodes';
+import CharCodes from './CharCodes';
 
 export const IsDigit = new Uint8Array(256);
 

package/src/core/syntax/Whitespace.ts

@@ -1,4 +1,4 @@
-import CharCodes from 'src/core/syntax/CharCodes';
+import CharCodes from './CharCodes';
 
 export const IsWhitespace = new Uint8Array(256);
 

package/src/core/writers/PDFStreamWriter.ts

@@ -12,7 +12,10 @@ import PDFCrossRefStream from 'src/core/structures/PDFCrossRefStream';
 import PDFObjectStream from 'src/core/structures/PDFObjectStream';
 import PDFWriter from 'src/core/writers/PDFWriter';
 import { last, waitForTick } from 'src/utils';
-import PDFDict from '../objects/PDFDict';
+import PDFDict from 'src/core/objects/PDFDict';
+import PDFCatalog from 'src/core/structures/PDFCatalog';
+import PDFPageTree from 'src/core/structures/PDFPageTree';
+import PDFPageLeaf from 'src/core/structures/PDFPageLeaf';
 
 class PDFStreamWriter extends PDFWriter {
   static forContext = (
@@ -80,6 +83,8 @@ class PDFStreamWriter extends PDFWriter {
     const compressedObjects: [PDFRef, PDFObject][][] = [];
     const objectStreamRefs: PDFRef[] = [];
 
+    const security = this.context.security;
+
     const indirectObjects = this.context.enumerateIndirectObjects();
     for (let idx = 0, len = indirectObjects.length; idx < len; idx++) {
       const indirectObject = indirectObjects[idx];
@@ -92,12 +97,16 @@ class PDFStreamWriter extends PDFWriter {
         ref === this.context.trailerInfo.Encrypt ||
         object instanceof PDFStream ||
         object instanceof PDFInvalidObject ||
+        object instanceof PDFCatalog ||
+        object instanceof PDFPageTree ||
+        object instanceof PDFPageLeaf ||
         ref.generationNumber !== 0 ||
         (object instanceof PDFDict &&
           (object as PDFDict).lookup(PDFName.of('Type')) === PDFName.of('Sig'));
 
       if (shouldNotCompress) {
         uncompressedObjects.push(indirectObject);
+        if (security) this.encrypt(ref, object, security);
         xrefStream.addUncompressedEntry(ref, size);
         size += this.computeIndirectObjectSize(indirectObject);
         if (this.shouldWaitForTick(1)) await waitForTick();
@@ -125,6 +134,8 @@ class PDFStreamWriter extends PDFWriter {
         this.encodeStreams,
       );
 
+      if (security) this.encrypt(ref, objectStream, security);
+
       xrefStream.addUncompressedEntry(ref, size);
       size += this.computeIndirectObjectSize([ref, objectStream]);
 

package/src/core/writers/PDFWriter.ts

@@ -1,8 +1,8 @@
 import {
   DefaultDocumentSnapshot,
   defaultDocumentSnapshot,
-  DocumentSnapshot,
 } from 'src/api/snapshot';
+import type { DocumentSnapshot } from 'src/api/snapshot';
 import PDFCrossRefSection from 'src/core/document/PDFCrossRefSection';
 import PDFHeader from 'src/core/document/PDFHeader';
 import PDFTrailer from 'src/core/document/PDFTrailer';
@@ -14,7 +14,9 @@ import PDFContext from 'src/core/PDFContext';
 import PDFObjectStream from 'src/core/structures/PDFObjectStream';
 import CharCodes from 'src/core/syntax/CharCodes';
 import { copyStringIntoBuffer, waitForTick } from 'src/utils';
-import PDFNumber from '../objects/PDFNumber';
+import PDFNumber from 'src/core/objects/PDFNumber';
+import PDFSecurity from 'src/core/security/PDFSecurity';
+import PDFStream from 'src/core/objects/PDFStream';
 
 export interface SerializationInfo {
   size: number;
@@ -53,14 +55,8 @@ class PDFWriter {
 
   async serializeToBuffer(): Promise<Uint8Array> {
     const incremental = !(this.snapshot instanceof DefaultDocumentSnapshot);
-    const {
-      size,
-      header,
-      indirectObjects,
-      xref,
-      trailerDict,
-      trailer,
-    } = await this.computeBufferSize(incremental);
+    const { size, header, indirectObjects, xref, trailerDict, trailer } =
+      await this.computeBufferSize(incremental);
 
     let offset = 0;
     const buffer = new Uint8Array(size);
@@ -157,14 +153,15 @@ class PDFWriter {
 
     const xref = PDFCrossRefSection.create();
 
+    const security = this.context.security;
+
     const indirectObjects = this.context.enumerateIndirectObjects();
 
     for (let idx = 0, len = indirectObjects.length; idx < len; idx++) {
       const indirectObject = indirectObjects[idx];
-      const [ref] = indirectObject;
-      if (!this.snapshot.shouldSave(ref.objectNumber)) {
-        continue;
-      }
+      const [ref, object] = indirectObject;
+      if (!this.snapshot.shouldSave(ref.objectNumber)) continue;
+      if (security) this.encrypt(ref, object, security);
       xref.addEntry(ref, size);
       size += this.computeIndirectObjectSize(indirectObject);
       if (this.shouldWaitForTick(1)) await waitForTick();
@@ -185,6 +182,18 @@ class PDFWriter {
     return { size, header, indirectObjects, xref, trailerDict, trailer };
   }
 
+  protected encrypt(ref: PDFRef, object: PDFObject, security: PDFSecurity) {
+    if (object instanceof PDFStream) {
+      const encryptFn = security.getEncryptFn(
+        ref.objectNumber,
+        ref.generationNumber,
+      );
+      const unencryptedContents = object.getContents();
+      const encryptedContents = encryptFn(unencryptedContents);
+      object.updateContents(encryptedContents);
+    }
+  }
+
   protected shouldWaitForTick = (n: number) => {
     this.parsedObjects += n;
     return this.parsedObjects % this.objectsPerTick === 0;

package/src/index.ts

@@ -1,4 +1,4 @@
-export * from 'src/api/index';
-export * from 'src/core/index';
-export * from 'src/types/index';
-export * from 'src/utils/index';
+export * from './api';
+export * from './core';
+export * from './types';
+export * from './utils';

package/src/types/index.ts

@@ -1 +1,38 @@
-export { TransformationMatrix } from 'src/types/matrix';
+import Arc from '../utils/elements/Arc';
+import Circle from '../utils/elements/Circle';
+import Ellipse from '../utils/elements/Ellipse';
+import Line from '../utils/elements/Line';
+import Plot from '../utils/elements/Plot';
+import Point from '../utils/elements/Point';
+import Rectangle from '../utils/elements/Rectangle';
+import Segment from '../utils/elements/Segment';
+export { TransformationMatrix } from './matrix';
+
+export type Size = {
+  width: number;
+  height: number;
+};
+
+export type Coordinates = {
+  x: number;
+  y: number;
+};
+
+export type GraphicElement =
+  | Arc
+  | Circle
+  | Ellipse
+  | Line
+  | Plot
+  | Point
+  | Rectangle
+  | Segment;
+
+export type Space = {
+  topLeft: Coordinates;
+  topRight: Coordinates;
+  bottomRight: Coordinates;
+  bottomLeft: Coordinates;
+};
+
+export type LinkElement = Rectangle | Ellipse;

package/src/utils/arrays.ts

@@ -1,5 +1,5 @@
-import { decodeFromBase64DataUri } from 'src/utils/base64';
-import { charFromCode } from 'src/utils/strings';
+import { decodeFromBase64DataUri } from './base64';
+import { charFromCode } from './strings';
 
 export const last = <T>(array: T[]): T => array[array.length - 1];
 
@@ -84,6 +84,18 @@ export const sortedUniq = <T>(array: T[], indexer: (elem: T) => any): T[] => {
   return uniq;
 };
 
+export const isArrayEqual = <T>(arr1: ArrayLike<T>, arr2: ArrayLike<T>) => {
+  if (arr1.length !== arr2.length) {
+    return false;
+  }
+  for (let i = 0, ii = arr1.length; i < ii; i++) {
+    if (arr1[i] !== arr2[i]) {
+      return false;
+    }
+  }
+  return true;
+};
+
 // Arrays and TypedArrays in JS both have .reverse() methods, which would seem
 // to negate the need for this function. However, not all runtimes support this
 // method (e.g. React Native). This function compensates for that fact.
@@ -144,3 +156,22 @@ export const toUint8Array = (input: string | ArrayBuffer | Uint8Array) => {
     );
   }
 };
+
+// Precompute hex octets for best performance
+// Credit: https://stackoverflow.com/questions/40031688/javascript-arraybuffer-to-hex/40031979#40031979
+
+const byteToHex: string[] = [];
+
+for (let byte = 0x00; byte <= 0xff; ++byte) {
+  byteToHex[byte] = byte.toString(16).padStart(2, '0');
+}
+
+export const byteArrayToHexString = (array: Uint8Array) => {
+  const hexOctets = new Array(array.length);
+
+  for (let idx = 0; idx < array.length; ++idx) {
+    hexOctets[idx] = byteToHex[array[idx]];
+  }
+
+  return hexOctets.join('');
+};

package/src/utils/base64.ts

@@ -73,7 +73,8 @@ export const decodeFromBase64 = (base64: string): Uint8Array => {
 
 // This regex is designed to be as flexible as possible. It will parse certain
 // invalid data URIs.
-const DATA_URI_PREFIX_REGEX = /^(data)?:?([\w\/\+]+)?;?(charset=[\w-]+|base64)?.*,/i;
+const DATA_URI_PREFIX_REGEX =
+  /^(data)?:?([\w/+]+)?;?(charset=[\w-]+|base64)?.*,/i;
 
 /**
  * If the `dataUri` input is a data URI, then the data URI prefix must not be