47 lines
2.3 KiB
TypeScript

import { NextRequest, NextResponse } from 'next/server'
import { getServerSession } from 'next-auth'
import { authOptions } from '@/lib/auth'
import { isSystemAdmin } from '@/lib/department-permission'
import { prisma } from '@/lib/prisma'
import { ensureMaterialPurgeWorkerStarted, MaterialPurgeError, retryMaterialPurge, toPurgeJobView } from '@/lib/material-purge'
export const dynamic = 'force-dynamic'
async function operator() {
const session = await getServerSession(authOptions)
if (!session?.user || !(session.user as any).id) return null
return isSystemAdmin((session.user as any).role) ? (session.user as any).id as string : null
}
export async function GET(request: NextRequest) {
if (!await operator()) return NextResponse.json({ error: '无权限' }, { status: 403 })
const ids = request.nextUrl.searchParams.getAll('id')
if (ids.length > 100 || ids.some(id => !/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(id))) {
return NextResponse.json({ error: '清理任务无效' }, { status: 400 })
}
try {
const [pending, tracked] = await Promise.all([
prisma.materialPurgeJob.findMany({ where: { status: { not: 'completed' } }, orderBy: { createdAt: 'asc' }, take: 100 }),
ids.length ? prisma.materialPurgeJob.findMany({ where: { id: { in: ids } } }) : Promise.resolve([]),
])
const jobs = [...tracked, ...pending.filter(job => !tracked.some(item => item.id === job.id))]
return NextResponse.json({ jobs: jobs.map(toPurgeJobView) }, { headers: { 'Cache-Control': 'no-store' } })
} catch {
return NextResponse.json({ error: '暂时无法读取清理进度,请重试' }, { status: 503 })
}
}
export async function POST(request: NextRequest) {
const operatorId = await operator()
if (!operatorId) return NextResponse.json({ error: '无权限' }, { status: 403 })
try {
const { jobId } = await request.json()
const job = await retryMaterialPurge(jobId, operatorId)
ensureMaterialPurgeWorkerStarted()
return NextResponse.json({ job }, { status: 202 })
} catch (error) {
if (error instanceof MaterialPurgeError) return NextResponse.json({ error: error.message }, { status: error.status })
return NextResponse.json({ error: '暂时无法重试,请稍后再试' }, { status: 503 })
}
}