feat: download page

2anki · Nov 12, 2023 · b8a626d · b8a626d
1 parent 43fe543
commit b8a626d
Show file tree

Hide file tree

Showing 4 changed files with 82 additions and 4 deletions.
diff --git a/src/controllers/DownloadController.ts b/src/controllers/DownloadController.ts
@@ -1,8 +1,12 @@
-import { Request, Response } from 'express';
+import fs from 'fs';
+
+import { static as serve, NextFunction, Request, Response } from 'express';
 
 import { sendError } from '../lib/error/sendError';
 import DownloadService from '../services/DownloadService';
 import StorageHandler from '../lib/storage/StorageHandler';
+import path from 'path';
+import { DownloadPage } from '../pages/DownloadPage';
 
 class DownloadController {
   constructor(private service: DownloadService) {}
@@ -34,6 +38,47 @@ class DownloadController {
       }
     }
   }
+
+  getDownloadPage(req: Request, res: Response) {
+    const { id } = req.params;
+    const workspace = path.join(process.env.WORKSPACE_BASE!, id);
+
+    if (!fs.existsSync(workspace)) {
+      return res.status(404).end();
+    }
+
+    if (fs.statSync(workspace).isDirectory()) {
+      fs.readdir(workspace, (err, files) => {
+        if (err) {
+          console.error(err);
+          res.status(500).send('Error reading directory');
+          return;
+        }
+
+        const page = DownloadPage({ id, files });
+        res.send(page);
+      });
+    } else {
+      const fileContent = fs.readFileSync(workspace, 'utf8');
+      return res.send(fileContent);
+    }
+  }
+
+  getAPKGFile(req: Request, res: Response) {
+    const { id, apkg } = req.params;
+    const workspace = path.join(process.env.WORKSPACE_BASE!, id);
+    const apkgFilePath = path.join(workspace, apkg);
+
+    if (!fs.existsSync(apkgFilePath)) {
+      return res.status(404).end();
+    }
+
+    const fileContent = fs.readFileSync(apkgFilePath, 'utf8');
+    const contentLength = Buffer.byteLength(fileContent);
+    res.set('Content-Type', 'application/apkg');
+    res.set('Content-Length', contentLength.toString());
+    return res.send(fileContent);
+  }
 }
 
 export default DownloadController;
diff --git a/src/controllers/UploadController.ts b/src/controllers/UploadController.ts
@@ -73,7 +73,6 @@ class UploadController {
   file(req: express.Request, res: express.Response) {
     try {
       console.info('uploading file');
-      console.time(req.path);
       const storage = new StorageHandler();
       const handleUploadEndpoint = this.service.getUploadHandler(res);
 
@@ -85,11 +84,9 @@ class UploadController {
           } else {
             sendError(error);
           }
-          console.timeEnd(req.path);
           return res.status(500).send(msg);
         }
         await this.service.handleUpload(storage, req, res);
-        console.timeEnd(req.path);
       });
     } catch (error) {
       sendError(error);

diff --git a/src/pages/DownloadPage.tsx b/src/pages/DownloadPage.tsx
@@ -0,0 +1,28 @@
+import path from 'path';
+
+import ReactDOMServer from 'react-dom/server';
+
+interface DownloadPageProps {
+  id: string;
+  files: string[];
+}
+
+export const DownloadPage = ({ id, files }: DownloadPageProps) => {
+  return ReactDOMServer.renderToStaticMarkup(
+    <html>
+      <header>
+        <title>Your download</title>
+      </header>
+      <body>
+        <h1>Your download is ready</h1>
+        {files.map((file) => (
+          <li key={file}>
+            <a download={`${path.basename(file)}`} href={`${id}/${file}`}>
+              {file}
+            </a>
+          </li>
+        ))}
+      </body>
+    </html>
+  );
+};
diff --git a/src/routes/DownloadRouter.ts b/src/routes/DownloadRouter.ts
@@ -18,6 +18,14 @@ const DownloadRouter = () => {
     controller.getFile(req, res, storage);
   });
 
+  router.get('/download/:id', (req, res) => {
+    controller.getDownloadPage(req, res);
+  });
+
+  router.get('/download/:id/:apkg', (req, res) => {
+    controller.getAPKGFile(req, res);
+  });
+
   return router;
 };