From 0de5ef8a985cc8d9f1fa4479eeaf7dbb20f5fd7c Mon Sep 17 00:00:00 2001
From: Fabio Di Stasio <fabio286@gmail.com>
Date: Mon, 11 Sep 2023 18:02:19 +0200
Subject: [PATCH] fix: empty workspace deleting connections in some conditions

---
 src/renderer/components/SettingBarContext.vue | 7 ++-----
 src/renderer/stores/connections.ts            | 4 ++++
 src/renderer/stores/workspaces.ts             | 4 ++++
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/renderer/components/SettingBarContext.vue b/src/renderer/components/SettingBarContext.vue
index 25c7865d..13b9eebd 100644
--- a/src/renderer/components/SettingBarContext.vue
+++ b/src/renderer/components/SettingBarContext.vue
@@ -55,7 +55,6 @@
 
 <script setup lang="ts">
 import { uidGen } from 'common/libs/uidGen';
-import { storeToRefs } from 'pinia';
 import { computed, Prop, ref } from 'vue';
 import { useI18n } from 'vue-i18n';
 
@@ -78,10 +77,8 @@ const {
 } = connectionsStore;
 
 const workspacesStore = useWorkspacesStore();
-const { getSelected: selectedWorkspace } = storeToRefs(workspacesStore);
 
 const {
-   selectWorkspace,
    removeConnected: disconnectWorkspace,
    getWorkspace
 } = workspacesStore;
@@ -101,8 +98,8 @@ const connectionName = computed(() => props.contextConnection.name || getConnect
 const isConnected = computed(() => getWorkspace(props.contextConnection.uid)?.connectionStatus === 'connected');
 
 const confirmDeleteConnection = () => {
-   if (selectedWorkspace.value === props.contextConnection.uid)
-      selectWorkspace(null);
+   if (isConnected.value)
+      disconnectWorkspace(props.contextConnection.uid);
    deleteConnection(props.contextConnection);
    closeContext();
 };
diff --git a/src/renderer/stores/connections.ts b/src/renderer/stores/connections.ts
index d1859f47..12b31dc1 100644
--- a/src/renderer/stores/connections.ts
+++ b/src/renderer/stores/connections.ts
@@ -3,6 +3,8 @@ import { uidGen } from 'common/libs/uidGen';
 import * as crypto from 'crypto';
 import * as Store from 'electron-store';
 import { defineStore } from 'pinia';
+
+import { useWorkspacesStore } from '@/stores/workspaces';
 const key = localStorage.getItem('key');
 
 export interface SidebarElement {
@@ -105,10 +107,12 @@ export const useConnectionsStore = defineStore('connections', {
             return el;
          });
          this.connectionsOrder = (this.connectionsOrder as SidebarElement[]).filter(el => el.uid !== connection.uid);
+         this.lastConnections = (this.lastConnections as SidebarElement[]).filter(el => el.uid !== connection.uid);
 
          this.connections = (this.connections as SidebarElement[]).filter(el => el.uid !== connection.uid);
          persistentStore.set('connections', this.connections);
          this.clearEmptyFolders();
+         useWorkspacesStore().removeWorkspace(connection.uid);
       },
       editConnection (connection: ConnectionParams) {
          const editedConnections = (this.connections as ConnectionParams[]).map(conn => {
diff --git a/src/renderer/stores/workspaces.ts b/src/renderer/stores/workspaces.ts
index 9ddcda5e..10743a6d 100644
--- a/src/renderer/stores/workspaces.ts
+++ b/src/renderer/stores/workspaces.ts
@@ -424,6 +424,10 @@ export const useWorkspacesStore = defineStore('workspaces', {
 
          this.workspaces.push(workspace);
       },
+      removeWorkspace (uid: string) {
+         this.workspaces = this.workspaces.filter((w: Workspace) => w.uid !== uid);
+         if (uid === this.selectedWorkspace) this.selectedWorkspace = 'NEW';
+      },
       changeBreadcrumbs (payload: Breadcrumb) {
          const breadcrumbsObj: Breadcrumb = {
             schema: null,