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'
}
});
}