From 0542b09e0f0de2ed8c3ca3472862a8891e174d2f Mon Sep 17 00:00:00 2001
From: Bauke <me@bauke.xyz>
Date: Sun, 25 Jun 2023 13:28:26 +0200
Subject: [PATCH] Rework the Hide Votes logic so it explicitly handles our own
 and other's votes.

---
 source/content-scripts/features/hide-votes.ts | 140 +++++++++++++-----
 1 file changed, 102 insertions(+), 38 deletions(-)

diff --git a/source/content-scripts/features/hide-votes.ts b/source/content-scripts/features/hide-votes.ts
index 90917f3..cce483a 100644
--- a/source/content-scripts/features/hide-votes.ts
+++ b/source/content-scripts/features/hide-votes.ts
@@ -9,52 +9,116 @@ export function runHideVotesFeature(data: HideVotesData) {
 function hideVotes(data: HideVotesData): number {
   let count = 0;
 
-  if (data.otherComments) {
-    const commentVotes = querySelectorAll(
-      '.btn-post-action[data-ic-put-to*="/vote"]:not(.trx-votes-hidden)',
-      '.btn-post-action[data-ic-delete-from*="/vote"]:not(.trx-votes-hidden)',
-    );
-    count += commentVotes.length;
+  // Get the username of the currently logged in user. When not logged in, set
+  // it to "<logged out>" which isn't a valid username, so matching against it
+  // will always return false. Meaning all comments are never the current user's.
+  const currentUser =
+    document.querySelector(".logged-in-user-username")?.textContent?.trim() ??
+    "<logged out>";
 
-    for (const vote of commentVotes) {
-      vote.classList.add("trx-votes-hidden");
-      if (!vote.textContent!.includes(" ")) {
-        continue;
-      }
-
-      vote.textContent = vote.textContent!.slice(
-        0,
-        vote.textContent!.indexOf(" "),
-      );
-    }
-  }
-
-  if (data.ownComments) {
-    const ownComments = querySelectorAll(".comment-votes");
-    count += ownComments.length;
-    for (const vote of ownComments) {
-      vote.classList.add("trx-hidden");
-    }
+  if (data.otherComments || data.ownComments) {
+    count += hideCommentVotes(data, currentUser);
   }
 
   if (data.otherTopics || data.ownTopics) {
-    const selectors: string[] = [];
+    count += hideTopicVotes(data, currentUser);
+  }
 
-    // Topics by other people will be encapsulated with a `<button>`.
-    if (data.otherTopics) {
-      selectors.push("button > .topic-voting-votes:not(.trx-votes-hidden)");
-    }
+  return count;
+}
 
-    // Topics by yourself will be encapsulated with a `<div>`.
-    if (data.ownTopics) {
-      selectors.push("div > .topic-voting-votes:not(.trx-votes-hidden)");
-    }
+/**
+ * Hide vote displays from comments.
+ * @param data The {@link HideVotesData}.
+ * @param currentUser The username of the currently logged in user.
+ * @returns The amount of vote displays that have been hidden.
+ */
+function hideCommentVotes(data: HideVotesData, currentUser: string): number {
+  let count = 0;
+  for (const comment of querySelectorAll(".comment-itself")) {
+    const postedBySelf =
+      comment.querySelector("header .link-user")?.textContent?.trim() ===
+      currentUser;
 
-    const topicVotes = querySelectorAll(...selectors);
-    count += topicVotes.length;
-    for (const vote of topicVotes) {
+    if (data.otherComments && !postedBySelf) {
+      // The vote element can be a "Vote" button, an "Unvote" button or a
+      // regular text display when you can't vote anymore.
+      const vote =
+        comment.querySelector(
+          '.btn-post-action[data-ic-put-to*="/vote"]:not(.trx-votes-hidden)',
+        ) ??
+        comment.querySelector(
+          '.btn-post-action[data-ic-delete-from*="/vote"]:not(.trx-votes-hidden)',
+        ) ??
+        comment.querySelector(".comment-votes:not(.trx-votes-hidden)");
+
+      if (vote === null) {
+        continue;
+      }
+
+      count++;
       vote.classList.add("trx-votes-hidden");
-      vote.textContent = "-";
+
+      if (vote.tagName === "BUTTON") {
+        // If the vote element is a button, only remove the number from the text.
+        if (!vote.textContent!.includes(" ")) {
+          // If there is no space in the text, it means there are no votes on
+          // this comment yet.
+          continue;
+        }
+
+        vote.textContent = vote.textContent!.slice(
+          0,
+          vote.textContent!.indexOf(" "),
+        );
+      } else {
+        // Otherwise if it's the regular text display, hide it entirely.
+        vote.classList.add("trx-hidden");
+      }
+    } else if (data.ownComments && postedBySelf) {
+      // Votes from our own comments are always the regular text display.
+      const vote = comment.querySelector(
+        ".comment-votes:not(.trx-votes-hidden)",
+      );
+      if (vote === null) {
+        continue;
+      }
+
+      count++;
+      vote.classList.add("trx-votes-hidden", "trx-hidden");
+    }
+  }
+
+  return count;
+}
+
+/**
+ * Hide vote displays from topics.
+ * @param data The {@link HideVotesData}.
+ * @param currentUser The username of the currently logged in user.
+ * @returns The amount of vote displays that have been hidden.
+ */
+function hideTopicVotes(data: HideVotesData, currentUser: string): number {
+  let count = 0;
+  for (const topic of querySelectorAll<HTMLElement>(".topic")) {
+    const postedBySelf = topic.dataset.topicPostedBy === currentUser;
+
+    let vote: Element | undefined;
+
+    // Select the vote number from the topic if we want to hide other's topics
+    // and it's not posted by the current user. Or if we want to hide our own
+    // topics and it *is* posted by the current user.
+    if (
+      (data.otherTopics && !postedBySelf) ||
+      (data.ownTopics && postedBySelf)
+    ) {
+      vote = topic.querySelector(".topic-voting-votes")!;
+    }
+
+    if (vote !== undefined) {
+      vote.classList.add("trx-votes-hidden");
+      vote.textContent = "-";
+      count++;
     }
   }