88 lines
1.9 KiB
TypeScript
88 lines
1.9 KiB
TypeScript
|
|
/// <reference lib="webworker" />
|
|
|
|
import PDFDocument from 'pdfkit/lib/document'
|
|
import SVGtoPDF from 'svg-to-pdfkit'
|
|
|
|
type ImgType = 'svg' | 'png'
|
|
|
|
const getDataURL = (blob: Blob): Promise<string> => {
|
|
return new Promise((resolve, reject) => {
|
|
const reader = new FileReader()
|
|
reader.onload = (): void => {
|
|
const result = reader.result
|
|
resolve(result as string)
|
|
}
|
|
reader.onerror = reject
|
|
reader.readAsDataURL(blob)
|
|
})
|
|
}
|
|
|
|
const fetchDataURL = async (imgUrl: string): Promise<string> => {
|
|
const r = await fetch(imgUrl)
|
|
const blob = await r.blob()
|
|
return getDataURL(blob)
|
|
}
|
|
|
|
const fetchText = async (imgUrl: string): Promise<string> => {
|
|
const r = await fetch(imgUrl)
|
|
return r.text()
|
|
}
|
|
|
|
const generatePDF = async (imgURLs: string[], imgType: ImgType, width: number, height: number): Promise<ArrayBuffer> => {
|
|
// @ts-ignore
|
|
const pdf = new (PDFDocument as typeof import('pdfkit'))({
|
|
// compress: true,
|
|
size: [width, height],
|
|
autoFirstPage: false,
|
|
margin: 0,
|
|
layout: 'portrait',
|
|
})
|
|
|
|
if (imgType === 'png') {
|
|
const imgDataUrlList: string[] = await Promise.all(imgURLs.map(fetchDataURL))
|
|
|
|
imgDataUrlList.forEach((data) => {
|
|
pdf.addPage()
|
|
pdf.image(data, {
|
|
width,
|
|
height,
|
|
})
|
|
})
|
|
} else { // imgType == "svg"
|
|
const svgList = await Promise.all(imgURLs.map(fetchText))
|
|
|
|
svgList.forEach((svg) => {
|
|
pdf.addPage()
|
|
SVGtoPDF(pdf, svg, 0, 0, {
|
|
preserveAspectRatio: 'none',
|
|
})
|
|
})
|
|
}
|
|
|
|
// @ts-ignore
|
|
const buf: Uint8Array = await pdf.getBuffer()
|
|
|
|
return buf.buffer
|
|
}
|
|
|
|
export type PDFWorkerMessage = [string[], ImgType, number, number];
|
|
|
|
onmessage = async (e): Promise<void> => {
|
|
const [
|
|
imgURLs,
|
|
imgType,
|
|
width,
|
|
height,
|
|
] = e.data as PDFWorkerMessage
|
|
|
|
const pdfBuf = await generatePDF(
|
|
imgURLs,
|
|
imgType,
|
|
width,
|
|
height,
|
|
)
|
|
|
|
postMessage(pdfBuf, [pdfBuf])
|
|
}
|