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: Add Bing Image Search fallback for empty results

Browse Source
This commit is contained in:
multipleof4
2026-03-11 17:12:59 -07:00
parent cd3a9a5446
commit 6a2f7baa01
1 changed files with 38 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -44,7 +44,6 @@ export async function onRequest(context) {
"surreal-db": "rate_limit", "surreal-db": "rate_limit",
}; };
// First ensure namespace and database exist
try { try {
const initSql = `DEFINE NAMESPACE IF NOT EXISTS direct_img; USE NS direct_img; DEFINE DATABASE IF NOT EXISTS rate_limit;`; const initSql = `DEFINE NAMESPACE IF NOT EXISTS direct_img; USE NS direct_img; DEFINE DATABASE IF NOT EXISTS rate_limit;`;
await fetch(`${env.SURREAL_URL}/sql`, { await fetch(`${env.SURREAL_URL}/sql`, {
@@ -141,8 +140,14 @@ export async function onRequest(context) {
return env.ASSETS.fetch(new Request(new URL("/bad.webp", url.origin))); return env.ASSETS.fetch(new Request(new URL("/bad.webp", url.origin)));
}; };
const imageUrls = await braveImageSearch(query, env.BRAVE_API_KEY); let imageUrls = await braveImageSearch(query, env.BRAVE_API_KEY);
if (!imageUrls || imageUrls.length === 0) return await fail("Search Failed", `Brave API returned no results for: ${query}`, "question", 3);
if (!imageUrls || imageUrls.length === 0) {
context.waitUntil(notify(env, { title: "Brave Search Empty", message: `No results for: ${query}. Trying Bing Fallback.`, tags: "warning,mag", priority: 3 }));
imageUrls = await bingImageSearchFallback(query);
}
if (!imageUrls || imageUrls.length === 0) return await fail("Search Failed", `Both Brave and Bing returned no results for: ${query}`, "question", 3);
const GLOBAL_DEADLINE = Date.now() + 20000; const GLOBAL_DEADLINE = Date.now() + 20000;
let imgResult = null; let imgResult = null;
@@ -212,6 +217,36 @@ async function braveImageSearch(query, apiKey) {
return data.results?.map(r => r.properties?.url || r.thumbnail?.src).filter(url => !!url) || null; return data.results?.map(r => r.properties?.url || r.thumbnail?.src).filter(url => !!url) || null;
} }
async function bingImageSearchFallback(query) {
try {
const res = await fetch(`https://www.bing.com/images/search?q=${encodeURIComponent(query)}&safesearch=moderate`, {
headers: {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36",
"Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8"
}
});
if (!res.ok) return null;
const html = await res.text();
const urls = [];
// Bing encodes the image payload inside HTML entity attributes and script tags
const regex1 = /"murl":"(https:[^&"]+)"/g;
const regex2 = /"murl":"(https:[^"]+)"/g;
let match;
while ((match = regex1.exec(html)) !== null) {
urls.push(match[1].replace(/\\\//g, '/'));
}
while ((match = regex2.exec(html)) !== null) {
urls.push(match[1].replace(/\\\//g, '/'));
}
return urls.length > 0 ? [...new Set(urls)] : null;
} catch (err) {
return null;
}
}
async function fetchImage(imageUrl, timeoutMs = 5000) { async function fetchImage(imageUrl, timeoutMs = 5000) {
try { try {
const res = await fetch(imageUrl, { const res = await fetch(imageUrl, {