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 }));
  },
);