fix: scope memory health diagnostics to allowed prefixes (#367)
When memoryAllowedPrefixes is configured, scan only those subdirectories instead of the entire MEMORY_PATH. Falls back to full scan if no prefixes are set.\n\nCloses #366
This commit is contained in:
parent
c537f31088
commit
8d671cb782
|
|
@ -1,4 +1,6 @@
|
|||
import { NextRequest, NextResponse } from 'next/server'
|
||||
import { existsSync } from 'fs'
|
||||
import { join } from 'path'
|
||||
import { config } from '@/lib/config'
|
||||
import { requireRole } from '@/lib/auth'
|
||||
import { readLimiter } from '@/lib/rate-limit'
|
||||
|
|
@ -6,6 +8,35 @@ import { runHealthDiagnostics } from '@/lib/memory-utils'
|
|||
import { logger } from '@/lib/logger'
|
||||
|
||||
const MEMORY_PATH = config.memoryDir
|
||||
const MEMORY_ALLOWED_PREFIXES = (config.memoryAllowedPrefixes || []).map((p) => p.replace(/\\/g, '/'))
|
||||
|
||||
function mergeReports(reports: Awaited<ReturnType<typeof runHealthDiagnostics>>[]) {
|
||||
const allCategories = reports.flatMap((report) => report.categories)
|
||||
const mergedCategories = Array.from(new Set(allCategories.map((category) => category.name))).map((name) => {
|
||||
const group = allCategories.filter((category) => category.name === name)
|
||||
const score = Math.round(group.reduce((sum, category) => sum + category.score, 0) / group.length)
|
||||
const status = score >= 80 ? 'healthy' : score >= 50 ? 'warning' : 'critical'
|
||||
return {
|
||||
name,
|
||||
status,
|
||||
score,
|
||||
issues: group.flatMap((category) => category.issues).slice(0, 10),
|
||||
suggestions: Array.from(new Set(group.flatMap((category) => category.suggestions))),
|
||||
}
|
||||
})
|
||||
|
||||
const overallScore = mergedCategories.length > 0
|
||||
? Math.round(mergedCategories.reduce((sum, category) => sum + category.score, 0) / mergedCategories.length)
|
||||
: 100
|
||||
const overall = overallScore >= 70 ? 'healthy' : overallScore >= 40 ? 'warning' : 'critical'
|
||||
|
||||
return {
|
||||
overall,
|
||||
overallScore,
|
||||
categories: mergedCategories,
|
||||
generatedAt: Date.now(),
|
||||
}
|
||||
}
|
||||
|
||||
export async function GET(request: NextRequest) {
|
||||
const auth = requireRole(request, 'viewer')
|
||||
|
|
@ -19,6 +50,17 @@ export async function GET(request: NextRequest) {
|
|||
}
|
||||
|
||||
try {
|
||||
if (MEMORY_ALLOWED_PREFIXES.length) {
|
||||
const reports = []
|
||||
for (const prefix of MEMORY_ALLOWED_PREFIXES) {
|
||||
const folder = prefix.replace(/\/$/, '')
|
||||
const fullPath = join(MEMORY_PATH, folder)
|
||||
if (!existsSync(fullPath)) continue
|
||||
reports.push(await runHealthDiagnostics(fullPath))
|
||||
}
|
||||
return NextResponse.json(reports.length > 0 ? mergeReports(reports) : await runHealthDiagnostics(MEMORY_PATH))
|
||||
}
|
||||
|
||||
const report = await runHealthDiagnostics(MEMORY_PATH)
|
||||
return NextResponse.json(report)
|
||||
} catch (err) {
|
||||
|
|
|
|||
Loading…
Reference in New Issue