fix(pdf): recognise .mjs deps, use UInt8Array

opengovsg · May 23, 2024 · 767ffcc · 767ffcc
1 parent e697ec2
commit 767ffcc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/pdf2md-cli.js b/lib/pdf2md-cli.js
@@ -54,7 +54,7 @@ async function createMarkdownFiles (filePaths, allOutputPaths) {
     const callbacks = allOutputPaths[i] && {}
     const pdfBuffer = fs.readFileSync(filePath)
     try {
-      const text = await pdf2md(pdfBuffer, callbacks)
+      const text = await pdf2md(new Uint8Array(pdfBuffer), callbacks)
       const outputFile = allOutputPaths[i] + '.md'
       console.log(`Writing to ${outputFile}...`)
       fs.writeFileSync(path.resolve(outputFile), text)

diff --git a/lib/util/pdf.js b/lib/util/pdf.js
@@ -1,6 +1,9 @@
 const path = require('path')
-const pdfjs = require('pdfjs-dist/legacy/build/pdf')
-pdfjs.GlobalWorkerOptions.workerSrc = `pdfjs-dist/legacy/build/pdf.worker`
+const pdfjsPromise = import('pdfjs-dist/build/pdf.mjs')
+
+pdfjsPromise.then((pdfjs) => {
+  pdfjs.GlobalWorkerOptions.workerSrc = `pdfjs-dist/build/pdf.worker.mjs`
+})
 
 const { findPageNumbers, findFirstPage, removePageNumber } = require('../../lib/util/page-number-functions')
 const TextItem = require('../models/TextItem')
@@ -22,6 +25,7 @@ exports.parse = async function parse (docOptions, callbacks) {
   } else if (docOptions instanceof ArrayBuffer || ArrayBuffer.isView(docOptions)) {
     docOptions = { data: docOptions };
   }
+  const pdfjs = await pdfjsPromise
   const pdfDocument = await pdfjs.getDocument(
     {
       ...docOptions,