mirror of
https://github.com/frappe/books.git
synced 2024-12-24 20:00:29 +00:00
54 lines
1.6 KiB
JavaScript
54 lines
1.6 KiB
JavaScript
const frappe = require('frappejs');
|
|
const puppeteer = require('puppeteer');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const { getTmpDir } = require('frappejs/server/utils');
|
|
const { getHTML } = require('frappejs/common/print');
|
|
const { getRandomString } = require('frappejs/utils');
|
|
|
|
async function makePDF(html, filepath) {
|
|
const browser = await puppeteer.launch();
|
|
const page = await browser.newPage();
|
|
await page.setContent(html);
|
|
await page.pdf({
|
|
path: filepath,
|
|
format: 'A4'
|
|
});
|
|
await browser.close();
|
|
}
|
|
|
|
async function getPDFForElectron(doctype, name, destination) {
|
|
const { shell } = require('electron');
|
|
const html = await getHTML(doctype, name);
|
|
const filepath = path.join(destination, name + '.pdf');
|
|
await makePDF(html, filepath);
|
|
shell.openItem(filepath);
|
|
}
|
|
|
|
function setupExpressRoute() {
|
|
if (!frappe.app) return;
|
|
frappe.app.post('/api/method/pdf', frappe.asyncHandler(handlePDFRequest));
|
|
}
|
|
|
|
async function handlePDFRequest(req, res) {
|
|
const args = req.body;
|
|
const { doctype, name } = args;
|
|
const html = await getHTML(doctype, name);
|
|
|
|
const filepath = path.join(getTmpDir(), `frappe-pdf-${getRandomString()}.pdf`);
|
|
await makePDF(html, filepath);
|
|
|
|
const file = fs.createReadStream(filepath);
|
|
const stat = fs.statSync(filepath);
|
|
res.setHeader('Content-Length', stat.size);
|
|
res.setHeader('Content-Type', 'application/pdf');
|
|
res.setHeader('Content-Disposition', `attachment; filename=${path.basename(filepath)}`);
|
|
file.pipe(res);
|
|
}
|
|
|
|
module.exports = {
|
|
makePDF,
|
|
setupExpressRoute,
|
|
getPDFForElectron
|
|
}
|