implement auto reconnect

src/Model/Chat/chatListModel.ts

@@ -1,5 +1,8 @@
 import ChatModel, { ChatMessage } from "./chatModel";
-import StorageModel, { StorageModelSubPath, filePaths } from "../Global/storageModel";
+import StorageModel, {
+    StorageModelSubPath,
+    filePaths,
+} from "../Global/storageModel";
 
 import ConnectionModel from "../Global/connectionModel";
 import { Message } from "udn-frontend";
@@ -7,104 +10,107 @@ import SettingsModel from "../Global/settingsModel";
 import { v4 } from "uuid";
 
 export default class ChatListModel {
-  storageModel: StorageModel;
-  settingsModel: SettingsModel;
-  connectionModel: ConnectionModel;
+    storageModel: StorageModel;
+    settingsModel: SettingsModel;
+    connectionModel: ConnectionModel;
 
-  // data
-  chatModels = new Set<ChatModel>();
+    // data
+    chatModels = new Set<ChatModel>();
 
-  // handlers
-  messageHandler = (data: Message): void => {
-    const channel: string | undefined = data.messageChannel;
-    const body: string | undefined = data.messageBody;
-    if (channel == undefined) return;
-    if (body == undefined) return;
+    // handlers
+    messageHandler = (data: Message): void => {
+        const channel: string | undefined = data.messageChannel;
+        const body: string | undefined = data.messageBody;
+        if (channel == undefined) return;
+        if (body == undefined) return;
 
-    this.routeMessageToCorrectChatModel(channel, (chatModel: ChatModel) =>
-      chatModel.handleMessage(body)
-    );
-  };
+        this.routeMessageToCorrectChatModel(channel, (chatModel: ChatModel) =>
+            chatModel.handleMessage(body),
+        );
+    };
 
-  messageSentHandler = (chatMessage: ChatMessage): void => {
-    const channel: string = chatMessage.channel;
+    messageSentHandler = (chatMessage: ChatMessage): void => {
+        const channel: string = chatMessage.channel;
 
-    this.routeMessageToCorrectChatModel(channel, (chatModel: ChatModel) =>
-      chatModel.handleMessageSent(chatMessage)
-    );
-  };
+        this.routeMessageToCorrectChatModel(channel, (chatModel: ChatModel) =>
+            chatModel.handleMessageSent(chatMessage),
+        );
+    };
 
-  // methods
-  routeMessageToCorrectChatModel = (
-    channel: string,
-    fn: (chatModel: ChatModel) => void
-  ): void => {
-    const allChannels = channel.split("/");
+    // methods
+    routeMessageToCorrectChatModel = (
+        channel: string,
+        fn: (chatModel: ChatModel) => void,
+    ): void => {
+        const allChannels = channel.split("/");
 
-    for (const chatModel of this.chatModels) {
-      for (const channel of allChannels) {
-        if (channel != chatModel.info.primaryChannel) continue;
-        fn(chatModel);
-        break;
-      }
+        for (const chatModel of this.chatModels) {
+            for (const channel of allChannels) {
+                if (channel != chatModel.info.primaryChannel) continue;
+                fn(chatModel);
+                break;
+            }
+        }
+    };
+
+    // storage
+    addChatModel = (chatModel: ChatModel) => {
+        this.chatModels.add(chatModel);
+    };
+
+    createChat = (primaryChannel: string): ChatModel => {
+        const id: string = v4();
+
+        const chatModel = new ChatModel(
+            this.storageModel,
+            this.connectionModel,
+            this.settingsModel,
+            this,
+            id,
+        );
+        chatModel.setPrimaryChannel(primaryChannel);
+
+        this.addChatModel(chatModel);
+        return chatModel;
+    };
+
+    untrackChat = (chat: ChatModel): void => {
+        this.chatModels.delete(chat);
+    };
+
+    // load
+    loadChats = (): void => {
+        const chatDir = StorageModel.getPath(
+            StorageModelSubPath.Chat,
+            filePaths.chat.base,
+        );
+        const chatIds = this.storageModel.list(chatDir);
+        for (const chatId of chatIds) {
+            const chatModel = new ChatModel(
+                this.storageModel,
+                this.connectionModel,
+                this.settingsModel,
+                this,
+                chatId,
+            );
+            this.addChatModel(chatModel);
+        }
+    };
+
+    // init
+    constructor(
+        storageModel: StorageModel,
+        settingsModel: SettingsModel,
+        connectionModel: ConnectionModel,
+    ) {
+        this.storageModel = storageModel;
+        this.settingsModel = settingsModel;
+        this.connectionModel = connectionModel;
+        this.loadChats();
+
+        connectionModel.messageHandlerManager.addHandler(this.messageHandler);
+        connectionModel.messageSentHandlerManager.addHandler(
+            this.messageSentHandler,
+        );
     }
-  };
-
-  // storage
-  addChatModel = (chatModel: ChatModel) => {
-    this.chatModels.add(chatModel);
-  };
-
-  createChat = (primaryChannel: string): ChatModel => {
-    const id: string = v4();
-
-    const chatModel = new ChatModel(
-      this.storageModel,
-      this.connectionModel,
-      this.settingsModel,
-      this,
-      id
-    );
-    chatModel.setPrimaryChannel(primaryChannel);
-
-    this.addChatModel(chatModel);
-    return chatModel;
-  };
-
-  untrackChat = (chat: ChatModel): void => {
-    this.chatModels.delete(chat);
-  };
-
-  // load
-  loadChats = (): void => {
-    const chatDir = StorageModel.getPath(StorageModelSubPath.Chat, filePaths.chat.base);
-    const chatIds = this.storageModel.list(chatDir);
-    for (const chatId of chatIds) {
-      const chatModel = new ChatModel(
-        this.storageModel,
-        this.connectionModel,
-        this.settingsModel,
-        this,
-        chatId
-      );
-      this.addChatModel(chatModel);
-    }
-  };
-
-  // init
-  constructor(
-    storageModel: StorageModel,
-    settingsModel: SettingsModel,
-    connectionModel: ConnectionModel
-  ) {
-    this.storageModel = storageModel;
-    this.settingsModel = settingsModel;
-    this.connectionModel = connectionModel;
-    this.loadChats();
-
-    connectionModel.messageHandlerManager.addHandler(this.messageHandler);
-    connectionModel.messageSentHandlerManager.addHandler(
-      this.messageSentHandler
-    );
-  }
 }