Added badges that show the amount of summaries found for every domain (resolve issue #10)

Author: joshwallerr

background.js

@@ -198,8 +198,6 @@ chrome.runtime.onMessage.addListener(
             }
           });
         });
-    } else if (request.showForm) {
-      // console.log("Received message to show form");
     } else if (request.type === "showPreloader") {
       console.log("RECIEVED showPreloader message for" + request.summaryName + " on " + request.domain);
       // Add the link to loadingSummaries
@@ -510,4 +508,82 @@ function rootDomain(hostname) {
       return parts.join('.');
 
   return parts.slice(-2).join('.');
-}
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+chrome.tabs.onActivated.addListener(activeInfo => {
+  updateBadgeOnTab(activeInfo.tabId);
+});
+
+chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
+  if (changeInfo.status === 'complete' && tab.active) {
+    updateBadgeOnTab(tabId);
+  }
+});
+
+function updateBadgeOnTab(tabId) {
+  chrome.tabs.get(tabId, (tab) => {
+    if (chrome.runtime.lastError || !tab || !tab.url) {
+      console.log(chrome.runtime.lastError ? chrome.runtime.lastError.message : 'No tab or URL found');
+      return;
+    }
+    const url = new URL(tab.url);
+    const domain = rootDomain(url.hostname);
+    checkAndFetchSummaryCount(domain, tabId);
+  });
+}
+
+function checkAndFetchSummaryCount(domain, tabId) {
+  chrome.storage.local.get([domain], function(result) {
+    if (result[domain] && new Date().getTime() - result[domain].timestamp < 86400000) { // 86400000 ms = 1 day
+      // Data is still valid
+      chrome.action.setBadgeText({text: result[domain].count.toString(), tabId: tabId});
+      chrome.action.setBadgeBackgroundColor({color: '#22c55e', tabId: tabId});
+    } else {
+      // Data is outdated or not present, fetch new data
+      fetchSummaryCount(domain, tabId);
+    }
+  });
+}
+
+function fetchSummaryCount(domain, tabId) {
+  fetch('https://docdecoder.app/getsumcount', {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json'
+    },
+    body: JSON.stringify({ domain: domain })
+  })
+  .then(response => response.json())
+  .then(data => {
+    if (data && data.summariesCount) {
+      chrome.storage.local.set({[domain]: {count: data.summariesCount, timestamp: new Date().getTime()}});
+      chrome.action.setBadgeText({text: data.summariesCount.toString(), tabId: tabId});
+      chrome.action.setBadgeBackgroundColor({color: '#22c55e', tabId: tabId});
+    } else {
+      chrome.action.setBadgeText({text: '', tabId: tabId});
+    }
+  })
+  .catch(error => console.error('Error fetching summary count:', error));
+}

content.js

@@ -159,4 +159,37 @@ function rootDomain(hostname) {
 
 
 
-// works now, but only one preloader is showing at a time. Not huge deal, but should be fixed eventually
+// works now, but only one preloader is showing at a time. Not huge deal, but should be fixed eventually
+
+
+
+
+
+
+
+
+// Add a route that just takes the domain and returns the summary count for that domain. Then display a badge on the extension icon with the number of summaries for that domain.
+
+// call to docdecoder.app/getsumcount with post and send the domain as the json body
+// response will be the number of summaries for that domain, as count
+// set the badge text to count
+
+// const currentDomain = rootDomain(new URL(window.location.href).hostname);
+// const summaryCountRequestData = {
+//   action: "getSummaryCount",
+//   domain: currentDomain
+// };
+// fetch('https://docdecoder.app/getsumcount', {
+//   method: 'POST',
+//   headers: {
+//     'Content-Type': 'application/json'
+//   },
+//   body: JSON.stringify(summaryCountRequestData)
+// })
+// .then(response => response.json())
+// .then(data => {
+//   console.log(data.summariesCount);
+//   if (data.summariesCount > 0) {
+//     chrome.runtime.sendMessage({badgeText: data.summariesCount.toString()});
+//   }
+// });

manifest.json

@@ -2,7 +2,7 @@
   "manifest_version": 3,
   "name": "DocDecoder - Summarize Web Policies With AI",
   "description": "DocDecoder uses GPT-4 to generate clear, concise summaries of any site's legal policies for you to skim over before you accept them.",
-  "version": "1.2.6",
+  "version": "1.2.7",
   "action": {
     "default_popup": "popup.html",
     "default_icon": {
@@ -19,7 +19,7 @@
   "background": {
     "service_worker": "background.js"
   },
-  "permissions": ["activeTab", "storage", "notifications", "tabs"],
+  "permissions": ["activeTab", "storage", "notifications", "tabs", "action"],
   "content_scripts": [
     {
       "matches": ["<all_urls>"],