direct-img/direct-img.link
1
0
Fork 0
mirror of https://github.com/direct-img/direct-img.link.git synced 2026-03-17 03:01:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Cache error state and serve bad.webp on fail

Browse Source
This commit is contained in:
multipleof4
2026-02-23 10:39:47 -08:00
parent cf7f35e184
commit 86f97ed406
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
functions/[[path]].js
View File

@@ -21,6 +21,7 @@ export async function onRequest(context) {
const cached = await env.DIRECT_IMG_CACHE.get(cacheKey, "json"); const cached = await env.DIRECT_IMG_CACHE.get(cacheKey, "json");
if (cached) { if (cached) {
if (cached.err) return env.ASSETS.fetch(new Request(new URL("/bad.webp", url.origin)));
const obj = await env.R2_IMAGES.get(r2Key); const obj = await env.R2_IMAGES.get(r2Key);
if (obj) { if (obj) {
const nowSec = Math.floor(Date.now() / 1000); const nowSec = Math.floor(Date.now() / 1000);
@@ -36,7 +37,6 @@ export async function onRequest(context) {
if (env.SURREAL_URL && env.SURREAL_USER && env.SURREAL_PASS) { if (env.SURREAL_URL && env.SURREAL_USER && env.SURREAL_PASS) {
const auth = btoa(`${env.SURREAL_USER}:${env.SURREAL_PASS}`); const auth = btoa(`${env.SURREAL_USER}:${env.SURREAL_PASS}`);
// Atomic upsert + increment, while recording the timestamp
const sql = `UPDATE rate:\`${rateId}\` SET count += 1, updated_at = time::now() RETURN count;`; const sql = `UPDATE rate:\`${rateId}\` SET count += 1, updated_at = time::now() RETURN count;`;
try { try {
@@ -51,7 +51,6 @@ export async function onRequest(context) {
if (data[0]?.status === "OK" && data[0]?.result?.length > 0) count = data[0].result[0].count; if (data[0]?.status === "OK" && data[0]?.result?.length > 0) count = data[0].result[0].count;
} }
// Background cleanup: ~5% chance to sweep records older than 25h asynchronously
if (Math.random() < 0.05) { if (Math.random() < 0.05) {
context.waitUntil( context.waitUntil(
fetch(`${env.SURREAL_URL}/sql`, { fetch(`${env.SURREAL_URL}/sql`, {
@@ -73,11 +72,14 @@ export async function onRequest(context) {
context.waitUntil(notify(env, { title: "New Search", message: `Query: ${query} (Search #${count} for ${ip})\n${url.origin}/${path}`, tags: "mag", priority: 3 })); context.waitUntil(notify(env, { title: "New Search", message: `Query: ${query} (Search #${count} for ${ip})\n${url.origin}/${path}`, tags: "mag", priority: 3 }));
const fail = async (t, m, tag, p) => {
context.waitUntil(notify(env, { title: t, message: m, tags: tag, priority: p }));
await env.DIRECT_IMG_CACHE.put(cacheKey, JSON.stringify({ t: Math.floor(Date.now() / 1000), err: true }), { expirationTtl: 86400 });
return env.ASSETS.fetch(new Request(new URL("/bad.webp", url.origin)));
};
const imageUrls = await braveImageSearch(query, env.BRAVE_API_KEY); const imageUrls = await braveImageSearch(query, env.BRAVE_API_KEY);
if (!imageUrls || imageUrls.length === 0) { if (!imageUrls || imageUrls.length === 0) return await fail("Search Failed", `No results for: ${query}`, "question", 3);
context.waitUntil(notify(env, { title: "Search Failed", message: `No results for: ${query}`, tags: "question", priority: 3 }));
return jsonResponse(404, { error: "No image found for query" });
}
const GLOBAL_DEADLINE = Date.now() + 20000; const GLOBAL_DEADLINE = Date.now() + 20000;
let imgResult = null; let imgResult = null;
@@ -89,10 +91,7 @@ export async function onRequest(context) {
if (imgResult) break; if (imgResult) break;
} }
if (!imgResult) { if (!imgResult) return await fail("Fetch Error (502)", `All sources failed for: ${query}`, "boom,x", 4);
context.waitUntil(notify(env, { title: "Fetch Error (502)", message: `All sources failed for: ${query}`, tags: "boom,x", priority: 4 }));
return jsonResponse(502, { error: "Failed to fetch image from all available sources" });
}
const { buffer: imgBuffer, contentType: finalContentType } = imgResult; const { buffer: imgBuffer, contentType: finalContentType } = imgResult;
await env.R2_IMAGES.put(r2Key, imgBuffer, { httpMetadata: { contentType: finalContentType } }); await env.R2_IMAGES.put(r2Key, imgBuffer, { httpMetadata: { contentType: finalContentType } });