underbbs/frontend/ts/thread.ts

import util from "./util"
import { Message } from "./message"

export class MessageNode {
  public parent: MessageNode | null = null;
  public children: MessageNode[] = [];
  public data: Message;
  
  constructor(msg: Message, parent: MessageNode | null = null) {
    this.data = msg;
    this.parent = parent;
  }
  
  findRoot(): MessageNode {
    let self: MessageNode | null = this;
    
    while(self.parent) {
      self = self.parent;
    }
    return self;
  }
}

export class MessageThread {
  public root: MessageNode;
  public messageCount: number;
  public visibility: string;
  public created: number;
  public latest: number;
  
  constructor(first: Message) {
    this.root = new MessageNode(first);
    this.messageCount = 1;
    this.visibility = first.visibility;
    this.created = first.created;
    this.latest = first.edited ? first.edited : first.created;
  }
  
  addReply(parentID: string, reply: Message): boolean {
    let node = this.findNode(this.root, parentID);
    if (node) {
      node.children.push(new MessageNode(reply, node));
      this.messageCount++;
      const mtime = reply.edited ? reply.edited : reply.created;
      if (this.latest < mtime) {
        this.latest = mtime;
      }
      return true;
    }
    return false;
  }
  
  findNode(node: MessageNode, id: string): MessageNode | null {
    if (node.data.id == id) {
      return node;
    } else {
      for (let n of node.children) {
        const x = this.findNode(n, id);
        if (x) {
          return x;
        }
      }
    }
    return null;
  }
}