disable button while working

mrjleo · Dec 28, 2019 · b9003a3 · b9003a3
1 parent 3981572
commit b9003a3
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/chrome-extension/src/background.js b/chrome-extension/src/background.js
@@ -57,4 +57,5 @@ async function classify(documentRepresentation, tab) {
     const inputs = getInputs(documentRepresentation, words, tags);
     const predictions = await model.predict(inputs).data();
     chrome.tabs.sendMessage(tab.id, {text: 'visualize', predictions: predictions});
+    chrome.runtime.sendMessage({text: 'enableButton'});
 }
diff --git a/chrome-extension/src/popup.html b/chrome-extension/src/popup.html
@@ -10,7 +10,7 @@
         </style>
     </head>
     <body>
-        <button id="extractContent">Extract content</button>
+        <button id="highlightContent">Highlight content</button>
         <script src="popup.js"></script>
     </body>
 </html>
diff --git a/chrome-extension/src/popup.js b/chrome-extension/src/popup.js
@@ -1,7 +1,18 @@
-const extractContentButton = document.getElementById('extractContent');
+const highlightContentButton = document.getElementById('highlightContent');
 
-extractContentButton.onclick = function(_element) {
-    chrome.tabs.query({active: true, currentWindow: true}, function(tabs){
-        chrome.tabs.sendMessage(tabs[0].id, {text: "sendDocumentRepresentation"});  
+highlightContentButton.onclick = function(_element) {
+    chrome.tabs.query({active: true, currentWindow: true}, function(tabs) {
+        chrome.tabs.sendMessage(tabs[0].id, {text: "sendDocumentRepresentation"});
+
+        // disable button
+        highlightContentButton.textContent = "Working...";
+        highlightContentButton.disabled = true;
     });
 };
+
+chrome.runtime.onMessage.addListener(function (msg, sender, _sendResponse) {
+    if (msg.text === 'enableButton') {
+        highlightContentButton.textContent = "Highlight content";
+        highlightContentButton.disabled = false;
+    }
+});