77 lines
1.8 KiB
TypeScript
77 lines
1.8 KiB
TypeScript
// Basic Screenshot Example
|
|
// Minimal example for taking screenshots with Cloudflare Browser Rendering
|
|
|
|
import puppeteer from "@cloudflare/puppeteer";
|
|
|
|
interface Env {
|
|
MYBROWSER: Fetcher;
|
|
}
|
|
|
|
export default {
|
|
async fetch(request: Request, env: Env): Promise<Response> {
|
|
const { searchParams } = new URL(request.url);
|
|
const url = searchParams.get("url");
|
|
|
|
if (!url) {
|
|
return new Response("Missing ?url parameter. Example: ?url=https://example.com", {
|
|
status: 400,
|
|
});
|
|
}
|
|
|
|
let normalizedUrl: string;
|
|
try {
|
|
normalizedUrl = new URL(url).toString();
|
|
} catch {
|
|
return new Response("Invalid URL", { status: 400 });
|
|
}
|
|
|
|
// Launch browser
|
|
const browser = await puppeteer.launch(env.MYBROWSER);
|
|
|
|
try {
|
|
// Create new page
|
|
const page = await browser.newPage();
|
|
|
|
// Navigate to URL
|
|
await page.goto(normalizedUrl, {
|
|
waitUntil: "networkidle0", // Wait for network to be idle
|
|
timeout: 30000, // 30 second timeout
|
|
});
|
|
|
|
// Take screenshot
|
|
const screenshot = await page.screenshot({
|
|
fullPage: true, // Capture full scrollable page
|
|
type: "png",
|
|
});
|
|
|
|
// Clean up
|
|
await browser.close();
|
|
|
|
return new Response(screenshot, {
|
|
headers: {
|
|
"content-type": "image/png",
|
|
"cache-control": "public, max-age=3600", // Cache for 1 hour
|
|
},
|
|
});
|
|
} catch (error) {
|
|
// Always close browser on error
|
|
await browser.close();
|
|
throw error;
|
|
}
|
|
},
|
|
};
|
|
|
|
/**
|
|
* Deploy:
|
|
* npx wrangler deploy
|
|
*
|
|
* Test:
|
|
* https://your-worker.workers.dev/?url=https://example.com
|
|
*
|
|
* Configuration (wrangler.jsonc):
|
|
* {
|
|
* "browser": { "binding": "MYBROWSER" },
|
|
* "compatibility_flags": ["nodejs_compat"]
|
|
* }
|
|
*/
|