CT/src/Model/chatListModel.ts

import StorageModel, { storageKeys } from "./storageModel";

import ChatModel from "./chatModel";
import ConnectionModel from "./connectionModel";
import { Message } from "udn-frontend";
import SettingsModel from "./settingsModel";
import { localeCompare } from "./Utility/utility";
import { v4 } from "uuid";

export default class ChatListModel {
  storageModel: StorageModel;
  settingsModel: SettingsModel;
  connectionModel: ConnectionModel;

  // data
  chatModels = new Set<ChatModel>();
  sortedPrimaryChannels: string[] = [];

  // handlers
  messageHandler = (data: Message): void => {
    for (const chatModel of this.chatModels) {
      const channel: string | undefined = data.messageChannel;
      const body: string | undefined = data.messageBody;
      if (channel == undefined) return;
      if (body == undefined) return;

      const allChannels = channel.split("/");
      for (const channel of allChannels) {
        if (channel != chatModel.info.primaryChannel) continue;
        chatModel.handleMessage(body);
        break;
      }
    }
  };

  // store & add
  addChatModel = (chatModel: ChatModel) => {
    this.chatModels.add(chatModel);
    this.updateIndices();
  };

  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);
    this.updateIndices();
  };

  // sorting
  updateIndices = () => {
    this.sortedPrimaryChannels = [];

    let allChannels: string[] = [];
    for (const chatModel of this.chatModels) {
      allChannels.push(chatModel.info.primaryChannel);
    }

    this.sortedPrimaryChannels = allChannels.sort(localeCompare);
  };

  getIndexOfPrimaryChannel(primaryChannel: string): number {
    return this.sortedPrimaryChannels.indexOf(primaryChannel);
  }

  // load
  loadChats = (): void => {
    const chatDir = storageKeys.chats;
    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.setMessageHandler(this.messageHandler);
  }
}