From 9d969929de90c6ce80e5d78ee230159781cf1585 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Sat, 5 May 2018 00:29:02 -0400
Subject: [PATCH] more tray icon options

---
 jslib                                    |  2 +-
 src/app/accounts/settings.component.html | 10 +++
 src/app/accounts/settings.component.ts   | 14 +++-
 src/desktopConstants.ts                  |  3 -
 src/locales/en/messages.json             | 13 +++-
 src/main.ts                              | 19 ++++--
 src/main/messaging.main.ts               | 17 +++++
 src/main/tray.main.ts                    | 86 ++++++++++++++++++------
 8 files changed, 127 insertions(+), 37 deletions(-)
 delete mode 100644 src/desktopConstants.ts

diff --git a/jslib b/jslib
index 61e13dae..c6c5dd6d 160000
--- a/jslib
+++ b/jslib
@@ -1 +1 @@
-Subproject commit 61e13daefafe53c4c4b33fa9419052bfc2cfc9cd
+Subproject commit c6c5dd6d46526a36b65de3f5069e56d131843f7d
diff --git a/src/app/accounts/settings.component.html b/src/app/accounts/settings.component.html
index 6577a3d9..951140a4 100644
--- a/src/app/accounts/settings.component.html
+++ b/src/app/accounts/settings.component.html
@@ -32,6 +32,16 @@
                             </div>
                             <small class="help-block">{{'disableFaviconDesc' | i18n}}</small>
                         </div>
+                        <div class="form-group">
+                            <div class="checkbox">
+                                <label for="enableTray">
+                                    <input id="enableTray" type="checkbox" name="EnableTray"
+                                           [(ngModel)]="enableTray" (change)="saveTray()">
+                                    {{'enableTray' | i18n}}
+                                </label>
+                            </div>
+                            <small class="help-block">{{'enableTrayDesc' | i18n}}</small>
+                        </div>
                         <div class="form-group">
                             <div class="checkbox">
                                 <label for="enableMinToTray">
diff --git a/src/app/accounts/settings.component.ts b/src/app/accounts/settings.component.ts
index e4782c36..c77d8034 100644
--- a/src/app/accounts/settings.component.ts
+++ b/src/app/accounts/settings.component.ts
@@ -15,7 +15,7 @@ import { StorageService } from 'jslib/abstractions/storage.service';
 
 import { ConstantsService } from 'jslib/services/constants.service';
 
-import { DesktopConstants } from '../../desktopConstants';
+import { ElectronConstants } from 'jslib/electron/electronConstants';
 
 @Component({
     selector: 'app-settings',
@@ -26,6 +26,7 @@ export class SettingsComponent implements OnInit {
     disableGa: boolean = false;
     disableFavicons: boolean = false;
     enableMinToTray: boolean = false;
+    enableTray: boolean = false;
     locale: string;
     lockOptions: any[];
     localeOptions: any[];
@@ -58,7 +59,8 @@ export class SettingsComponent implements OnInit {
     async ngOnInit() {
         this.lockOption = await this.storageService.get<number>(ConstantsService.lockOptionKey);
         this.disableFavicons = await this.storageService.get<boolean>(ConstantsService.disableFaviconKey);
-        this.enableMinToTray = await this.storageService.get<boolean>(DesktopConstants.enableMinimizeToTrayKey);
+        this.enableMinToTray = await this.storageService.get<boolean>(ElectronConstants.enableMinimizeToTrayKey);
+        this.enableTray = await this.storageService.get<boolean>(ElectronConstants.enableTrayKey);
         this.locale = await this.storageService.get<string>(ConstantsService.localeKey);
 
         const disableGa = await this.storageService.get<boolean>(ConstantsService.disableGaKey);
@@ -88,10 +90,16 @@ export class SettingsComponent implements OnInit {
     }
 
     async saveMinToTray() {
-        await this.storageService.save(DesktopConstants.enableMinimizeToTrayKey, this.enableMinToTray);
+        await this.storageService.save(ElectronConstants.enableMinimizeToTrayKey, this.enableMinToTray);
         this.callAnalytics('MinimizeToTray', this.enableMinToTray);
     }
 
+    async saveTray() {
+        await this.storageService.save(ElectronConstants.enableTrayKey, this.enableTray);
+        this.callAnalytics('Tray', this.enableTray);
+        this.messagingService.send(this.enableTray ? 'showTray' : 'removeTray');
+    }
+
     async saveLocale() {
         await this.storageService.save(ConstantsService.localeKey, this.locale);
         this.analytics.eventTrack.next({ action: 'Set Locale ' + this.locale });
diff --git a/src/desktopConstants.ts b/src/desktopConstants.ts
deleted file mode 100644
index 5ccf3921..00000000
--- a/src/desktopConstants.ts
+++ /dev/null
@@ -1,3 +0,0 @@
-export class DesktopConstants {
-    static readonly enableMinimizeToTrayKey: string = 'enableMinimizeToTray';
-}
diff --git a/src/locales/en/messages.json b/src/locales/en/messages.json
index 4cb0c304..d8e0642f 100644
--- a/src/locales/en/messages.json
+++ b/src/locales/en/messages.json
@@ -771,10 +771,16 @@
     "message": "Website Icons provide a recognizable image next to each login item in your vault."
   },
   "enableMinToTray": {
-    "message": "Minimize to Tray"
+    "message": "Minimize to Tray Icon"
   },
   "enableMinToTrayDesc": {
-    "message": "When minimizing the window, run Bitwarden as a system tray icon."
+    "message": "When minimizing the window, show an icon in the system tray instead."
+  },
+  "enableTray": {
+    "message": "Enable Tray Icon"
+  },
+  "enableTrayDesc": {
+    "message": "Always show an icon in the system tray."
   },
   "language": {
     "message": "Language"
@@ -1037,5 +1043,8 @@
   },
   "default": {
     "message": "Default"
+  },
+  "exit": {
+    "message": "Exit"
   }
 }
diff --git a/src/main.ts b/src/main.ts
index dd21170c..adf41301 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -1,4 +1,8 @@
-import { app, BrowserWindow } from 'electron';
+import {
+    app,
+    BrowserWindow,
+    MenuItemConstructorOptions,
+} from 'electron';
 import * as path from 'path';
 
 import { I18nService } from './services/i18n.service';
@@ -17,8 +21,6 @@ import { ElectronStorageService } from 'jslib/electron/services/electronStorage.
 import { UpdaterMain } from 'jslib/electron/updater.main';
 import { WindowMain } from 'jslib/electron/window.main';
 
-import { DesktopConstants } from './desktopConstants';
-
 export class Main {
     logService: ElectronLogService;
     i18nService: I18nService;
@@ -76,9 +78,7 @@ export class Main {
         });
         this.menuMain = new MenuMain(this);
         this.powerMonitorMain = new PowerMonitorMain(this);
-        this.trayMain = new TrayMain(this.windowMain, 'Bitwarden', async () => {
-            return await this.storageService.get<boolean>(DesktopConstants.enableMinimizeToTrayKey);
-        });
+        this.trayMain = new TrayMain(this.windowMain, this.i18nService, this.storageService, 'Bitwarden');
 
         this.messagingService = new ElectronMainMessagingService(this.windowMain, (message) => {
             this.messagingMain.onMessage(message);
@@ -95,7 +95,12 @@ export class Main {
             this.messagingMain.init();
             this.menuMain.init();
             this.powerMonitorMain.init();
-            this.trayMain.init();
+            this.trayMain.init([{
+                label: this.i18nService.t('lockNow'),
+                enabled: false,
+                id: 'lockNow',
+                click: () => this.messagingService.send('lockVault'),
+            }]);
             await this.updaterMain.init();
         }, (e: any) => {
             // tslint:disable-next-line
diff --git a/src/main/messaging.main.ts b/src/main/messaging.main.ts
index ca757130..695fbcc5 100644
--- a/src/main/messaging.main.ts
+++ b/src/main/messaging.main.ts
@@ -21,6 +21,16 @@ export class MessagingMain {
                 break;
             case 'updateAppMenu':
                 this.main.menuMain.updateApplicationMenuState(message.isAuthenticated, message.isLocked);
+                this.updateTrayMenu(message.isAuthenticated, message.isLocked);
+                break;
+            case 'showTray':
+                this.main.trayMain.showTray();
+                break;
+            case 'removeTray':
+                this.main.trayMain.removeTray();
+                break;
+            case 'hideToTray':
+                this.main.trayMain.hideToTray();
                 break;
             default:
                 break;
@@ -42,4 +52,11 @@ export class MessagingMain {
             });
         }, SyncInterval);
     }
+
+    private updateTrayMenu(isAuthenticated: boolean, isLocked: boolean) {
+        const lockNowTrayMenuItem = this.main.trayMain.contextMenu.getMenuItemById('lockNow');
+        if (lockNowTrayMenuItem != null) {
+            lockNowTrayMenuItem.enabled = isAuthenticated && !isLocked;
+        }
+    }
 }
diff --git a/src/main/tray.main.ts b/src/main/tray.main.ts
index 83258991..adb16f86 100644
--- a/src/main/tray.main.ts
+++ b/src/main/tray.main.ts
@@ -1,21 +1,26 @@
 import {
     Menu,
+    MenuItem,
+    MenuItemConstructorOptions,
     nativeImage,
     Tray,
 } from 'electron';
 import * as path from 'path';
 
+import { I18nService } from 'jslib/abstractions/i18n.service';
+import { StorageService } from 'jslib/abstractions/storage.service';
+import { ElectronConstants } from 'jslib/electron/electronConstants';
 import { WindowMain } from 'jslib/electron/window.main';
 
-import { DesktopConstants } from '../desktopConstants';
-
 export class TrayMain {
+    contextMenu: Menu;
+
     private tray: Tray;
-    private menu: Menu;
     private icon: string | Electron.NativeImage;
     private pressedIcon: Electron.NativeImage;
 
-    constructor(private windowMain: WindowMain, private appName: string, private minToTray: () => Promise<boolean>) {
+    constructor(private windowMain: WindowMain, private i18nService: I18nService,
+        private storageService: StorageService, private appName: string) {
         if (process.platform === 'win32') {
             this.icon = path.join(__dirname, '/images/icon.ico');
         } else if (process.platform === 'darwin') {
@@ -28,52 +33,91 @@ export class TrayMain {
         }
     }
 
-    init() {
-        if (process.platform === 'linux') {
-            this.menu = Menu.buildFromTemplate([{
-                label: this.appName,
-                click: () => this.open(),
-            }]);
+    async init(additionalMenuItems: MenuItemConstructorOptions[] = null) {
+        const menuItemOptions: MenuItemConstructorOptions[] = [{
+            label: this.appName,
+            click: () => this.toggleWindow(),
+        },
+        { type: 'separator' },
+        {
+            label: this.i18nService.t('exit'),
+            click: () => this.closeWindow(),
+        }];
+
+        if (additionalMenuItems != null) {
+            menuItemOptions.splice(1, 0, ...additionalMenuItems);
+        }
+
+        this.contextMenu = Menu.buildFromTemplate(menuItemOptions);
+        if (await this.storageService.get<boolean>(ElectronConstants.enableTrayKey)) {
+            this.showTray();
         }
 
         this.windowMain.win.on('minimize', async (e: Event) => {
-            if (await this.minToTray()) {
+            if (await this.storageService.get<boolean>(ElectronConstants.enableMinimizeToTrayKey)) {
                 e.preventDefault();
-                await this.handleHideEvent();
+                await this.hideToTray();
             }
         });
 
         this.windowMain.win.on('show', async (e: Event) => {
-            await this.handleShowEvent();
+            const enableTray = await this.storageService.get<boolean>(ElectronConstants.enableTrayKey);
+            if (!enableTray) {
+                await this.removeTray(false);
+            }
         });
     }
 
-    private handleShowEvent() {
+    removeTray(showWindow = true) {
         if (this.tray != null) {
             this.tray.destroy();
             this.tray = null;
         }
+
+        if (this.windowMain.win != null && !this.windowMain.win.isVisible()) {
+            this.windowMain.win.show();
+        }
     }
 
-    private handleHideEvent() {
+    hideToTray() {
+        this.showTray();
+        if (this.windowMain.win != null) {
+            this.windowMain.win.hide();
+        }
+    }
+
+    showTray() {
+        if (this.tray != null) {
+            return;
+        }
+
         this.tray = new Tray(this.icon);
         this.tray.setToolTip(this.appName);
+        this.tray.on('click', () => this.toggleWindow());
+
         if (this.pressedIcon != null) {
             this.tray.setPressedImage(this.pressedIcon);
         }
-        if (this.menu != null) {
-            this.tray.setContextMenu(this.menu);
+        if (this.contextMenu != null) {
+            this.tray.setContextMenu(this.contextMenu);
         }
-
-        this.tray.on('click', () => open());
-        this.windowMain.win.hide();
     }
 
-    private open() {
+    private toggleWindow() {
+        if (this.windowMain.win == null) {
+            return;
+        }
+
         if (this.windowMain.win.isVisible()) {
             this.windowMain.win.hide();
         } else {
             this.windowMain.win.show();
         }
     }
+
+    private closeWindow() {
+        if (this.windowMain.win != null) {
+            this.windowMain.win.close();
+        }
+    }
 }