add socket

Author: khanh1902

server/model.nlp

@@ -62,6 +62,7 @@
     "react": "^18.2.0",
     "reflect-metadata": "^0.1.13",
     "slugify": "^1.6.6",
+    "socket.io": "^4.7.5",
     "swagger-jsdoc": "^6.2.1",
     "swagger-ui-express": "^4.5.0",
     "typedi": "^0.10.0",

server/package-lock.json

@@ -18,12 +18,19 @@ import { LocaleService } from './i18n/ctx'
 import { getPreferredLocale } from './i18n/get-preferred-locale'
 import { loadAllLocales } from './i18n/i18n-util.sync'
 
+// socket.io
+import { createServer } from 'http'
+import { Server } from 'socket.io'
+import { SocketController } from './controllers/socket.controller'
+
 Container.set(LOCALE_KEY, new LocaleService('en'))
 
 export class App {
   public app: express.Application
   public env: string
   public port: string | number
+  public static io: Server
+
 
   constructor(routes: Routes[]) {
     this.app = express()
@@ -37,12 +44,28 @@ export class App {
   }
 
   public listen() {
-    this.app.listen(this.port, () => {
-      logger.info(`=================================`)
-      logger.info(`======= ENV: ${this.env} =======`)
-      logger.info(`🚀 App listening on the port ${this.port}`)
-      logger.info(`=================================`)
-    })
+    try {
+      const socketController = new SocketController();
+      const server = createServer(this.app);
+      App.io = new Server(server);
+
+      App.io.on('connection', (socket) => {
+        const query = socket.handshake.query;
+        const userId = query.userId;
+        socket.join(userId);
+        socketController.handleSocketEvents(socket);
+      });
+
+      server.listen(this.port, () => {
+        logger.info(`=================================`)
+        logger.info(`======= ENV: ${this.env} =======`)
+        logger.info(`🚀 App listening on the port ${this.port}`)
+        logger.info(`=================================`)
+      })
+
+    } catch (error) {
+      console.error('Error initializing server:', error);
+    }
   }
 
   public getServer() {

server/package.json

@@ -90,7 +90,7 @@ export class LineChannel extends BaseChannel {
           data: null,
         })
       }
-    } catch (e) {}
+    } catch (e) { }
   }
 
   public async sendMessageToUser({ userId, text }) {

server/src/app.ts

@@ -0,0 +1,34 @@
+import { logger } from '@/utils/logger'
+import { Request, Response } from 'express'
+import { App } from '../app'
+import { BaseChannel } from './base.channel'
+
+export class WebChannel extends BaseChannel {
+    credentials: string
+
+    constructor(
+        id: string,
+        contactId: string,
+        contactName: string,
+        channelType: string,
+        credentials: string,
+    ) {
+        super(id, contactId, contactName, channelType)
+        this.channelType = channelType;
+        this.credentials = credentials;
+    }
+
+    async prepareMessage(req: Request, res: Response) { }
+
+    public async sendMessageToUser({ userId, text }) {
+        try {
+            if (App.io) {
+                App.io.to(userId).emit('message', { userId, message: text });
+            }
+        } catch (e) {
+            logger.info(
+                `[WEB] Bot send message to User failed - Error: ${e.message}`,
+            )
+        }
+    }
+}

server/src/channels/line.channel.ts

@@ -23,6 +23,7 @@ export class IntentController {
     })
   })
 
+  // for bot
   public predictIntent = catchAsync(async (req: RequestWithUser, res) => {
     req.body.userId = req.user?.id
     const data = await this.intentService.PredictTrainIntent(req.body)

server/src/channels/web.channel.ts

@@ -0,0 +1,14 @@
+import { LOCALE_KEY } from '@/constants'
+import { LocaleService } from '@/i18n/ctx'
+import { SocketService } from '@/services/socket.service'
+import { Socket } from 'socket.io'
+import Container from 'typedi'
+
+export class SocketController {
+    public socketService = Container.get(SocketService)
+    public localeService = Container.get<LocaleService>(LOCALE_KEY)
+
+    public handleSocketEvents(socket: Socket) {
+        this.socketService.handleSocketEvents(socket);
+    }
+}