Add badge with link count

There is a badge on the linkdump button to show how many links are in the dump.
This way, there is no need to open it to check if it's empty or not.

See #32

_locales/en/messages.json

@@ -11,10 +11,6 @@
     "message": "Download complete",
     "description": "Notify the user that the download is complete."
   },
-  "notificationStorageCleared": {
-    "message": "Storage cleared",
-    "description": "Notify the user that the storage has been cleared."
-  },
   "notificationStorageCopied": {
     "message": "Storage copied",
     "description": "Notify the user that the storage has been copied to the clipboard."

_locales/fr/messages.json

@@ -11,10 +11,6 @@
     "message": "Téléchargement terminé",
     "description": "Notify the user that the download is complete."
   },
-  "notificationStorageCleared": {
-    "message": "Entrepôt nettoyé",
-    "description": "Notify the user that the storage has been cleared."
-  },
   "notificationStorageCopied": {
     "message": "Entrepôt copié",
     "description": "Notify the user that the storage has been copied to the clipboard."

background.js

@@ -17,6 +17,22 @@ function notification(message) {
       });
 }
 
+async function updateBadge(items) {
+  let count = items;
+
+  if (items === undefined) {
+    const storage = await browser.storage.local.get('urls');
+    const urls = storage.urls || [];
+    count = urls.length || '';
+  } else if (items === 0) {
+    count = '';
+  }
+
+  browser.browserAction.setBadgeText({text: count.toString()});
+  browser.browserAction.setBadgeTextColor({color: 'white'});
+  browser.browserAction.setBadgeBackgroundColor({color: '#007bc5'});
+}
+
 async function addLink(link) {
   const storage = await browser.storage.local.get('urls');
   let urls = storage.urls || [];
@@ -36,7 +52,8 @@ async function addLink(link) {
     }
   })
 
-  await browser.storage.local.set({ "urls": urls});
+  await browser.storage.local.set({ urls });
+  await updateBadge(urls.length);
 }
 
 function getLinks(bookmark, initialLinks = []) {
@@ -142,31 +159,31 @@ function download(downloadOptions) {
   });
 }
 
-function deleteLink(link) {
-  browser.storage.local.get('urls').then(obj => {
-    if (!obj.urls) return;
-    let { urls } = obj;
+async function deleteLink(link) {
+  const obj = await browser.storage.local.get('urls');
+  if (!obj.urls) return;
+  let { urls } = obj;
 
-    urls = urls.filter((item) => item.url !== link.url && item.title !== link.title);
+  urls = urls.filter((item) => item.url !== link.url && item.title !== link.title);
 
-    browser.storage.local.set({ urls });
-  });
+  await browser.storage.local.set({ urls });
+  await updateBadge(urls.length);
 }
 
-function clear() {
-  browser.storage.local.remove('urls').then(
-    notification('notificationStorageCleared')
-  );
+async function clear() {
+  await browser.storage.local.remove('urls');
+  await updateBadge(0);
 }
 
 function handleChanged(delta) {
   if (delta.id !== downloadId) {
     return;
   }
   if (delta.state && delta.state.current === 'complete') {
-    browser.storage.local.get('options').then(obj => {
+    browser.storage.local.get('options').then(async obj => {
       if (obj.options !== undefined && obj.options.clear !== undefined && obj.options.clear.download) {
-        browser.storage.local.remove('urls');
+        await browser.storage.local.remove('urls');
+        await updateBadge(0);
       }
     }).then(notification('notificationDownloadComplete'));
   }
@@ -185,9 +202,10 @@ function handleMessage(message) {
       break;
     }
     case 'copied': {
-      browser.storage.local.get('options').then(obj => {
+      browser.storage.local.get('options').then(async obj => {
         if (obj.options !== undefined && obj.options.clear !== undefined && obj.options.clear.copy) {
-          browser.storage.local.remove('urls');
+          await browser.storage.local.remove('urls');
+          await updateBadge(0);
         }
       }).then(notification('notificationStorageCopied'));
       break;
@@ -239,3 +257,5 @@ browser.pageAction.onClicked.addListener(tab => {
 browser.downloads.onChanged.addListener(handleChanged);
 
 browser.runtime.onMessage.addListener(handleMessage);
+
+updateBadge();

manifest.json

@@ -7,7 +7,7 @@
 
   "manifest_version": 2,
   "name": "LinkDump",
-  "version": "1.14",
+  "version": "1.15",
   "description": "__MSG_extensionDescription__",
   "icons": {
     "48": "icons/linkdump-48.png"

package.json

@@ -1,6 +1,6 @@
 {
   "name": "linkdump",
-  "version": "1.14",
+  "version": "1.15",
   "description": "Store links and dump them",
   "main": "background.js",
   "scripts": {