mirror of
https://github.com/shadcn-ui/ui.git
synced 2026-09-14 20:00:35 +08:00
352f2094bd
* feat(registry): add health monitoring * fix(registry): cache health directory route * fix(registry): harden health state transitions
64 lines
1.7 KiB
TypeScript
64 lines
1.7 KiB
TypeScript
import { NextResponse } from "next/server"
|
|
|
|
import {
|
|
createPublicRegistryDirectory,
|
|
registryDirectorySchema,
|
|
} from "@/lib/registry-directory"
|
|
import { loadRegistryHealthSnapshot } from "@/lib/registry-health/blob"
|
|
import directory from "@/registry/directory.json"
|
|
|
|
export const dynamic = "force-static"
|
|
export const revalidate = 300
|
|
|
|
const SNAPSHOT_STALE_AFTER_MS = 6 * 60 * 60 * 1000
|
|
const SNAPSHOT_TIMEOUT_MS = 2500
|
|
const registries = createPublicRegistryDirectory(
|
|
registryDirectorySchema.parse(directory)
|
|
)
|
|
|
|
async function getHealthSnapshot() {
|
|
const controller = new AbortController()
|
|
const timeout = setTimeout(() => controller.abort(), SNAPSHOT_TIMEOUT_MS)
|
|
|
|
try {
|
|
const snapshot = await loadRegistryHealthSnapshot({
|
|
token: process.env.BLOB_READ_WRITE_TOKEN,
|
|
abortSignal: controller.signal,
|
|
})
|
|
if (!snapshot) throw new Error("missing_snapshot")
|
|
|
|
const age = Date.now() - new Date(snapshot.generatedAt).getTime()
|
|
if (age < 0 || age > SNAPSHOT_STALE_AFTER_MS) {
|
|
throw new Error("stale_snapshot")
|
|
}
|
|
|
|
return snapshot
|
|
} catch (error) {
|
|
console.warn(
|
|
"Registry health snapshot unavailable:",
|
|
error instanceof Error ? error.message : "unknown_error"
|
|
)
|
|
return null
|
|
} finally {
|
|
clearTimeout(timeout)
|
|
}
|
|
}
|
|
|
|
export async function GET() {
|
|
if (process.env.REGISTRY_HEALTH_ENABLED !== "1") {
|
|
return NextResponse.json(registries)
|
|
}
|
|
|
|
const snapshot = await getHealthSnapshot()
|
|
if (!snapshot) {
|
|
return NextResponse.json(registries)
|
|
}
|
|
|
|
return NextResponse.json(
|
|
registries.map((registry) => {
|
|
const health = snapshot.registries[registry.name]
|
|
return health ? { ...registry, health } : registry
|
|
})
|
|
)
|
|
}
|