Turn WsMessage into an object

src/controllers/UserWebSocketController.ts

@@ -11,6 +11,7 @@ import { randomStr } from '../utils/CryptUtils';
 import { EventsHelper } from '../helpers/EventsHelper';
 import { NotificationsHelper } from '../helpers/NotificationsHelper';
 import { ConversationsHelper } from '../helpers/ConversationsHelper';
+import { WsMessage } from '../entities/WsMessage';
 
 interface PendingRequests {
 	time: number,
@@ -26,12 +27,6 @@ export interface ActiveClient {
 	ws: ws
 }
 
-export interface WsMessage {
-	id: string,
-	title: string,
-	data: any
-}
-
 // Tokens are valid only 10 seconds after they are generated
 const TOKENS_DURATION = 10
 const TOKEN_LENGTH = 20
@@ -201,11 +196,11 @@ export class UserWebSocketController {
 				continue;
 
 			// Notify user
-			this.Send(userID, "", {
+			this.Send(userID, "", new WsMessage({
 				title: "number_notifs",
 				id: "",
 				data: await NotificationsHelper.CountUnread(userID)
-			});
+			}));
 		}
 
 	}
@@ -222,11 +217,11 @@ export class UserWebSocketController {
 				continue;
 			
 			// Notify user
-			this.Send(userID, "", {
+			this.Send(userID, "", new WsMessage({
 				title: "number_unread_conversations",
 				id: "",
 				data: await ConversationsHelper.CountUnreadForUser(userID)
-			});
+			}));
 		}
 	}
 }

src/entities/WebSocketRequestHandler.ts

@@ -5,7 +5,8 @@
  */
 
 import { BaseRequestsHandler } from "./BaseRequestsHandler";
-import { ActiveClient, WsMessage } from "../controllers/UserWebSocketController";
+import { ActiveClient } from "../controllers/UserWebSocketController";
+import { WsMessage } from "./WsMessage";
 
 export class UserWebSocketRequestsHandler extends BaseRequestsHandler {
 
@@ -17,7 +18,7 @@ export class UserWebSocketRequestsHandler extends BaseRequestsHandler {
 		return this.wsClient.userID;
 	}
 
-	
+
 	protected getPostParam(name: string) {
 		throw new Error("Method not implemented.");
 	}

src/entities/WsMessage.ts

@@ -0,0 +1,24 @@
+/**
+ * WebSocket message
+ * 
+ * @author Pierre Hubert
+ */
+
+export interface WsMessageBuilder {
+	id: string,
+	title: string,
+	data: any
+}
+
+export class WsMessage implements WsMessageBuilder {
+	id: string;
+	title: string;
+	data: any;
+	
+	public constructor(info: WsMessageBuilder) {
+		for (const key in info) {
+			if (info.hasOwnProperty(key))
+				this[key] = info[key];
+		}
+	}
+}