UDN/static/node_modules/udn-frontend/index.ts

export interface Message {
  // connecting to server
  requestingServerConnection?: boolean;

  // mailbox
  requestingMailboxSetup?: boolean;
  requestedMailbox?: string;
  deletingMailbox?: string;
  assignedMailboxId?: string; // sent by server
  connectedMailboxId?: string; // sent by server

  // subscribing to channel
  subscribeChannel?: string;
  unsubscribeChannel?: string;
  subscribed?: boolean; // sent by server to confirm subscription

  //sending message
  messageChannel?: string;
  messageBody?: string;
}

export default class UDNFrontend {
  ws: WebSocket | undefined;

  // HANDLERS
  private connectionHandler = () => {};
  private disconnectionHandler = () => {};
  private messageHandler = (data: Message) => {};
  private mailboxHandler = (mailboxId: string) => {};
  private mailboxConnectionHandler = (mailboxId: string) => {};

  // INIT
  set onconnect(handler: () => void) {
    this.connectionHandler = handler;
  }
  set ondisconnect(handler: () => void) {
    this.disconnectionHandler = handler;
  }
  set onmessage(handler: (data: Message) => void) {
    this.messageHandler = handler;
  }
  set onmailbox(handler: (mailboxId: string) => void) {
    this.mailboxHandler = handler;
  }
  set onmailboxconnect(handler: (mailboxId: string) => void) {
    this.mailboxConnectionHandler = handler;
  }

  // UTILITY METHODS
  private send(messageObject: Message): boolean {
    if (this.ws == undefined) return false;

    const messageString = JSON.stringify(messageObject);
    this.ws.send(messageString);
    return true;
  }

  // PUBLIC METHODS
  // connection
  connect(address: string): void {
    this.disconnect();

    this.ws = new WebSocket(address);
    this.ws.addEventListener("open", this.connectionHandler);
    this.ws.addEventListener("close", this.disconnectionHandler);
    this.ws.addEventListener("message", (message) => {
      const dataString = message.data.toString();
      const data = JSON.parse(dataString);

      if (data.assignedMailboxId) {
        return this.mailboxHandler(data.assignedMailboxId);
      } else if (data.connectedMailboxId) {
        return this.mailboxConnectionHandler(data.assignedMailboxId);
      } else {
        this.messageHandler(data);
      }
    });
  }

  disconnect(): void {
    this.ws?.close();
  }

  // message
  sendMessage(channel: string, body: string): boolean {
    const messageObject = {
      messageChannel: channel,
      messageBody: body,
    };
    return this.send(messageObject);
  }

  // subscription
  subscribe(channel: string): boolean {
    const messageObject = {
      subscribeChannel: channel,
    };
    return this.send(messageObject);
  }

  unsubscribe(channel: string): boolean {
    const messageObject = {
      unsubscribeChannel: channel,
    };
    return this.send(messageObject);
  }

  // mailbox
  requestMailbox(): boolean {
    const messageObject = {
      requestingMailboxSetup: true,
    };
    return this.send(messageObject);
  }

  connectMailbox(mailboxId: string): boolean {
    const messageObject = {
      requestedMailbox: mailboxId,
    };
    return this.send(messageObject);
  }

  deleteMailbox(mailboxId: string): boolean {
    const messageObject = {
      deletingMailbox: mailboxId,
    };
    return this.send(messageObject);
  }
}