From 27769f204f731d20c7ba2f838c02b7c2f28fa0c3 Mon Sep 17 00:00:00 2001
From: Fabio Di Stasio <fabio286@gmail.com>
Date: Fri, 20 Nov 2020 17:24:02 +0100
Subject: [PATCH] feat: display all keys in properties tab

---
 src/main/ipc-handlers/tables.js               | 11 +++++++++
 src/main/libs/clients/MySQLClient.js          | 24 +++++++++++++++++++
 src/renderer/components/WorkspacePropsTab.vue | 19 ++++++++++++++-
 .../components/WorkspacePropsTable.vue        |  5 ++++
 .../components/WorkspacePropsTableRow.vue     | 18 ++++++++------
 src/renderer/ipc-api/Tables.js                |  4 ++++
 src/renderer/scss/_table-keys.scss            | 13 +++++++---
 7 files changed, 83 insertions(+), 11 deletions(-)

diff --git a/src/main/ipc-handlers/tables.js b/src/main/ipc-handlers/tables.js
index a153452a..3e97a200 100644
--- a/src/main/ipc-handlers/tables.js
+++ b/src/main/ipc-handlers/tables.js
@@ -30,6 +30,17 @@ export default (connections) => {
       }
    });
 
+   ipcMain.handle('get-table-indexes', async (event, params) => {
+      try {
+         const result = await connections[params.uid].getTableIndexes(params);
+
+         return { status: 'success', response: result };
+      }
+      catch (err) {
+         return { status: 'error', response: err.toString() };
+      }
+   });
+
    ipcMain.handle('get-key-usage', async (event, params) => {
       try {
          const result = await connections[params.uid].getKeyUsage(params);
diff --git a/src/main/libs/clients/MySQLClient.js b/src/main/libs/clients/MySQLClient.js
index 59a387fc..64b6b9fb 100644
--- a/src/main/libs/clients/MySQLClient.js
+++ b/src/main/libs/clients/MySQLClient.js
@@ -198,6 +198,30 @@ export class MySQLClient extends AntaresCore {
       });
    }
 
+   /**
+    * @param {Object} params
+    * @param {String} params.schema
+    * @param {String} params.table
+    * @returns {Object} table indexes
+    * @memberof MySQLClient
+    */
+   async getTableIndexes ({ schema, table }) {
+      const { rows } = await this.raw(`SHOW INDEXES FROM \`${table}\` FROM \`${schema}\``);
+
+      return rows.map(row => {
+         return {
+            unique: !row.Non_unique,
+            name: row.Key_name,
+            column: row.Column_name,
+            indexType: row.Index_type,
+            type: row.Key_name === 'PRIMARY' ? 'PRIMARY' : !row.Non_unique ? 'UNIQUE' : row.Index_type === 'FULLTEXT' ? 'FULLTEXT' : 'INDEX',
+            cardinality: row.Cardinality,
+            comment: row.Comment,
+            indexComment: row.Index_comment
+         };
+      });
+   }
+
    /**
     * @param {Object} params
     * @param {String} params.schema
diff --git a/src/renderer/components/WorkspacePropsTab.vue b/src/renderer/components/WorkspacePropsTab.vue
index 39af2933..cc0d79c1 100644
--- a/src/renderer/components/WorkspacePropsTab.vue
+++ b/src/renderer/components/WorkspacePropsTab.vue
@@ -32,7 +32,7 @@
                   <span>{{ $t('word.add') }}</span>
                   <i class="mdi mdi-24px mdi-playlist-plus ml-1" />
                </button>
-               <button class="btn btn-dark btn-sm d-none">
+               <button class="btn btn-dark btn-sm">
                   <span>{{ $t('word.indexes') }}</span>
                   <i class="mdi mdi-24px mdi-key mdi-rotate-45 ml-1" />
                </button>
@@ -52,6 +52,7 @@
             v-if="localFields"
             ref="queryTable"
             :fields="localFields"
+            :indexes="localIndexes"
             :tab-uid="tabUid"
             :conn-uid="connection.uid"
             :table="table"
@@ -178,6 +179,8 @@ export default {
          localFields: [],
          originalKeyUsage: [],
          localKeyUsage: [],
+         originalIndexes: [],
+         localIndexes: [],
          localOptions: [],
          lastTable: null
       };
@@ -257,6 +260,20 @@ export default {
             this.addNotification({ status: 'error', message: err.stack });
          }
 
+         try { // Indexes
+            const { status, response } = await Tables.getTableIndexes(params);
+
+            if (status === 'success') {
+               this.originalIndexes = response;
+               this.localIndexes = JSON.parse(JSON.stringify(response));
+            }
+            else
+               this.addNotification({ status: 'error', message: response });
+         }
+         catch (err) {
+            this.addNotification({ status: 'error', message: err.stack });
+         }
+
          try { // Key usage (foreign keys)
             const { status, response } = await Tables.getKeyUsage(params);
 
diff --git a/src/renderer/components/WorkspacePropsTable.vue b/src/renderer/components/WorkspacePropsTable.vue
index 434417a3..e8505805 100644
--- a/src/renderer/components/WorkspacePropsTable.vue
+++ b/src/renderer/components/WorkspacePropsTable.vue
@@ -99,6 +99,7 @@
                v-for="row in fields"
                :key="row._id"
                :row="row"
+               :indexes="getIndexes(row.name)"
                :data-types="dataTypes"
                @contextmenu="contextMenu"
             />
@@ -122,6 +123,7 @@ export default {
    },
    props: {
       fields: Array,
+      indexes: Array,
       tabUid: [String, Number],
       connUid: String,
       table: String,
@@ -195,6 +197,9 @@ export default {
       },
       removeField () {
          this.$emit('remove-field', this.selectedField);
+      },
+      getIndexes (field) {
+         return this.indexes.filter(index => index.column === field);
       }
    }
 };
diff --git a/src/renderer/components/WorkspacePropsTableRow.vue b/src/renderer/components/WorkspacePropsTableRow.vue
index 23338e71..c2ecfdff 100644
--- a/src/renderer/components/WorkspacePropsTableRow.vue
+++ b/src/renderer/components/WorkspacePropsTableRow.vue
@@ -7,12 +7,15 @@
          </div>
       </div>
       <div class="td" tabindex="0">
-         <i
-            v-if="localRow.key"
-            :title="keyName(localRow.key)"
-            class="mdi mdi-key column-key c-help pl-1"
-            :class="`key-${localRow.key}`"
-         />
+         <div class="text-center">
+            <i
+               v-for="index in indexes"
+               :key="index.name"
+               :title="index.type"
+               class="d-inline-block mdi mdi-key column-key c-help"
+               :class="`key-${index.type}`"
+            />
+         </div>
       </div>
       <div class="td">
          <span
@@ -279,7 +282,8 @@ export default {
    },
    props: {
       row: Object,
-      dataTypes: Array
+      dataTypes: Array,
+      indexes: Array
    },
    data () {
       return {
diff --git a/src/renderer/ipc-api/Tables.js b/src/renderer/ipc-api/Tables.js
index f2141bac..fe3eb5b0 100644
--- a/src/renderer/ipc-api/Tables.js
+++ b/src/renderer/ipc-api/Tables.js
@@ -10,6 +10,10 @@ export default class {
       return ipcRenderer.invoke('get-table-data', params);
    }
 
+   static getTableIndexes (params) {
+      return ipcRenderer.invoke('get-table-indexes', params);
+   }
+
    static getKeyUsage (params) {
       return ipcRenderer.invoke('get-key-usage', params);
    }
diff --git a/src/renderer/scss/_table-keys.scss b/src/renderer/scss/_table-keys.scss
index 6bd88dc2..ab72b5f4 100644
--- a/src/renderer/scss/_table-keys.scss
+++ b/src/renderer/scss/_table-keys.scss
@@ -4,15 +4,22 @@
   line-height: 1.5;
   margin-right: 0.2rem;
 
-  &.key-pri {
+  &.key-pri,
+  &.key-PRIMARY {
     color: goldenrod;
   }
 
-  &.key-uni {
+  &.key-uni,
+  &.key-UNIQUE {
     color: deepskyblue;
   }
 
-  &.key-mul {
+  &.key-mul,
+  &.key-INDEX {
     color: palegreen;
   }
+
+  &.key-FULLTEXT {
+    color: mediumvioletred;
+  }
 }