From a04d2299d439b38684a95663878585ae2b0b9ae2 Mon Sep 17 00:00:00 2001
From: Aaron Yarborough <me@aaronjy.me>
Date: Wed, 1 Apr 2020 19:31:55 +0100
Subject: [PATCH] Hide stats in big tweets

---
 content.js |  7 +------
 hider.js   | 49 ++++++++++++++++++++++++++++++++++++-------------
 2 files changed, 37 insertions(+), 19 deletions(-)

diff --git a/content.js b/content.js
index 08912cc..ae3dc28 100644
--- a/content.js
+++ b/content.js
@@ -8,16 +8,13 @@ class Background {
 	}
 
 	static async listen() {
-		console.log("LISTENING")
-		
 		const preferences = await this.getSavedPreferences();
 		const hider = new TwitterHider(preferences);
 
 		hider.init();
 	}
 
-	static checkIfFirstStart(callback) {
-		console.log("CHECKING IF FIRST START");
+	static checkIfFirstStart() {
 		return new Promise(resolve => {
 			chrome.storage.local.get('setup', data => {
 				resolve(!data.setup);
@@ -26,8 +23,6 @@ class Background {
 	}
 
 	static setup() {
-		console.log("SETUP");
-
 		const defaults = {
 			hideLikes: true,
 			hideRetweets: true,
diff --git a/hider.js b/hider.js
index f477e80..80ae06c 100644
--- a/hider.js
+++ b/hider.js
@@ -1,12 +1,9 @@
-console.log("LOADED HIDER");
-
-class TwitterHider
-{
-	notProcessedAttribute = 'data-twitterhider-processed';
-	notProcessedSelector = `:not([${this.notProcessedAttribute}])`
+class TwitterHider {
+	processedAttribute = 'data-twitterhider-processed';
+	notProcessedSelector = `:not([${this.processedAttribute}])`;
 
 	selectors = {
-		countText: '.css-901oao.css-16my406.r-1qd0xha.r-ad9z0x.r-bcqeeo.r-qvutc0',
+		countText: `span.css-901oao.css-16my406.r-1qd0xha.r-ad9z0x.r-bcqeeo.r-qvutc0${this.notProcessedSelector}`,
 		replyContainer: `div[data-testid="reply"]${this.notProcessedSelector}`,
 		retweetContainer: `div[data-testid="retweet"]${this.notProcessedSelector},div[data-testid="unretweet"]${this.notProcessedSelector}`,
 		likeContainer: `div[data-testid="like"]${this.notProcessedSelector},div[data-testid="unlike"]${this.notProcessedSelector}`
@@ -19,8 +16,6 @@ class TwitterHider
 	}
 
 	init() {
-		console.table(this.preferences);
-
 		const hideLikes = this.getPreference("hideLikes");
 		const hideRetweets = this.getPreference("hideRetweets");
 		const hideReplies = this.getPreference("hideReplies");
@@ -39,12 +34,20 @@ class TwitterHider
 	}
 
 	hideLikeCounts() {
-		const elements = document.querySelectorAll(this.selectors.likeContainer);
+		const elements = [
+			...document.querySelectorAll(this.selectors.likeContainer),
+			...this.findMainTweetCountByLabelText("Likes")
+		];
+
 		this.omitElements(elements);
 	}
 
 	hideRetweetCounts() {
-		const elements = document.querySelectorAll(this.selectors.retweetContainer);
+		const elements = [
+			...document.querySelectorAll(this.selectors.retweetContainer),
+			...this.findMainTweetCountByLabelText("Retweets")
+		];
+
 		this.omitElements(elements);
 	}
 
@@ -55,17 +58,37 @@ class TwitterHider
 
 	omitElements(elements) {
 		elements.forEach((element) => {
-			element.setAttribute(this.notProcessedAttribute, true);
+			this.markElementAsProcessed(element);
+
 			const textElement = element.querySelector(this.selectors.countText);
 
 			if (textElement !== null)
-				textElement.innerHTML = "?";
+				this.omitElement(textElement);
 		});
 	}
 
+	omitElement(element) {
+		element.innerHTML = "???";
+	}
+
+	markElementAsProcessed(element) {
+		element.setAttribute(this.processedAttribute, true);
+	}
+
 	getPreference(preference) {
 		if (this.preferences[preference] === undefined)
 			throw `preference ${preference} wasn't given in the constructor.`;
 		return this.preferences[preference];
 	}
+
+	findMainTweetCountByLabelText(labelText) {
+		return [...document.querySelectorAll(this.selectors.countText)]
+			.filter(element => element.innerHTML === labelText)
+			.map(element => {
+				const wrappingDiv = element.parentElement.parentElement.parentElement;
+				const countContainer = wrappingDiv.children[0];
+				
+				return countContainer;
+			});
+	}
 }
\ No newline at end of file