34 lines
995 B
JavaScript
34 lines
995 B
JavaScript
const { execFile } = require('child_process');
|
|
const path = require('path');
|
|
const fs = require('fs');
|
|
|
|
const SCREENSHOTS_DIR = path.join(__dirname, '../public/screenshots');
|
|
|
|
function takeScreenshot(slug, port, useLocalhost = false) {
|
|
return new Promise((resolve, reject) => {
|
|
const outPath = path.join(SCREENSHOTS_DIR, `${slug}.png`);
|
|
const host = useLocalhost ? 'localhost' : '127.0.0.1';
|
|
const url = `http://${host}:${port}`;
|
|
|
|
const args = [
|
|
'screenshot',
|
|
'--viewport-size', '1280,800',
|
|
'--wait-for-timeout', '3000',
|
|
url,
|
|
outPath,
|
|
];
|
|
|
|
execFile('/usr/bin/playwright', args, { timeout: 30000 }, (err, stdout, stderr) => {
|
|
if (err) {
|
|
reject(new Error(`Screenshot failed: ${err.message}\n${stderr}`));
|
|
} else if (fs.existsSync(outPath)) {
|
|
resolve(`/screenshots/${slug}.png`);
|
|
} else {
|
|
reject(new Error('Screenshot file not created'));
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
module.exports = { takeScreenshot };
|