2026-03-30 16:58:14 +08:00

52 lines
1.7 KiB
TypeScript

import express from "express";
import u from "@/utils";
import { z } from "zod";
import { success } from "@/lib/responseFormat";
import { validateFields } from "@/middleware/middleware";
const router = express.Router();
import compressing from "compressing";
import path from "path";
import getPath from "@/utils/getPath";
export default router.post(
"/",
validateFields({
shotId: z.array(
z.object({
id: z.string(),
}),
),
}),
async (req, res) => {
const { shotId } = req.body;
const id = shotId.map((item: { id: string }) => item.id);
const data = await u.db("o_storyboard").whereIn("id", id);
const result = await Promise.all(
data.map(async (item) => {
const url = await u.oss.getFileUrl(item.filePath!);
return { ...item, url };
}),
);
const zipStream = new compressing.zip.Stream();
result.forEach((item) => {
const ext = (item.filePath?.split(".").pop() || "png").toLowerCase();
const absPath = path.join(getPath("oss"), item.filePath!);
zipStream.addEntry(absPath, { relativePath: `${item.title}.${ext}` });
});
const fileName = `分镜.zip`;
const zipFilePath = getPath(["oss", "temp", fileName]);
await new Promise<void>((resolve, reject) => {
const fs = require("fs");
fs.mkdirSync(getPath(["oss", "temp"]), { recursive: true });
const writeStream = fs.createWriteStream(zipFilePath);
zipStream.pipe(writeStream);
writeStream.on("finish", resolve);
writeStream.on("error", reject);
});
const downloadUrl = `${process.env.OSSURL || "http://127.0.0.1:10588/"}temp/${fileName}`;
res.json(success({ url: downloadUrl }));
},
);