fix pr comments

Author: egor-zalenski

src/extension.ts

@@ -84,7 +84,7 @@ export async function activate(context: vscode.ExtensionContext) {
         viewId: ViewId.AddKey,
         handleMessage: (message) => handleMessage(message),
         message: args,
-      })
+      }).postMessage({ action: 'SetDatabase', data: { database: args.data?.database } })
     }),
 
     vscode.commands.registerCommand('RedisForVSCode.addDatabase', (args) => {

src/webviews/src/modules/config/Config.tsx

@@ -3,7 +3,7 @@ import { useShallow } from 'zustand/react/shallow'
 
 import { useLocation } from 'react-router-dom'
 import { isDifferentConsentsExists } from 'uiSrc/utils'
-import { fetchDatabases } from 'uiSrc/store'
+import { fetchDatabases, useDatabasesStore } from 'uiSrc/store'
 import { fetchAppInfo, useAppInfoStore } from 'uiSrc/store/hooks/use-app-info-store/useAppInfoStore'
 import { vscodeApi } from 'uiSrc/services'
 import { VscodeMessageAction } from 'uiSrc/constants'
@@ -23,6 +23,10 @@ export const Config = () => {
     appInfo.setInitialStateAppInfo()
     fetchDatabases()
 
+    if (window.ri?.database) {
+      useDatabasesStore.getState().setConnectedDatabase(window.ri.database)
+    }
+
     if (window.ri?.appInfo) {
       appInfo.setAppInfo(window.ri?.appInfo)
       return