76 lines
2.6 KiB
TypeScript
76 lines
2.6 KiB
TypeScript
import process from "node:process"
|
|
import { TTL } from "@shared/consts"
|
|
import type { SourceID, SourceResponse } from "@shared/types"
|
|
import { sources } from "@shared/sources"
|
|
import { getters } from "#/getters"
|
|
import { useCache } from "#/hooks/useCache"
|
|
|
|
export default defineEventHandler(async (event): Promise<SourceResponse> => {
|
|
try {
|
|
let id = getRouterParam(event, "id") as SourceID
|
|
const query = getQuery(event)
|
|
const latest = query.latest !== undefined && query.latest !== "false"
|
|
const isValid = (id: SourceID) => !id || !sources[id] || !getters[id]
|
|
|
|
if (isValid(id)) {
|
|
const redirectID = sources?.[id]?.redirect
|
|
if (redirectID) id = redirectID
|
|
if (isValid(id)) throw new Error("Invalid source id")
|
|
}
|
|
|
|
const cacheTable = useCache()
|
|
const now = Date.now()
|
|
if (cacheTable) {
|
|
if (process.env.INIT_TABLE !== "false") await cacheTable.init()
|
|
const cache = await cacheTable.get(id)
|
|
if (cache) {
|
|
// interval 刷新间隔，对于缓存失效也要执行的。本质上表示本来内容更新就很慢，这个间隔内可能内容压根不会更新。
|
|
// 默认 10 分钟，是低于 TTL 的，但部分 Source 的更新间隔会超过 TTL，甚至有的一天更新一次。
|
|
const interval = sources[id].interval
|
|
if (now - cache.updated < interval) {
|
|
return {
|
|
status: "success",
|
|
updatedTime: now,
|
|
items: cache.data,
|
|
}
|
|
}
|
|
|
|
// 而 TTL 缓存失效时间，在时间范围内，就算内容更新了也要用这个缓存。
|
|
// 复用缓存是不会更新时间的。
|
|
if (now - cache.updated < TTL) {
|
|
// 有 latest
|
|
// 没有 latest，但服务器禁止登录
|
|
|
|
// 没有 latest
|
|
// 有 latest，服务器可以登录但没有登录
|
|
if (!latest || (!event.context.disabledLogin && !event.context.user)) {
|
|
return {
|
|
status: "cache",
|
|
updatedTime: cache.updated,
|
|
items: cache.data,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
const data = (await getters[id]()).slice(0, 30)
|
|
logger.success(`fetch ${id} latest`)
|
|
if (cacheTable) {
|
|
if (event.context.waitUntil) event.context.waitUntil(cacheTable.set(id, data))
|
|
else await cacheTable.set(id, data)
|
|
}
|
|
return {
|
|
status: "success",
|
|
updatedTime: now,
|
|
items: data,
|
|
}
|
|
} catch (e: any) {
|
|
logger.error(e)
|
|
throw createError({
|
|
statusCode: 500,
|
|
message: e instanceof Error ? e.message : "Internal Server Error",
|
|
})
|
|
}
|
|
})
|