CT/src/Model/chatListModel.ts

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

import { ChatModel } from "./chatModel";
import ConnectionModel from "./connectionModel";
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[] = [];

  // 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);
  }
  
  // restore
  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();
  }
}