feat: socketio event

Services/server/src/app.ts

@@ -2,6 +2,8 @@ import "dotenv/config";
 import express from "express";
 import swaggerUi from "swagger-ui-express";
 import cors from "cors";
+import { createServer } from "http";
+import { Server } from "socket.io";
 
 import { RegisterRoutes } from "./routes";
 import errorHandler from "./middlewares/exception";
@@ -9,6 +11,7 @@ import rabbitmq from "./rabbitmq";
 
 import swaggerSpecs from "./swagger.json";
 import { handler as amqHandler } from "./rabbitmq/handler";
+import { setInstance } from "./lib/websocket";
 
 const PORT = +(process.env.PORT || 80);
 
@@ -31,7 +34,24 @@ app.use((_req, res, _next) => {
   res.sendFile(`${process.cwd()}/static/index.html`);
 });
 
-app.listen(PORT, "0.0.0.0", () =>
+const server = createServer(app);
+const io = new Server(server, {
+  cors: {
+    origin: "*",
+  },
+});
+
+setInstance(io);
+
+io.on("connection", (socket) => {
+  console.log("User Connected");
+
+  socket.on("disconnected", () => {
+    console.log("User Disconnected");
+  });
+});
+
+server.listen(PORT, "0.0.0.0", () =>
   console.log(`[APP] Application is running on http://localhost:${PORT}`),
 );
 

Services/server/src/controllers/cabinetController.ts

@@ -22,6 +22,7 @@ import { copyCond, listFolder, listItem, replaceIllegalChars } from "../utils/mi
 
 import HttpStatusCode from "../interfaces/http-status";
 import { StorageFile, StorageFolder } from "../interfaces/storage-fs";
+import { getInstance } from "../lib/websocket";
 
 const DEFAULT_BUCKET = process.env.MINIO_BUCKET;
 const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;
@@ -86,6 +87,9 @@ export class CabinetController extends Controller {
 
     if (!created) throw new Error("เกิดข้อผิดพลาดกับระบบจัดการไฟล์");
 
+    const io = getInstance();
+    io?.emit("CreateFolder", { pathname: `${replaceIllegalChars(body.name)}/` });
+
     return this.setStatus(HttpStatusCode.CREATED);
   }
 
@@ -155,6 +159,9 @@ export class CabinetController extends Controller {
       }),
     );
 
+    const io = getInstance();
+    io?.emit("EditFolder", { from: `${cabinetName}/`, to: `${replaceIllegalChars(body.name)}/` });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 
@@ -180,6 +187,9 @@ export class CabinetController extends Controller {
       stream.on("error", () => reject(new Error("เกิดข้อผิดพลาด ไม่สามารถลบไฟล์ได้")));
     });
 
+    const io = getInstance();
+    io?.emit("DeleteFolder", { pathname: `${cabinetName}/` });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 

Services/server/src/controllers/drawerController.ts

@@ -23,6 +23,7 @@ import { copyCond, listFolder, listItem, pathExist, replaceIllegalChars } from "
 import HttpStatusCode from "../interfaces/http-status";
 import { StorageFile, StorageFolder } from "../interfaces/storage-fs";
 import HttpError from "../interfaces/http-error";
+import { getInstance } from "../lib/websocket";
 
 const DEFAULT_BUCKET = process.env.MINIO_BUCKET;
 const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;
@@ -101,6 +102,9 @@ export class DrawerController extends Controller {
 
     if (!created) throw new Error("เกิดข้อผิดพลาดกับระบบจัดการไฟล์");
 
+    const io = getInstance();
+    io?.emit("CreateFolder", { pathname: `${basePath}${replaceIllegalChars(body.name)}/` });
+
     return this.setStatus(HttpStatusCode.CREATED);
   }
 
@@ -172,6 +176,12 @@ export class DrawerController extends Controller {
       }),
     );
 
+    const io = getInstance();
+    io?.emit("EditFolder", {
+      from: `${cabinetName}/${drawerName}/`,
+      to: `${cabinetName}/${replaceIllegalChars(body.name)}/`,
+    });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 
@@ -201,6 +211,11 @@ export class DrawerController extends Controller {
       stream.on("error", () => reject(new Error("เกิดข้อผิดพลาด ไม่สามารถลบไฟล์ได้")));
     });
 
+    const io = getInstance();
+    io?.emit("DeleteFolder", {
+      pathname: `${cabinetName}/${drawerName}/`,
+    });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 

Services/server/src/controllers/folderController.ts

@@ -23,6 +23,7 @@ import { copyCond, listFolder, listItem, pathExist, replaceIllegalChars } from "
 import HttpStatusCode from "../interfaces/http-status";
 import { StorageFile, StorageFolder } from "../interfaces/storage-fs";
 import HttpError from "../interfaces/http-error";
+import { getInstance } from "../lib/websocket";
 
 const DEFAULT_BUCKET = process.env.MINIO_BUCKET;
 const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;
@@ -106,6 +107,9 @@ export class FolderController extends Controller {
 
     if (!created) throw new Error("เกิดข้อผิดพลาดกับระบบจัดการไฟล์");
 
+    const io = getInstance();
+    io?.emit("CreateFolder", { pathname: `${basePath}${replaceIllegalChars(body.name)}/` });
+
     return this.setStatus(HttpStatusCode.CREATED);
   }
 
@@ -179,6 +183,12 @@ export class FolderController extends Controller {
       }),
     );
 
+    const io = getInstance();
+    io?.emit("EditFolder", {
+      from: `${cabinetName}/${drawerName}/${folderName}`,
+      to: `${cabinetName}/${drawerName}/${replaceIllegalChars(body.name)}/`,
+    });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 
@@ -213,6 +223,9 @@ export class FolderController extends Controller {
       stream.on("error", () => reject(new Error("เกิดข้อผิดพลาด ไม่สามารถลบไฟล์ได้")));
     });
 
+    const io = getInstance();
+    io?.emit("DeleteFolder", { pathname: `${cabinetName}/${drawerName}/${folderName}/` });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 

Services/server/src/controllers/subFolderController.ts

@@ -23,6 +23,7 @@ import { copyCond, listFolder, listItem, pathExist, replaceIllegalChars } from "
 import HttpStatusCode from "../interfaces/http-status";
 import { StorageFile, StorageFolder } from "../interfaces/storage-fs";
 import HttpError from "../interfaces/http-error";
+import { getInstance } from "../lib/websocket";
 
 const DEFAULT_BUCKET = process.env.MINIO_BUCKET;
 const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;
@@ -105,6 +106,9 @@ export class SubFolderController extends Controller {
 
     if (!created) throw new Error("เกิดข้อผิดพลาดกับระบบจัดการไฟล์");
 
+    const io = getInstance();
+    io?.emit("CreateFolder", { pathname: `${basePath}${replaceIllegalChars(body.name)}/` });
+
     return this.setStatus(HttpStatusCode.CREATED);
   }
 
@@ -182,6 +186,12 @@ export class SubFolderController extends Controller {
       }),
     );
 
+    const io = getInstance();
+    io?.emit("EditFolder", {
+      from: `${cabinetName}/${drawerName}/${folderName}/${subFolderName}/`,
+      to: `${cabinetName}/${drawerName}/${folderName}/${replaceIllegalChars(body.name)}/`,
+    });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 
@@ -218,6 +228,11 @@ export class SubFolderController extends Controller {
       stream.on("error", () => reject(new Error("เกิดข้อผิดพลาด ไม่สามารถลบไฟล์ได้")));
     });
 
+    const io = getInstance();
+    io?.emit("DeleteFolder", {
+      pathname: `${cabinetName}/${drawerName}/${folderName}/${subFolderName}`,
+    });
+
     return this.setStatus(HttpStatusCode.NO_CONTENT);
   }
 

Services/server/src/lib/websocket.ts

@@ -0,0 +1,11 @@
+import { Server } from "socket.io";
+
+let io: Server | null = null;
+
+export function setInstance(server: Server) {
+  io = server;
+}
+
+export function getInstance() {
+  return io;
+}

Services/server/src/rabbitmq/handler.ts

@@ -1,6 +1,7 @@
 import { StorageFile } from "../interfaces/storage-fs";
 import esClient from "../elasticsearch";
 import minioClient from "../minio";
+import { getInstance } from "../lib/websocket";
 
 const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;
 
@@ -27,7 +28,7 @@ export async function handler(key: string, event: string): Promise<boolean> {
       cachedBuffer[key] = buffer;
     } catch (e: any) {
       if (e.code === "NoSuchKey") {
-        console.info(`[AMQ] Key: ${key} received but cannot be found.`)
+        console.info(`[AMQ] Key: ${key} received but cannot be found.`);
         delete cachedBuffer[key];
         delete cachedMetadata[key];
         await ensureDelete(pathname);
@@ -43,7 +44,7 @@ export async function handler(key: string, event: string): Promise<boolean> {
 
   const rec = await popInfo(pathname);
 
-  console.info(`[AMQ] Key: ${key} - ${rec ?? 'Not Found.'}`)
+  console.info(`[AMQ] Key: ${key} - ${rec ?? "Not Found."}`);
 
   const result = rec
     ? await handleFoundRecord(rec, cachedBuffer[key], cachedMetadata[key])
@@ -94,6 +95,10 @@ async function ensureDelete(pathname: string) {
       conflicts: "proceed",
     })
     .catch((e) => console.error(e));
+
+  const io = getInstance();
+  io?.send("FileDelete", { pathname });
+
   return true;
 }
 
@@ -132,12 +137,16 @@ async function handleNotFoundRecord(
       pipeline: "attachment",
       index: DEFAULT_INDEX!,
       document: { data: base64, ...metadata },
+      refresh: "wait_for",
     })
     .catch((e) => console.error(e));
 
-  if (result) return true;
+  if (!result) return false;
 
-  return false;
+  const io = getInstance();
+  io?.send("FileUpdate", metadata);
+
+  return true;
 }
 
 async function handleFoundRecord(
@@ -154,10 +163,14 @@ async function handleFoundRecord(
       pipeline: "attachment",
       index: DEFAULT_INDEX!,
       document: { data: Buffer.from(buffer).toString("base64"), ...metadata },
+      refresh: "wait_for",
     })
     .catch((e) => console.error(e));
 
-  if (result) return true;
+  if (!result) return false;
 
-  return false;
+  const io = getInstance();
+  io?.send("FileUpdate", metadata);
+
+  return true;
 }