feat: videoInfo API in backend

This commit is contained in:
alikia2x (寒寒) 2025-04-07 21:29:53 +08:00
parent b4205049cb
commit 71ed0bd66b
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
10 changed files with 100 additions and 36 deletions

View File

@ -9,18 +9,18 @@ export const db = pool;
export const dbCred = poolCred; export const dbCred = poolCred;
export const dbMiddleware = createMiddleware(async (c, next) => { export const dbMiddleware = createMiddleware(async (c, next) => {
const connection = await pool.connect(); const connection = await pool.connect();
c.set("db", connection); c.set("db", connection);
await next(); await next();
connection.release(); connection.release();
}); });
export const dbCredMiddleware = createMiddleware(async (c, next) => { export const dbCredMiddleware = createMiddleware(async (c, next) => {
const connection = await poolCred.connect(); const connection = await poolCred.connect();
c.set("dbCred", connection); c.set("dbCred", connection);
await next(); await next();
connection.release(); connection.release();
}) });
declare module "hono" { declare module "hono" {
interface ContextVariableMap { interface ContextVariableMap {

View File

@ -4,11 +4,12 @@
"@rabbit-company/argon2id": "jsr:@rabbit-company/argon2id@^2.1.0", "@rabbit-company/argon2id": "jsr:@rabbit-company/argon2id@^2.1.0",
"hono": "jsr:@hono/hono@^4.7.5", "hono": "jsr:@hono/hono@^4.7.5",
"zod": "npm:zod", "zod": "npm:zod",
"yup": "npm:yup" "yup": "npm:yup",
"@crawler/net/videoInfo": "../crawler/net/getVideoInfo.ts"
}, },
"tasks": { "tasks": {
"dev": "deno serve --env-file=.env --allow-env --allow-net --watch main.ts", "dev": "deno serve --env-file=.env --allow-env --allow-net --allow-read --allow-write --allow-run --watch main.ts",
"start": "deno serve --env-file=.env --allow-env --allow-net --host 127.0.0.1 main.ts" "start": "deno serve --env-file=.env --allow-env --allow-net --allow-read --allow-write --allow-run --host 127.0.0.1 main.ts"
}, },
"compilerOptions": { "compilerOptions": {
"jsx": "precompile", "jsx": "precompile",

View File

@ -3,16 +3,19 @@ import { dbCredMiddleware, dbMiddleware } from "./database.ts";
import { rootHandler } from "./root.ts"; import { rootHandler } from "./root.ts";
import { getSnapshotsHanlder } from "./snapshots.ts"; import { getSnapshotsHanlder } from "./snapshots.ts";
import { registerHandler } from "./register.ts"; import { registerHandler } from "./register.ts";
import { videoInfoHandler } from "./videoInfo.ts";
export const app = new Hono(); export const app = new Hono();
app.use('/video/*', dbMiddleware); app.use("/video/*", dbMiddleware);
app.use('/user', dbCredMiddleware); app.use("/user", dbCredMiddleware);
app.get("/", ...rootHandler); app.get("/", ...rootHandler);
app.get('/video/:id/snapshots', ...getSnapshotsHanlder); app.get("/video/:id/snapshots", ...getSnapshotsHanlder);
app.post('/user', ...registerHandler); app.post("/user", ...registerHandler);
app.get("/video/:id/info", ...videoInfoHandler);
const fetch = app.fetch; const fetch = app.fetch;
@ -20,4 +23,4 @@ export default {
fetch, fetch,
} satisfies Deno.ServeDefaultExport; } satisfies Deno.ServeDefaultExport;
export const VERSION = "0.3.0"; export const VERSION = "0.4.0";

View File

@ -8,7 +8,7 @@ import type { Client } from "https://deno.land/x/postgres@v0.19.3/mod.ts";
const RegistrationBodySchema = object({ const RegistrationBodySchema = object({
username: string().trim().required("Username is required").max(50, "Username cannot exceed 50 characters"), username: string().trim().required("Username is required").max(50, "Username cannot exceed 50 characters"),
password: string().required("Password is required"), password: string().required("Password is required"),
nickname: string().optional(), nickname: string().optional(),
}); });
type ContextType = Context<BlankEnv & { Bindings: Bindings }, "/user", BlankInput>; type ContextType = Context<BlankEnv & { Bindings: Bindings }, "/user", BlankInput>;
@ -19,7 +19,7 @@ export const userExists = async (username: string, client: Client) => {
`; `;
const result = await client.queryObject(query, [username]); const result = await client.queryObject(query, [username]);
return result.rows.length > 0; return result.rows.length > 0;
} };
export const registerHandler = createHandlers(async (c: ContextType) => { export const registerHandler = createHandlers(async (c: ContextType) => {
const client = c.get("dbCred"); const client = c.get("dbCred");
@ -28,11 +28,11 @@ export const registerHandler = createHandlers(async (c: ContextType) => {
const body = await RegistrationBodySchema.validate(await c.req.json()); const body = await RegistrationBodySchema.validate(await c.req.json());
const { username, password, nickname } = body; const { username, password, nickname } = body;
if (await userExists(username, client)) { if (await userExists(username, client)) {
return c.json({ return c.json({
message: `User "${username}" already exists.`, message: `User "${username}" already exists.`,
}, 400); }, 400);
} }
const hash = await Argon2id.hashEncoded(password); const hash = await Argon2id.hashEncoded(password);
@ -49,7 +49,7 @@ export const registerHandler = createHandlers(async (c: ContextType) => {
return c.json({ return c.json({
message: "Invalid registration data.", message: "Invalid registration data.",
errors: e.errors, errors: e.errors,
}, 400); }, 400);
} else if (e instanceof SyntaxError) { } else if (e instanceof SyntaxError) {
return c.json({ return c.json({
message: "Invalid JSON in request body.", message: "Invalid JSON in request body.",

View File

@ -5,27 +5,25 @@ import { createHandlers } from "./utils.ts";
export const rootHandler = createHandlers((c) => { export const rootHandler = createHandlers((c) => {
let singer: Singer | Singer[] | null = null; let singer: Singer | Singer[] | null = null;
const shouldShowSpecialSinger = Math.random() < 0.016; const shouldShowSpecialSinger = Math.random() < 0.016;
if (getSingerForBirthday().length !== 0){ if (getSingerForBirthday().length !== 0) {
singer = getSingerForBirthday(); singer = getSingerForBirthday();
for (const s of singer) { for (const s of singer) {
delete s.birthday; delete s.birthday;
s.message = `${s.name}生日快乐~` s.message = `${s.name}生日快乐~`;
} }
} } else if (shouldShowSpecialSinger) {
else if (shouldShowSpecialSinger) { singer = pickSpecialSinger();
singer = pickSpecialSinger(); } else {
} singer = pickSinger();
else {
singer = pickSinger();
} }
return c.json({ return c.json({
"project": { "project": {
"name": "中V档案馆", "name": "中V档案馆",
"motto": "一起唱吧,心中的歌!" "motto": "一起唱吧,心中的歌!",
}, },
"status": 200, "status": 200,
"version": VERSION, "version": VERSION,
"time": Date.now(), "time": Date.now(),
"singer": singer "singer": singer,
}) });
}) });

View File

@ -70,7 +70,7 @@ export interface Singer {
name: string; name: string;
color?: string; color?: string;
birthday?: string; birthday?: string;
message?: string; message?: string;
} }
export const specialSingers = [ export const specialSingers = [

View File

@ -12,12 +12,12 @@ const SnapshotQueryParamsSchema = object({
reverse: boolean().optional(), reverse: boolean().optional(),
}); });
const idSchema = mixed().test( export const idSchema = mixed().test(
"is-valid-id", "is-valid-id",
'id must be a string starting with "av" followed by digits, or "BV" followed by 10 alphanumeric characters, or a positive integer', 'id must be a string starting with "av" followed by digits, or "BV" followed by 10 alphanumeric characters, or a positive integer',
async (value) => { async (value) => {
if (value && await number().integer().isValid(value)) { if (value && await number().integer().isValid(value)) {
const v = parseInt(value as string); const v = parseInt(value as string);
return Number.isInteger(v) && v > 0; return Number.isInteger(v) && v > 0;
} }
@ -46,10 +46,9 @@ export const getSnapshotsHanlder = createHandlers(async (c: ContextType) => {
let videoId: string | number = idParam as string; let videoId: string | number = idParam as string;
if (videoId.startsWith("av")) { if (videoId.startsWith("av")) {
videoId = parseInt(videoId.slice(2)); videoId = parseInt(videoId.slice(2));
} } else if (await number().isValid(videoId)) {
else if (await number().isValid(videoId)) {
videoId = parseInt(videoId); videoId = parseInt(videoId);
} }
const queryParams = await SnapshotQueryParamsSchema.validate(c.req.query()); const queryParams = await SnapshotQueryParamsSchema.validate(c.req.query());
const { ps, pn, offset, reverse = false } = queryParams; const { ps, pn, offset, reverse = false } = queryParams;

View File

@ -1,5 +1,5 @@
import { createFactory } from 'hono/factory' import { createFactory } from "hono/factory";
const factory = createFactory(); const factory = createFactory();
export const createHandlers = factory.createHandlers; export const createHandlers = factory.createHandlers;

View File

@ -0,0 +1,39 @@
import type { Context } from "hono";
import { createHandlers } from "./utils.ts";
import type { BlankEnv, BlankInput } from "hono/types";
import { number, ValidationError } from "yup";
import { getVideoInfo, getVideoInfoByBV } from "@crawler/net/videoInfo";
import { idSchema } from "./snapshots.ts";
import type { VideoInfoData } from "../crawler/net/bilibili.d.ts";
type ContextType = Context<BlankEnv, "/video/:id/info", BlankInput>;
export const videoInfoHandler = createHandlers(async (c: ContextType) => {
try {
const id = await idSchema.validate(c.req.param("id"));
let videoId: string | number = id as string;
if (videoId.startsWith("av")) {
videoId = parseInt(videoId.slice(2));
} else if (await number().isValid(videoId)) {
videoId = parseInt(videoId);
}
let result: VideoInfoData | number;
if (typeof videoId === "number") {
result = await getVideoInfo(videoId, "getVideoInfo");
} else {
result = await getVideoInfoByBV(videoId, "getVideoInfo");
}
if (typeof result === "number") {
return c.json({ message: "Error fetching video info", code: result }, 500);
}
return c.json(result);
} catch (e) {
if (e instanceof ValidationError) {
return c.json({ message: "Invalid query parameters", errors: e.errors }, 400);
} else {
return c.json({ message: "Unhandled error", error: e }, 500);
}
}
});

View File

@ -25,3 +25,27 @@ export async function getVideoInfo(aid: number, task: string): Promise<VideoInfo
} }
return data.data; return data.data;
} }
/*
* Fetch video metadata from bilibili API by BVID
* @param {string} bvid - The video's BVID
* @param {string} task - The task name used in scheduler. It can be one of the following:
* - snapshotVideo
* - getVideoInfo
* - snapshotMilestoneVideo
* @returns {Promise<VideoInfoData | number>} VideoInfoData or the error code returned by bilibili API
* @throws {NetSchedulerError} - The error will be thrown in following cases:
* - No proxy is available currently: with error code `NO_PROXY_AVAILABLE`
* - The native `fetch` function threw an error: with error code `FETCH_ERROR`
* - The alicloud-fc threw an error: with error code `ALICLOUD_FC_ERROR`
*/
export async function getVideoInfoByBV(bvid: string, task: string): Promise<VideoInfoData | number> {
const url = `https://api.bilibili.com/x/web-interface/view?bvid=${bvid}`;
const data = await netScheduler.request<VideoInfoResponse>(url, task);
const errMessage = `Error fetching metadata for ${bvid}:`;
if (data.code !== 0) {
logger.error(errMessage + data.code + "-" + data.message, "net", "fn:getVideoInfoByBV");
return data.code;
}
return data.data;
}