screenshoteer/index.js

59 lines
1.9 KiB
JavaScript
Raw Normal View History

2018-12-02 12:06:38 +01:00
#!/usr/bin/env node
const puppeteer = require('puppeteer')
const devices = require('puppeteer/DeviceDescriptors')
const program = require('commander')
var urlvalue, emulate = "";
program
.option('--url, [url]', 'The url')
.option('--emulate, [emulate]', 'emulate device')
.option('--fullpage, [fullpage]', 'Full Page')
.option('--pdf, [pdf]', 'Generate PDF')
.option('--w, [w]', 'width')
.option('--h, [h]', 'height')
2018-12-04 13:38:32 +01:00
.option('--waitfor, [waitfor]', 'Wait time in milliseconds')
2018-12-04 16:23:06 +01:00
.option('--el, [el]', 'element css selector')
2018-12-02 12:06:38 +01:00
.parse(process.argv);
if (program.url) urlvalue = program.url
else process.exit(console.log("Please add --url parameter. Something like this: $ screenshoteer --url http:www.example.com"));
2018-12-06 10:14:59 +01:00
!program.fullpage ? fullPage = true : fullPage = JSON.parse(program.fullpage);
2018-12-02 12:06:38 +01:00
console.log(urlvalue);
console.log(fullPage);
(async () => {
2018-12-05 17:00:18 +01:00
try {
await execute();
} catch(e) {
console.error(e);
process.exit(1);
}
async function execute() {
const browser = await puppeteer.launch()
const page = await browser.newPage()
2018-12-06 10:14:59 +01:00
const timestamp = new Date().getTime()
2018-12-05 17:00:18 +01:00
if (program.w && program.h) await page.setViewport({width: Number(program.w), height: Number(program.h)})
if (program.emulate) await page.emulate(devices[program.emulate]);
await page.goto(urlvalue)
2018-12-06 10:14:59 +01:00
const title = (await page.title()).replace(/[/\\?%*:|"<>]/g, '-')
2018-12-05 17:00:18 +01:00
if (program.waitfor) await page.waitFor(Number(program.waitfor))
if (program.el) {
const el = await page.$(program.el);
2018-12-06 10:14:59 +01:00
await el.screenshot({path: `${title} ${program.emulate} ${program.el} ${timestamp}.png`});
2018-12-05 17:00:18 +01:00
} else {
2018-12-06 10:14:59 +01:00
await page.screenshot({path: `${title} ${program.emulate} ${timestamp}.png`, fullPage: fullPage})
2018-12-05 17:00:18 +01:00
}
await page.emulateMedia('screen')
2018-12-06 10:14:59 +01:00
if (program.pdf) await page.pdf({path: `${title} ${program.emulate} ${timestamp}.pdf`})
console.log(title)
2018-12-05 17:00:18 +01:00
await browser.close()
2018-12-04 16:23:06 +01:00
}
2018-12-05 17:00:18 +01:00
2018-12-02 12:06:38 +01:00
})()