Feat/upload html file

package.json

@@ -15,12 +15,14 @@
     "dependencies": {
         "cheerio": "^1.0.0",
         "express": "^4.21.1",
+        "multer": "^1.4.5-lts.1",
         "nodemon": "^3.1.7"
     },
     "devDependencies": {
         "@types/cheerio": "^0.22.35",
         "@types/express": "^5.0.0",
         "@types/jest": "^29.5.13",
+        "@types/multer": "^1.4.12",
         "jest": "^29.7.0",
         "ts-jest": "^29.2.5",
         "ts-node": "^10.9.2"

src/controllers/processHtmlController.ts

@@ -1,6 +1,5 @@
 import { NextFunction, Request, Response } from "express";
 import { ProcessHtmlService } from "../services/processHtmlService";
-import path from "path";
 import { CssCode } from "../types";
 
 const service = new ProcessHtmlService();
@@ -10,10 +9,11 @@ export async function processHtmlInputController(
     res: Response,
     next: NextFunction,
 ) {
-    const cssCode = req.body as CssCode;
-    const htmlCode = path.resolve(__dirname, path.join("../data/index.html"));
+    console.log(JSON.parse(req.body.data));
+    const cssCode = JSON.parse(req.body.data) as CssCode;
     try {
-        const response = await service.extractData(htmlCode, cssCode);
+        const htmlContent = req.file!.buffer.toString();
+        const response = await service.extractData(htmlContent, cssCode);
         res.json(response);
     } catch (err) {
         next(err);

src/middlewares/validateFile.ts

@@ -0,0 +1,17 @@
+export const validateFile = (
+    req: Express.Request,
+    file: Express.Multer.File,
+    cb: Function,
+) => {
+    const filetypes = /html|htm/;
+    const mimetype = filetypes.test(file.mimetype);
+    const extname = filetypes.test(
+        file.originalname.split(".").pop()!.toLowerCase(),
+    );
+
+    if (mimetype && extname) {
+        return cb(null, true);
+    } else {
+        return cb(new Error("Only HTML files are allowed!"), false);
+    }
+};

src/routes/processHtmlRouter.ts

@@ -1,8 +1,17 @@
 import { Router } from "express";
 import { processHtmlInputController } from "../controllers/processHtmlController";
+import { validateFile } from "../middlewares/validateFile";
+
+import multer from "multer";
 
 const r = Router();
 
-r.post("/extract", processHtmlInputController);
+const storage = multer.memoryStorage();
+const upload = multer({
+    storage,
+    fileFilter: validateFile,
+});
+
+r.post("/extract", upload.single("htmlFile"), processHtmlInputController);
 
 export default r;

src/services/processHtmlService.test.ts

@@ -1,5 +1,6 @@
 import path from "path";
 import { ProcessHtmlService } from "../services/processHtmlService";
+import {readFile} from "fs/promises";
 
 describe("Process html api", () => {
     test("Extract html: Example 1", async () => {
@@ -13,7 +14,8 @@ describe("Process html api", () => {
             __dirname,
             path.join("../data/index.html"),
         );
-        const r = await service.extractData(htmlCode, cssCode);
+        const html = await readFile(htmlCode, "utf-8");
+        const r = await service.extractData(html, cssCode);
         expect(r).toBeTruthy();
         expect(r).toEqual({
             title: "This is the title of the page",
@@ -34,7 +36,8 @@ describe("Process html api", () => {
             __dirname,
             path.join("../data/index.html"),
         );
-        const r = await service.extractData(htmlCode, cssCode);
+        const html = await readFile(htmlCode, "utf-8");
+        const r = await service.extractData(html, cssCode);
         expect(r).toBeTruthy();
         expect(r).toEqual({
             firstParagraph:
@@ -65,7 +68,8 @@ describe("Process html api", () => {
             __dirname,
             path.join("../data/index.html"),
         );
-        const r = await service.extractData(htmlCode, cssCode);
+        const html = await readFile(htmlCode, "utf-8");
+        const r = await service.extractData(html, cssCode);
         expect(r).toBeTruthy();
         expect(r).toEqual({
             title: "This is the title of the page",