From 9046b858b1e4608af4c01bc4d69e1a49d4009c07 Mon Sep 17 00:00:00 2001
From: Giulio Ganci <giulioganci@gmail.com>
Date: Sat, 16 Oct 2021 17:05:26 +0200
Subject: [PATCH] feat(UI): ctrl|cmd+t, ctrl|cmd+w shortcut to open/close
 workspace tabs

---
 src/main/index.js                     | 57 ++++++++++++++++++---------
 src/renderer/components/Workspace.vue | 24 +++++++++++
 2 files changed, 62 insertions(+), 19 deletions(-)

diff --git a/src/main/index.js b/src/main/index.js
index 5f1a805d..753a3122 100644
--- a/src/main/index.js
+++ b/src/main/index.js
@@ -41,18 +41,18 @@ async function createMainWindow () {
    remoteMain.enable(window.webContents);
 
    try {
-      if (isDevelopment) { //
+      if (isDevelopment) {
+      //
          await window.loadURL(`http://localhost:${process.env.ELECTRON_WEBPACK_WDS_PORT}`);
 
          // const { default: installExtension, VUEJS3_DEVTOOLS } = require('electron-devtools-installer');
 
-         // const oldDevToolsID = session.defaultSession.getAllExtensions().find(ext => ext.name === 'Vue.js devtools').id;
-         // session.defaultSession.removeExtension(oldDevToolsID);
-         // const toolName = await installExtension(VUEJS3_DEVTOOLS);
-         // console.log(toolName, 'installed');
+      // const oldDevToolsID = session.defaultSession.getAllExtensions().find(ext => ext.name === 'Vue.js devtools').id;
+      // session.defaultSession.removeExtension(oldDevToolsID);
+      // const toolName = await installExtension(VUEJS3_DEVTOOLS);
+      // console.log(toolName, 'installed');
       }
-      else
-         await window.loadURL(new URL(`file:///${path.join(__dirname, 'index.html')}`).href);
+      else await window.loadURL(new URL(`file:///${path.join(__dirname, 'index.html')}`).href);
    }
    catch (err) {
       console.log(err);
@@ -70,10 +70,9 @@ async function createMainWindow () {
    });
 
    return window;
-};
+}
 
-if (!gotTheLock)
-   app.quit();
+if (!gotTheLock) app.quit();
 else {
    require('@electron/remote/main').initialize();
 
@@ -83,33 +82,53 @@ else {
    // quit application when all windows are closed
    app.on('window-all-closed', () => {
       // on macOS it is common for applications to stay open until the user explicitly quits
-      if (process.platform !== 'darwin')
-         app.quit();
+      if (process.platform !== 'darwin') app.quit();
    });
 
    app.on('activate', async () => {
       // on macOS it is common to re-create a window even after all windows have been closed
       if (mainWindow === null) {
          mainWindow = await createMainWindow();
-         if (isDevelopment)
-            mainWindow.webContents.openDevTools();
+         if (isDevelopment) mainWindow.webContents.openDevTools();
       }
    });
 
    // create main BrowserWindow when electron is ready
    app.on('ready', async () => {
       mainWindow = await createMainWindow();
-      Menu.setApplicationMenu(null);
+      createAppMenu();
 
-      if (isDevelopment)
-         mainWindow.webContents.openDevTools();
+      if (isDevelopment) mainWindow.webContents.openDevTools();
 
-      process.on('uncaughtException', error => {
+      process.on('uncaughtException', (error) => {
          mainWindow.webContents.send('unhandled-exception', error);
       });
 
-      process.on('unhandledRejection', error => {
+      process.on('unhandledRejection', (error) => {
          mainWindow.webContents.send('unhandled-exception', error);
       });
    });
 }
+
+function createAppMenu () {
+   let menu = null;
+
+   if (process.platform === 'darwin') {
+      menu = Menu.buildFromTemplate([
+         {
+            label: app.name,
+            submenu: [
+               {
+                  role: 'about'
+               },
+               { type: 'separator' },
+               {
+                  role: 'quit'
+               }
+            ]
+         }
+      ]);
+   }
+
+   Menu.setApplicationMenu(menu);
+}
diff --git a/src/renderer/components/Workspace.vue b/src/renderer/components/Workspace.vue
index bd437c56..da4869ad 100644
--- a/src/renderer/components/Workspace.vue
+++ b/src/renderer/components/Workspace.vue
@@ -586,11 +586,16 @@ export default {
       }
    },
    async created () {
+      window.addEventListener('keydown', this.onKey);
       await this.addWorkspace(this.connection.uid);
       const isInitiated = await Connection.checkConnection(this.connection.uid);
       if (isInitiated)
          this.connectWorkspace(this.connection);
    },
+   beforeDestroy () {
+      window.removeEventListener('keydown', this.onKey);
+   },
+
    methods: {
       ...mapActions({
          addWorkspace: 'workspaces/addWorkspace',
@@ -604,6 +609,25 @@ export default {
       addQueryTab () {
          this.newTab({ uid: this.connection.uid, type: 'query' });
       },
+      getSelectedTab () {
+         return this.workspace.tabs.find(tab => tab.uid === this.selectedTab);
+      },
+      onKey (e) {
+         e.stopPropagation();
+
+         if (!this.isSelected)
+            return;
+
+         if ((e.ctrlKey || e.metaKey) && e.keyCode === 84) { // CTRL|Command + t
+            this.addQueryTab();
+         }
+
+         if ((e.ctrlKey || e.metaKey) && e.keyCode === 87) { // CTRL|Command + w
+            const currentTab = this.getSelectedTab();
+            if (currentTab)
+               this.closeTab(currentTab);
+         }
+      },
       openAsPermanentTab (tab) {
          const permanentTabs = {
             table: 'data',