Rework & type relations

api/src/controllers/shows/logic.ts

@@ -17,16 +17,15 @@ import {
 	sqlarr,
 } from "~/db/utils";
 import type { MovieStatus } from "~/models/movie";
-import { SerieStatus, SerieTranslation } from "~/models/serie";
-import { Studio } from "~/models/studio";
+import { SerieStatus, type SerieTranslation } from "~/models/serie";
+import type { Studio } from "~/models/studio";
 import {
 	type FilterDef,
 	Genre,
 	type Image,
 	Sort,
-	isUuid,
+	buildRelations,
 	keysetPaginate,
-	selectTranslationQuery,
 	sortToSql,
 } from "~/models/utils";
 
@@ -69,14 +68,60 @@ 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
-	>;
+const showRelations = {
+	translations: () => {
+		const { pk, language, ...trans } = getColumns(showTranslations);
+		return db
+			.select({
+				json: jsonbObjectAgg(
+					language,
+					jsonbBuildObject<SerieTranslation>(trans),
+				).as("json"),
+			})
+			.from(showTranslations)
+			.where(eq(showTranslations.pk, shows.pk))
+			.as("translations");
+	},
+	studios: ({ languages }: { languages: string[] }) => {
+		const { pk: _, ...studioCol } = getColumns(studios);
+		const studioTransQ = db
+			.selectDistinctOn([studioTranslations.pk])
+			.from(studioTranslations)
+			.orderBy(
+				studioTranslations.pk,
+				sql`array_position(${sqlarr(languages)}, ${studioTranslations.language}`,
+			)
+			.as("t");
+		const { pk, language, ...studioTrans } = getColumns(studioTransQ);
+
+		return db
+			.select({
+				json: coalesce(
+					jsonbAgg(jsonbBuildObject<Studio>({ ...studioTrans, ...studioCol })),
+					sql`'[]'::jsonb`,
+				).as("json"),
+			})
+			.from(studios)
+			.leftJoin(studioTransQ, eq(studios.pk, studioTransQ.pk))
+			.where(
+				exists(
+					db
+						.select()
+						.from(showStudioJoin)
+						.where(
+							and(
+								eq(showStudioJoin.studioPk, studios.pk),
+								eq(showStudioJoin.showPk, shows.pk),
+							),
+						),
+				),
+			)
+			.as("studios");
+	},
+	// only available for movies
+	videos: () => {
+		throw new Error();
+	},
 };
 
 export async function getShows({
@@ -115,61 +160,6 @@ export async function getShows({
 		.as("t");
 	const { pk, ...transCol } = getColumns(transQ);
 
-	const relationsSql = buildRelations(relations, (x) => {
-		switch (x) {
-			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<SerieTranslation[]>`${db
-					.select({ json: jsonbObjectAgg(language, jsonbBuildObject(trans)) })
-					.from(showTranslations)
-					.where(eq(showTranslations.pk, shows.pk))}`;
-			}
-			case "studios": {
-				const { pk: _, ...studioCol } = getColumns(studios);
-				const studioTransQ = db
-					.selectDistinctOn([studioTranslations.pk])
-					.from(studioTranslations)
-					.where(
-						!fallbackLanguage
-							? eq(showTranslations.language, sql`any(${sqlarr(languages)})`)
-							: undefined,
-					)
-					.orderBy(
-						studioTranslations.pk,
-						sql`array_position(${sqlarr(languages)}, ${studioTranslations.language}`,
-					)
-					.as("t");
-				const { pk, language, ...studioTrans } = getColumns(studioTransQ);
-
-				return sql<Studio>`${db
-					.select({
-						json: coalesce(
-							jsonbAgg(jsonbBuildObject({ ...studioTrans, ...studioCol })),
-							sql`'[]'::jsonb`,
-						),
-					})
-					.from(studios)
-					.leftJoin(studioTransQ, eq(studios.pk, studioTransQ.pk))
-					.where(
-						exists(
-							db
-								.select()
-								.from(showStudioJoin)
-								.where(
-									and(
-										eq(showStudioJoin.studioPk, studios.pk),
-										eq(showStudioJoin.showPk, shows.pk),
-									),
-								),
-						),
-					)}`;
-			}
-		}
-	});
-
 	return await db
 		.select({
 			...getColumns(shows),
@@ -189,7 +179,7 @@ export async function getShows({
 				logo: sql<Image>`coalesce(nullif(${shows.original}->'logo', 'null'::jsonb), ${transQ.logo})`,
 			}),
 
-			...relationsSql,
+			...buildRelations(relations, showRelations, { languages }),
 		})
 		.from(shows)
 		[fallbackLanguage ? "innerJoin" : "leftJoin"](

api/src/db/utils.ts

@@ -1,5 +1,6 @@
 import {
 	type ColumnsSelection,
+	InferColumnsDataTypes,
 	type SQL,
 	type SQLWrapper,
 	type Subquery,
@@ -94,24 +95,26 @@ export function values(items: Record<string, unknown>[]) {
 	};
 }
 
-export const coalesce = (val: SQLWrapper, def: SQLWrapper) => {
-	return sql`coalesce(${val}, ${def})`;
+export const coalesce = <T>(val: SQL<T>, def: SQLWrapper) => {
+	return sql<T>`coalesce(${val}, ${def})`;
 };
 
-export const jsonbObjectAgg = (key: SQLWrapper, value: SQLWrapper) => {
-	return sql`jsonb_object_agg(${sql.join([key, value], sql.raw(","))})`;
+export const jsonbObjectAgg = <T>(key: SQLWrapper, value: SQL<T>) => {
+	return sql<
+		Record<string, T>
+	>`jsonb_object_agg(${sql.join([key, value], sql.raw(","))})`;
 };
 
-export const jsonbAgg = (val: SQLWrapper) => {
-	return sql`jsonb_agg(${val})`;
+export const jsonbAgg = <T>(val: SQL<T>) => {
+	return sql<T>`jsonb_agg(${val})`;
 };
 
-export const jsonbBuildObject = (select: Record<string, SQLWrapper>) => {
+export const jsonbBuildObject = <T>(select: Record<string, SQLWrapper>) => {
 	const query = sql.join(
 		Object.entries(select).flatMap(([k, v]) => {
 			return [sql.raw(`'${k}'`), v];
 		}),
 		sql.raw(", "),
 	);
-	return sql`jsonb_build_object(${query})`;
+	return sql<T>`jsonb_build_object(${query})`;
 };

api/src/models/utils/index.ts

@@ -9,3 +9,4 @@ export * from "./sort";
 export * from "./keyset-paginate";
 export * from "./db-metadata";
 export * from "./original";
+export * from "./relations";

api/src/models/utils/relations.ts

@@ -0,0 +1,26 @@
+import { type SQL, type Subquery, sql } from "drizzle-orm";
+import type { SelectResultField } from "drizzle-orm/query-builders/select.types";
+
+export const buildRelations = <
+	R extends string,
+	P extends object,
+	Rel extends Record<R, (languages: P) => Subquery>,
+>(
+	enabled: R[],
+	relations: Rel,
+	params: P,
+) => {
+	// we wrap that in a sql`` instead of using the builder because of this issue
+	// https://github.com/drizzle-team/drizzle-orm/pull/1674
+	return Object.fromEntries(
+		enabled.map((x) => [x, sql`${relations[x](params)}`]),
+	) as {
+		[P in R]?: SQL<
+			ReturnType<Rel[P]>["_"]["selectedFields"] extends {
+				[key: string]: infer TValue;
+			}
+				? SelectResultField<TValue>
+				: never
+		>;
+	};
+};