IO-1998 LMS Zip Download.

jobs/jobsDownloadMedia.ts

@@ -0,0 +1,77 @@
+import { Request, Response } from "express";
+import fs from "fs-extra";
+import multer from "multer";
+import path from "path";
+import { logger } from "../server";
+import GenerateThumbnail from "../util/generateThumbnail";
+import generateUniqueFilename from "../util/generateUniqueFilename";
+import { ConvertHeicFiles } from "../util/heicConverter";
+import { PathToRoFolder } from "../util/pathGenerators";
+import { JobsListMedia } from "./jobsListMedia";
+import JSZip from "jszip";
+import ListableChecker from "../util/listableChecker";
+import { JobRelativeFilePath } from "../util/serverInit";
+
+//param: files: string[] | array of filenames.
+export async function jobsDownloadMedia(req: Request, res: Response) {
+  const jobid: string = (req.body.jobid || "").trim();
+
+  try {
+    //Do we need all files or just some files?
+    const files: string[] = req.body.files || [];
+    const zip: JSZip = new JSZip();
+    await fs.ensureDir(PathToRoFolder(jobid));
+
+    logger.debug(`Generating batch download for Job ID ${jobid}`, files);
+    //Prepare the zip file.
+
+    const filesList: fs.Dirent[] = (
+      await fs.readdir(PathToRoFolder(jobid), {
+        withFileTypes: true,
+      })
+    ).filter((f) => f.isFile() && ListableChecker(f));
+
+    if (files.length === 0) {
+      //Get everything.
+
+      await Promise.all(
+        filesList.map(async (file) => {
+          //Do something async
+          const fileOnDisk: Buffer = await fs.readFile(
+            JobRelativeFilePath(jobid, file.name)
+          );
+          zip.file(path.parse(path.basename(file.name)).base, fileOnDisk);
+        })
+      );
+    } else {
+      //Get the files that are in the list and see which are requested.
+      await Promise.all(
+        filesList.map(async (file) => {
+          if (files.includes(path.parse(path.basename(file.name)).base)) {
+            // File is in the set of requested files.
+            const fileOnDisk: Buffer = await fs.readFile(
+              JobRelativeFilePath(jobid, file.name)
+            );
+            zip.file(path.parse(path.basename(file.name)).base, fileOnDisk);
+          }
+        })
+      );
+    }
+    //Send it as a response to download it automatically.
+    // res.setHeader("Content-disposition", "attachment; filename=" + filename);
+
+    zip
+      .generateNodeStream({
+        type: "nodebuffer",
+        streamFiles: true,
+        //encodeFileName: (filename) => `${jobid}.zip`,
+      })
+      .pipe(res);
+  } catch (error) {
+    logger.error("Error downloading job media.", {
+      jobid,
+      error: (error as Error).message,
+    });
+    res.status(500).json((error as Error).message);
+  }
+}

jobs/jobsListMedia.ts

@@ -7,7 +7,7 @@ import MediaFile from "../util/interfaces/MediaFile";
 import ListableChecker from "../util/listableChecker";
 import GenerateUrl from "../util/MediaUrlGen";
 import { PathToRoFolder } from "../util/pathGenerators";
-import { FolderPaths } from "../util/serverInit";
+import { FolderPaths, JobRelativeFilePath } from "../util/serverInit";
 
 export async function JobsListMedia(req: Request, res: Response) {
   const jobid: string = (req.body.jobid || "").trim();
@@ -19,8 +19,13 @@ export async function JobsListMedia(req: Request, res: Response) {
       //We just uploaded files, we're going to send only those back.
       ret = await Promise.all(
         (req.files as Express.Multer.File[]).map(async (file) => {
+          const relativeFilePath: string = JobRelativeFilePath(
+            jobid,
+            file.filename
+          );
+
           const relativeThumbPath: string = await GenerateThumbnail(
-            path.join(FolderPaths.Jobs, jobid, file.filename)
+            relativeFilePath
           );
           return {
             src: GenerateUrl([
@@ -38,6 +43,7 @@ export async function JobsListMedia(req: Request, res: Response) {
             thumbnailHeight: 250,
             thumbnailWidth: 250,
             filename: file.filename,
+            relativeFilePath,
           };
         })
       );
@@ -46,15 +52,17 @@ export async function JobsListMedia(req: Request, res: Response) {
         await fs.readdir(PathToRoFolder(jobid), {
           withFileTypes: true,
         })
-      ).filter(
-        (f) =>
-          f.isFile() && !/(^|\/)\.[^\/\.]/g.test(f.name) && ListableChecker(f)
-      );
+      ).filter((f) => f.isFile() && ListableChecker(f));
 
       ret = await Promise.all(
         filesList.map(async (file) => {
+          const relativeFilePath: string = JobRelativeFilePath(
+            jobid,
+            file.name
+          );
+
           const relativeThumbPath: string = await GenerateThumbnail(
-            path.join(FolderPaths.Jobs, jobid, file.name)
+            relativeFilePath
           );
           return {
             src: GenerateUrl([
@@ -72,6 +80,7 @@ export async function JobsListMedia(req: Request, res: Response) {
             thumbnailHeight: 250,
             thumbnailWidth: 250,
             filename: file.name,
+            relativeFilePath,
           };
         })
       );