feat(extensions): simplify context menu setup if only 1 domain is con…

…figured
edmundhung · Aug 13, 2021 · b6308be · b6308be
1 parent 39a8f56
commit b6308be
Showing 1 changed file with 24 additions and 9 deletions.
diff --git a/packages/extensions/src/background.ts b/packages/extensions/src/background.ts
@@ -167,18 +167,33 @@ async function updateContextMenu(
 ): Promise<void> {
   await browser.contextMenus.removeAll();
 
-  browser.contextMenus.create({
-    id: 'maildog',
-    title: 'maildog',
-    contexts: ['all'],
-  });
+  const list = Object.entries(emailsByDomain);
+
+  if (list.length === 0) {
+    return;
+  }
 
-  for (const [domain, emails] of Object.entries(emailsByDomain)) {
+  if (list.length > 1) {
     browser.contextMenus.create({
-      id: `maildog-${domain}`,
-      parentId: 'maildog',
-      title: domain,
+      id: 'maildog',
+      title: 'maildog',
+      contexts: ['all'],
     });
+  }
+
+  for (const [domain, emails] of list) {
+    if (list.length > 1) {
+      browser.contextMenus.create({
+        id: `maildog-${domain}`,
+        parentId: 'maildog',
+        title: domain,
+      });
+    } else {
+      browser.contextMenus.create({
+        id: `maildog-${domain}`,
+        title: `maildog (${domain})`,
+      });
+    }
 
     for (const email of emails) {
       browser.contextMenus.create({