LiuShen / 截图API

import HOME_HTML from './home.html'; import SCREENSHOT_HTML from './screenshot.html'; export default { async fetch(request, env, ctx) { // 获取请求的 URL 和路径 const url = new URL(request.url); const path = url.pathname; // 路由不同的 API 接口 if (path.startsWith('/api/screenshot')) { return handleScreenshotRequest(url); } // 处理主页的请求 if (path === '/') { return new Response(HOME_HTML, { status: 200, headers: { "content-type": "text/html" } }); } if (path === '/screenshot') { return new Response(SCREENSHOT_HTML, { status: 200, headers: { "content-type": "text/html" } }); } return new Response('Invalid API endpoint', { status: 404 }); } } // 处理 /api/screenshot 请求 async function handleScreenshotRequest(url) { const targetUrl = url.searchParams.get('url') || 'https://www.liushen.fun/'; const width = url.searchParams.get('width') || 400; const height = url.searchParams.get('height') || 250; const method = url.searchParams.get('method') || '1'; // 默认为 mshot 方法 console.log(targetUrl, width, height, method); // 构造第一个截图 API 请求地址 (mshots) const screenshotUrl1 = `https://s0.wp.com/mshots/v1/${encodeURIComponent(targetUrl)}?w=${width}&h=${height}`; // 构造第二个截图 API 请求地址 (urlscan) const screenshotUrl2 = `https://urlscan.io/liveshot/?width=${width}&height=${height}&url=${encodeURIComponent(targetUrl)}`; try { let response; // 根据 method 参数选择首选的截图方式 if (method === '1') { // 请求 mshot response = await fetchWithUA(screenshotUrl1); if (!response.ok) { console.log("mshots failed, trying urlscan"); response = await fetchWithUA(screenshotUrl2); // 如果 mshots 失败，尝试 urlscan } } else { // 请求 urlscan response = await fetchWithUA(screenshotUrl2); if (!response.ok) { console.log("urlscan failed, trying mshots"); response = await fetchWithUA(screenshotUrl1); // 如果 urlscan 失败，尝试 mshots } } // 如果两个 API 都失败，返回 403 错误 if (!response.ok) { console.log("Both APIs failed, returning 403"); return new Response("Both screenshot APIs failed.", { status: 403 }); } return new Response(response.body, response); } catch (error) { // 捕获异常并返回 403 错误 console.log("Error occurred:", error); return new Response("Error occurred while processing the request.", { status: 403 }); } } // 添加带有 User-Agent 的 fetch 方法 async function fetchWithUA(url) { return await fetch(url, { headers: { 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36' } }); }