From 9a732ea1971d223f3278ad02d3dd77837fecb377 Mon Sep 17 00:00:00 2001
From: Fabio Di Stasio <fabio286@gmail.com>
Date: Mon, 25 Dec 2023 10:54:41 +0100
Subject: [PATCH] feat: open saved queries in a tab

---
 .github/workflows/test-e2e-win.yml            |  2 +-
 src/renderer/components/ScratchpadNote.vue    | 14 ++++++-
 src/renderer/components/TheScratchpad.vue     | 31 ++++++++++++++-
 src/renderer/components/WorkspaceTabQuery.vue |  9 +++++
 src/renderer/stores/workspaces.ts             | 39 +++++++++++++++----
 5 files changed, 85 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/test-e2e-win.yml b/.github/workflows/test-e2e-win.yml
index bca35269..77a3b36a 100644
--- a/.github/workflows/test-e2e-win.yml
+++ b/.github/workflows/test-e2e-win.yml
@@ -3,7 +3,7 @@ name: Test end-to-end [WINDOWS]
 on: 
   push:
     branches:
-        - master
+        - develop
 
 jobs:
   release:
diff --git a/src/renderer/components/ScratchpadNote.vue b/src/renderer/components/ScratchpadNote.vue
index 46b71c3c..e958bb41 100644
--- a/src/renderer/components/ScratchpadNote.vue
+++ b/src/renderer/components/ScratchpadNote.vue
@@ -67,6 +67,17 @@
                      :size="22"
                   /> {{ t('general.undo') }}
                </button>
+               <button
+                  v-if="note.type === 'query'"
+                  class="btn btn-link pl-1"
+                  @click.stop="$emit('select-query', note.note)"
+               >
+                  <BaseIcon
+                     icon-name="mdiOpenInApp"
+                     class="pr-1"
+                     :size="22"
+                  /> {{ t('general.select') }}
+               </button>
                <button
                   v-if="note.type === 'query'"
                   class="btn btn-link pl-1"
@@ -139,7 +150,8 @@ defineEmits([
    'select-note',
    'toggle-note',
    'archive-note',
-   'restore-note'
+   'restore-note',
+   'select-query'
 ]);
 
 const noteParagraph: Ref<HTMLDivElement> = ref(null);
diff --git a/src/renderer/components/TheScratchpad.vue b/src/renderer/components/TheScratchpad.vue
index a87f0ac4..36cd29f0 100644
--- a/src/renderer/components/TheScratchpad.vue
+++ b/src/renderer/components/TheScratchpad.vue
@@ -113,6 +113,7 @@
                               @delete-note="deleteNote"
                               @archive-note="archiveNote"
                               @restore-note="restoreNote"
+                              @select-query="selectQuery"
                            />
                         </template>
                      </BaseVirtualScroll>
@@ -179,13 +180,15 @@ const { t } = useI18n();
 
 const applicationStore = useApplicationStore();
 const scratchpadStore = useScratchpadStore();
+const workspacesStore = useWorkspacesStore();
 
 const { connectionNotes, selectedTag } = storeToRefs(scratchpadStore);
 const { changeNotes } = scratchpadStore;
 const { hideScratchpad } = applicationStore;
 const { getConnectionName } = useConnectionsStore();
 const { connections } = storeToRefs(useConnectionsStore());
-const { getSelected: selectedWorkspace } = storeToRefs(useWorkspacesStore());
+const { getSelected: selectedWorkspace } = storeToRefs(workspacesStore);
+const { getWorkspaceTab, getWorkspace, newTab, updateTabContent } = workspacesStore;
 
 const localConnection = ref(null);
 const table: Ref<HTMLDivElement> = ref(null);
@@ -276,6 +279,32 @@ const deleteNote = (uid: string) => {
    changeNotes(otherNotes);
 };
 
+const selectQuery = (query: string) => {
+   const workspace = getWorkspace(selectedWorkspace.value);
+   const selectedTab = getWorkspaceTab(workspace.selectedTab);
+
+   if (selectedTab.type === 'query') {
+      updateTabContent({
+         tab: selectedTab.uid,
+         uid: selectedWorkspace.value,
+         type: 'query',
+         content: query,
+         schema: workspace.breadcrumbs.schema
+      });
+   }
+   else {
+      newTab({
+         uid: selectedWorkspace.value,
+         type: 'query',
+         content: query,
+         autorun: false,
+         schema: workspace.breadcrumbs.schema
+      });
+   }
+
+   hideScratchpad();
+};
+
 const closeEditModal = () => {
    isEditModal.value = false;
    noteToEdit.value = null;
diff --git a/src/renderer/components/WorkspaceTabQuery.vue b/src/renderer/components/WorkspaceTabQuery.vue
index 80930dc4..f8a3ef7b 100644
--- a/src/renderer/components/WorkspaceTabQuery.vue
+++ b/src/renderer/components/WorkspaceTabQuery.vue
@@ -345,6 +345,7 @@ watch(query, (val) => {
          schema: selectedSchema.value,
          content: val
       });
+
       isQuerySaved.value = false;
    }, 200);
 });
@@ -368,6 +369,14 @@ watch(databaseSchemas, () => {
       selectedSchema.value = null;
 }, { deep: true });
 
+watch(() => props.tab.content, () => {
+   query.value = props.tab.content;
+   const editorValue = queryEditor.value.editor.session.getValue();
+
+   if (editorValue !== query.value)// If change not rendered in editor
+      queryEditor.value.editor.session.setValue(query.value);
+});
+
 const runQuery = async (query: string) => {
    if (!query || isQuering.value) return;
    isQuering.value = true;
diff --git a/src/renderer/stores/workspaces.ts b/src/renderer/stores/workspaces.ts
index 10743a6d..44b02508 100644
--- a/src/renderer/stores/workspaces.ts
+++ b/src/renderer/stores/workspaces.ts
@@ -67,7 +67,7 @@ export interface Workspace {
    client?: ClientCode;
    database?: string;
    connectionStatus: string;
-   selectedTab: string | number;
+   selectedTab: string;
    searchTerm: string;
    tabs: WorkspaceTab[];
    structure: WorkspaceStructure[];
@@ -119,12 +119,12 @@ export const useWorkspacesStore = defineStore('workspaces', {
          return state.workspaces.find(workspace => workspace.uid === uid).variables.find(variable => variable.name === name);
       },
       getWorkspaceTab (state) {
-         return (tUid: string) => {
+         return (tUid: string): WorkspaceTab => {
             if (!this.getSelected) return;
             const workspace = state.workspaces.find(workspace => workspace.uid === this.getSelected);
             if ('tabs' in workspace)
                return workspace.tabs.find(tab => tab.uid === tUid);
-            return {};
+            return null;
          };
       },
       getConnected: state => {
@@ -410,7 +410,7 @@ export const useWorkspacesStore = defineStore('workspaces', {
          const workspace: Workspace = {
             uid,
             connectionStatus: 'disconnected',
-            selectedTab: 0,
+            selectedTab: '0',
             searchTerm: '',
             tabs: [],
             structure: [],
@@ -629,18 +629,43 @@ export const useWorkspacesStore = defineStore('workspaces', {
                   : false;
 
                if (existentTab) {
-                  this._replaceTab({ uid, tab: existentTab.uid, type, database: workspaceTabs.database, schema, elementName, elementType });
+                  this._replaceTab({ uid,
+                     tab: existentTab.uid,
+                     type,
+                     database: workspaceTabs.database,
+                     schema,
+                     elementName,
+                     elementType
+                  });
                   tabUid = existentTab.uid;
                }
                else {
                   tabUid = uidGen('T');
-                  this._addTab({ uid, tab: tabUid, content, type, autorun, database: workspaceTabs.database, schema, elementName, elementType });
+                  this._addTab({ uid,
+                     tab: tabUid,
+                     content,
+                     type,
+                     autorun,
+                     database: workspaceTabs.database,
+                     schema,
+                     elementName,
+                     elementType
+                  });
                }
             }
                break;
             default:
                tabUid = uidGen('T');
-               this._addTab({ uid, tab: tabUid, content, type, autorun, database: workspaceTabs.database, schema, elementName, elementType });
+               this._addTab({ uid,
+                  tab: tabUid,
+                  content,
+                  type,
+                  autorun,
+                  database: workspaceTabs.database,
+                  schema,
+                  elementName,
+                  elementType
+               });
                break;
          }