From eb1f62bb1cc2120762ae7c4254e30b8a926b1595 Mon Sep 17 00:00:00 2001
From: Gregoire Parant <g.parant@thecodingmachine.com>
Date: Tue, 20 Oct 2020 08:30:11 +0200
Subject: [PATCH] Add banned message and close ws

---
 back/src/Controller/IoSocketController.ts | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/back/src/Controller/IoSocketController.ts b/back/src/Controller/IoSocketController.ts
index c4d882a2..0425aa40 100644
--- a/back/src/Controller/IoSocketController.ts
+++ b/back/src/Controller/IoSocketController.ts
@@ -77,16 +77,17 @@ export class IoSocketController {
                         JSON.parse(new TextDecoder("utf-8").decode(new Uint8Array(arrayBuffer)));
 
                     if(message.event === 'user-message') {
-                        if (message.message.type === 'ban') {
-                            const messageToEmit = (message.message as {message: string, type: string, userUuid: string});
-                            socketManager.emitSendUserMessage(messageToEmit);
-                        }
-                        if (message.message.type === 'banned') {
-                            const messageToEmit = (message.message as {message: string, type: string, userUuid: string});
-                            const socketUser = socketManager.emitSendUserMessage(messageToEmit);
-                            setTimeout(() => {
-                                socketUser.close();
-                            }, 10000);
+                        const messageToEmit = (message.message as { message: string, type: string, userUuid: string });
+                        switch (message.message.type) {
+                            case 'ban':
+                                socketManager.emitSendUserMessage(messageToEmit);
+                                break;
+                            case 'banned':
+                                const socketUser = socketManager.emitSendUserMessage(messageToEmit);
+                                setTimeout(() => {
+                                    socketUser.close();
+                                }, 10000);
+                                break;
                         }
                     }
                 }catch (err) {