From edf1632cabf1fa1eb588ac3d66ef3b9594e98f8f Mon Sep 17 00:00:00 2001
From: "SPRINX0\\prochazka" <jan.prochazka@sprinx.com>
Date: Mon, 23 Jun 2025 13:28:37 +0200
Subject: [PATCH] SYNC:  fixed connection for scripts

---
 .../web/src/appobj/SavedFileAppObject.svelte  | 24 ++++++++++++++++++-
 packages/web/src/tabs/QueryTab.svelte         |  9 +++----
 2 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/packages/web/src/appobj/SavedFileAppObject.svelte b/packages/web/src/appobj/SavedFileAppObject.svelte
index 1cf2ec306..8132f9da2 100644
--- a/packages/web/src/appobj/SavedFileAppObject.svelte
+++ b/packages/web/src/appobj/SavedFileAppObject.svelte
@@ -1,5 +1,6 @@
 <script lang="ts" context="module">
-  import { filterName, getConnectionLabel } from 'dbgate-tools';
+  import { filterName, getConnectionLabel, getSqlFrontMatter } from 'dbgate-tools';
+  import yaml from 'js-yaml';
 
   interface FileTypeHandler {
     icon: string;
@@ -9,6 +10,7 @@
     currentConnection: boolean;
     extension: string;
     label: string;
+    switchDatabaseOnOpen?: (data: any) => Promise<boolean>;
   }
 
   const sql: FileTypeHandler = {
@@ -19,6 +21,21 @@
     currentConnection: true,
     extension: 'sql',
     label: 'SQL file',
+    switchDatabaseOnOpen: async data => {
+      const frontMatter = getSqlFrontMatter(data, yaml);
+      if (frontMatter?.connectionId) {
+        const connection = await getConnectionInfo({ conid: frontMatter.connectionId });
+        // console.log('Switching database to', frontMatter.databaseName, 'on connection', connection);
+        if (connection && frontMatter.databaseName) {
+          currentDatabase.set({
+            connection,
+            name: frontMatter.databaseName,
+          });
+          return true;
+        }
+      }
+      return false;
+    },
   };
 
   const shell: FileTypeHandler = {
@@ -161,6 +178,7 @@
   import AppObjectCore from './AppObjectCore.svelte';
   import { isProApp } from '../utility/proTools';
   import { saveFileToDisk } from '../utility/exportFileTools';
+  import { getConnectionInfo } from '../utility/metadataLoaders';
 
   export let data;
 
@@ -281,6 +299,10 @@
     let tooltip = undefined;
     const connProps: any = {};
 
+    if (handler.switchDatabaseOnOpen) {
+      await handler.switchDatabaseOnOpen(dataContent);
+    }
+
     if (handler.currentConnection) {
       const connection = _.get($currentDatabase, 'connection') || {};
       const database = _.get($currentDatabase, 'name');
diff --git a/packages/web/src/tabs/QueryTab.svelte b/packages/web/src/tabs/QueryTab.svelte
index 7aba2ceb7..db2eca3ae 100644
--- a/packages/web/src/tabs/QueryTab.svelte
+++ b/packages/web/src/tabs/QueryTab.svelte
@@ -616,16 +616,13 @@
 
   export function toggleFixedConnection() {
     const frontMatter = getSqlFrontMatter($editorValue, yaml);
-    const currentDatabase = getCurrentDatabase();
     setEditorData(
       setSqlFrontMatter(
         $editorValue,
-        frontMatter?.connectionId &&
-          frontMatter?.connectionId == currentDatabase?.connection?._id &&
-          frontMatter?.databaseName == currentDatabase?.name
+        frontMatter?.connectionId && frontMatter?.connectionId == conid && frontMatter?.databaseName == database
           ? { ...frontMatter, connectionId: undefined, databaseName: undefined }
-          : currentDatabase?.connection?._id
-            ? { ...frontMatter, connectionId: currentDatabase.connection._id, databaseName: currentDatabase.name }
+          : conid
+            ? { ...frontMatter, connectionId: conid, databaseName: database }
             : { ...frontMatter, connectionId: undefined, databaseName: undefined },
         yaml
       )