-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
43 lines (31 loc) · 1.16 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env nodejs
const puppeteer = require ('puppeteer');
const express = require('express');
var app = express(exports);
const main = async (url, landscape) => {
const browser = await puppeteer.launch({ headless: true });
const page = await browser.newPage();
let pdf = ''
console.log(url);
await page.goto(url, {waitUntil: 'networkidle0', timeout: 30000}).catch((res) => {
console.log('fails', res)
});
const pageStyle = landscape ? '@page { size: A4 landscape; } .print-hide { display: none !important; }': '@page { size: A4; } .print-hide { display: none !important; }'
page.addStyleTag(
{'content': pageStyle}
)
pdf = await page.pdf({ format: 'A4' }).catch((res) => {
console.log('fails', res)
});
await browser.close().catch((res) => {
console.log('fails', res)
});
return pdf
}
app.get('/headless-pdf', async function (req, res) {
const pdf = await main(req.query.url, req.query.landscape);
res.set("Content-Disposition", "attachment;filename=" + (req.query.filename || "print") + ".pdf");
res.contentType("application/pdf");
res.send(pdf);
});
app.listen(8080, function(){ console.log('Listening on 8080') });