53 lines
1.4 KiB
JavaScript
53 lines
1.4 KiB
JavaScript
|
const puppeteer = require('puppeteer');
|
||
|
|
const path = require('path');
|
||
|
|
const { URL } = require('url');
|
||
|
|
|
||
|
|
// Check if a file path or URL is provided as an argument
|
||
|
|
const inputPath = process.argv[2];
|
||
|
|
if (!inputPath) {
|
||
|
|
console.error('Please provide the path to the HTML file or a URL as an argument.');
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Determine if the input is a URL
|
||
|
|
let isUrl;
|
||
|
|
try {
|
||
|
|
new URL(inputPath);
|
||
|
|
isUrl = true;
|
||
|
|
} catch (error) {
|
||
|
|
isUrl = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set the output file name (optional argument)
|
||
|
|
const outputFile = process.argv[3] || 'output.png';
|
||
|
|
|
||
|
|
(async () => {
|
||
|
|
const browser = await puppeteer.launch();
|
||
|
|
const page = await browser.newPage();
|
||
|
|
|
||
|
|
// Set viewport size and device scale factor for better readability
|
||
|
|
await page.setViewport({
|
||
|
|
width: 1240, // Set width of the viewport
|
||
|
|
height: 1754, // Set height of the viewport
|
||
|
|
deviceScaleFactor: 2 // Increase the device scale factor for a zoom effect
|
||
|
|
});
|
||
|
|
|
||
|
|
// Load the page, either from a URL or a local file
|
||
|
|
if (isUrl) {
|
||
|
|
await page.goto(inputPath, { waitUntil: 'networkidle0' });
|
||
|
|
} else {
|
||
|
|
const fullPath = path.resolve(inputPath);
|
||
|
|
await page.goto(`file://${fullPath}`, { waitUntil: 'networkidle0' });
|
||
|
|
}
|
||
|
|
|
||
|
|
// Capture the screenshot
|
||
|
|
await page.screenshot({
|
||
|
|
path: outputFile,
|
||
|
|
fullPage: true
|
||
|
|
});
|
||
|
|
|
||
|
|
await browser.close();
|
||
|
|
console.log(`Screenshot taken and saved as ${outputFile}`);
|
||
|
|
})();
|
||
|
|
|