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

17
db.ts
View File

@@ -1062,12 +1062,24 @@ export class EhDb {
get_random_file(
is_nsfw: boolean | null = null,
is_ad: boolean | null = null,
gids: Set<number> | null = null,
) {
const args = [];
let join_sql = "";
const with_sql = [];
const where_sql = [];
if (gids) {
const gids_sql = [];
for (const gid of gids) {
gids_sql.push("(?)");
args.push(gid);
}
with_sql.push(`WITH gids AS (VALUES ${gids_sql.join(",")})`);
join_sql += " INNER JOIN pmeta ON file.token = pmeta.token";
where_sql.push("pmeta.gid IN gids");
}
if (is_nsfw !== null || is_ad !== null) {
join_sql = " LEFT JOIN filemeta ON file.token = filemeta.token";
join_sql += " LEFT JOIN filemeta ON file.token = filemeta.token";
if (is_nsfw !== null) {
where_sql.push("IFNULL(filemeta.is_nsfw, 0) = ?");
args.push(is_nsfw);
@@ -1080,9 +1092,10 @@ export class EhDb {
const wsql = where_sql.length
? ` WHERE ${where_sql.join(" AND ")}`
: "";
const wisql = with_sql.length ? `${with_sql.join(" ")} ` : "";
const s = this.convert_file(
this.db.queryEntries<EhFileRaw>(
`SELECT file.* FROM file${join_sql}${wsql} ORDER BY RANDOM() LIMIT 1;`,
`${wisql}SELECT file.* FROM file${join_sql}${wsql} ORDER BY RANDOM() LIMIT 1;`,
args,
),
);