antares/src/renderer/components/WorkspaceQueryTab.vue

<template>
   <div v-show="isSelected" class="workspace-query-tab column col-12 columns col-gapless">
      <div class="workspace-query-runner column col-12">
         <QueryEditor v-if="isSelected" :value.sync="query" />
         <div class="workspace-query-runner-footer">
            <div class="workspace-query-buttons">
               <button
                  class="btn btn-link btn-sm"
                  :class="{'loading':isQuering}"
                  :disabled="!query"
                  @click="runQuery(query)"
               >
                  <span>{{ $t('word.run') }}</span>
                  <i class="mdi mdi-24px mdi-play text-success" />
               </button>
            </div>
            <div class="workspace-query-info">
               <div v-if="resultsCount !== false">
                  {{ $t('word.results') }}: <b>{{ resultsCount }}</b>
               </div>
               <div v-if="affectedCount !== false">
                  {{ $t('message.affectedRows') }}: <b>{{ affectedCount }}</b>
               </div>
               <div v-if="workspace.breadcrumbs.schema">
                  {{ $t('word.schema') }}: <b>{{ workspace.breadcrumbs.schema }}</b>
               </div>
            </div>
         </div>
      </div>
      <div class="workspace-query-results column col-12">
         <WorkspaceQueryTable
            v-if="results"
            v-show="!isQuering"
            ref="queryTable"
            :results="results"
            :tab-uid="tabUid"
            @update-field="updateField"
            @delete-selected="deleteSelected"
         />
      </div>
   </div>
</template>

<script>
import Connection from '@/ipc-api/Connection';
import Tables from '@/ipc-api/Tables';
import QueryEditor from '@/components/QueryEditor';
import WorkspaceQueryTable from '@/components/WorkspaceQueryTable';
import { mapGetters, mapActions } from 'vuex';
import tableTabs from '@/mixins/tableTabs';

export default {
   name: 'WorkspaceQueryTab',
   components: {
      QueryEditor,
      WorkspaceQueryTable
   },
   mixins: [tableTabs],
   props: {
      connection: Object,
      tabUid: String,
      isSelected: Boolean
   },
   data () {
      return {
         query: '',
         lastQuery: '',
         isQuering: false,
         results: [],
         resultsCount: false,
         affectedCount: false
      };
   },
   computed: {
      ...mapGetters({
         getWorkspace: 'workspaces/getWorkspace'
      }),
      workspace () {
         return this.getWorkspace(this.connection.uid);
      }
   },
   methods: {
      ...mapActions({
         addNotification: 'notifications/addNotification',
         setTabFields: 'workspaces/setTabFields',
         setTabKeyUsage: 'workspaces/setTabKeyUsage'
      }),
      getTable (index) {
         const resultsWithRows = this.results.filter(result => result.rows);

         if (resultsWithRows[index] && resultsWithRows[index].fields && resultsWithRows[index].fields.length)
            return resultsWithRows[index].fields[0].orgTable;
         return '';
      },
      async runQuery (query) {
         if (!query) return;
         this.isQuering = true;
         this.clearTabData();

         try { // Query Data
            const params = {
               uid: this.connection.uid,
               schema: this.schema,
               query
            };

            const { status, response } = await Connection.rawQuery(params);

            if (status === 'success') {
               this.results = Array.isArray(response) ? response : [response];

               let selectedFields = [];
               const fieldsArr = [];
               const keysArr = [];
               let index = 0;

               for (let i = 0; i < this.results.length; i++) {
                  if (this.results[i].rows) { // if is a select
                     const table = this.getTable(index);

                     selectedFields = this.results[i].fields.map(field => field.orgName);
                     const selectedSchema = this.results[i].fields[0].db;
                     this.resultsCount += this.results[i].rows.length;

                     try { // Table data
                        const params = {
                           uid: this.connection.uid,
                           schema: selectedSchema,
                           table
                        };

                        const { status, response } = await Tables.getTableColumns(params);

                        if (status === 'success') {
                           let fields = response.filter(field => selectedFields.includes(field.name));
                           if (selectedFields.length) {
                              fields = fields.map((field, index) => {
                                 return { ...field, alias: this.results[i].fields[index].name };
                              });
                           }

                           fieldsArr.push(fields);
                        }
                        else
                           this.addNotification({ status: 'error', message: response });
                     }
                     catch (err) {
                        this.addNotification({ status: 'error', message: err.stack });
                     }

                     try { // Key usage (foreign keys)
                        const params = {
                           uid: this.connection.uid,
                           schema: selectedSchema,
                           table
                        };

                        const { status, response } = await Tables.getKeyUsage(params);
                        if (status === 'success')
                           keysArr.push(response);
                        else
                           this.addNotification({ status: 'error', message: response });
                     }
                     catch (err) {
                        this.addNotification({ status: 'error', message: err.stack });
                     }
                  }
                  else if (this.results[i].report) { // if is a query without output
                     this.affectedCount += this.results[i].report.affectedRows;
                  }

                  index++;
               }

               this.setTabFields({ cUid: this.connection.uid, tUid: this.tabUid, fields: fieldsArr });
               this.setTabKeyUsage({ cUid: this.connection.uid, tUid: this.tabUid, keyUsage: keysArr });
            }
            else
               this.addNotification({ status: 'error', message: response });
         }
         catch (err) {
            this.addNotification({ status: 'error', message: err.stack });
         }

         this.isQuering = false;
         this.lastQuery = query;
      },
      reloadTable () {
         this.runQuery(this.lastQuery);
      },
      clearTabData () {
         this.results = [];
         this.resultsCount = false;
         this.affectedCount = false;
         this.setTabFields({ cUid: this.connection.uid, tUid: this.tabUid, fields: [] });
      }
   }
};
</script>

<style lang="scss">
.workspace-tabs {
  align-content: baseline;

  .workspace-query-runner {
    .workspace-query-runner-footer {
      display: flex;
      justify-content: space-between;
      padding: 0.3rem 0.6rem 0.4rem;
      align-items: center;

      .workspace-query-buttons {
        display: flex;

        .btn {
          display: flex;
          align-self: center;
          color: $body-font-color;
          margin-right: 0.4rem;
        }
      }

      .workspace-query-info {
        display: flex;

        > div + div {
          padding-left: 0.6rem;
        }
      }
    }
  }
}

</style>