85 lines
2.8 KiB
TypeScript
85 lines
2.8 KiB
TypeScript
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";
|
|
import { Redis } from "ioredis";
|
|
import { NetSchedulerError } from "../crawler/mq/scheduler.ts";
|
|
import logger from "../crawler/log/logger.ts";
|
|
import type { Client } from "https://deno.land/x/postgres@v0.19.3/mod.ts";
|
|
|
|
const redis = new Redis({ maxRetriesPerRequest: null });
|
|
const CACHE_EXPIRATION_SECONDS = 60;
|
|
|
|
type ContextType = Context<BlankEnv, "/video/:id/info", BlankInput>;
|
|
|
|
async function insertVideoSnapshot(client: Client, data: VideoInfoData) {
|
|
const views = data.stat.view;
|
|
const danmakus = data.stat.danmaku;
|
|
const replies = data.stat.reply;
|
|
const likes = data.stat.like;
|
|
const coins = data.stat.coin;
|
|
const shares = data.stat.share;
|
|
const favorites = data.stat.favorite;
|
|
const aid = data.aid;
|
|
|
|
const query: string = `
|
|
INSERT INTO video_snapshot (aid, views, danmakus, replies, likes, coins, shares, favorites)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
|
|
`;
|
|
|
|
await client.queryObject(
|
|
query,
|
|
[aid, views, danmakus, replies, likes, coins, shares, favorites],
|
|
);
|
|
|
|
logger.log(`Inserted into snapshot for video ${aid} by videoInfo API.`, "api", "fn:insertVideoSnapshot");
|
|
}
|
|
|
|
|
|
export const videoInfoHandler = createHandlers(async (c: ContextType) => {
|
|
const client = c.get("db");
|
|
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);
|
|
}
|
|
|
|
const cacheKey = `cvsa:videoInfo:${videoId}`;
|
|
const cachedData = await redis.get(cacheKey);
|
|
|
|
if (cachedData) {
|
|
return c.json(JSON.parse(cachedData));
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
await redis.setex(cacheKey, CACHE_EXPIRATION_SECONDS, JSON.stringify(result));
|
|
|
|
await insertVideoSnapshot(client, result);
|
|
|
|
return c.json(result);
|
|
} catch (e) {
|
|
if (e instanceof ValidationError) {
|
|
return c.json({ message: "Invalid query parameters", errors: e.errors }, 400);
|
|
} else if (e instanceof NetSchedulerError) {
|
|
return c.json({ message: "Error fetching video info", code: e.code }, 500);
|
|
} else {
|
|
return c.json({ message: "Unhandled error", error: e }, 500);
|
|
}
|
|
}
|
|
}); |