bitvid/js/services/discussionCountService.js

import { devLogger } from "../utils/logger.js";
import { buildVideoAddressPointer } from "../utils/videoPointer.js";
import { getAllowedCommentKinds } from "../nostr/commentTargetNormalizer.js";

const DEFAULT_MAX_DISCUSSION_COUNT_VIDEOS = 24;
const COUNT_UNSUPPORTED_TITLE = "Relay does not support NIP-45 COUNT queries.";
const COUNT_PARTIAL_TITLE =
  "Some relays timed out. Comment counts may be partial.";

function toPositiveInteger(value, fallback) {
  const numeric = Number(value);
  if (!Number.isFinite(numeric) || numeric <= 0) {
    return fallback;
  }
  return Math.floor(numeric);
}

export default class DiscussionCountService {
  constructor({
    maxVideos = DEFAULT_MAX_DISCUSSION_COUNT_VIDEOS,
    logger = devLogger,
  } = {}) {
    this.maxVideos = toPositiveInteger(maxVideos, DEFAULT_MAX_DISCUSSION_COUNT_VIDEOS);
    this.logger = logger && typeof logger.warn === "function" ? logger : devLogger;
    this.videoDiscussionCountCache = new Map();
    this.inFlightDiscussionCounts = new Map();
  }

  clearCache() {
    this.videoDiscussionCountCache.clear();
    this.inFlightDiscussionCounts.clear();
  }

  getCachedCount(videoId) {
    if (!videoId) {
      return undefined;
    }
    return this.videoDiscussionCountCache.get(videoId);
  }

  refreshCounts(videos = [], options = {}) {
    const { videoListRoot = null, nostrClient = null } = options;

    if (!Array.isArray(videos) || !videos.length) {
      return;
    }

    const root =
      videoListRoot && typeof videoListRoot.querySelector === "function"
        ? videoListRoot
        : null;
    if (!root || !nostrClient?.pool || typeof nostrClient.countEventsAcrossRelays !== "function") {
      return;
    }

    const eligibleVideos = videos
      .filter(
        (video) =>
          video &&
          typeof video.id === "string" &&
          video.id &&
          video.enableComments !== false
      )
      .slice(0, this.maxVideos);

    eligibleVideos.forEach((video) => {
      const container = root.querySelector(`[data-discussion-count="${video.id}"]`);
      if (!container) {
        return;
      }

      const cachedCount = this.getCachedCount(video.id);
      if (typeof cachedCount === "number") {
        this.updateDiscussionCountElement(container, cachedCount);
        return;
      }

      const filters = this.buildDiscussionCountFilters(video);
      if (!filters.length) {
        this.markDiscussionCountError(container, { unsupported: true });
        return;
      }

      if (this.inFlightDiscussionCounts.has(video.id)) {
        this.setDiscussionCountPending(container);
        return;
      }

      this.setDiscussionCountPending(container);

      const request = Promise.resolve()
        .then(() => nostrClient.countEventsAcrossRelays(filters))
        .then((result) => {
          const perRelay = Array.isArray(result?.perRelay)
            ? result.perRelay.filter((entry) => entry && entry.ok)
            : [];

          if (!perRelay.length) {
            this.markDiscussionCountError(container, { unsupported: true });
            return result;
          }

          const total = Number(result?.total);
          const isPartial = Boolean(result?.partial);
          const normalized =
            Number.isFinite(total) && total >= 0 ? Math.floor(total) : 0;
          this.videoDiscussionCountCache.set(video.id, normalized);
          this.updateDiscussionCountElement(container, normalized, { partial: isPartial });
          return result;
        })
        .catch((error) => {
          if (this.logger?.warn) {
            this.logger.warn(
              `[counts] Failed to fetch discussion count for ${video.id}:`,
              error
            );
          }
          this.markDiscussionCountError(container);
          throw error;
        })
        .finally(() => {
          this.inFlightDiscussionCounts.delete(video.id);
        });

      this.inFlightDiscussionCounts.set(video.id, request);
    });
  }

  buildDiscussionCountFilters(video) {
    if (!video || typeof video !== "object") {
      return [];
    }

    // Count the same kinds the thread actually displays (NIP-22 kind 1111 plus
    // legacy kind 1). Counting only 1111 made the badge undercount threads that
    // contain legacy comments.
    const commentKinds = getAllowedCommentKinds();

    const filters = [];
    const eventId = typeof video.id === "string" ? video.id.trim() : "";
    if (eventId) {
      filters.push({ kinds: commentKinds, "#E": [eventId] });
    }

    const address = buildVideoAddressPointer(video);
    const rootIdentifier =
      typeof video.videoRootId === "string" ? video.videoRootId.trim() : "";

    const uppercaseFilter = { kinds: commentKinds };
    let hasUppercasePointer = false;

    if (rootIdentifier) {
      uppercaseFilter["#I"] = [rootIdentifier];
      hasUppercasePointer = true;
    } else if (address) {
      uppercaseFilter["#A"] = [address];
      hasUppercasePointer = true;
    } else if (eventId) {
      uppercaseFilter["#E"] = [eventId];
      hasUppercasePointer = true;
    }

    const rootKind = (() => {
      if (Number.isFinite(video.kind)) {
        return String(Math.floor(video.kind));
      }
      if (typeof video.kind === "string" && video.kind.trim()) {
        const parsed = Number(video.kind);
        if (Number.isFinite(parsed)) {
          return String(Math.floor(parsed));
        }
        return video.kind.trim();
      }
      return "";
    })();

    if (rootKind) {
      uppercaseFilter["#K"] = [rootKind];
    }

    const rootAuthor =
      typeof video.pubkey === "string" ? video.pubkey.trim() : "";
    if (rootAuthor) {
      uppercaseFilter["#P"] = [rootAuthor];
    }

    if (hasUppercasePointer) {
      filters.push(uppercaseFilter);
    }

    if (address) {
      filters.push({ kinds: commentKinds, "#A": [address] });
    }

    return filters;
  }

  getVideoAddressPointer(video) {
    return buildVideoAddressPointer(video);
  }

  setDiscussionCountPending(element) {
    if (!element) {
      return;
    }

    element.dataset.countState = "pending";
    const valueEl = element.querySelector("[data-discussion-count-value]");
    if (valueEl) {
      valueEl.textContent = "…";
    }
    element.removeAttribute("title");
  }

  updateDiscussionCountElement(element, count, { partial = false } = {}) {
    if (!element) {
      return;
    }

    const valueEl = element.querySelector("[data-discussion-count-value]");
    if (!valueEl) {
      return;
    }

    const numeric = Number(count);
    const safeValue =
      Number.isFinite(numeric) && numeric >= 0 ? Math.floor(numeric) : 0;
    element.dataset.countState = partial ? "partial" : "ready";
    valueEl.textContent = safeValue.toLocaleString();
    if (partial) {
      element.title = COUNT_PARTIAL_TITLE;
    } else {
      element.removeAttribute("title");
    }
  }

  markDiscussionCountError(element, { unsupported = false } = {}) {
    if (!element) {
      return;
    }

    const valueEl = element.querySelector("[data-discussion-count-value]");
    if (valueEl) {
      valueEl.textContent = "—";
    }

    element.dataset.countState = unsupported ? "unsupported" : "error";

    if (unsupported) {
      element.title = COUNT_UNSUPPORTED_TITLE;
    } else {
      element.removeAttribute("title");
    }
  }
}

export { COUNT_UNSUPPORTED_TITLE };