snort/src/nostr/Thread.js

import Event from "./Event";
import EventKind from "./EventKind";

export default class Thread {
    constructor() {
        /** @type {Tag} */
        this.Root = null;
        /** @type {Tag} */
        this.ReplyTo = null;
        /** @type {Array<Tag>} */
        this.Mentions = [];
        /** @type {Array<String>} */
        this.PubKeys = [];
    }

    /**
     * Extract thread information from an Event
     * @param {Event} ev Event to extract thread from
     */
    static ExtractThread(ev) {
        let isThread = ev.Tags.some(a => a.Key === "e");
        if (!isThread) {
            return null;
        }

        let shouldWriteMarkers = ev.Kind === EventKind.TextNote;
        let ret = new Thread();
        let eTags = ev.Tags.filter(a => a.Key === "e");
        let marked = eTags.some(a => a.Marker !== null);
        if (!marked) {
            ret.Root = eTags[0];
            ret.Root.Marker = shouldWriteMarkers ? "root" : null;
            if (eTags.length > 1) {
                ret.ReplyTo = eTags[1];
                ret.ReplyTo.Marker = shouldWriteMarkers ? "reply" : null;
            }
            if (eTags.length > 2) {
                ret.Mentions = eTags.slice(2);
                if (shouldWriteMarkers) {
                    ret.Mentions.forEach(a => a.Marker = "mention");
                }
            }
        } else {
            let root = eTags.find(a => a.Marker === "root");
            let reply = eTags.find(a => a.Marker === "reply");
            ret.Root = root;
            ret.ReplyTo = reply;
            ret.Mentions = eTags.filter(a => a.Marker === "mention");
        }
        ret.PubKeys = [...new Set(ev.Tags.filter(a => a.Key === "p").map(a => a.PubKey))]
        return ret;
    }
}