mirror of
https://github.com/direct-img/direct-img.link.git
synced 2026-10-15 09:17:54 +00:00
multipleof4andClaude Opus 5.5
fce6fdfd0f
The AuroraIMG-6M API takes a 32-bit seed, so FNV-1a produces it directly instead of truncating a SHA-256 digest. It's ~28x faster than crypto.subtle (0.35us vs 9.7us per seed in workerd) and seeds don't need a cryptographic hash. Every query gets new AI images; cached ones are unaffected until they expire. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
38 lines
1.8 KiB
JavaScript
38 lines
1.8 KiB
JavaScript
// AI images for ai.direct-img.link, and the fallback when a search finds nothing
|
|
const SOURCES = { "auroraimg-6m": auroraImage };
|
|
export const DEFAULT_AI_SOURCE = "auroraimg-6m";
|
|
|
|
// ?src= on ai.*, case-insensitive. Returns null if invalid.
|
|
export function parseAiSource(searchParams) {
|
|
const src = searchParams.get("src")?.toLowerCase() || DEFAULT_AI_SOURCE;
|
|
return Object.hasOwn(SOURCES, src) ? src : null;
|
|
}
|
|
|
|
// Same result shape as fetchImage: { success, buffer, contentType } or { success: false, reason }
|
|
export async function aiImage(src, query, i, env) {
|
|
try {
|
|
return await SOURCES[src](query, seedFor(query, i), env);
|
|
} catch (err) {
|
|
return { success: false, reason: err.name === "TimeoutError" ? "Timeout" : err.message };
|
|
}
|
|
}
|
|
|
|
// Deterministic per query and i, so each ?i= is a different image of the same prompt and repeats are identical.
|
|
// 32-bit FNV-1a: the API takes a 32-bit seed, and this is ~28x faster than crypto.subtle (seeds don't need security)
|
|
function seedFor(query, i) {
|
|
let h = 0x811c9dc5, s = `${query}\n${i}`;
|
|
for (let k = 0; k < s.length; k++) h = Math.imul(h ^ s.charCodeAt(k), 0x01000193);
|
|
return h >>> 0;
|
|
}
|
|
|
|
// AuroraIMG-6M on our Dokploy server: https://github.com/direct-img/AuroraIMG-6M
|
|
async function auroraImage(query, seed, env) {
|
|
if (!env.AURORA_API_KEY) return { success: false, reason: "AURORA_API_KEY not set" };
|
|
const res = await fetch(`https://aurora.direct-img.link/generate?${new URLSearchParams({ prompt: query, seed })}`, {
|
|
headers: { Authorization: `Bearer ${env.AURORA_API_KEY}` },
|
|
signal: AbortSignal.timeout(15000),
|
|
});
|
|
if (!res.ok) return { success: false, reason: `HTTP ${res.status}` };
|
|
return { success: true, buffer: await res.arrayBuffer(), contentType: res.headers.get("content-type") };
|
|
}
|