103 lines
3.9 KiB
TypeScript
103 lines
3.9 KiB
TypeScript
import { zipSync, strToU8 } from "fflate";
|
||
import { adminForbidden, isAdminRequest } from "../../../lib/admin-auth";
|
||
import { ensureSchema, getRawDb, getUploadBucket } from "../../../lib/mvp-db";
|
||
import { parseResultScreenshotKeys } from "../../../lib/result-screenshots";
|
||
|
||
function safeName(value: string) {
|
||
return value.replace(/[\\/:*?"<>|\r\n]/g, "_").trim().slice(0, 60) || "KOC";
|
||
}
|
||
|
||
function csvCell(value: unknown) {
|
||
const text = String(value ?? "");
|
||
return `"${text.replaceAll('"', '""')}"`;
|
||
}
|
||
|
||
export async function GET(request: Request) {
|
||
if (!(await isAdminRequest(request))) return adminForbidden();
|
||
try {
|
||
await ensureSchema();
|
||
const taskId = new URL(request.url).searchParams.get("task")?.trim();
|
||
if (!taskId) {
|
||
return Response.json({ error: "缺少任务参数" }, { status: 400 });
|
||
}
|
||
const task = await getRawDb()
|
||
.prepare("SELECT name, task_type FROM tasks WHERE id = ?")
|
||
.bind(taskId)
|
||
.first<{ name: string; task_type: string }>();
|
||
if (!task || task.task_type !== "screenshot_collect") {
|
||
return Response.json({ error: "没有找到截图回收任务" }, { status: 404 });
|
||
}
|
||
const rows = await getRawDb()
|
||
.prepare(
|
||
`SELECT d.id, d.result_screenshot_key, d.result_submitted_at,
|
||
d.claimed_at, c.source_row, c.title, p.name AS partner_name,
|
||
cl.claimant_name
|
||
FROM distributions d
|
||
JOIN contents c ON c.id = d.content_id
|
||
JOIN partners p ON p.id = d.partner_id
|
||
LEFT JOIN claims cl ON cl.id = d.claim_id
|
||
WHERE d.task_id = ?
|
||
ORDER BY COALESCE(c.source_row, 999999), d.claimed_at, d.id`,
|
||
)
|
||
.bind(taskId)
|
||
.all<{
|
||
id: string;
|
||
result_screenshot_key: string | null;
|
||
result_submitted_at: string | null;
|
||
claimed_at: string;
|
||
source_row: number | null;
|
||
title: string;
|
||
partner_name: string;
|
||
claimant_name: string | null;
|
||
}>();
|
||
const files: Record<string, Uint8Array> = {};
|
||
const manifest = [
|
||
["序号", "搜索关键词", "领取人", "领取时间", "提交时间", "文件名"],
|
||
];
|
||
let exported = 0;
|
||
for (const [rowIndex, row] of rows.results.entries()) {
|
||
const fileNames: string[] = [];
|
||
const screenshotKeys = row.result_submitted_at
|
||
? parseResultScreenshotKeys(row.result_screenshot_key)
|
||
: [];
|
||
for (const [imageIndex, screenshotKey] of screenshotKeys.entries()) {
|
||
const object = await getUploadBucket().get(screenshotKey);
|
||
if (!object) continue;
|
||
const extension = screenshotKey.split(".").at(-1) || "jpg";
|
||
const fileName = `${String(row.source_row ?? rowIndex + 1).padStart(3, "0")}-${safeName(row.claimant_name || row.partner_name)}-${imageIndex + 1}.${extension}`;
|
||
files[fileName] = new Uint8Array(await object.arrayBuffer());
|
||
fileNames.push(fileName);
|
||
exported += 1;
|
||
}
|
||
manifest.push([
|
||
String(row.source_row ?? ""),
|
||
row.title,
|
||
row.claimant_name || row.partner_name,
|
||
row.claimed_at,
|
||
row.result_submitted_at || "",
|
||
fileNames.join("；"),
|
||
]);
|
||
}
|
||
files["回收清单.csv"] = strToU8(
|
||
`\uFEFF${manifest.map((cells) => cells.map(csvCell).join(",")).join("\r\n")}`,
|
||
);
|
||
const archive = zipSync(files, { level: 0 });
|
||
const headers = new Headers({
|
||
"Content-Type": "application/zip",
|
||
"Content-Disposition": `attachment; filename*=UTF-8''${encodeURIComponent(`${safeName(task.name)}-截图回收.zip`)}`,
|
||
"Cache-Control": "private, no-store",
|
||
"X-KOC-Exported-Count": String(exported),
|
||
});
|
||
const body = archive.buffer.slice(
|
||
archive.byteOffset,
|
||
archive.byteOffset + archive.byteLength,
|
||
) as ArrayBuffer;
|
||
return new Response(body, { headers });
|
||
} catch (error) {
|
||
return Response.json(
|
||
{ error: error instanceof Error ? error.message : "截图打包失败" },
|
||
{ status: 500 },
|
||
);
|
||
}
|
||
}
|