Start the smart video controller

This commit is contained in:
Zoe Roux 2025-01-28 18:22:57 +01:00
parent 015f58226a
commit 1a2ab48c73
No known key found for this signature in database
4 changed files with 103 additions and 18 deletions

View File

@ -1,16 +1,23 @@
import { and, eq, inArray, sql } from "drizzle-orm";
import { Elysia, t } from "elysia";
import { db } from "~/db";
import { videos as videosT } from "~/db/schema";
import { entries, entryVideoJoin, shows, videos } from "~/db/schema";
import { bubbleVideo } from "~/models/examples";
import { SeedVideo, Video } from "~/models/video";
import { comment } from "~/utils";
import { computeVideoSlug } from "./seed/insert/entries";
const CreatedVideo = t.Object({
id: t.String({ format: "uuid" }),
path: t.String({ example: bubbleVideo.path }),
path: t.String({ examples: [bubbleVideo.path] }),
// entries: t.Array(
// t.Object({
// slug: t.String({ format: "slug", examples: ["bubble-v2"] }),
// }),
// ),
});
export const videos = new Elysia({ prefix: "/videos", tags: ["videos"] })
export const videosH = new Elysia({ prefix: "/videos", tags: ["videos"] })
.model({
video: Video,
"created-videos": t.Array(CreatedVideo),
@ -22,20 +29,101 @@ export const videos = new Elysia({ prefix: "/videos", tags: ["videos"] })
.post(
"",
async ({ body, error }) => {
const ret = await db
.insert(videosT)
const oldRet = await db
.insert(videos)
.values(body)
.onConflictDoNothing()
.returning({ id: videosT.id, path: videosT.path });
return error(201, ret);
.returning({
pk: videos.pk,
id: videos.id,
path: videos.path,
guess: videos.guess,
});
return error(201, oldRet);
// TODO: this is a huge untested wip
const vidsI = db.$with("vidsI").as(
db.insert(videos).values(body).onConflictDoNothing().returning({
pk: videos.pk,
id: videos.id,
path: videos.path,
guess: videos.guess,
}),
);
const findEntriesQ = db
.select({
guess: videos.guess,
entryPk: entries.pk,
showSlug: shows.slug,
// TODO: handle extras here
// guessit can't know if an episode is a special or not. treat specials like a normal episode.
kind: sql`
case when ${entries.kind} = 'movie' then 'movie' else 'episode' end
`.as("kind"),
season: entries.seasonNumber,
episode: entries.episodeNumber,
})
.from(entries)
.leftJoin(entryVideoJoin, eq(entryVideoJoin.entry, entries.pk))
.leftJoin(videos, eq(videos.pk, entryVideoJoin.video))
.leftJoin(shows, eq(shows.pk, entries.showPk))
.as("find_entries");
const hasRenderingQ = db
.select()
.from(entryVideoJoin)
.where(eq(entryVideoJoin.entry, findEntriesQ.entryPk));
const ret = await db
.with(vidsI)
.insert(entryVideoJoin)
.select(
db
.select({
entry: findEntriesQ.entryPk,
video: vidsI.pk,
slug: computeVideoSlug(
findEntriesQ.showSlug,
sql`exists(${hasRenderingQ})`,
),
})
.from(vidsI)
.leftJoin(
findEntriesQ,
and(
eq(
sql`${findEntriesQ.guess}->'title'`,
sql`${vidsI.guess}->'title'`,
),
// TODO: find if @> with a jsonb created on the fly is
// better than multiples checks
sql`${vidsI.guess} @> {"kind": }::jsonb`,
inArray(findEntriesQ.kind, sql`${vidsI.guess}->'type'`),
inArray(findEntriesQ.episode, sql`${vidsI.guess}->'episode'`),
inArray(findEntriesQ.season, sql`${vidsI.guess}->'season'`),
),
),
)
.onConflictDoNothing()
.returning({
slug: entryVideoJoin.slug,
entryPk: entryVideoJoin.entry,
id: vidsI.id,
path: vidsI.path,
});
return error(201, ret as any);
},
{
body: t.Array(SeedVideo),
response: { 201: "created-videos" },
response: { 201: t.Array(CreatedVideo) },
detail: {
description: comment`
Create videos in bulk.
Duplicated videos will simply be ignored.
If a videos has a \`guess\` field, it will be used to automatically register the video under an existing
movie or entry.
`,
},
},

View File

@ -7,7 +7,7 @@ import { movies } from "./controllers/movies";
import { seasonsH } from "./controllers/seasons";
import { seed } from "./controllers/seed";
import { series } from "./controllers/series";
import { videos } from "./controllers/videos";
import { videosH } from "./controllers/videos";
import { migrate } from "./db";
import { Image } from "./models/utils";
import { comment } from "./utils";
@ -75,7 +75,7 @@ const app = new Elysia()
.use(series)
.use(entries)
.use(seasonsH)
.use(videos)
.use(videosH)
.use(seed)
.listen(3000);

View File

@ -2,9 +2,6 @@ import { type TSchema, t } from "elysia";
import { comment } from "../utils";
import { bubbleVideo, registerExamples } from "./examples";
const Guess = <T extends TSchema>(schema: T) =>
t.Optional(t.Union([schema, t.Array(schema)]));
export const Video = t.Object({
id: t.String({ format: "uuid" }),
slug: t.String({ format: "slug" }),
@ -40,9 +37,9 @@ export const Video = t.Object({
t.Object(
{
title: t.String(),
year: Guess(t.Integer()),
season: Guess(t.Integer()),
episode: Guess(t.Integer()),
year: t.Array(t.Integer(), { default: [] }),
season: t.Array(t.Integer(), { default: [] }),
episode: t.Array(t.Integer(), { default: [] }),
// TODO: maybe replace "extra" with the `extraKind` value (aka behind-the-scene, trailer, etc)
type: t.Optional(t.UnionEnum(["episode", "movie", "extra"])),

View File

@ -4,7 +4,7 @@ import { base } from "~/base";
import { movies } from "~/controllers/movies";
import { seed } from "~/controllers/seed";
import { series } from "~/controllers/series";
import { videos } from "~/controllers/videos";
import { videosH } from "~/controllers/videos";
import type { SeedMovie } from "~/models/movie";
import type { SeedVideo } from "~/models/video";
@ -12,7 +12,7 @@ export const app = new Elysia()
.use(base)
.use(movies)
.use(series)
.use(videos)
.use(videosH)
.use(seed);
export const getMovie = async (