Fix #148

LiveTL · Feb 13, 2021 · 9372560 · 9372560
1 parent bdc1552
commit 9372560
Showing 1 changed file with 27 additions and 1 deletion.
diff --git a/LiveTL/js/lib/translator-mode.js b/LiveTL/js/lib/translator-mode.js
@@ -5,7 +5,7 @@ import { getStorage, setupDefaultTranslatorMode } from './storage.js';
 
 const TranslatorMode = (() => {
 
-  const [container, chatbox] = document.querySelectorAll('#input');
+  let [container, chatbox] = document.querySelectorAll('#input');
   const defaultt = false;
   let sendButton = document.querySelector('#send-button');
   const postMessage = window.parent.postMessage;
@@ -20,9 +20,35 @@ const TranslatorMode = (() => {
   async function run() {
     if (is_ytc()) {
       await ytc_run();
+      window.addEventListener('click', e => {
+        if (switchedChat(e)) {
+          rebindChatbox();
+        }
+      });
     }
   }
 
+  function switchedChat(e) {
+    e = e.target;
+    while (e.parentElement) {
+      if (e.tagName === 'A' && e.getAttribute('aria-selected')) {
+        return true;
+      }
+      e = e.parentElement;
+    }
+    return false;
+  }
+
+  function rebindChatbox() {
+    setTimeout(async () => {
+      chatbox.removeEventListener('keydown', checkAndAddTLTag);
+      sendButton.removeEventListener('mouseup', checkAndAddTLTag);
+      chatbox.removeEventListener('focus', checkAndAddTLTag);
+      [container, chatbox] = document.querySelectorAll('#input');
+      await ytc_run();
+    }, 1000);
+  }
+
   async function ytc_run() {
     if (chatbox) {
       chatbox.style.whiteSpace = `pre-wrap`;