Add with=translations in the sql builder logic

This commit is contained in:
Zoe Roux
2025-03-09 01:16:50 +01:00
parent 3f77a1bda5
commit c161d680e3
7 changed files with 104 additions and 11 deletions
+40 -5
View File
@@ -2,7 +2,12 @@ import type { StaticDecode } from "@sinclair/typebox";
import { type SQL, and, eq, sql } from "drizzle-orm";
import { db } from "~/db";
import { showTranslations, shows, studioTranslations } from "~/db/schema";
import { getColumns, sqlarr } from "~/db/utils";
import {
getColumns,
jsonbBuildObject,
jsonbObjectAgg,
sqlarr,
} from "~/db/utils";
import type { MovieStatus } from "~/models/movie";
import { SerieStatus } from "~/models/serie";
import {
@@ -55,6 +60,16 @@ export const showSort = Sort(
},
);
const buildRelations = <R extends string>(
relations: R[],
toSql: (relation: R) => SQL,
) => {
return Object.fromEntries(relations.map((x) => [x, toSql(x)])) as Record<
R,
SQL
>;
};
export async function getShows({
after,
limit,
@@ -64,15 +79,17 @@ export async function getShows({
languages,
fallbackLanguage = true,
preferOriginal = false,
relations = [],
}: {
after: string | undefined;
after?: string;
limit: number;
query: string | undefined;
sort: StaticDecode<typeof showSort>;
filter: SQL | undefined;
query?: string;
sort?: StaticDecode<typeof showSort>;
filter?: SQL;
languages: string[];
fallbackLanguage?: boolean;
preferOriginal?: boolean;
relations?: ("translations" | "studios" | "videos")[];
}) {
const transQ = db
.selectDistinctOn([showTranslations.pk])
@@ -89,6 +106,22 @@ export async function getShows({
.as("t");
const { pk, ...transCol } = getColumns(transQ);
const relationsSql = buildRelations(relations, (x) => {
switch (x) {
case "studios":
case "videos":
case "translations": {
// we wrap that in a sql`` instead of using the builder because of this issue
// https://github.com/drizzle-team/drizzle-orm/pull/1674
const { pk, language, ...trans } = getColumns(showTranslations);
return sql`${db
.select({ json: jsonbObjectAgg(language, jsonbBuildObject(trans)) })
.from(showTranslations)
.where(eq(showTranslations.pk, shows.pk))}`;
}
}
});
return await db
.select({
...getColumns(shows),
@@ -107,6 +140,8 @@ export async function getShows({
banner: sql<Image>`coalesce(nullif(${shows.original}->'banner', 'null'::jsonb), ${transQ.banner})`,
logo: sql<Image>`coalesce(nullif(${shows.original}->'logo', 'null'::jsonb), ${transQ.logo})`,
}),
...relationsSql,
})
.from(shows)
[fallbackLanguage ? "innerJoin" : "leftJoin"](
+1
View File
@@ -32,6 +32,7 @@ export const movies = new Elysia({ prefix: "/movies", tags: ["movies"] })
}) => {
const langs = processLanguages(languages);
const [ret] = await getShows({
limit: 1,
filter: and(
isUuid(id) ? eq(shows.id, id) : eq(shows.slug, id),
eq(shows.kind, "movie"),
+9 -3
View File
@@ -10,6 +10,7 @@ import {
Filter,
Page,
createPage,
isUuid,
processLanguages,
} from "~/models/utils";
import { desc } from "~/models/utils/descriptions";
@@ -30,11 +31,16 @@ export const series = new Elysia({ prefix: "/series", tags: ["series"] })
set,
}) => {
const langs = processLanguages(languages);
const ret = await getShow(id, {
const [ret] = await getShows({
limit: 1,
filter: and(
isUuid(id) ? eq(shows.id, id) : eq(shows.slug, id),
eq(shows.kind, "serie"),
),
languages: langs,
fallbackLanguage: langs.includes("*"),
preferOriginal,
relations,
filters: eq(shows.kind, "serie"),
});
if (!ret) {
return error(404, {
@@ -49,7 +55,7 @@ export const series = new Elysia({ prefix: "/series", tags: ["series"] })
});
}
set.headers["content-language"] = ret.language;
return ret.show;
return ret;
},
{
detail: {