lifegpc/eh-downloader
1
0
Fork 0
mirror of https://github.com/lifegpc/eh-downloader.git synced 2026-07-08 01:31:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for filtering files by gids in

Browse Source 
get_random_file() function
This commit is contained in:
lifegpc
2023-12-14 19:35:49 +08:00
parent 788606ae7d
commit efcb276c45
2 changed files with 49 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
routes/api/file/random.ts
View File

@@ -10,7 +10,40 @@ export const handler: Handlers = {
const is_nsfw = await parse_bool(u.searchParams.get("is_nsfw"), null);
const is_ad = await parse_bool(u.searchParams.get("is_ad"), null);
const thumb = await parse_bool(u.searchParams.get("thumb"), false);
const f = m.db.get_random_file(is_nsfw, is_ad);
const tgids = u.searchParams.get("gids");
let gids = tgids
? new Set(
tgids.split(",").map((x) => parseInt(x)).filter((v) =>
!isNaN(v)
),
)
: null;
const meili_query = u.searchParams.get("meili_query");
const meili_filter = u.searchParams.get("meili_filter");
if (meili_query || meili_filter) {
if (!m.meilisearch) {
return new Response("Meilisearch is not enabled.", {
status: 400,
});
}
try {
const index = await m.meilisearch.gmeta;
const re = await index.search(meili_query, {
filter: meili_filter || undefined,
hitsPerPage: 1000,
attributesToRetrieve: ["gid"],
});
if (gids === null) {
gids = new Set<number>();
}
re.hits.forEach((d) => {
gids?.add(d.gid);
});
} catch (e) {
return new Response(e.message, { status: 400 });
}
}
const f = m.db.get_random_file(is_nsfw, is_ad, gids);
if (!f) return new Response("File not found.", { status: 404 });
const t = thumb ? "thumbnail" : "file";
return Response.redirect(`${get_host(req)}/api/${t}/${f.id}`);