49 lines
1.5 KiB
JavaScript
49 lines
1.5 KiB
JavaScript
import PuppeteerHTMLPDF from "puppeteer-html-pdf";
|
|
import { nanoid } from 'nanoid';
|
|
import fs from 'fs';
|
|
|
|
const pluginDefaults = {
|
|
// Plugin defaults
|
|
tempDir: "dist", // Directory where the temporary files are created
|
|
debug: false, // Keep the generated temporary html file for debugging
|
|
format: "A4", // Default pdf format
|
|
margin: { // PDF file margins
|
|
top: "1cm",
|
|
right: "1.25cm",
|
|
bottom: "1cm",
|
|
left: "1.25cm",
|
|
}
|
|
};
|
|
|
|
export default function (eleventyConfig, options = {}) {
|
|
const pluginConfig = Object.assign(pluginDefaults, options);
|
|
|
|
eleventyConfig.addTransform("toPDF", async (content) => {
|
|
if (this.outputPath && this.outputPath.toLowerCase().endsWith('.pdf')) {
|
|
let tempFileName = pluginConfig.debug ? this.page.fileSlug : nanoid();
|
|
let htmlFile = `./${pluginConfig.tempDir}/${tempFileName}.html`;
|
|
let pdfFile = `./${pluginConfig.tempDir}/${tempFileName}.pdf`;
|
|
|
|
fs.writeFileSync(htmlFile, content);
|
|
|
|
const htmlPDF = new PuppeteerHTMLPDF();
|
|
const pdfOptions = {
|
|
format: pluginConfig.format,
|
|
margin: pluginConfig.margin,
|
|
path: pdfFile,
|
|
};
|
|
htmlPDF.setOptions(pdfOptions);
|
|
|
|
const pdfContent = await htmlPDF.readFile(htmlFile, "utf8");
|
|
await htmlPDF.create(pdfContent);
|
|
let contents = fs.readFileSync(pdfFile, 'binary');
|
|
|
|
if (!pluginConfig.debug) {
|
|
fs.unlinkSync(htmlFile);
|
|
fs.unlinkSync(pdfFile);
|
|
}
|
|
|
|
return Buffer.from(contents,'binary');
|
|
} else return content;
|
|
});
|
|
}; |