newsnow/server/api/s/[id].ts

import process from "node:process"
import { TTL } from "@shared/consts"
import type { SourceID, SourceResponse } from "@shared/types"
import { sources } from "@shared/sources"
import { sourcesGetters } from "#/sources"
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] || !sourcesGetters[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 sourcesGetters[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",
    })
  }
})