remodel

src/Model/chatModel.ts

@@ -3,46 +3,42 @@ import * as React from "bloatless-react";
 import { UDN, isConnected } from "./model";
 import { decryptString, encryptString } from "../cryptUtility";
 
+import { Message } from "udn-frontend";
 import { senderName } from "./viewModel";
 import { storageKeys } from "../utility";
 
 // TYPES
 // message
-export interface Message {
+export interface ChatMessage {
   channel: string;
   sender: string;
   body: string;
   isoDate: string;
 }
 
-export class Message {
-  constructor(
-    public channel: string,
-    public sender: string,
-    public body: string,
-    public isoDate: string
-  ) {}
-}
-
 // chat
 export class Chat {
   id: string;
 
-  primaryChannel: React.State<string>;
+  isSubscribed = new React.State(false);
+  currentChannel = new React.State("");
+
+  primaryChannelInput: React.State<string>;
   secondaryChannels: React.ListState<string>;
 
   encryptionKey: React.State<string>;
-  messages: React.ListState<Message>;
+  messages: React.ListState<ChatMessage>;
 
   composingMessage: React.State<string>;
-  cannotSendMessage: React.State<boolean>;
-
   newSecondaryChannelName: React.State<string>;
+
+  cannotSendMessage: React.State<boolean>;
   cannotAddSecondaryChannel: React.State<boolean>;
+  cannotSetChannel: React.State<boolean>;
 
   // init
   constructor(id: string = React.UUID()) {
-    this.primaryChannel = React.restoreState(
+    this.primaryChannelInput = React.restoreState(
       storageKeys.primaryChannel(id),
       ""
     );
@@ -60,16 +56,48 @@ export class Chat {
     this.newSecondaryChannelName = new React.State("");
 
     this.cannotSendMessage = React.createProxyState(
-      [this.primaryChannel, this.composingMessage, isConnected],
+      [
+        this.primaryChannelInput,
+        this.composingMessage,
+        this.isSubscribed,
+        isConnected,
+      ],
       () =>
-        this.primaryChannel.value == "" ||
+        this.primaryChannelInput.value == "" ||
         this.composingMessage.value == "" ||
+        this.isSubscribed.value == false ||
         isConnected.value == false
     );
     this.cannotAddSecondaryChannel = React.createProxyState(
       [this.newSecondaryChannelName],
       () => this.newSecondaryChannelName.value == ""
     );
+    this.cannotSetChannel = React.createProxyState(
+      [this.primaryChannelInput, this.currentChannel],
+      () =>
+        this.primaryChannelInput.value == "" ||
+        this.primaryChannelInput.value == this.currentChannel.value
+    );
+  }
+
+  // handlers
+  onmessage(data: Message): void {
+    if (data.messageChannel && data.messageChannel != this.currentChannel.value)
+      return;
+
+    if (data.subscribed != undefined) this.handleSubscription(data.subscribed);
+
+    if (!data.messageBody) return;
+    const { sender, body, channel, isoDate } = JSON.parse(data.messageBody);
+    this.handleMessage({ sender, body, channel, isoDate });
+  }
+
+  handleSubscription(isSubscribed: boolean): void {
+    this.isSubscribed.value = isSubscribed;
+  }
+
+  handleMessage(chatMessage: ChatMessage): void {
+    this.messages.add(chatMessage);
   }
 
   // messages
@@ -81,7 +109,7 @@ export class Chat {
       ...this.secondaryChannels.value.values(),
     ].map((channel) => channel);
     const allChannelNames: string[] = [
-      this.primaryChannel.value,
+      this.primaryChannelInput.value,
       ...secondaryChannelNames,
     ];
 
@@ -95,12 +123,12 @@ export class Chat {
 
     // create object
     const joinedChannelName = allChannelNames.join("/");
-    const messageObject = new Message(
-      joinedChannelName,
-      senderName.value,
-      encrypted,
-      new Date().toISOString()
-    );
+    const messageObject: ChatMessage = {
+      channel: joinedChannelName,
+      sender: senderName.value,
+      body: encrypted,
+      isoDate: new Date().toISOString(),
+    };
     const messageString = JSON.stringify(messageObject);
 
     // send
@@ -114,17 +142,21 @@ export class Chat {
     this.messages.clear();
   }
 
-  deleteMessage(message: Message): void {
+  deleteMessage(message: ChatMessage): void {
     this.messages.remove(message);
   }
 
-  async decryptReceivedMessage(message: Message): Promise<void> {
+  async decryptReceivedMessage(message: ChatMessage): Promise<void> {
     message.body = await decryptString(message.body, this.encryptionKey.value);
     this.messages.callSubscriptions();
   }
 
   // channel
-  setChannel(): void {}
+  setChannel(): void {
+    if (this.cannotSetChannel.value == true) return;
+    this.currentChannel.value = this.primaryChannelInput.value;
+    UDN.subscribe(this.currentChannel.value);
+  }
 
   addSecondaryChannel(): void {
     if (this.cannotAddSecondaryChannel.value == true) return;
@@ -146,7 +178,7 @@ export const chats = new React.ListState<Chat>(chatArray);
 // METHODS
 export function createChatWithName(name: string): void {
   const newChat = new Chat();
-  newChat.primaryChannel.value = name;
+  newChat.primaryChannelInput.value = name;
   chatIds.add(newChat.id);
 }