155 lines
4.4 KiB
TypeScript
155 lines
4.4 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server'
|
|
import { getServerSession } from 'next-auth'
|
|
import { authOptions } from '@/lib/auth'
|
|
import { prisma } from '@/lib/prisma'
|
|
import path from 'path'
|
|
import JSZip from 'jszip'
|
|
import { ensureMaterialDownloadLogsTable } from '@/lib/material-download-log'
|
|
|
|
// SKU批量下载API
|
|
export async function POST(
|
|
request: NextRequest,
|
|
{ params }: { params: { id: string } }
|
|
) {
|
|
try {
|
|
const session = await getServerSession(authOptions)
|
|
if (!session?.user || !(session.user as any).id) {
|
|
return NextResponse.json({ error: '请先登录' }, { status: 401 })
|
|
}
|
|
|
|
const userId = (session.user as any).id
|
|
|
|
const user = await prisma.user.findUnique({
|
|
where: { id: userId },
|
|
select: { canDownload: true },
|
|
})
|
|
|
|
if (!user?.canDownload) {
|
|
return NextResponse.json({ error: '您没有下载权限，请联系管理员' }, { status: 403 })
|
|
}
|
|
|
|
const materialId = params.id
|
|
const body = await request.json()
|
|
const { skuIds } = body
|
|
|
|
if (!skuIds || !Array.isArray(skuIds) || skuIds.length === 0) {
|
|
return NextResponse.json({ error: '请选择要下载的SKU' }, { status: 400 })
|
|
}
|
|
|
|
// 检查素材是否存在（排除已删除的素材）
|
|
const material = await prisma.material.findFirst({
|
|
where: { id: materialId, deletedAt: null },
|
|
select: { id: true, productModel: true, materialType: true }
|
|
})
|
|
|
|
if (!material) {
|
|
return NextResponse.json({ error: '素材不存在或已被删除' }, { status: 404 })
|
|
}
|
|
|
|
if (material.materialType !== '标准模型') {
|
|
return NextResponse.json({ error: '只有标准模型类型才支持SKU批量下载' }, { status: 400 })
|
|
}
|
|
|
|
// 获取选中的SKU列表
|
|
const skus = await prisma.materialSku.findMany({
|
|
where: {
|
|
id: { in: skuIds },
|
|
materialId,
|
|
deletedAt: null // 只获取未删除的SKU
|
|
},
|
|
select: {
|
|
id: true,
|
|
skuName: true,
|
|
filePath: true,
|
|
fileName: true
|
|
}
|
|
})
|
|
|
|
if (skus.length === 0) {
|
|
return NextResponse.json({ error: '没有找到可下载的SKU' }, { status: 404 })
|
|
}
|
|
|
|
// 创建ZIP文件
|
|
const zip = new JSZip()
|
|
let addedCount = 0
|
|
|
|
const fs = await import('fs/promises')
|
|
|
|
for (const sku of skus) {
|
|
try {
|
|
const filePath = path.join(process.cwd(), 'uploads', sku.filePath)
|
|
|
|
// 检查文件是否存在
|
|
try {
|
|
await fs.access(filePath)
|
|
} catch {
|
|
console.warn(`SKU文件不存在: ${filePath}`)
|
|
continue
|
|
}
|
|
|
|
// 读取文件并添加到ZIP
|
|
const fileBuffer = await fs.readFile(filePath)
|
|
const fileName = sku.fileName || `${sku.skuName}.zip`
|
|
zip.file(fileName, fileBuffer)
|
|
addedCount++
|
|
} catch (error) {
|
|
console.error(`添加SKU文件失败: ${sku.id}`, error)
|
|
// 继续处理其他文件
|
|
}
|
|
}
|
|
|
|
if (addedCount === 0) {
|
|
return NextResponse.json({ error: '没有可下载的文件' }, { status: 404 })
|
|
}
|
|
|
|
// 生成ZIP文件
|
|
const zipBuffer = await zip.generateAsync({
|
|
type: 'nodebuffer',
|
|
compression: 'DEFLATE',
|
|
compressionOptions: { level: 6 }
|
|
})
|
|
|
|
const filename = `${material.productModel}_SKU批量下载_${Date.now()}.zip`
|
|
|
|
recordBatchDownloadLog(materialId, userId).catch((error) => {
|
|
console.error('[SKU Batch Download] 记录下载日志失败:', error)
|
|
})
|
|
|
|
return new NextResponse(zipBuffer as any, {
|
|
headers: {
|
|
'Content-Type': 'application/zip',
|
|
'Content-Length': zipBuffer.length.toString(),
|
|
'Content-Disposition': `attachment; filename="${encodeURIComponent(filename)}"`,
|
|
},
|
|
})
|
|
} catch (error) {
|
|
console.error('SKU批量下载失败:', error)
|
|
return NextResponse.json(
|
|
{ error: '下载失败: ' + (error instanceof Error ? error.message : '未知错误') },
|
|
{ status: 500 }
|
|
)
|
|
}
|
|
}
|
|
|
|
async function recordBatchDownloadLog(materialId: string, userId: string) {
|
|
await ensureMaterialDownloadLogsTable()
|
|
await Promise.all([
|
|
prisma.materialDownloadLog.create({
|
|
data: {
|
|
materialId,
|
|
userId,
|
|
downloadedAt: new Date(),
|
|
},
|
|
}),
|
|
prisma.material.update({
|
|
where: { id: materialId },
|
|
data: {
|
|
downloadCount: {
|
|
increment: 1,
|
|
},
|
|
},
|
|
}),
|
|
])
|
|
}
|
|
|