From adde4c51c006ad2547463505cf519b1e60c5efb8 Mon Sep 17 00:00:00 2001
From: Simon <simobilleter@gmail.com>
Date: Fri, 25 Aug 2023 19:28:42 +0700
Subject: [PATCH] check channel subscribed

---
 extension/background.js | 10 ++++++++-
 extension/script.js     | 45 +++++++++++++++++++++++++++++++++--------
 2 files changed, 46 insertions(+), 9 deletions(-)

diff --git a/extension/background.js b/extension/background.js
index 80eefe6..f649b25 100644
--- a/extension/background.js
+++ b/extension/background.js
@@ -138,10 +138,15 @@ async function subscribe(url) {
 async function videoExists(id) {
   const path = `api/video/${id}/`;
   let response = await sendGet(path);
-  console.log(response);
   return Boolean(response.data);
 }
 
+async function getChannel(channelHandle) {
+  const path = `api/channel/search/?q=${channelHandle}`;
+  let response = await sendGet(path);
+  return response.data;
+}
+
 async function cookieStr(cookieLines) {
   const path = 'api/cookie/';
   let payload = {
@@ -226,6 +231,9 @@ function handleMessage(request, sender, sendResponse) {
       case 'videoExists': {
         return await videoExists(request.videoId);
       }
+      case 'getChannel': {
+        return await getChannel(request.channelHandle);
+      }
       default: {
         let err = new Error(`unknown message type ${JSON.stringify(request.type)}`);
         console.log(err);
diff --git a/extension/script.js b/extension/script.js
index 6d4e4df..6d8f846 100644
--- a/extension/script.js
+++ b/extension/script.js
@@ -106,8 +106,20 @@ function getBrowser() {
 }
 
 function getChannelContainers() {
-  let nodes = document.querySelectorAll('#inner-header-container, #owner');
-  return nodes;
+  const elements = document.querySelectorAll('#inner-header-container, #owner');
+  const channelContainerNodes = [];
+
+  elements.forEach(element => {
+    if (isElementVisible(element)) {
+      channelContainerNodes.push(element);
+    }
+  });
+
+  return channelContainerNodes;
+}
+
+function isElementVisible(element) {
+  return element.offsetWidth > 0 || element.offsetHeight > 0 || element.getClientRects().length > 0;
 }
 
 function ensureTALinks() {
@@ -201,7 +213,7 @@ function buildChannelButtonDiv() {
 
 function buildChannelSubButton(channelHandle) {
   let channelSubButton = document.createElement('span');
-  channelSubButton.innerText = 'Subscribe';
+  channelSubButton.innerText = 'Checking...';
   channelSubButton.title = `TA Subscribe: ${channelHandle}`;
   channelSubButton.setAttribute('data-id', channelHandle);
   channelSubButton.setAttribute('data-type', 'channel');
@@ -211,19 +223,36 @@ function buildChannelSubButton(channelHandle) {
     console.log(`subscribe to: ${channelHandle}`);
     sendUrl(channelHandle, 'subscribe', channelSubButton);
   });
-  channelSubButton.addEventListener('mouseover', () => {
-    checkChannelSubscribed(channelHandle);
-  });
   Object.assign(channelSubButton.style, {
     padding: '5px',
     cursor: 'pointer',
   });
+  checkChannelSubscribed(channelSubButton);
 
   return channelSubButton;
 }
 
-function checkChannelSubscribed(channelHandle) {
-  console.log(`check channel subscribed: ${channelHandle}`);
+function checkChannelSubscribed(channelSubButton) {
+  function handleResponse(message) {
+    if (
+      message === false ||
+      (typeof message === 'object' && message.channel_subscribed === false)
+    ) {
+      channelSubButton.innerText = 'Subscribe';
+    } else if (typeof message === 'object' && message.channel_subscribed === true) {
+      channelSubButton.innerText = 'Unsubscribe';
+    } else {
+      console.log('Unknown state');
+    }
+  }
+  function handleError() {
+    console.log('error');
+  }
+
+  let channelHandle = channelSubButton.dataset.id;
+  let message = { type: 'getChannel', channelHandle };
+  let sending = sendMessage(message);
+  sending.then(handleResponse, handleError);
 }
 
 function buildSpacer() {