From 02635d3f938a57353dee6eace99e3136379ee84b Mon Sep 17 00:00:00 2001
From: Ilnar Agliamzianov <ilnar.agliamzianov@jetbrains.com>
Date: Mon, 19 Aug 2024 10:35:28 +0300
Subject: [PATCH 1/4] Refactor the code related to 'Database Tools' integration

---
 .../.idea/gradle.xml                          |   1 +
 .../build.gradle.kts                          |   1 +
 .../IntelliJCosmosActionsContributor.java     |  14 +-
 ...eCosmosDbAccountConnectionInterceptor.java |  44 +--
 .../AzureCosmosDbAccountParamEditor.java      | 278 ++++------------
 .../dbtools/CosmosDbAccountComboBox.java      |  33 ++
 .../dbtools/CosmosNotSignedInLabel.java       |  26 ++
 .../cosmos/dbtools/DbToolsWorkaround.java     |  15 +-
 .../cosmos/dbtools/NoAccountsTipLabel.java    |  44 +++
 .../build.gradle.kts                          |   1 +
 .../dbtools/DatabaseDbToolsWorkaround.java    |  60 ++--
 .../dbtools/DatabaseNotSignedInLabel.java     |  25 ++
 .../database/dbtools/DatabaseServerClass.java |  39 +++
 .../DatabaseServerConnectionInterceptor.java  |  44 +--
 .../dbtools/DatabaseServerParamEditor.java    | 301 +++---------------
 .../dbtools/DatabaseServerTypeUIFactory.java  |   6 +-
 .../database/dbtools/NoServersTipLabel.java   |  55 ++++
 .../dbtools/OpenWithDatabaseToolsAction.java  |  11 +-
 .../database/dbtools/SqlDbServerComboBox.java |  44 +++
 .../IntellijMySqlActionsContributor.java      |  13 +-
 .../IntellijPostgreSqlActionsContributor.java |  13 +-
 .../IntellijSqlServerActionsContributor.java  |  14 +-
 .../build.gradle.kts                          |  10 +
 .../dbtools/AzureParamsEditorBase.java        | 162 ++++++++++
 .../intellij/dbtools/DatabasePlugin.java      |  25 ++
 .../intellij/dbtools/DatabaseTools.java       |  14 +
 .../intellij/dbtools/DbToolsWorkarounds.java  |  42 +++
 .../intellij/dbtools/DbUtilsComboBoxBase.java |  56 ++++
 .../intellij/dbtools/NoResourceTipLabel.java  |  54 ++++
 .../intellij/dbtools/NotSignedInTipLabel.java |  44 +++
 .../TelemetryConnectionInterceptor.java       |  50 +++
 .../build.gradle.kts                          |   1 +
 .../settings.gradle.kts                       |   3 +-
 33 files changed, 895 insertions(+), 648 deletions(-)
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/build.gradle.kts
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java

diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/.idea/gradle.xml b/PluginsAndFeatures/azure-toolkit-for-intellij/.idea/gradle.xml
index 3ffccecde10..47a02f1861c 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/.idea/gradle.xml
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/.idea/gradle.xml
@@ -22,6 +22,7 @@
             <option value="$PROJECT_DIR$/azure-intellij-plugin-cosmos" />
             <option value="$PROJECT_DIR$/azure-intellij-plugin-database" />
             <option value="$PROJECT_DIR$/azure-intellij-plugin-database-java" />
+            <option value="$PROJECT_DIR$/azure-intellij-plugin-dbtools" />
             <option value="$PROJECT_DIR$/azure-intellij-plugin-eventhubs" />
             <option value="$PROJECT_DIR$/azure-intellij-plugin-guidance" />
             <option value="$PROJECT_DIR$/azure-intellij-plugin-guidance-java" />
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/build.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/build.gradle.kts
index 4019ae664bb..1ffaf2bb819 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/build.gradle.kts
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/build.gradle.kts
@@ -12,6 +12,7 @@ dependencies {
     implementation(project(":azure-intellij-resource-connector-lib"))
     // runtimeOnly project(path: ":azure-intellij-resource-connector-lib", configuration: "instrumentedJar")
     implementation(project(":azure-intellij-resource-connector-lib-java"))
+    implementation(project(":azure-intellij-plugin-dbtools"))
     // runtimeOnly project(path: ":azure-intellij-resource-connector-lib-java", configuration: "instrumentedJar")
     implementation("com.microsoft.azure:azure-toolkit-cosmos-lib")
     implementation("com.microsoft.azure:azure-toolkit-ide-common-lib")
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/IntelliJCosmosActionsContributor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/IntelliJCosmosActionsContributor.java
index f6a4bdf289e..c30bf7046b3 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/IntelliJCosmosActionsContributor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/IntelliJCosmosActionsContributor.java
@@ -25,6 +25,7 @@
 import com.microsoft.azure.toolkit.intellij.cosmos.creation.CreateCosmosContainerAction;
 import com.microsoft.azure.toolkit.intellij.cosmos.creation.CreateCosmosDBAccountAction;
 import com.microsoft.azure.toolkit.intellij.cosmos.creation.CreateCosmosDatabaseAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabasePlugin;
 import com.microsoft.azure.toolkit.lib.common.action.Action;
 import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
 import com.microsoft.azure.toolkit.lib.common.bundle.AzureString;
@@ -125,22 +126,13 @@ public void registerHandlers(AzureActionManager am) {
         am.registerHandler(ResourceCommonActionsContributor.CREATE, (r, e) -> r instanceof CassandraKeyspace && ((CassandraKeyspace) r).getFormalStatus().isRunning(), (Object r, AnActionEvent e) ->
             CreateCosmosContainerAction.createCassandraTable(e.getProject(), (CassandraKeyspace) r, CassandraTableDraft.CassandraTableConfig.getDefaultConfig()));
 
-        final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
-        if (PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null) {
-            final BiConsumer<CosmosDBAccount, AnActionEvent> openDatabaseHandler = (c, e) -> openDatabaseTool(e.getProject(), c);
+        if (!DatabasePlugin.isInstalled()) {
+            final BiConsumer<CosmosDBAccount, AnActionEvent> openDatabaseHandler = (c, e) -> DatabasePlugin.throwTryUltimateIfNotInstalled(c);
             final boolean cassandraOn = Registry.is("azure.toolkit.cosmos_cassandra.dbtools.enabled");
             am.registerHandler(CosmosActionsContributor.OPEN_DATABASE_TOOL, (r, e) -> r instanceof MongoCosmosDBAccount || (r instanceof CassandraCosmosDBAccount && cassandraOn), openDatabaseHandler);
         }
     }
 
-    private void openDatabaseTool(Project project, CosmosDBAccount account) {
-        final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
-        final String DATABASE_PLUGIN_NOT_INSTALLED = "\"Database tools and SQL\" plugin is not installed.";
-        final String NOT_SUPPORT_ERROR_ACTION = "\"Database tools and SQL\" plugin is only provided in IntelliJ Ultimate edition.";
-        final Action<Object> tryUltimate = AzureActionManager.getInstance().getAction(IntellijActionsContributor.TRY_ULTIMATE).bind(account);
-        throw new AzureToolkitRuntimeException(DATABASE_PLUGIN_NOT_INSTALLED, NOT_SUPPORT_ERROR_ACTION, tryUltimate);
-    }
-
     private <T extends AzResource> void openResourceConnector(@Nonnull final T resource, @Nonnull final AzureServiceResource.Definition<T> definition, Project project) {
         AzureTaskManager.getInstance().runLater(() -> {
             final ConnectorDialog dialog = new ConnectorDialog(project);
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
index 46003ae727b..5d74960f338 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
@@ -1,43 +1,11 @@
-/*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for license information.
- */
-
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
-import com.azure.resourcemanager.resources.fluentcore.arm.ResourceId;
-import com.intellij.database.dataSource.DatabaseConnectionInterceptor;
-import com.intellij.database.dataSource.DatabaseConnectionPoint;
-import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
-import com.microsoft.azure.toolkit.lib.common.operation.Operation;
-import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
-import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
-import org.apache.commons.lang3.StringUtils;
-import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
-
-import java.util.HashMap;
-import java.util.Map;
-import java.util.concurrent.CompletionStage;
+import com.microsoft.azure.toolkit.intellij.dbtools.TelemetryConnectionInterceptor;
 
-import static com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter.*;
+import static com.microsoft.azure.toolkit.intellij.cosmos.dbtools.AzureCosmosDbAccountParamEditor.KEY_COSMOS_ACCOUNT_ID;
 
-@SuppressWarnings("UnstableApiUsage")
-public class AzureCosmosDbAccountConnectionInterceptor implements DatabaseConnectionInterceptor {
-    @Nullable
-    public CompletionStage<ProtoConnection> intercept(@NotNull DatabaseConnectionInterceptor.ProtoConnection proto, boolean silent) {
-        final DatabaseConnectionPoint point = proto.getConnectionPoint();
-        final String accountId = point.getAdditionalProperty(AzureCosmosDbAccountParamEditor.KEY_COSMOS_ACCOUNT_ID);
-        if (StringUtils.isNotBlank(accountId) && !StringUtils.equalsIgnoreCase(accountId, AzureCosmosDbAccountParamEditor.NONE)) {
-            final Map<String, String> properties = new HashMap<>();
-            final ResourceId id = ResourceId.fromString(accountId);
-            properties.put("subscriptionId", id.subscriptionId());
-            properties.put(SERVICE_NAME, "cosmos");
-            properties.put(OPERATION_NAME, "connect_jdbc_from_dbtools");
-            properties.put(OP_NAME, "cosmos.connect_jdbc_from_dbtools");
-            properties.put(OP_TYPE, Operation.Type.USER);
-            AzureTelemeter.log(AzureTelemetry.Type.OP_END, properties);
-        }
-        return null;
+public class AzureCosmosDbAccountConnectionInterceptor extends TelemetryConnectionInterceptor {
+    protected AzureCosmosDbAccountConnectionInterceptor() {
+        super(KEY_COSMOS_ACCOUNT_ID, "cosmos", "cosmos.connect_jdbc_from_dbtools");
     }
-}
\ No newline at end of file
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
index 22947591ec6..69f53bf96c6 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
@@ -1,258 +1,100 @@
-/*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for license information.
- */
-
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.intellij.credentialStore.OneTimeString;
-import com.intellij.database.dataSource.DataSourceConfigurable;
 import com.intellij.database.dataSource.LocalDataSource;
 import com.intellij.database.dataSource.url.DataInterchange;
 import com.intellij.database.dataSource.url.FieldSize;
-import com.intellij.database.dataSource.url.template.UrlEditorModel;
-import com.intellij.database.dataSource.url.ui.ParamEditorBase;
-import com.intellij.icons.AllIcons;
-import com.intellij.ide.DataManager;
-import com.intellij.openapi.actionSystem.AnActionEvent;
-import com.intellij.openapi.actionSystem.CommonDataKeys;
-import com.intellij.openapi.actionSystem.DataContext;
-import com.intellij.openapi.actionSystem.EmptyAction;
-import com.intellij.openapi.project.Project;
-import com.intellij.openapi.wm.ToolWindow;
-import com.intellij.openapi.wm.ToolWindowManager;
-import com.intellij.ui.ComponentUtil;
-import com.intellij.ui.HyperlinkLabel;
-import com.intellij.ui.components.JBCheckBox;
-import com.microsoft.azure.toolkit.ide.common.action.ResourceCommonActionsContributor;
-import com.microsoft.azure.toolkit.intellij.common.AzureComboBox;
-import com.microsoft.azure.toolkit.intellij.cosmos.creation.CreateCosmosDBAccountAction;
-import com.microsoft.azure.toolkit.lib.Azure;
-import com.microsoft.azure.toolkit.lib.auth.AzureAccount;
-import com.microsoft.azure.toolkit.lib.common.action.Action;
-import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
-import com.microsoft.azure.toolkit.lib.common.cache.CacheManager;
-import com.microsoft.azure.toolkit.lib.common.model.AbstractAzResource;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.AzureParamsEditorBase;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
 import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
-import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
-import com.microsoft.azure.toolkit.lib.common.task.AzureTask;
-import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
-import com.microsoft.azure.toolkit.lib.cosmos.AzureCosmosService;
 import com.microsoft.azure.toolkit.lib.cosmos.CosmosDBAccount;
 import com.microsoft.azure.toolkit.lib.cosmos.model.CosmosDBAccountConnectionString;
 import com.microsoft.azure.toolkit.lib.cosmos.model.DatabaseAccountKind;
-import lombok.Getter;
-import lombok.RequiredArgsConstructor;
-import lombok.Setter;
-import lombok.SneakyThrows;
 import org.apache.commons.lang3.StringUtils;
-import org.apache.commons.lang3.reflect.FieldUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
-import javax.swing.*;
-import java.awt.*;
-import java.awt.event.InputEvent;
-import java.util.Collections;
-import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
-import java.util.function.Consumer;
-import java.util.stream.Collectors;
 
-public class AzureCosmosDbAccountParamEditor extends ParamEditorBase<AzureCosmosDbAccountParamEditor.CosmosDbAccountComboBox> {
+public class AzureCosmosDbAccountParamEditor extends AzureParamsEditorBase<CosmosDbAccountComboBox, CosmosDBAccount> {
     public static final String KEY_COSMOS_ACCOUNT_ID = "AZURE_COSMOS_ACCOUNT";
-    public static final String NONE = "<NONE>";
     public static final String KEY_FROM_AZURE_EXPLORER = "FROM_EXPLORER";
-    public static final String NO_ACCOUNT_TIPS_TEMPLATE = "<html>No Azure Cosmos DB accounts (%s). You can <a href=''>create one</a> first.</html>";
-    public static final String NOT_SIGNIN_TIPS = "<html><a href=\"\">Sign in</a> to select an existing Azure Cosmos DB account.</html>";
-    private final DatabaseAccountKind kind;
-    @Getter
-    @Setter
-    private String text = "";
+    @Nonnull
+    private final DatabaseAccountKind databaseAccountKind;
     @Nullable
     private CosmosDBAccountConnectionString connectionString;
-    private boolean updating;
 
-    public AzureCosmosDbAccountParamEditor(@Nonnull DatabaseAccountKind kind, @Nonnull String label, @Nonnull DataInterchange interchange) {
-        super(new CosmosDbAccountComboBox(kind), interchange, FieldSize.LARGE, label);
-        this.kind = kind;
-        final LocalDataSource dataSource = getDataSourceConfigurable().getDataSource();
-        final CosmosDbAccountComboBox combox = this.getEditorComponent();
-        combox.addValueChangedListener(this::setAccount);
-        interchange.addPersistentProperty(KEY_COSMOS_ACCOUNT_ID);
+    public AzureCosmosDbAccountParamEditor(
+            @Nonnull DatabaseAccountKind databaseAccountKind,
+            @Nullable String caption,
+            @NotNull DataInterchange interchange,
+            AnAction @NotNull ... actions) {
+        super(new CosmosDbAccountComboBox(databaseAccountKind), interchange, FieldSize.LARGE, caption, KEY_COSMOS_ACCOUNT_ID, actions);
+        this.databaseAccountKind = databaseAccountKind;
+        final CosmosDbAccountComboBox comboBox = this.getEditorComponent();
         final String accountId = interchange.getProperty(KEY_COSMOS_ACCOUNT_ID);
         if (StringUtils.isNotBlank(accountId)) {
-            combox.setValue(new AzureComboBox.ItemReference<>(i -> i.getId().equals(accountId)));
+            comboBox.setByResourceId(accountId);
         }
-
-        interchange.addPropertyChangeListener((evt -> onPropertiesChanged(evt.getPropertyName(), evt.getNewValue())), this);
     }
 
     @Override
-    protected @Nonnull JComponent createComponent(CosmosDbAccountComboBox combox) {
-        final JPanel container = new JPanel();
-        final BoxLayout layout = new BoxLayout(container, BoxLayout.Y_AXIS);
-        container.setLayout(layout);
-
-        combox.setAlignmentX(Component.LEFT_ALIGNMENT);
-        container.add(combox);
-
-        if (!Azure.az(AzureAccount.class).isLoggedIn()) {
-            final HyperlinkLabel notSignInTips = initNotSignInTipsLabel(combox);
-            container.add(notSignInTips);
-        }
-
-        final HyperlinkLabel noAccountsTips = initNoAccountTipsLabel(combox);
-        container.add(noAccountsTips);
-        return container;
+    protected NotSignedInTipLabel createNotSignedInTipLabel() {
+        return new CosmosNotSignedInLabel(databaseAccountKind);
     }
 
-    @AzureOperation(name = "user/cosmos.signin_from_dbtools")
-    private void signInAndReloadItems(CosmosDbAccountComboBox combox, HyperlinkLabel notSignInTips) {
-        OperationContext.action().setTelemetryProperty("kind", this.kind.getValue());
-        AzureActionManager.getInstance().getAction(Action.REQUIRE_AUTH).handle((a) -> {
-            notSignInTips.setVisible(false);
-            combox.reloadItems();
-        });
-    }
-
-    @Nonnull
-    private HyperlinkLabel initNotSignInTipsLabel(CosmosDbAccountComboBox combox) {
-        final HyperlinkLabel label = new HyperlinkLabel();
-        label.setHtmlText(NOT_SIGNIN_TIPS);
-        label.setIcon(AllIcons.General.Information);
-        label.addHyperlinkListener(e -> signInAndReloadItems(combox, label));
-        label.setAlignmentX(Component.LEFT_ALIGNMENT);
-        return label;
-    }
-
-    private HyperlinkLabel initNoAccountTipsLabel(CosmosDbAccountComboBox combox) {
-        final HyperlinkLabel label = new HyperlinkLabel();
-        label.setHtmlText(String.format(NO_ACCOUNT_TIPS_TEMPLATE, combox.getKind().getValue()));
-        label.setIcon(AllIcons.General.Information);
-        label.addHyperlinkListener(e -> createAccountInIde(e.getInputEvent()));
-        label.setAlignmentX(Component.LEFT_ALIGNMENT);
-        combox.setAccountsListener(label::setVisible);
-        return label;
-    }
-
-    @AzureOperation(name = "user/cosmos.create_account_from_dbtools")
-    private void createAccountInIde(InputEvent e) {
-        OperationContext.action().setTelemetryProperty("kind", this.kind.getValue());
-        final DataContext context = DataManager.getInstance().getDataContext(e.getComponent());
-        final Project project = context.getData(CommonDataKeys.PROJECT);
-        final Window window = ComponentUtil.getActiveWindow();
-        window.setVisible(false);
-        window.dispose();
-        final ToolWindow explorer = ToolWindowManager.getInstance(Objects.requireNonNull(project)).getToolWindow("Azure Explorer");
-        Objects.requireNonNull(explorer).activate(() -> {
-            final AnActionEvent event = AnActionEvent.createFromAnAction(new EmptyAction(), e, "cosmos.dbtools", context);
-            AzureActionManager.getInstance().getAction(ResourceCommonActionsContributor.SELECT_RESOURCE_IN_EXPLORER).handle(Azure.az(AzureCosmosService.class), event);
-            CreateCosmosDBAccountAction.create(null, null);
-        });
+    @Override
+    protected NoResourceTipLabel createNoResourceTipLabel() {
+        return new NoAccountsTipLabel(databaseAccountKind);
     }
 
-    private void onPropertiesChanged(String propertyName, Object newValue) {
-        if (!this.updating && StringUtils.isNotEmpty((String) newValue) && StringUtils.equals(propertyName, "host") && Objects.nonNull(this.connectionString)) {
-            final AzureCosmosDbAccountParamEditor.CosmosDbAccountComboBox combox = this.getEditorComponent();
-            final CosmosDBAccount account = combox.getValue();
-            if (Objects.nonNull(account) && !Objects.equals(this.connectionString.getHost(), newValue)) {
-                this.getEditorComponent().setValue((CosmosDBAccount) null);
-                this.setAccount(null);
-            }
+    @Override
+    @AzureOperation(name = "user/cosmos.select_account_dbtools.account", params = {"value.getName()"}, source = "value")
+    protected void setResource(
+            @Nonnull DataInterchange interchange,
+            @Nullable Object fromBackground,
+            @Nullable CosmosDBAccount value,
+            @Nullable String oldResourceId,
+            @Nullable String newResourceId) {
+        this.connectionString = (CosmosDBAccountConnectionString) fromBackground;
+        final boolean fromExplorer = Objects.nonNull(interchange.getProperty(KEY_FROM_AZURE_EXPLORER));
+        interchange.putProperty(KEY_FROM_AZURE_EXPLORER, null);
+        interchange.putProperty(KEY_COSMOS_ACCOUNT_ID, Optional.ofNullable(newResourceId).orElse(NONE));
+        if (Objects.isNull(value) || Objects.isNull(connectionString) || StringUtils.equalsIgnoreCase(oldResourceId, newResourceId) && !fromExplorer) {
+            return;
         }
-    }
-
-    @AzureOperation(name = "user/cosmos.select_account_dbtools.account", params = {"account.getName()"}, source = "account")
-    private void setAccount(@Nullable CosmosDBAccount account) {
-        final DataInterchange interchange = this.getInterchange();
-        final String oldAccountId = interchange.getProperty(KEY_COSMOS_ACCOUNT_ID);
-        final String newAccountId = Optional.ofNullable(account).map(AbstractAzResource::getId).orElse(null);
-        final AzureTaskManager manager = AzureTaskManager.getInstance();
-        this.updating = true;
-        manager.runOnPooledThread(() -> {
-            this.connectionString = Optional.ofNullable(account).map(CosmosDBAccount::getCosmosDBAccountPrimaryConnectionString).orElse(null);
-            manager.runLater(() -> {
-                final boolean fromExplorer = Objects.nonNull(interchange.getProperty(KEY_FROM_AZURE_EXPLORER));
-                interchange.putProperty(KEY_FROM_AZURE_EXPLORER, null);
-                interchange.putProperty(KEY_COSMOS_ACCOUNT_ID, Optional.ofNullable(newAccountId).orElse(NONE));
-                if (Objects.isNull(account) || Objects.isNull(connectionString) || StringUtils.equalsIgnoreCase(oldAccountId, newAccountId) && !fromExplorer) {
-                    this.updating = false;
-                    return;
-                }
-                final LocalDataSource dataSource = interchange.getDataSource();
-                final String host = connectionString.getHost();
-                final String port = String.valueOf(connectionString.getPort());
-                final String user = connectionString.getUsername();
-                final String password = String.valueOf(connectionString.getPassword());
-                LocalDataSource.setUsername(dataSource, user);
-                interchange.getCredentials().storePassword(dataSource, new OneTimeString(password));
-                this.setUseSsl(true);
-                interchange.putProperties(consumer -> {
-                    consumer.consume("host", host);
-                    consumer.consume("user", user);
-                    consumer.consume("port", port);
-                });
-                this.setUsername(user);
-                this.updating = false;
-            }, AzureTask.Modality.ANY);
+        final LocalDataSource dataSource = interchange.getDataSource();
+        final String host = connectionString.getHost();
+        final String port = String.valueOf(connectionString.getPort());
+        final String user = connectionString.getUsername();
+        final String password = String.valueOf(connectionString.getPassword());
+        LocalDataSource.setUsername(dataSource, user);
+        interchange.getCredentials().storePassword(dataSource, new OneTimeString(password));
+        this.setUseSsl(true);
+        interchange.putProperties(consumer -> {
+            consumer.consume("host", host);
+            consumer.consume("user", user);
+            consumer.consume("port", port);
         });
+        this.setUsername(user);
     }
 
-    private void setUsername(String user) {
-        final UrlEditorModel model = this.getDataSourceConfigurable().getUrlEditor().getEditorModel();
-        model.setParameter("user", user);
-        model.commit(true);
-    }
-
-    @SneakyThrows
-    private void setUseSsl(boolean useSsl) {
-        final DataSourceConfigurable configurable = this.getDataSourceConfigurable();
-        final JBCheckBox useSSLCheckBox = (JBCheckBox) FieldUtils.readField(configurable.getSshSslPanel(), "myUseSSLJBCheckBox", true);
-        useSSLCheckBox.setSelected(useSsl);
+    @Override
+    protected Object doBackgroundUpdateForSetResource(@Nullable CosmosDBAccount value) {
+        return Optional.ofNullable(value).map(CosmosDBAccount::getCosmosDBAccountPrimaryConnectionString).orElse(null);
     }
 
-    @SneakyThrows
-    private DataSourceConfigurable getDataSourceConfigurable() {
-        return (DataSourceConfigurable) FieldUtils.readField(this.getInterchange(), "myConfigurable", true);
+    @Override
+    protected boolean needsBackgroundUpdateForSetResource() {
+        return true;
     }
 
-    @Getter
-    @RequiredArgsConstructor
-    static class CosmosDbAccountComboBox extends AzureComboBox<CosmosDBAccount> {
-        private final DatabaseAccountKind kind;
-        private boolean noAccounts;
-        private Consumer<Boolean> accountsListener;
-
-        @Nullable
-        @Override
-        protected CosmosDBAccount doGetDefaultValue() {
-            return CacheManager.getUsageHistory(CosmosDBAccount.class)
-                .peek(v -> Objects.isNull(kind) || Objects.equals(kind, v.getKind()));
-        }
-
-        @Nonnull
-        @Override
-        protected List<CosmosDBAccount> loadItems() {
-            if (!Azure.az(AzureAccount.class).isLoggedIn()) {
-                return Collections.emptyList();
-            }
-            final List<CosmosDBAccount> accounts = Azure.az(AzureCosmosService.class).getDatabaseAccounts(kind).stream()
-                .filter(m -> !m.isDraftForCreating()).collect(Collectors.toList());
-            this.noAccounts = accounts.size() == 0;
-            Optional.ofNullable(this.accountsListener).ifPresent(l -> l.accept(this.noAccounts));
-            return accounts;
-        }
-
-        public void setAccountsListener(@Nonnull Consumer<Boolean> accountsListener) {
-            this.accountsListener = accountsListener;
-            accountsListener.accept(this.noAccounts);
-        }
-
-        @Override
-        protected String getItemText(Object item) {
-            return Optional.ofNullable(item).map(i -> ((CosmosDBAccount) i)).map(AbstractAzResource::getName).orElse("");
-        }
+    @Override
+    protected String getHostFromValue(CosmosDBAccount value) {
+        return Optional.ofNullable(this.connectionString).map(CosmosDBAccountConnectionString::getHost).orElse(null);
     }
-}
+}
\ No newline at end of file
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
new file mode 100644
index 00000000000..c0022cb83f9
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
@@ -0,0 +1,33 @@
+package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.common.cache.CacheManager;
+import com.microsoft.azure.toolkit.lib.cosmos.AzureCosmosService;
+import com.microsoft.azure.toolkit.lib.cosmos.CosmosDBAccount;
+import com.microsoft.azure.toolkit.lib.cosmos.model.DatabaseAccountKind;
+import lombok.RequiredArgsConstructor;
+
+import javax.annotation.Nullable;
+import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
+
+@RequiredArgsConstructor
+public class CosmosDbAccountComboBox extends DbUtilsComboBoxBase<CosmosDBAccount> {
+    private final DatabaseAccountKind kind;
+
+    @Nullable
+    @Override
+    protected CosmosDBAccount doGetDefaultValue() {
+        return CacheManager.getUsageHistory(CosmosDBAccount.class)
+                .peek(v -> Objects.isNull(kind) || Objects.equals(kind, v.getKind()));
+    }
+
+    @Override
+    protected List<CosmosDBAccount> load() {
+        return Azure.az(AzureCosmosService.class).getDatabaseAccounts(kind).stream()
+                .filter(m -> !m.isDraftForCreating()).collect(Collectors.toList());
+    }
+}
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
new file mode 100644
index 00000000000..043330a2647
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
@@ -0,0 +1,26 @@
+package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
+import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
+import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
+import com.microsoft.azure.toolkit.lib.cosmos.model.DatabaseAccountKind;
+
+import javax.annotation.Nonnull;
+
+public class CosmosNotSignedInLabel extends NotSignedInTipLabel {
+    private static final String NOT_SIGNIN_TIPS = "<html><a href=\"\">Sign in</a> to select an existing Azure Cosmos DB account.</html>";
+    @Nonnull
+    private final DatabaseAccountKind kind;
+
+    public CosmosNotSignedInLabel(@Nonnull DatabaseAccountKind kind) {
+        super(NOT_SIGNIN_TIPS);
+        this.kind = kind;
+    }
+
+    @Override
+    @AzureOperation(name = "user/cosmos.signin_from_dbtools")
+    protected void signIn() {
+        OperationContext.action().setTelemetryProperty("kind", this.kind.getValue());
+        super.signIn();
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/DbToolsWorkaround.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/DbToolsWorkaround.java
index 32c7f665194..14766e1e737 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/DbToolsWorkaround.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/DbToolsWorkaround.java
@@ -16,6 +16,7 @@
 import com.intellij.openapi.util.JDOMUtil;
 import com.intellij.openapi.util.registry.Registry;
 import com.microsoft.azure.toolkit.intellij.common.IntelliJAzureIcons;
+import com.microsoft.azure.toolkit.intellij.dbtools.DbToolsWorkarounds;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
 import kotlin.Unit;
@@ -102,19 +103,7 @@ private static void addAsUserDriver(@Nonnull final String configUri) {
         driver.setIcon(IntelliJAzureIcons.getIcon(icon));
     }
 
-    @SuppressWarnings("unchecked")
     private static void makeAccountShowAtTop() {
-        try {
-            final Field HEADS = FieldUtils.getField(ParametersLayoutUtils.class, "HEADS", true);
-            final List<String> heads = (List<String>) FieldUtils.readStaticField(HEADS, true);
-            if (!heads.contains(PARAM_NAME)) {
-                final Object[] old = heads.toArray();
-                heads.set(0, PARAM_NAME);
-                for (int i = 0; i < old.length - 1; i++) {
-                    heads.set(i + 1, (String) old[i]);
-                }
-            }
-        } catch (final Throwable ignored) {
-        }
+        DbToolsWorkarounds.makeParameterShowAtTop(PARAM_NAME);
     }
 }
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
new file mode 100644
index 00000000000..d0de745f609
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
@@ -0,0 +1,44 @@
+package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
+
+import com.intellij.openapi.project.Project;
+import com.microsoft.azure.toolkit.intellij.cosmos.creation.CreateCosmosDBAccountAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.lib.AzService;
+import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
+import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
+import com.microsoft.azure.toolkit.lib.cosmos.AzureCosmosService;
+import com.microsoft.azure.toolkit.lib.cosmos.model.DatabaseAccountKind;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.awt.event.InputEvent;
+
+public class NoAccountsTipLabel extends NoResourceTipLabel {
+    private static final String NO_ACCOUNT_TIPS_TEMPLATE = "<html>No Azure Cosmos DB accounts (%s). You can <a href=''>create one</a> first.</html>";
+    @Nonnull
+    private final DatabaseAccountKind kind;
+
+    public NoAccountsTipLabel(@Nonnull DatabaseAccountKind kind) {
+        super(String.format(NO_ACCOUNT_TIPS_TEMPLATE, kind.getValue()));
+        this.kind = kind;
+    }
+
+    @Override
+    @AzureOperation(name = "user/cosmos.create_account_from_dbtools")
+    protected void createResourceInIde(InputEvent e) {
+        OperationContext.action().setTelemetryProperty("kind", this.kind.getValue());
+        super.createResourceInIde(e);
+    }
+
+    @Nullable
+    @Override
+    protected Class<? extends AzService> getClazzForNavigationToExplorer() {
+        return AzureCosmosService.class;
+    }
+
+    @Override
+    protected void createResource(Project project) {
+        CreateCosmosDBAccountAction.create(null, null);
+    }
+}
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/build.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/build.gradle.kts
index 948aaef10ce..1164a13259e 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/build.gradle.kts
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/build.gradle.kts
@@ -2,6 +2,7 @@ dependencies {
     implementation(project(":azure-intellij-plugin-lib"))
     // runtimeOnly project(path: ":azure-intellij-plugin-lib", configuration: "instrumentedJar")
     implementation(project(":azure-intellij-resource-connector-lib"))
+    implementation(project(":azure-intellij-plugin-dbtools"))
     // runtimeOnly project(path: ":azure-intellij-resource-connector-lib", configuration: "instrumentedJar")
     implementation("com.microsoft.azure:azure-toolkit-database-lib")
     implementation("com.microsoft.azure:azure-toolkit-mysql-lib")
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseDbToolsWorkaround.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseDbToolsWorkaround.java
index 6f7304e7e05..afce3ecf175 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseDbToolsWorkaround.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseDbToolsWorkaround.java
@@ -5,13 +5,11 @@
 
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
-import com.intellij.database.dataSource.DatabaseDriverImpl;
-import com.intellij.database.dataSource.DatabaseDriverManager;
-import com.intellij.database.dataSource.url.template.UrlTemplate;
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.project.DumbAware;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.startup.ProjectActivity;
+import com.microsoft.azure.toolkit.intellij.dbtools.DbToolsWorkarounds;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
 import kotlin.Unit;
@@ -20,9 +18,6 @@
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 import java.util.HashMap;
-import java.util.LinkedList;
-import java.util.List;
-import java.util.Objects;
 
 public class DatabaseDbToolsWorkaround implements ProjectActivity, DumbAware {
 
@@ -44,46 +39,35 @@ public Object execute(@Nonnull Project project, @Nonnull Continuation<? super Un
     }
 
     private static void loadMySqlAzureTemplates() {
-        final DatabaseDriverManager manager = DatabaseDriverManager.getInstance();
-        final DatabaseDriverImpl driver = ((DatabaseDriverImpl) manager.getDriver("mysql.8"));
-        if (Objects.nonNull(driver)) {
-            final List<UrlTemplate> templates = new LinkedList<>(driver.getUrlTemplates());
-            templates.removeIf(t -> t.getTemplate().contains(DatabaseServerTypeUIFactory.MYSQL));
-            templates.add(0, new UrlTemplate("Azure", "jdbc:mysql://{host::localhost}?[:{port::3306}][/{database}?][/{account:az_mysql_server}?][\\?<&,user={user},password={password},{:identifier}={:param}>]"));
-            driver.setURLTemplates(templates);
-        }
+        DbToolsWorkarounds.loadDriverTemplate(
+                "mysql.8",
+                DatabaseServerTypeUIFactory.MYSQL,
+                "Azure",
+                "jdbc:mysql://{host::localhost}?[:{port::3306}][/{database}?][/{account:az_mysql_server}?][\\?<&,user={user},password={password},{:identifier}={:param}>]");
+
     }
 
     private static void loadPostgreSqlAzureTemplates() {
-        final DatabaseDriverManager manager = DatabaseDriverManager.getInstance();
-        final DatabaseDriverImpl driver = ((DatabaseDriverImpl) manager.getDriver("postgresql"));
-        if (Objects.nonNull(driver)) {
-            final List<UrlTemplate> templates = new LinkedList<>(driver.getUrlTemplates());
-            templates.removeIf(t -> t.getTemplate().contains(DatabaseServerTypeUIFactory.POSTGRE));
-            templates.add(0, new UrlTemplate("Azure", "jdbc:postgresql://[{host::localhost}[:{port::5432}]][/{database:database/[^?]+:postgres}?][/{account:az_postgre_server}?][\\?<&,user={user:param},password={password:param},{:identifier}={:param}>]"));
-            driver.setURLTemplates(templates);
-        }
+        DbToolsWorkarounds.loadDriverTemplate(
+                "postgresql",
+                DatabaseServerTypeUIFactory.POSTGRE,
+                "Azure",
+                "jdbc:postgresql://[{host::localhost}[:{port::5432}]][/{database:database/[^?]+:postgres}?][/{account:az_postgre_server}?][\\?<&,user={user:param},password={password:param},{:identifier}={:param}>]");
     }
 
     private static void loadSqlServerAzureTemplates() {
-        final DatabaseDriverManager manager = DatabaseDriverManager.getInstance();
-        final DatabaseDriverImpl driver = ((DatabaseDriverImpl) manager.getDriver("sqlserver.ms"));
-        if (Objects.nonNull(driver)) {
-            final List<UrlTemplate> templates = new LinkedList<>(driver.getUrlTemplates());
-            templates.removeIf(t -> t.getTemplate().contains(DatabaseServerTypeUIFactory.SQLSERVER));
-            templates.add(0, new UrlTemplate("Azure", "jdbc:sqlserver://{host:ssrp_host:localhost}[\\\\{instance:ssrp_instance}][:{port:ssrp_port}][/{account:az_sqlserver_server}?][;<;,user[Name]={user:param},password={password:param},database[Name]={database},{:identifier}={:param}>];?"));
-            driver.setURLTemplates(templates);
-        }
+        DbToolsWorkarounds.loadDriverTemplate(
+                "sqlserver.ms",
+                DatabaseServerTypeUIFactory.SQLSERVER,
+                "Azure",
+                "jdbc:sqlserver://{host:ssrp_host:localhost}[\\\\{instance:ssrp_instance}][:{port:ssrp_port}][/{account:az_sqlserver_server}?][;<;,user[Name]={user:param},password={password:param},database[Name]={database},{:identifier}={:param}>];?");
     }
 
     private static void loadAzureSqlDatabaseAzureTemplates() {
-        final DatabaseDriverManager manager = DatabaseDriverManager.getInstance();
-        final DatabaseDriverImpl driver = ((DatabaseDriverImpl) manager.getDriver("azure.ms"));
-        if (Objects.nonNull(driver)) {
-            final List<UrlTemplate> templates = new LinkedList<>(driver.getUrlTemplates());
-            templates.removeIf(t -> t.getTemplate().contains(DatabaseServerTypeUIFactory.SQLSERVER));
-            templates.add(0, new UrlTemplate("Azure", "jdbc:sqlserver://{host:host_ipv6:server.database.windows.net}[:{port::1433}][/{account:az_sqlserver_server}?][;<;,user[Name]={user:param},password={password:param},database[Name]={database},{:identifier}={:param}>];?"));
-            driver.setURLTemplates(templates);
-        }
+        DbToolsWorkarounds.loadDriverTemplate(
+                "azure.ms",
+                DatabaseServerTypeUIFactory.SQLSERVER,
+                "Azure",
+                "jdbc:sqlserver://{host:host_ipv6:server.database.windows.net}[:{port::1433}][/{account:az_sqlserver_server}?][;<;,user[Name]={user:param},password={password:param},database[Name]={database},{:identifier}={:param}>];?");
     }
 }
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
new file mode 100644
index 00000000000..2a262679aa9
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
@@ -0,0 +1,25 @@
+package com.microsoft.azure.toolkit.intellij.database.dbtools;
+
+
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
+import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
+import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
+import javax.annotation.Nonnull;
+
+public class DatabaseNotSignedInLabel extends NotSignedInTipLabel {
+    private static final String NOT_SIGNIN_TIPS = "<html><a href=\"\">Sign in</a> to select an existing %s server in Azure.</html>";
+    @Nonnull
+    private final DatabaseServerClass databaseServerClass;
+
+    public DatabaseNotSignedInLabel(@Nonnull DatabaseServerClass databaseServerClass) {
+        super(String.format(NOT_SIGNIN_TIPS, databaseServerClass.getServerName()));
+        this.databaseServerClass = databaseServerClass;
+    }
+
+    @Override
+    @AzureOperation(name = "user/$database.signin_from_dbtools")
+    protected void signIn() {
+        OperationContext.current().setTelemetryProperty("serviceName", databaseServerClass.getServiceName());
+        super.signIn();
+    }
+}
\ No newline at end of file
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
new file mode 100644
index 00000000000..8bc3de36c3e
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
@@ -0,0 +1,39 @@
+package com.microsoft.azure.toolkit.intellij.database.dbtools;
+
+import com.microsoft.azure.toolkit.lib.database.entity.IDatabaseServer;
+import com.microsoft.azure.toolkit.lib.mysql.MySqlServer;
+import com.microsoft.azure.toolkit.lib.postgre.PostgreSqlServer;
+import com.microsoft.azure.toolkit.lib.sqlserver.MicrosoftSqlServer;
+import lombok.Getter;
+
+public enum DatabaseServerClass {
+    MySql(MySqlServer.class),
+    MsSql(MicrosoftSqlServer.class),
+    Postgres(PostgreSqlServer.class);
+
+    @Getter
+    private final Class<? extends IDatabaseServer<?>> clazz;
+
+    DatabaseServerClass(Class<? extends IDatabaseServer<?>> clazz) {
+        this.clazz = clazz;
+    }
+
+    public String getServerName() {
+        return switch (this) {
+            case MySql -> "MySQL";
+            case MsSql -> "Microsoft SQL";
+            case Postgres -> "PostgreSQL";
+            default -> "Database";
+        };
+    }
+
+    @javax.annotation.Nullable
+    public String getServiceName() {
+        return switch (this) {
+            case MySql -> "mysql";
+            case MsSql -> "sqlserver";
+            case Postgres -> "postgre";
+            default -> null;
+        };
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerConnectionInterceptor.java
index aabc01bd122..21ae1a97c2d 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerConnectionInterceptor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerConnectionInterceptor.java
@@ -1,43 +1,9 @@
-/*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for license information.
- */
-
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
-import com.azure.resourcemanager.resources.fluentcore.arm.ResourceId;
-import com.intellij.database.dataSource.DatabaseConnectionInterceptor;
-import com.intellij.database.dataSource.DatabaseConnectionPoint;
-import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
-import com.microsoft.azure.toolkit.lib.common.operation.Operation;
-import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
-import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
-import org.apache.commons.lang3.StringUtils;
-
-import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.concurrent.CompletionStage;
-
-import static com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter.*;
+import com.microsoft.azure.toolkit.intellij.dbtools.TelemetryConnectionInterceptor;
 
-@SuppressWarnings("UnstableApiUsage")
-public class DatabaseServerConnectionInterceptor implements DatabaseConnectionInterceptor {
-    @Nullable
-    public CompletionStage<ProtoConnection> intercept(@Nonnull DatabaseConnectionInterceptor.ProtoConnection proto, boolean silent) {
-        final DatabaseConnectionPoint point = proto.getConnectionPoint();
-        final String accountId = point.getAdditionalProperty(DatabaseServerParamEditor.KEY_DB_SERVER_ID);
-        if (StringUtils.isNotBlank(accountId) && !StringUtils.equalsIgnoreCase(accountId, DatabaseServerParamEditor.NONE)) {
-            final Map<String, String> properties = new HashMap<>();
-            final ResourceId id = ResourceId.fromString(accountId);
-            properties.put("subscriptionId", id.subscriptionId());
-            properties.put(SERVICE_NAME, "database");
-            properties.put(OPERATION_NAME, "connect_jdbc_from_dbtools");
-            properties.put(OP_NAME, "database.connect_jdbc_from_dbtools");
-            properties.put(OP_TYPE, Operation.Type.USER);
-            AzureTelemeter.log(AzureTelemetry.Type.OP_END, properties);
-        }
-        return null;
+public class DatabaseServerConnectionInterceptor extends TelemetryConnectionInterceptor {
+    protected DatabaseServerConnectionInterceptor() {
+        super(DatabaseServerParamEditor.KEY_DB_SERVER_ID, "database", "database.connect_jdbc_from_dbtools");
     }
-}
\ No newline at end of file
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerParamEditor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerParamEditor.java
index 0a34bde7e73..e22333a3c01 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerParamEditor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerParamEditor.java
@@ -1,286 +1,81 @@
-/*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for license information.
- */
-
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
-import com.intellij.database.dataSource.DataSourceConfigurable;
 import com.intellij.database.dataSource.LocalDataSource;
 import com.intellij.database.dataSource.url.DataInterchange;
 import com.intellij.database.dataSource.url.FieldSize;
-import com.intellij.database.dataSource.url.template.UrlEditorModel;
-import com.intellij.database.dataSource.url.ui.ParamEditorBase;
-import com.intellij.icons.AllIcons;
-import com.intellij.ide.DataManager;
-import com.intellij.openapi.actionSystem.AnActionEvent;
-import com.intellij.openapi.actionSystem.CommonDataKeys;
-import com.intellij.openapi.actionSystem.DataContext;
-import com.intellij.openapi.actionSystem.EmptyAction;
-import com.intellij.openapi.project.Project;
-import com.intellij.openapi.wm.ToolWindow;
-import com.intellij.openapi.wm.ToolWindowManager;
-import com.intellij.ui.ComponentUtil;
-import com.intellij.ui.HyperlinkLabel;
-import com.microsoft.azure.toolkit.ide.common.action.ResourceCommonActionsContributor;
-import com.microsoft.azure.toolkit.intellij.common.AzureComboBox;
-import com.microsoft.azure.toolkit.intellij.database.mysql.creation.CreateMySqlAction;
-import com.microsoft.azure.toolkit.intellij.database.postgre.creation.CreatePostgreSqlAction;
-import com.microsoft.azure.toolkit.intellij.database.sqlserver.creation.CreateSqlServerAction;
-import com.microsoft.azure.toolkit.lib.Azure;
-import com.microsoft.azure.toolkit.lib.auth.AzureAccount;
-import com.microsoft.azure.toolkit.lib.common.action.Action;
-import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
-import com.microsoft.azure.toolkit.lib.common.cache.CacheManager;
-import com.microsoft.azure.toolkit.lib.common.model.AzResource;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.AzureParamsEditorBase;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
 import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
-import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
-import com.microsoft.azure.toolkit.lib.common.task.AzureTask;
-import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
 import com.microsoft.azure.toolkit.lib.database.entity.IDatabaseServer;
-import com.microsoft.azure.toolkit.lib.mysql.AzureMySql;
-import com.microsoft.azure.toolkit.lib.mysql.MySqlServer;
-import com.microsoft.azure.toolkit.lib.postgre.AzurePostgreSql;
-import com.microsoft.azure.toolkit.lib.postgre.PostgreSqlServer;
-import com.microsoft.azure.toolkit.lib.sqlserver.AzureSqlServer;
-import com.microsoft.azure.toolkit.lib.sqlserver.MicrosoftSqlServer;
-import lombok.Getter;
-import lombok.RequiredArgsConstructor;
-import lombok.Setter;
-import lombok.SneakyThrows;
-import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
-import org.apache.commons.lang3.reflect.FieldUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
-import javax.swing.*;
-import java.awt.*;
-import java.awt.event.InputEvent;
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
-import java.util.function.Consumer;
-import java.util.stream.Collectors;
 
-public class DatabaseServerParamEditor extends ParamEditorBase<DatabaseServerParamEditor.SqlDbServerComboBox> {
+public class DatabaseServerParamEditor extends AzureParamsEditorBase<SqlDbServerComboBox, IDatabaseServer<?>> {
     public static final String KEY_DB_SERVER_ID = "AZURE_SQL_DB_SERVER";
-    public static final String NONE = "<NONE>";
-    public static final String NO_SERVERS_TIPS = "<html>No existing %s servers in Azure. You can <a href=''>create one</a> first.</html>";
-    public static final String NOT_SIGNIN_TIPS = "<html><a href=\"\">Sign in</a> to select an existing %s server in Azure.</html>";
-    private final Class<? extends IDatabaseServer<?>> clazz;
-    @Getter
-    @Setter
-    private String text = "";
-    private boolean updating;
-
-    public DatabaseServerParamEditor(@Nonnull Class<? extends IDatabaseServer<?>> clazz, @Nonnull String label, @Nonnull DataInterchange interchange) {
-        super(new SqlDbServerComboBox(clazz), interchange, FieldSize.LARGE, label);
-        this.clazz = clazz;
+    @Nonnull
+    private final DatabaseServerClass databaseServerClass;
+
+    public DatabaseServerParamEditor(
+            @Nonnull DatabaseServerClass databaseServerClass,
+            @Nullable String caption,
+            @NotNull DataInterchange interchange,
+            AnAction @NotNull ... actions) {
+        super(new SqlDbServerComboBox(databaseServerClass), interchange, FieldSize.LARGE, caption, KEY_DB_SERVER_ID, actions);
+        this.databaseServerClass = databaseServerClass;
         final LocalDataSource dataSource = getDataSourceConfigurable().getDataSource();
-        final SqlDbServerComboBox combox = this.getEditorComponent();
-        combox.addValueChangedListener(this::setServer);
-        interchange.addPersistentProperty(KEY_DB_SERVER_ID);
-        final String serverId = interchange.getProperty(KEY_DB_SERVER_ID);
+
+        final SqlDbServerComboBox comboBox = getEditorComponent();
         final boolean isModifying = StringUtils.isNotBlank(dataSource.getUsername());
+        final String serverId = interchange.getProperty(KEY_DB_SERVER_ID);
+
         if (isModifying && Objects.nonNull(serverId)) {
-            combox.setValue(new AzureComboBox.ItemReference<>(i -> i.getId().equals(serverId)));
+            comboBox.setByResourceId(serverId);
         } else if (Objects.isNull(serverId)) {
-            combox.setValue((IDatabaseServer<?>) null);
+            comboBox.setNull();
         }
-
-        interchange.addPropertyChangeListener((evt -> onPropertiesChanged(evt.getPropertyName(), evt.getNewValue())), this);
     }
 
     @Override
-    protected @Nonnull JComponent createComponent(SqlDbServerComboBox combox) {
-        final JPanel container = new JPanel();
-        final BoxLayout layout = new BoxLayout(container, BoxLayout.Y_AXIS);
-        container.setLayout(layout);
-
-        combox.setAlignmentX(Component.LEFT_ALIGNMENT);
-        container.add(combox);
-
-        if (!Azure.az(AzureAccount.class).isLoggedIn()) {
-            final HyperlinkLabel notSignInTips = initNotSignInTipsLabel(combox);
-            container.add(notSignInTips);
-        }
-
-        final HyperlinkLabel noServersTips = initNoServerTipsLabel(combox);
-        container.add(noServersTips);
-        return container;
-    }
-
-    @AzureOperation(name = "user/$database.signin_from_dbtools")
-    private void signInAndReloadItems(SqlDbServerComboBox combox, HyperlinkLabel notSignInTips) {
-        OperationContext.current().setTelemetryProperty("serviceName", getServiceName());
-        AzureActionManager.getInstance().getAction(Action.REQUIRE_AUTH).handle((a) -> {
-            notSignInTips.setVisible(false);
-            combox.reloadItems();
-        });
-    }
-
-    @Nonnull
-    private HyperlinkLabel initNotSignInTipsLabel(SqlDbServerComboBox combox) {
-        final HyperlinkLabel label = new HyperlinkLabel();
-        label.setHtmlText(String.format(NOT_SIGNIN_TIPS, getName(combox.getClazz())));
-        label.setIcon(AllIcons.General.Information);
-        label.addHyperlinkListener(e -> signInAndReloadItems(combox, label));
-        label.setAlignmentX(Component.LEFT_ALIGNMENT);
-        return label;
+    protected NotSignedInTipLabel createNotSignedInTipLabel() {
+        return new DatabaseNotSignedInLabel(databaseServerClass);
     }
 
-    private HyperlinkLabel initNoServerTipsLabel(SqlDbServerComboBox combox) {
-        final HyperlinkLabel label = new HyperlinkLabel();
-        label.setHtmlText(String.format(NO_SERVERS_TIPS, getName(combox.getClazz())));
-        label.setIcon(AllIcons.General.Information);
-        label.addHyperlinkListener(e -> createServerInIde(e.getInputEvent()));
-        label.setAlignmentX(Component.LEFT_ALIGNMENT);
-        combox.setServerListener(label::setVisible);
-        return label;
-    }
-
-    @AzureOperation(name = "user/$database.create_server_from_dbtools")
-    private void createServerInIde(InputEvent e) {
-        OperationContext.current().setTelemetryProperty("serviceName", getServiceName());
-        final DataContext context = DataManager.getInstance().getDataContext(e.getComponent());
-        final Project project = context.getData(CommonDataKeys.PROJECT);
-        final Window window = ComponentUtil.getActiveWindow();
-        window.setVisible(false);
-        window.dispose();
-        final ToolWindow explorer = ToolWindowManager.getInstance(Objects.requireNonNull(project)).getToolWindow("Azure Explorer");
-        Objects.requireNonNull(explorer).activate(() -> {
-            final AnActionEvent event = AnActionEvent.createFromAnAction(new EmptyAction(), e, "database.dbtools", context);
-            if (MySqlServer.class.isAssignableFrom(clazz)) {
-                AzureActionManager.getInstance().getAction(ResourceCommonActionsContributor.SELECT_RESOURCE_IN_EXPLORER).handle(Azure.az(AzureMySql.class), event);
-                CreateMySqlAction.create(project, null);
-            } else if (PostgreSqlServer.class.isAssignableFrom(clazz)) {
-                AzureActionManager.getInstance().getAction(ResourceCommonActionsContributor.SELECT_RESOURCE_IN_EXPLORER).handle(Azure.az(AzurePostgreSql.class), event);
-                CreatePostgreSqlAction.create(project, null);
-            } else if (MicrosoftSqlServer.class.isAssignableFrom(clazz)) {
-                AzureActionManager.getInstance().getAction(ResourceCommonActionsContributor.SELECT_RESOURCE_IN_EXPLORER).handle(Azure.az(AzureSqlServer.class), event);
-                CreateSqlServerAction.create(project, null);
-            }
-        });
+    @Override
+    protected NoResourceTipLabel createNoResourceTipLabel() {
+        return new NoServersTipLabel(databaseServerClass);
     }
 
-    private void onPropertiesChanged(String propertyName, Object newValue) {
-        if (!this.updating && StringUtils.isNotEmpty((String) newValue) && StringUtils.equals(propertyName, "host")) {
-            final SqlDbServerComboBox combox = this.getEditorComponent();
-            final IDatabaseServer<?> server = combox.getValue();
-            if (Objects.nonNull(server) && !Objects.equals(server.getJdbcUrl().getServerHost(), newValue)) {
-                combox.setValue((IDatabaseServer<?>) null);
-                this.setServer(null);
-            }
+    @Override
+    @AzureOperation(name = "user/$database.select_server_dbtools.server", params = {"value.getName()"}, source = "value")
+    protected void setResource(
+            @Nonnull DataInterchange interchange,
+            @Nullable Object fromBackground,
+            @Nullable IDatabaseServer<?> value,
+            @Nullable String oldResourceId,
+            @Nullable String newResourceId) {
+        // null will not be saved at com/intellij/database/dataSource/url/ui/DynamicJdbcUrlEditor#storeProperties
+        interchange.putProperty(KEY_DB_SERVER_ID, Optional.ofNullable(newResourceId).orElse(NONE));
+        if (Objects.isNull(value) || Objects.equals(oldResourceId, newResourceId)) {
+            return;
         }
+        final String user = value.getFullAdminName();
+        LocalDataSource.setUsername(interchange.getDataSource(), user);
+        this.setUsername(user);
+        this.setJdbcUrl(value.getJdbcUrl().toString());
     }
 
-    @AzureOperation(name = "user/$database.select_server_dbtools.server", params = {"server.getName()"}, source = "server")
-    private void setServer(@Nullable IDatabaseServer<?> server) {
-        final DataInterchange interchange = this.getInterchange();
-        final String oldServerId = interchange.getProperty(KEY_DB_SERVER_ID);
-        final String newServerId = Optional.ofNullable(server).map(IDatabaseServer::getId).orElse(null);
-        this.updating = true;
-        AzureTaskManager.getInstance().runLater(() -> {
-            // null will not be saved at com/intellij/database/dataSource/url/ui/DynamicJdbcUrlEditor#storeProperties
-            interchange.putProperty(KEY_DB_SERVER_ID, Optional.ofNullable(newServerId).orElse(NONE));
-            if (Objects.isNull(server) || Objects.equals(oldServerId, newServerId)) {
-                this.updating = false;
-                return;
-            }
-            final String user = server.getFullAdminName();
-            LocalDataSource.setUsername(interchange.getDataSource(), user);
-            this.setUsername(user);
-            this.setJdbcUrl(server.getJdbcUrl().toString());
-            this.updating = false;
-        }, AzureTask.Modality.ANY);
-    }
-
-    private void setUsername(String user) {
-        final UrlEditorModel model = this.getDataSourceConfigurable().getUrlEditor().getEditorModel();
-        model.setParameter("user", user);
-        model.commit(true);
-    }
-
-    private void setJdbcUrl(String url) {
-        final UrlEditorModel model = this.getDataSourceConfigurable().getUrlEditor().getEditorModel();
-        model.setUrl(url);
-        model.commit(true);
-    }
-
-    @SneakyThrows
-    private DataSourceConfigurable getDataSourceConfigurable() {
-        return (DataSourceConfigurable) FieldUtils.readField(this.getInterchange(), "myConfigurable", true);
-    }
-
-    private String getName(Class<? extends IDatabaseServer<?>> clazz) {
-        if (MySqlServer.class.isAssignableFrom(clazz)) {
-            return "MySQL";
-        } else if (PostgreSqlServer.class.isAssignableFrom(clazz)) {
-            return "PostgreSQL";
-        } else if (MicrosoftSqlServer.class.isAssignableFrom(clazz)) {
-            return "Microsoft SQL";
-        }
-        return "Database";
+    @Override
+    protected String getHostFromValue(IDatabaseServer<?> value) {
+        return value.getJdbcUrl().getServerHost();
     }
+}
 
-    @Getter
-    @RequiredArgsConstructor
-    static class SqlDbServerComboBox extends AzureComboBox<IDatabaseServer<?>> {
-        private final Class<? extends IDatabaseServer<?>> clazz;
-        private boolean noServers;
-        private Consumer<Boolean> serverListener;
-
-        @Nullable
-        @Override
-        protected IDatabaseServer<?> doGetDefaultValue() {
-            return CacheManager.getUsageHistory(clazz).peek();
-        }
-
-        @Nonnull
-        @Override
-        protected List<IDatabaseServer<?>> loadItems() {
-            if (!Azure.az(AzureAccount.class).isLoggedIn()) {
-                return Collections.emptyList();
-            }
-            List<IDatabaseServer<?>> servers = new ArrayList<>();
-            if (MySqlServer.class.isAssignableFrom(clazz)) {
-                servers = Azure.az(AzureMySql.class).servers().stream()
-                    .filter(s -> s.getFormalStatus().isRunning()).collect(Collectors.toList());
-            } else if (MicrosoftSqlServer.class.isAssignableFrom(clazz)) {
-                servers = Azure.az(AzureSqlServer.class).servers().stream()
-                    .filter(s -> s.getFormalStatus().isRunning()).collect(Collectors.toList());
-            } else if (PostgreSqlServer.class.isAssignableFrom(clazz)) {
-                servers = Azure.az(AzurePostgreSql.class).servers().stream()
-                    .filter(s -> s.getFormalStatus().isRunning()).collect(Collectors.toList());
-            }
-            this.noServers = CollectionUtils.isEmpty(servers);
-            Optional.ofNullable(this.serverListener).ifPresent(l -> l.accept(this.noServers));
-            return servers;
-        }
-
-        public void setServerListener(@Nonnull Consumer<Boolean> serverListener) {
-            this.serverListener = serverListener;
-            serverListener.accept(this.noServers);
-        }
 
-        @Override
-        protected String getItemText(Object item) {
-            return Optional.ofNullable(item).map(i -> ((IDatabaseServer<?>) i)).map(AzResource::getName).orElse("");
-        }
-    }
 
-    @Nullable
-    private String getServiceName() {
-        if (MySqlServer.class.isAssignableFrom(clazz)) {
-            return "mysql";
-        } else if (PostgreSqlServer.class.isAssignableFrom(clazz)) {
-            return "postgre";
-        } else if (MicrosoftSqlServer.class.isAssignableFrom(clazz)) {
-            return "sqlserver";
-        }
-        return null;
-    }
-}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerTypeUIFactory.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerTypeUIFactory.java
index 3afae1f05f0..c57d1d17ece 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerTypeUIFactory.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerTypeUIFactory.java
@@ -29,17 +29,17 @@ public class DatabaseServerTypeUIFactory implements TypesRegistryUi.TypeDescript
     public void createTypeDescriptor(@NotNull Consumer<? super TypeDescriptorUi> consumer) {
         final BaseTypeDescriptor mysql = new BaseTypeDescriptor(MYSQL) {
             protected ParamEditor createFieldImpl(@Nonnull String caption, @Nullable String configuration, @Nonnull DataInterchange interchange) {
-                return new DatabaseServerParamEditor(MySqlServer.class, formatFieldCaption("Server"), interchange);
+                return new DatabaseServerParamEditor(DatabaseServerClass.MySql, formatFieldCaption("Server"), interchange);
             }
         };
         final BaseTypeDescriptor sqlserver = new BaseTypeDescriptor(SQLSERVER) {
             protected ParamEditor createFieldImpl(@Nonnull String caption, @Nullable String configuration, @Nonnull DataInterchange interchange) {
-                return new DatabaseServerParamEditor(MicrosoftSqlServer.class, formatFieldCaption("Server"), interchange);
+                return new DatabaseServerParamEditor(DatabaseServerClass.MsSql, formatFieldCaption("Server"), interchange);
             }
         };
         final BaseTypeDescriptor postgre = new BaseTypeDescriptor(POSTGRE) {
             protected ParamEditor createFieldImpl(@Nonnull String caption, @Nullable String configuration, @Nonnull DataInterchange interchange) {
-                return new DatabaseServerParamEditor(PostgreSqlServer.class, formatFieldCaption("Server"), interchange);
+                return new DatabaseServerParamEditor(DatabaseServerClass.Postgres, formatFieldCaption("Server"), interchange);
             }
         };
         consumer.consume(mysql);
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
new file mode 100644
index 00000000000..3fdf4229d8c
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
@@ -0,0 +1,55 @@
+package com.microsoft.azure.toolkit.intellij.database.dbtools;
+
+import com.intellij.openapi.project.Project;
+import com.microsoft.azure.toolkit.intellij.database.mysql.creation.CreateMySqlAction;
+import com.microsoft.azure.toolkit.intellij.database.postgre.creation.CreatePostgreSqlAction;
+import com.microsoft.azure.toolkit.intellij.database.sqlserver.creation.CreateSqlServerAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.lib.AzService;
+import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
+import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
+import com.microsoft.azure.toolkit.lib.mysql.AzureMySql;
+import com.microsoft.azure.toolkit.lib.postgre.AzurePostgreSql;
+import com.microsoft.azure.toolkit.lib.sqlserver.AzureSqlServer;
+import org.jetbrains.annotations.Nullable;
+
+import javax.annotation.Nonnull;
+import java.awt.event.InputEvent;
+
+public class NoServersTipLabel extends NoResourceTipLabel {
+    private static final String NO_SERVERS_TIPS = "<html>No existing %s servers in Azure. You can <a href=''>create one</a> first.</html>";
+    @Nonnull
+    private final DatabaseServerClass databaseServerClass;
+
+    public NoServersTipLabel(@Nonnull DatabaseServerClass databaseServerClass) {
+        super(String.format(NO_SERVERS_TIPS, databaseServerClass.getServerName()));
+        this.databaseServerClass = databaseServerClass;
+    }
+
+    @Override
+    @AzureOperation(name = "user/$database.create_server_from_dbtools")
+    protected void createResourceInIde(InputEvent e) {
+        OperationContext.current().setTelemetryProperty("serviceName", databaseServerClass.getServiceName());
+        super.createResourceInIde(e);
+    }
+
+    @Nullable
+    @Override
+    protected Class<? extends AzService> getClazzForNavigationToExplorer() {
+        return switch (databaseServerClass) {
+            case MySql -> AzureMySql.class;
+            case MsSql -> AzureSqlServer.class;
+            case Postgres -> AzurePostgreSql.class;
+            default -> null;
+        };
+    }
+
+    @Override
+    protected void createResource(Project project) {
+        switch (databaseServerClass) {
+            case MySql -> CreateMySqlAction.create(project, null);
+            case MsSql -> CreateSqlServerAction.create(project, null);
+            case Postgres -> CreatePostgreSqlAction.create(project, null);
+        };
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/OpenWithDatabaseToolsAction.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/OpenWithDatabaseToolsAction.java
index 03132adccd4..9a6dd902f84 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/OpenWithDatabaseToolsAction.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/OpenWithDatabaseToolsAction.java
@@ -11,16 +11,25 @@
 import com.intellij.database.psi.DbPsiFacade;
 import com.intellij.database.view.ui.DataSourceManagerDialog;
 import com.intellij.openapi.project.Project;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabasePlugin;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabaseTools;
 import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
 import com.microsoft.azure.toolkit.lib.common.operation.OperationContext;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
 import com.microsoft.azure.toolkit.lib.database.entity.IDatabaseServer;
 import com.microsoft.azure.toolkit.lib.mysql.MySqlServer;
 import com.microsoft.azure.toolkit.lib.postgre.PostgreSqlServer;
 import com.microsoft.azure.toolkit.lib.sqlserver.MicrosoftSqlServer;
 
+import javax.annotation.Nonnull;
+
 public class OpenWithDatabaseToolsAction {
+    public static void openDatabaseTool(Project project, @Nonnull IDatabaseServer<?> server) {
+        DatabaseTools.openDatabaseTool(project, server, OpenWithDatabaseToolsAction::openDataSourceManagerDialog);
+    }
+
     @AzureOperation(name = "boundary/$database.open_database_tools.server", params = {"server.getName()"}, source = "server")
-    public static void openDataSourceManagerDialog(IDatabaseServer<?> server, Project project) {
+    private static void openDataSourceManagerDialog(IDatabaseServer<?> server, Project project) {
         final DataSourceRegistry registry = new DataSourceRegistry(project);
         final DbPsiFacade dbPsiFacade = DbPsiFacade.getInstance(project);
         final DataSourceDetector.Builder builder = registry.getBuilder()
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
new file mode 100644
index 00000000000..d6632d6c409
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
@@ -0,0 +1,44 @@
+package com.microsoft.azure.toolkit.intellij.database.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.common.cache.CacheManager;
+import com.microsoft.azure.toolkit.lib.database.entity.IDatabaseServer;
+import com.microsoft.azure.toolkit.lib.mysql.AzureMySql;
+import com.microsoft.azure.toolkit.lib.postgre.AzurePostgreSql;
+import com.microsoft.azure.toolkit.lib.sqlserver.AzureSqlServer;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Getter
+@RequiredArgsConstructor
+public class SqlDbServerComboBox extends DbUtilsComboBoxBase<IDatabaseServer<?>> {
+    @Nonnull
+    private final DatabaseServerClass databaseServerClass;
+
+    @Nullable
+    @Override
+    protected IDatabaseServer<?> doGetDefaultValue() {
+        return CacheManager.getUsageHistory(databaseServerClass.getClazz()).peek();
+    }
+
+    @Override
+    protected List<IDatabaseServer<?>> load() {
+        final List<? extends IDatabaseServer<?>> allServers = switch (databaseServerClass) {
+            case MySql -> Azure.az(AzureMySql.class).servers();
+            case MsSql -> Azure.az(AzureSqlServer.class).servers();
+            case Postgres -> Azure.az(AzurePostgreSql.class).servers();
+        };
+
+        return getRunningServers(allServers);
+    }
+
+    private List<IDatabaseServer<?>> getRunningServers(List<? extends IDatabaseServer<?>> allServers) {
+        return allServers.stream().filter(s -> s.getFormalStatus().isRunning()).collect(Collectors.toList());
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/mysql/IntellijMySqlActionsContributor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/mysql/IntellijMySqlActionsContributor.java
index a327ad7a88b..db6c0e20cd6 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/mysql/IntellijMySqlActionsContributor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/mysql/IntellijMySqlActionsContributor.java
@@ -33,6 +33,8 @@
 import java.util.function.BiConsumer;
 import java.util.function.BiPredicate;
 
+import static com.microsoft.azure.toolkit.intellij.database.dbtools.OpenWithDatabaseToolsAction.openDatabaseTool;
+
 public class IntellijMySqlActionsContributor implements IActionsContributor {
     private static final String NAME_PREFIX = "MySQL - %s";
     private static final String DEFAULT_DRIVER_CLASS_NAME = "com.mysql.cj.jdbc.Driver";
@@ -65,17 +67,6 @@ public void registerHandlers(AzureActionManager am) {
         am.registerHandler(MySqlActionsContributor.OPEN_DATABASE_TOOL, (r, e) -> true, openDatabaseHandler);
     }
 
-    private void openDatabaseTool(Project project, @Nonnull MySqlServer server) {
-        final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
-        final String DATABASE_PLUGIN_NOT_INSTALLED = "\"Database tools and SQL\" plugin is not installed.";
-        final String NOT_SUPPORT_ERROR_ACTION = "\"Database tools and SQL\" plugin is only provided in IntelliJ Ultimate edition.";
-        if (PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null) {
-            final Action<Object> tryUltimate = AzureActionManager.getInstance().getAction(IntellijActionsContributor.TRY_ULTIMATE).bind(server);
-            throw new AzureToolkitRuntimeException(DATABASE_PLUGIN_NOT_INSTALLED, NOT_SUPPORT_ERROR_ACTION, tryUltimate);
-        }
-        AzureTaskManager.getInstance().runLater(() -> OpenWithDatabaseToolsAction.openDataSourceManagerDialog(server, project));
-    }
-
     @Override
     public int getOrder() {
         return MySqlActionsContributor.INITIALIZE_ORDER + 1;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/postgre/IntellijPostgreSqlActionsContributor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/postgre/IntellijPostgreSqlActionsContributor.java
index 3999ea214cc..4fdf1609785 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/postgre/IntellijPostgreSqlActionsContributor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/postgre/IntellijPostgreSqlActionsContributor.java
@@ -33,6 +33,8 @@
 import java.util.function.BiConsumer;
 import java.util.function.BiPredicate;
 
+import static com.microsoft.azure.toolkit.intellij.database.dbtools.OpenWithDatabaseToolsAction.openDatabaseTool;
+
 public class IntellijPostgreSqlActionsContributor implements IActionsContributor {
     private static final String NAME_PREFIX = "PostgreSQL - %s";
     private static final String DEFAULT_DRIVER_CLASS_NAME = "org.postgresql.Driver";
@@ -65,17 +67,6 @@ public void registerHandlers(AzureActionManager am) {
         am.registerHandler(PostgreSqlActionsContributor.OPEN_DATABASE_TOOL, (r, e) -> true, openDatabaseHandler);
     }
 
-    private void openDatabaseTool(Project project, @Nonnull PostgreSqlServer server) {
-        final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
-        final String DATABASE_PLUGIN_NOT_INSTALLED = "\"Database tools and SQL\" plugin is not installed.";
-        final String NOT_SUPPORT_ERROR_ACTION = "\"Database tools and SQL\" plugin is only provided in IntelliJ Ultimate edition.";
-        if (PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null) {
-            final Action<Object> tryUltimate = AzureActionManager.getInstance().getAction(IntellijActionsContributor.TRY_ULTIMATE).bind(server);
-            throw new AzureToolkitRuntimeException(DATABASE_PLUGIN_NOT_INSTALLED, NOT_SUPPORT_ERROR_ACTION, tryUltimate);
-        }
-        AzureTaskManager.getInstance().runLater(() -> OpenWithDatabaseToolsAction.openDataSourceManagerDialog(server, project));
-    }
-
     @Override
     public int getOrder() {
         return PostgreSqlActionsContributor.INITIALIZE_ORDER + 1;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/sqlserver/IntellijSqlServerActionsContributor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/sqlserver/IntellijSqlServerActionsContributor.java
index f658c2c7bac..21a96a4b6d9 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/sqlserver/IntellijSqlServerActionsContributor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/sqlserver/IntellijSqlServerActionsContributor.java
@@ -19,6 +19,7 @@
 import com.microsoft.azure.toolkit.intellij.database.sqlserver.connection.SqlServerDatabaseResourceDefinition;
 import com.microsoft.azure.toolkit.intellij.database.sqlserver.creation.CreateSqlServerAction;
 import com.microsoft.azure.toolkit.intellij.database.sqlserver.creation.SqlServerCreationDialog;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabasePlugin;
 import com.microsoft.azure.toolkit.lib.common.action.Action;
 import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
 import com.microsoft.azure.toolkit.lib.common.exception.AzureToolkitRuntimeException;
@@ -33,6 +34,8 @@
 import java.util.function.BiConsumer;
 import java.util.function.BiPredicate;
 
+import static com.microsoft.azure.toolkit.intellij.database.dbtools.OpenWithDatabaseToolsAction.openDatabaseTool;
+
 public class IntellijSqlServerActionsContributor implements IActionsContributor {
     private static final String NAME_PREFIX = "SQL Server - %s";
     private static final String DEFAULT_DRIVER_CLASS_NAME = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
@@ -65,17 +68,6 @@ public void registerHandlers(AzureActionManager am) {
         am.registerHandler(SqlServerActionsContributor.OPEN_DATABASE_TOOL, (r, e) -> true, openDatabaseHandler);
     }
 
-    private void openDatabaseTool(Project project, @Nonnull MicrosoftSqlServer server) {
-        final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
-        final String DATABASE_PLUGIN_NOT_INSTALLED = "\"Database tools and SQL\" plugin is not installed.";
-        final String NOT_SUPPORT_ERROR_ACTION = "\"Database tools and SQL\" plugin is only provided in IntelliJ Ultimate edition.";
-        if (PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null) {
-            final Action<Object> tryUltimate = AzureActionManager.getInstance().getAction(IntellijActionsContributor.TRY_ULTIMATE).bind(server);
-            throw new AzureToolkitRuntimeException(DATABASE_PLUGIN_NOT_INSTALLED, NOT_SUPPORT_ERROR_ACTION, tryUltimate);
-        }
-        AzureTaskManager.getInstance().runLater(() -> OpenWithDatabaseToolsAction.openDataSourceManagerDialog(server, project));
-    }
-
     @Override
     public int getOrder() {
         return SqlServerActionsContributor.INITIALIZE_ORDER + 1;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/build.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/build.gradle.kts
new file mode 100644
index 00000000000..fed754bd128
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/build.gradle.kts
@@ -0,0 +1,10 @@
+dependencies {
+    implementation(project(":azure-intellij-plugin-lib"))
+    implementation(project(":azure-intellij-resource-connector-lib"))
+    implementation("com.microsoft.azure:azure-toolkit-ide-common-lib")
+    intellijPlatform {
+        // Plugin Dependencies. Uses `platformBundledPlugins` property from the gradle.properties file for bundled IntelliJ Platform plugins.
+        bundledPlugin("com.intellij.database")
+    }
+
+}
\ No newline at end of file
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
new file mode 100644
index 00000000000..baf8dc9ea2e
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
@@ -0,0 +1,162 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.database.dataSource.DataSourceConfigurable;
+import com.intellij.database.dataSource.url.DataInterchange;
+import com.intellij.database.dataSource.url.FieldSize;
+import com.intellij.database.dataSource.url.template.UrlEditorModel;
+import com.intellij.database.dataSource.url.ui.ParamEditorBase;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.util.NlsContexts;
+import com.intellij.ui.components.JBCheckBox;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.auth.AzureAccount;
+import com.microsoft.azure.toolkit.lib.common.model.AzComponent;
+import com.microsoft.azure.toolkit.lib.common.model.AzResource;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTask;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.SneakyThrows;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.commons.lang3.reflect.FieldUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.swing.*;
+import java.awt.*;
+import java.util.Objects;
+import java.util.Optional;
+import java.util.function.Consumer;
+
+public abstract class AzureParamsEditorBase<TCombo extends DbUtilsComboBoxBase<TItem>, TItem extends AzResource> extends ParamEditorBase<TCombo> {
+    public static final String NONE = "<NONE>";
+    private final String dataSourceKey;
+    @Getter
+    @Setter
+    private String text = "";
+    private boolean updating;
+
+    public AzureParamsEditorBase(
+            @NotNull TCombo editorComponent,
+            @NotNull DataInterchange interchange,
+            @NotNull FieldSize fieldSize,
+            @NlsContexts.Label @Nullable String caption,
+            @NotNull String dataSourceKey,
+            AnAction @NotNull ... actions) {
+        super(editorComponent, interchange, fieldSize, caption, actions);
+        this.dataSourceKey = dataSourceKey;
+
+        final DbUtilsComboBoxBase<TItem> comboBox = this.getEditorComponent();
+        comboBox.addValueChangedListener(this::setResource);
+        interchange.addPersistentProperty(dataSourceKey);
+
+        interchange.addPropertyChangeListener((evt -> onPropertiesChanged(evt.getPropertyName(), evt.getNewValue())), this);
+    }
+
+    @Override
+    protected @NotNull JComponent createComponent(TCombo component) {
+        final JPanel container = new JPanel();
+        final BoxLayout layout = new BoxLayout(container, BoxLayout.Y_AXIS);
+        container.setLayout(layout);
+
+        component.setAlignmentX(Component.LEFT_ALIGNMENT);
+        container.add(component);
+
+        if (!Azure.az(AzureAccount.class).isLoggedIn()) {
+            final NotSignedInTipLabel notSignInTips = createNotSignedInTipLabel();
+            notSignInTips.addSignedInListener(() -> {
+                notSignInTips.setVisible(false);
+                component.reloadItems();
+            }, this);
+
+            container.add(notSignInTips);
+        }
+
+        final NoResourceTipLabel noResourceTipLabel = createNoResourceTipLabel();
+        component.setResourcesLoadedListener(noResourceTipLabel::setVisible);
+
+        container.add(noResourceTipLabel);
+        return container;
+    }
+
+    protected abstract NotSignedInTipLabel createNotSignedInTipLabel();
+
+    protected abstract NoResourceTipLabel createNoResourceTipLabel();
+
+    protected void setUsername(String user) {
+        final UrlEditorModel model = this.getDataSourceConfigurable().getUrlEditor().getEditorModel();
+        model.setParameter("user", user);
+        model.commit(true);
+    }
+
+    @SneakyThrows
+    protected void setUseSsl(boolean useSsl) {
+        final DataSourceConfigurable configurable = this.getDataSourceConfigurable();
+        final JBCheckBox useSSLCheckBox = (JBCheckBox) FieldUtils.readField(configurable.getSshSslPanel(), "myUseSSLJBCheckBox", true);
+        useSSLCheckBox.setSelected(useSsl);
+    }
+
+    protected void setJdbcUrl(@NotNull String url) {
+        final UrlEditorModel model = this.getDataSourceConfigurable().getUrlEditor().getEditorModel();
+        model.setUrl(url);
+        model.commit(true);
+    }
+
+    protected void onPropertiesChanged(String propertyName, Object newValue) {
+        if (updating || StringUtils.isEmpty((String) newValue) || !StringUtils.equals(propertyName, "host")) return;
+
+        final DbUtilsComboBoxBase<TItem> comboBox = this.getEditorComponent();
+        final TItem resource = comboBox.getValue();
+        if (Objects.nonNull(resource) && !Objects.equals(getHostFromValue(resource), newValue)) {
+            comboBox.setNull();
+            this.setResource(null);
+        }
+    }
+
+    @SneakyThrows
+    protected DataSourceConfigurable getDataSourceConfigurable() {
+        return (DataSourceConfigurable) FieldUtils.readField(this.getInterchange(), "myConfigurable", true);
+    }
+
+    private void setResource(@Nullable TItem resource) {
+        final DataInterchange interchange = this.getInterchange();
+        final String oldResourceId = interchange.getProperty(dataSourceKey);
+        final String newResourceId = Optional.ofNullable(resource).map(AzComponent::getId).orElse(null);
+        this.updating = true;
+
+        final AzureTaskManager manager = AzureTaskManager.getInstance();
+
+        final Consumer<Object> runSetResource = (fromBackground) -> {
+            setResource(interchange, fromBackground, resource, oldResourceId, newResourceId);
+            this.updating = false;
+        };
+
+        if (needsBackgroundUpdateForSetResource()) {
+            manager.runOnPooledThread(() -> {
+                final Object fromBackground = doBackgroundUpdateForSetResource(resource);
+                manager.runLater(() -> runSetResource.accept(fromBackground), AzureTask.Modality.ANY);
+            });
+        } else {
+            manager.runLater(() -> runSetResource.accept(null), AzureTask.Modality.ANY);
+        }
+    }
+
+    protected abstract void setResource(@NotNull DataInterchange interchange,
+                                        @Nullable Object fromBackground,
+                                        @Nullable TItem value,
+                                        @Nullable String oldResourceId,
+                                        @Nullable String newResourceId);
+
+    @Nullable
+    protected Object doBackgroundUpdateForSetResource(@Nullable TItem value) {
+        return null;
+    }
+
+    protected boolean needsBackgroundUpdateForSetResource() {
+        return false;
+    }
+
+    @Nullable
+    protected abstract String getHostFromValue(@NotNull TItem value);
+}
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
new file mode 100644
index 00000000000..ef4874b279c
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
@@ -0,0 +1,25 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.ide.plugins.PluginManagerCore;
+import com.intellij.openapi.extensions.PluginId;
+import com.microsoft.azure.toolkit.intellij.common.action.IntellijActionsContributor;
+import com.microsoft.azure.toolkit.lib.common.action.Action;
+import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
+import com.microsoft.azure.toolkit.lib.common.exception.AzureToolkitRuntimeException;
+
+public class DatabasePlugin {
+    private static final String DATABASE_TOOLS_PLUGIN_ID = "com.intellij.database";
+    private static final String DATABASE_PLUGIN_NOT_INSTALLED = "\"Database tools and SQL\" plugin is not installed.";
+    private static final String NOT_SUPPORT_ERROR_ACTION = "\"Database tools and SQL\" plugin is only provided in IntelliJ Ultimate edition.";
+
+    public static boolean isInstalled() {
+        return PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null;
+    }
+
+    public static void throwTryUltimateIfNotInstalled(Object binder) {
+        if (PluginManagerCore.getPlugin(PluginId.findId(DATABASE_TOOLS_PLUGIN_ID)) == null) {
+            final Action<Object> tryUltimate = AzureActionManager.getInstance().getAction(IntellijActionsContributor.TRY_ULTIMATE).bind(binder);
+            throw new AzureToolkitRuntimeException(DATABASE_PLUGIN_NOT_INSTALLED, NOT_SUPPORT_ERROR_ACTION, tryUltimate);
+        }
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
new file mode 100644
index 00000000000..453e1e5c165
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
@@ -0,0 +1,14 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.openapi.project.Project;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
+
+import javax.annotation.Nonnull;
+import java.util.function.BiConsumer;
+
+public class DatabaseTools {
+    public static <T> void openDatabaseTool(Project project, @Nonnull T server, BiConsumer<T, Project> dataSourceManagerDialogOpener) {
+        DatabasePlugin.throwTryUltimateIfNotInstalled(server);
+        AzureTaskManager.getInstance().runLater(() -> dataSourceManagerDialogOpener.accept(server, project));
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
new file mode 100644
index 00000000000..1f196ecb19e
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
@@ -0,0 +1,42 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.database.dataSource.DatabaseDriverImpl;
+import com.intellij.database.dataSource.DatabaseDriverManager;
+import com.intellij.database.dataSource.url.template.UrlTemplate;
+import com.intellij.database.dataSource.url.ui.ParametersLayoutUtils;
+import org.apache.commons.lang3.reflect.FieldUtils;
+
+import java.lang.reflect.Field;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Objects;
+
+public class DbToolsWorkarounds  {
+    public static void loadDriverTemplate(String driverId, String templateKey, String templateName, String template) {
+        final DatabaseDriverManager manager = DatabaseDriverManager.getInstance();
+        final DatabaseDriverImpl driver = ((DatabaseDriverImpl) manager.getDriver(driverId));
+        if (Objects.nonNull(driver)) {
+            final List<UrlTemplate> templates = new LinkedList<>(driver.getUrlTemplates());
+            templates.removeIf(t -> t.getTemplate().contains(templateKey));
+            templates.add(0, new UrlTemplate(templateName, template));
+            driver.setURLTemplates(templates);
+        }
+    }
+
+    @SuppressWarnings("unchecked")
+    public static void makeParameterShowAtTop(String parameterName) {
+        try {
+            final Field HEADS = FieldUtils.getField(ParametersLayoutUtils.class, "HEADS", true);
+            final List<String> heads = (List<String>) FieldUtils.readStaticField(HEADS, true);
+            if (!heads.contains(parameterName)) {
+                final Object[] old = heads.toArray();
+                heads.set(0, parameterName);
+                for (int i = 0; i < old.length - 1; i++) {
+                    heads.set(i + 1, (String) old[i]);
+                }
+            }
+        } catch (final Throwable ignored) {
+        }
+    }
+}
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
new file mode 100644
index 00000000000..819fdfa5372
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
@@ -0,0 +1,56 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.common.AzureComboBox;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.auth.AzureAccount;
+import com.microsoft.azure.toolkit.lib.common.model.AzResource;
+import org.apache.commons.collections4.CollectionUtils;
+
+import javax.annotation.Nonnull;
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+import java.util.function.Consumer;
+
+
+public abstract class DbUtilsComboBoxBase<T extends AzResource> extends AzureComboBox<T> {
+    private boolean noResources;
+    private Consumer<Boolean> resourcesLoadedListener;
+
+    public void setResourcesLoadedListener(@Nonnull Consumer<Boolean> serverListener) {
+        this.resourcesLoadedListener = serverListener;
+        serverListener.accept(this.noResources);
+    }
+
+    public void setByResourceId(@Nonnull String resourceId) {
+        setValue(new AzureComboBox.ItemReference<>(i -> i.getId().equals(resourceId)));
+    }
+
+    public void setNull() {
+        setValue((T)null);
+    }
+
+    @Nonnull
+    @Override
+    protected List<T> loadItems() {
+        if (!Azure.az(AzureAccount.class).isLoggedIn()) {
+            return Collections.emptyList();
+        }
+
+        final List<T> resources = load();
+        this.noResources = CollectionUtils.isEmpty(resources);
+        Optional.ofNullable(this.resourcesLoadedListener).ifPresent(l -> l.accept(this.noResources));
+        return resources;
+    }
+
+    protected abstract List<T> load();
+
+    @Override
+    protected String getItemText(Object item) {
+        return Optional.ofNullable(item)
+                .filter(AzResource.class::isInstance)
+                .map(AzResource.class::cast)
+                .map(AzResource::getName)
+                .orElse("");
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
new file mode 100644
index 00000000000..df23421573b
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
@@ -0,0 +1,54 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.icons.AllIcons;
+import com.intellij.ide.DataManager;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.actionSystem.CommonDataKeys;
+import com.intellij.openapi.actionSystem.DataContext;
+import com.intellij.openapi.actionSystem.EmptyAction;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.wm.ToolWindow;
+import com.intellij.openapi.wm.ToolWindowManager;
+import com.intellij.ui.ComponentUtil;
+import com.intellij.ui.HyperlinkLabel;
+import com.microsoft.azure.toolkit.ide.common.action.ResourceCommonActionsContributor;
+import com.microsoft.azure.toolkit.lib.AzService;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.awt.*;
+import java.awt.event.InputEvent;
+import java.util.Objects;
+
+public abstract class NoResourceTipLabel extends HyperlinkLabel {
+    public NoResourceTipLabel(@Nonnull String text) {
+        super();
+
+        setHtmlText(text);
+        setIcon(AllIcons.General.Information);
+        addHyperlinkListener(e -> createResourceInIde(e.getInputEvent()));
+        setAlignmentX(Component.LEFT_ALIGNMENT);
+    }
+
+    protected void createResourceInIde(InputEvent e) {
+        final DataContext context = DataManager.getInstance().getDataContext(e.getComponent());
+        final Project project = context.getData(CommonDataKeys.PROJECT);
+        final Window window = ComponentUtil.getActiveWindow();
+        window.setVisible(false);
+        window.dispose();
+        final ToolWindow explorer = ToolWindowManager.getInstance(Objects.requireNonNull(project)).getToolWindow("Azure Explorer");
+        Objects.requireNonNull(explorer).activate(() -> {
+            final AnActionEvent event = AnActionEvent.createFromAnAction(new EmptyAction(), e, "database.dbtools", context);
+            final Class<? extends AzService> clazz = getClazzForNavigationToExplorer();
+            if (clazz == null) return;
+            AzureActionManager.getInstance().getAction(ResourceCommonActionsContributor.SELECT_RESOURCE_IN_EXPLORER).handle(Azure.az(clazz), event);
+            createResource(project);
+        });
+    }
+
+    @Nullable
+    protected abstract Class<? extends AzService> getClazzForNavigationToExplorer();
+    protected abstract void createResource(Project project);
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
new file mode 100644
index 00000000000..37610d88c65
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
@@ -0,0 +1,44 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.intellij.icons.AllIcons;
+import com.intellij.openapi.Disposable;
+import com.intellij.ui.HyperlinkLabel;
+import com.intellij.util.EventDispatcher;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.auth.AzureAccount;
+import com.microsoft.azure.toolkit.lib.common.action.Action;
+import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
+
+import javax.annotation.Nonnull;
+import java.awt.*;
+import java.util.EventListener;
+
+public class NotSignedInTipLabel extends HyperlinkLabel {
+    private final EventDispatcher<SignedInListener> dispatcher = EventDispatcher.create(SignedInListener.class);
+
+    public NotSignedInTipLabel(@Nonnull String text) {
+        super();
+
+        setHtmlText(text);
+        setIcon(AllIcons.General.Information);
+        addHyperlinkListener(e -> signIn());
+        setAlignmentX(Component.LEFT_ALIGNMENT);
+    }
+
+    public void addSignedInListener(@Nonnull SignedInListener listener, @Nonnull Disposable parentDisposable) {
+        dispatcher.addListener(listener, parentDisposable);
+    }
+
+    protected void signIn() {
+        AzureActionManager.getInstance().getAction(Action.REQUIRE_AUTH).handle((a) -> {
+            if (Azure.az(AzureAccount.class).isLoggedIn()) {
+                dispatcher.getMulticaster().onSignedIn();
+            }
+        });
+    }
+
+    public interface SignedInListener extends EventListener {
+        void onSignedIn();
+    }
+}
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java
new file mode 100644
index 00000000000..88c8edfe925
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java
@@ -0,0 +1,50 @@
+package com.microsoft.azure.toolkit.intellij.dbtools;
+
+import com.azure.resourcemanager.resources.fluentcore.arm.ResourceId;
+import com.intellij.database.dataSource.DatabaseConnectionInterceptor;
+import com.intellij.database.dataSource.DatabaseConnectionPoint;
+import com.microsoft.azure.toolkit.lib.common.operation.Operation;
+import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
+import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
+import org.apache.commons.lang3.StringUtils;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.CompletionStage;
+
+import static com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter.*;
+
+@SuppressWarnings("UnstableApiUsage")
+public abstract class TelemetryConnectionInterceptor implements DatabaseConnectionInterceptor {
+    @Nonnull
+    private final String resourceKey;
+    @Nonnull
+    private final String serviceName;
+    @Nonnull
+    private final String opName;
+
+    protected TelemetryConnectionInterceptor(@Nonnull String resourceKey, @Nonnull String serviceName, @Nonnull String opName) {
+        this.resourceKey = resourceKey;
+        this.serviceName = serviceName;
+        this.opName = opName;
+    }
+
+    @Nullable
+    public CompletionStage<DatabaseConnectionInterceptor.ProtoConnection> intercept(@Nonnull DatabaseConnectionInterceptor.ProtoConnection proto, boolean silent) {
+        final DatabaseConnectionPoint point = proto.getConnectionPoint();
+        final String resourceId = point.getAdditionalProperty(resourceKey);
+        if (StringUtils.isNotBlank(resourceId) && !StringUtils.equalsIgnoreCase(resourceId, AzureParamsEditorBase.NONE)) {
+            final Map<String, String> properties = new HashMap<>();
+            final ResourceId id = ResourceId.fromString(resourceId);
+            properties.put("subscriptionId", id.subscriptionId());
+            properties.put(SERVICE_NAME, serviceName);
+            properties.put(OPERATION_NAME, "connect_jdbc_from_dbtools");
+            properties.put(OP_NAME, opName);
+            properties.put(OP_TYPE, Operation.Type.USER);
+            AzureTelemeter.log(AzureTelemetry.Type.OP_END, properties);
+        }
+        return null;
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/build.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/build.gradle.kts
index 85adf440c0a..b977e8d448a 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/build.gradle.kts
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/build.gradle.kts
@@ -187,6 +187,7 @@ dependencies {
     implementation(project(":azure-intellij-plugin-containerregistry"))
     implementation(project(":azure-intellij-plugin-containerservice"))
     implementation(project(":azure-intellij-plugin-containerapps"))
+    implementation(project(":azure-intellij-plugin-dbtools"))
     implementation(project(":azure-intellij-plugin-database"))
     implementation(project(":azure-intellij-plugin-database-java"))
     implementation(project(":azure-intellij-plugin-vm"))
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/settings.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/settings.gradle.kts
index c6da38d916a..e88a3cb3f31 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/settings.gradle.kts
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/settings.gradle.kts
@@ -47,4 +47,5 @@ include("azure-intellij-plugin-sparkoncosmos")
 include("azure-intellij-plugin-sqlserverbigdata")
 include("azure-intellij-plugin-keyvault")
 include("azure-intellij-plugin-keyvault-java")
-include("azure-intellij-plugin-integration-services")
\ No newline at end of file
+include("azure-intellij-plugin-integration-services")
+include("azure-intellij-plugin-dbtools")

From 05b2b52b5f22c24b47891433375c59757d274f8d Mon Sep 17 00:00:00 2001
From: Ilnar Agliamzianov <ilnar.agliamzianov@jetbrains.com>
Date: Mon, 19 Aug 2024 10:37:03 +0300
Subject: [PATCH 2/4] Support opening Redis cache in "Database" tool window

---
 .../build.gradle.kts                          |   5 +
 .../IntellijRedisActionsContributor.java      |  58 ++++++++++
 .../redis/dbtools/NoCachesTipLabel.java       |  34 ++++++
 .../dbtools/RedisAccessKeyAuthProvider.java   | 102 ++++++++++++++++++
 .../dbtools/RedisAccountTypeFactory.java      |  22 ++++
 .../redis/dbtools/RedisCacheComboBox.java     |  36 +++++++
 .../redis/dbtools/RedisCacheParamEditor.java  |  70 ++++++++++++
 .../dbtools/RedisCacheTypeUIFactory.java      |  28 +++++
 .../dbtools/RedisConnectionInterceptor.java   |  10 ++
 .../redis/dbtools/RedisDbToolsWorkaround.java |  45 ++++++++
 .../intellij/redis/dbtools/RedisJdbcUrl.java  |  10 ++
 .../redis/dbtools/RedisNotSignedInLabel.java  |  17 +++
 .../azure-intellij-plugin-redis-dbtools.xml   |  11 ++
 .../META-INF/azure-intellij-plugin-redis.xml  |   1 +
 .../azure/toolkit/operation.properties        |   1 +
 15 files changed, 450 insertions(+)
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccountTypeFactory.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheTypeUIFactory.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
 create mode 100644 PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis-dbtools.xml

diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/build.gradle.kts b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/build.gradle.kts
index e5c8088d541..c69035f7a9d 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/build.gradle.kts
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/build.gradle.kts
@@ -2,9 +2,14 @@ dependencies {
     implementation(project(":azure-intellij-plugin-lib"))
     // runtimeOnly project(path: ":azure-intellij-plugin-lib", configuration: "instrumentedJar")
     implementation(project(":azure-intellij-resource-connector-lib"))
+    implementation(project(":azure-intellij-plugin-dbtools"))
     // runtimeOnly project(path: ":azure-intellij-resource-connector-lib", configuration: "instrumentedJar")
     implementation("com.microsoft.azure:azure-toolkit-redis-lib")
     implementation("com.microsoft.azure:azure-toolkit-ide-common-lib")
     implementation("com.microsoft.azure:azure-toolkit-ide-redis-lib")
     implementation("redis.clients:jedis:3.6.3")
+
+    intellijPlatform {
+        bundledPlugin("com.intellij.database")
+    }
 }
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/IntellijRedisActionsContributor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/IntellijRedisActionsContributor.java
index 3c36661b7a2..f61ded00a4c 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/IntellijRedisActionsContributor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/IntellijRedisActionsContributor.java
@@ -5,7 +5,13 @@
 
 package com.microsoft.azure.toolkit.intellij.redis;
 
+import com.intellij.database.Dbms;
+import com.intellij.database.autoconfig.DataSourceDetector;
+import com.intellij.database.autoconfig.DataSourceRegistry;
+import com.intellij.database.psi.DbPsiFacade;
+import com.intellij.database.view.ui.DataSourceManagerDialog;
 import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.Project;
 import com.microsoft.azure.toolkit.ide.common.IActionsContributor;
 import com.microsoft.azure.toolkit.ide.common.action.ResourceCommonActionsContributor;
 import com.microsoft.azure.toolkit.ide.common.icon.AzureIcons;
@@ -13,21 +19,53 @@
 import com.microsoft.azure.toolkit.intellij.common.IntelliJAzureIcons;
 import com.microsoft.azure.toolkit.intellij.common.properties.AzureResourceEditorViewManager;
 import com.microsoft.azure.toolkit.intellij.common.properties.AzureResourceEditorViewManager.AzureResourceFileType;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabasePlugin;
+import com.microsoft.azure.toolkit.intellij.dbtools.DatabaseTools;
 import com.microsoft.azure.toolkit.intellij.redis.creation.CreateRedisCacheAction;
+import com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisCacheParamEditor;
+import com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisJdbcUrl;
 import com.microsoft.azure.toolkit.intellij.redis.explorer.RedisCacheExplorerProvider;
+import com.microsoft.azure.toolkit.lib.common.action.Action;
 import com.microsoft.azure.toolkit.lib.common.action.AzureActionManager;
+import com.microsoft.azure.toolkit.lib.common.action.IActionGroup;
 import com.microsoft.azure.toolkit.lib.common.model.AzResource;
+import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
 import com.microsoft.azure.toolkit.lib.resource.ResourceGroup;
 import com.microsoft.azure.toolkit.redis.AzureRedis;
 import com.microsoft.azure.toolkit.redis.RedisCache;
 import com.microsoft.azure.toolkit.redis.model.RedisConfig;
 
+import javax.annotation.Nonnull;
 import javax.swing.*;
 import java.util.Objects;
 import java.util.function.BiConsumer;
 import java.util.function.BiPredicate;
 
+import static com.microsoft.azure.toolkit.ide.redis.RedisActionsContributor.REDIS_ACTIONS;
+import static com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisAccessKeyAuthProvider.RedisAccessKey;
+
 public class IntellijRedisActionsContributor implements IActionsContributor {
+    public static final Action.Id<AzResource> OPEN_DATABASE_TOOL = Action.Id.of("user/redis.open_database_tools.name");
+
+    @Override
+    public void registerActions(AzureActionManager am) {
+        new Action<>(OPEN_DATABASE_TOOL)
+                .visibleWhen(s -> s instanceof RedisCache)
+                .enableWhen(s -> s.getFormalStatus().isRunning())
+                .withIcon(AzureIcons.Action.OPEN_DATABASE_TOOL.getIconPath())
+                .withLabel("Open with Database Tools")
+                .withIdParam(AzResource::getName)
+                .register(am);
+    }
+
+    @Override
+    public void registerGroups(AzureActionManager am) {
+        final IActionGroup group = am.getGroup(REDIS_ACTIONS);
+        group.addAction("---");
+        group.addAction(OPEN_DATABASE_TOOL);
+    }
+
     @Override
     public void registerHandlers(AzureActionManager am) {
         final BiPredicate<Object, AnActionEvent> condition = (r, e) -> r instanceof AzureRedis;
@@ -49,10 +87,30 @@ public void registerHandlers(AzureActionManager am) {
             CreateRedisCacheAction.create(e.getProject(), config);
         };
         am.registerHandler(RedisActionsContributor.GROUP_CREATE_REDIS, (r, e) -> true, groupCreateServerHandler);
+
+        final BiConsumer<AzResource, AnActionEvent> openDatabaseHandler = (c, e) -> openDatabaseTool(e.getProject(), (RedisCache) c);
+        am.registerHandler(OPEN_DATABASE_TOOL, (r, e) -> true, openDatabaseHandler);
     }
 
     @Override
     public int getOrder() {
         return RedisActionsContributor.INITIALIZE_ORDER + 1;
     }
+
+    public static void openDatabaseTool(Project project, @Nonnull RedisCache cache) {
+        DatabaseTools.openDatabaseTool(project, cache, IntellijRedisActionsContributor::openDataSourceManagerDialog);
+    }
+
+    public static void openDataSourceManagerDialog(RedisCache cache, Project project) {
+        final DataSourceRegistry registry = new DataSourceRegistry(project);
+        final DbPsiFacade dbPsiFacade = DbPsiFacade.getInstance(project);
+        final DataSourceDetector.Builder builder = registry.getBuilder()
+                .withJdbcAdditionalProperty(RedisCacheParamEditor.KEY_REDIS_CACHE, cache.getId())
+                .withDbms(Dbms.REDIS)
+                .withUrl(RedisJdbcUrl.from(cache))
+                .withAuthProviderId(RedisAccessKey)
+                .commit();
+
+        DataSourceManagerDialog.showDialog(dbPsiFacade, registry);
+    }
 }
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
new file mode 100644
index 00000000000..672d0de14b7
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
@@ -0,0 +1,34 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.openapi.project.Project;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.intellij.redis.creation.CreateRedisCacheAction;
+import com.microsoft.azure.toolkit.lib.AzService;
+import com.microsoft.azure.toolkit.redis.AzureRedis;
+import org.jetbrains.annotations.Nullable;
+
+import java.awt.event.InputEvent;
+
+public class NoCachesTipLabel extends NoResourceTipLabel {
+    private static final String NO_SERVERS_TIPS = "<html>No existing Redis caches in Azure. You can <a href=''>create one</a> first.</html>";
+
+    public NoCachesTipLabel() {
+        super(NO_SERVERS_TIPS);
+    }
+
+    @Override
+    protected void createResourceInIde(InputEvent e) {
+        super.createResourceInIde(e);
+    }
+
+    @Nullable
+    @Override
+    protected Class<? extends AzService> getClazzForNavigationToExplorer() {
+        return AzureRedis.class;
+    }
+
+    @Override
+    protected void createResource(Project project) {
+        CreateRedisCacheAction.create(project, null);
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
new file mode 100644
index 00000000000..dac6367be7d
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
@@ -0,0 +1,102 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.credentialStore.Credentials;
+import com.intellij.database.Dbms;
+import com.intellij.database.dataSource.DatabaseAuthProvider;
+import com.intellij.database.dataSource.DatabaseConnectionPoint;
+import com.intellij.database.dataSource.DatabaseCredentialsAuthProvider;
+import com.microsoft.azure.toolkit.intellij.dbtools.AzureParamsEditorBase;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.common.task.AzureTaskManager;
+import com.microsoft.azure.toolkit.redis.AzureRedis;
+import com.microsoft.azure.toolkit.redis.RedisCache;
+import com.microsoft.azure.toolkit.redis.RedisCacheModule;
+import com.microsoft.azure.toolkit.redis.RedisServiceSubscription;
+import kotlin.coroutines.Continuation;
+import org.apache.commons.lang3.StringUtils;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+
+public class RedisAccessKeyAuthProvider implements DatabaseAuthProvider {
+    public static final String RedisAccessKey = "redis_access_key";
+
+    @Nullable
+    @Override
+    public Object interceptConnection(@Nonnull ProtoConnection proto, boolean silent, @Nonnull Continuation<? super Boolean> $completion) {
+        final String cacheId = getCacheIdIfValid(proto);
+        if (cacheId == null) return false;
+
+        final AzureTaskManager manager = AzureTaskManager.getInstance();
+
+        manager.runInBackground("Loading Redis Cache Primary Key", () -> {
+            final String primaryKey = getPrimaryKeyForCache(cacheId);
+            if (primaryKey == null) {
+                $completion.resumeWith(false);
+                return;
+            }
+
+            manager.runLater(() -> {
+                setPrimaryKeyAsPassword(proto, primaryKey);
+                $completion.resumeWith(true);
+            });
+        });
+
+        return kotlin.coroutines.intrinsics.IntrinsicsKt.getCOROUTINE_SUSPENDED();
+    }
+
+    @Nonnull
+    @Override
+    public ApplicabilityLevel.Result getApplicability(@Nonnull DatabaseConnectionPoint point, @Nonnull ApplicabilityLevel level) {
+        final boolean isRedis = point.getDbms().in(Dbms.REDIS);
+        final String cacheId = point.getAdditionalProperty(RedisCacheParamEditor.KEY_REDIS_CACHE);
+        if (isRedis && isValidCacheId(cacheId))
+            return ApplicabilityLevel.Result.PREFERRED;
+
+        return isRedis ? ApplicabilityLevel.Result.DEFAULT : ApplicabilityLevel.Result.NOT_APPLICABLE;
+    }
+
+    @Nonnull
+    @Override
+    public String getId() {
+        return RedisAccessKey;
+    }
+
+    @Nonnull
+    @Override
+    public String getDisplayName() {
+        return "Redis Access Key";
+    }
+
+    private void setPrimaryKeyAsPassword(@Nonnull ProtoConnection proto, @Nonnull String primaryKey) {
+        final var credentials = new Credentials(null, primaryKey);
+        DatabaseCredentialsAuthProvider.applyCredentials(proto, credentials, false);
+    }
+
+    @Nullable
+    private String getPrimaryKeyForCache(@Nonnull String cacheId) {
+        final RedisServiceSubscription subscription = Azure.az(AzureRedis.class).get(cacheId);
+        if (subscription == null)  {
+            return null;
+        }
+
+        final RedisCacheModule caches = subscription.caches();
+
+        final RedisCache cache = caches.get(cacheId);
+        if (cache == null) {
+            return null;
+        }
+
+        return cache.getPrimaryKey();
+    }
+
+    @Nullable
+    private String getCacheIdIfValid(@Nonnull ProtoConnection proto) {
+        final String cacheId = proto.getConnectionPoint().getAdditionalProperty(RedisCacheParamEditor.KEY_REDIS_CACHE);
+        return isValidCacheId(cacheId) ? cacheId : null;
+    }
+
+    private boolean isValidCacheId(@Nullable String cacheId) {
+        return StringUtils.isNotBlank(cacheId) && !StringUtils.equalsIgnoreCase(cacheId, AzureParamsEditorBase.NONE);
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccountTypeFactory.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccountTypeFactory.java
new file mode 100644
index 00000000000..56f5588a647
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccountTypeFactory.java
@@ -0,0 +1,22 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.database.dataSource.url.TypeDescriptor;
+import com.intellij.database.dataSource.url.TypesRegistry;
+import com.intellij.util.Consumer;
+import org.jetbrains.annotations.NotNull;
+
+public class RedisAccountTypeFactory implements TypesRegistry.TypeDescriptorFactory {
+    public static final String TYPE_NAME = "redis_cache";
+    public static final String CAPTION = "Cache";
+    public static final String PARAM_NAME = "cache";
+
+    @Override
+    public void createTypeDescriptor(@NotNull Consumer<? super TypeDescriptor> consumer) {
+        consumer.consume(TypesRegistry.createTypeDescriptor(TYPE_NAME, ".", CAPTION));
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
new file mode 100644
index 00000000000..36088ee53f3
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
@@ -0,0 +1,36 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
+import com.microsoft.azure.toolkit.lib.Azure;
+import com.microsoft.azure.toolkit.lib.common.cache.CacheManager;
+import com.microsoft.azure.toolkit.redis.AzureRedis;
+import com.microsoft.azure.toolkit.redis.RedisCache;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+@Getter
+@RequiredArgsConstructor
+public class RedisCacheComboBox extends DbUtilsComboBoxBase<RedisCache> {
+
+    @Nullable
+    @Override
+    protected RedisCache doGetDefaultValue() {
+        return CacheManager.getUsageHistory(RedisCache.class).peek();
+    }
+
+    @Override
+    protected List<RedisCache> load() {
+        final Stream<RedisCache> allCaches = Azure.az(AzureRedis.class).list().stream().flatMap(x -> x.caches().list().stream());
+        return filterDrafts(allCaches);
+    }
+
+    private List<RedisCache> filterDrafts(@Nonnull Stream<RedisCache> caches) {
+        return caches.filter(m -> !m.isDraftForCreating()).collect(Collectors.toList());
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
new file mode 100644
index 00000000000..886af326c15
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
@@ -0,0 +1,70 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.database.dataSource.LocalDataSource;
+import com.intellij.database.dataSource.url.DataInterchange;
+import com.intellij.database.dataSource.url.FieldSize;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.microsoft.azure.toolkit.intellij.dbtools.AzureParamsEditorBase;
+import com.microsoft.azure.toolkit.intellij.dbtools.NoResourceTipLabel;
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
+import com.microsoft.azure.toolkit.redis.RedisCache;
+import org.apache.commons.lang3.StringUtils;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.annotation.Nonnull;
+import java.util.Objects;
+import java.util.Optional;
+
+public class RedisCacheParamEditor extends AzureParamsEditorBase<RedisCacheComboBox, RedisCache> {
+    public static final String KEY_REDIS_CACHE = "AZURE_REDIS_CACHE";
+
+    public RedisCacheParamEditor(
+            @NotNull DataInterchange interchange,
+            @NotNull FieldSize fieldSize,
+            @Nullable String caption,
+            AnAction @NotNull ... actions) {
+        super(new RedisCacheComboBox(), interchange, fieldSize, caption, KEY_REDIS_CACHE, actions);
+        final LocalDataSource dataSource = getDataSourceConfigurable().getDataSource();
+
+        final RedisCacheComboBox comboBox = getEditorComponent();
+        final String cacheId = interchange.getProperty(KEY_REDIS_CACHE);
+
+        if (StringUtils.isNotBlank(cacheId)) {
+            comboBox.setByResourceId(cacheId);
+        }
+    }
+
+    @Override
+    protected NotSignedInTipLabel createNotSignedInTipLabel() {
+        return new RedisNotSignedInLabel();
+    }
+
+    @Override
+    protected NoResourceTipLabel createNoResourceTipLabel() {
+        return new NoCachesTipLabel();
+    }
+
+    @Override
+    protected void setResource(
+            @Nonnull DataInterchange interchange,
+            @Nullable Object fromBackground,
+            @Nullable RedisCache value,
+            @Nullable String oldResourceId,
+            @Nullable String newResourceId) {
+        interchange.putProperty(KEY_REDIS_CACHE, Optional.ofNullable(newResourceId).orElse(NONE));
+        if (Objects.isNull(value) || Objects.equals(oldResourceId, newResourceId)) {
+            return;
+        }
+
+        setJdbcUrl(RedisJdbcUrl.from(value));
+    }
+
+    @Override
+    protected String getHostFromValue(RedisCache value) {
+        return value.getHostName();
+    }
+}
+
+
+
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheTypeUIFactory.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheTypeUIFactory.java
new file mode 100644
index 00000000000..ed4d554e978
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheTypeUIFactory.java
@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.database.dataSource.url.*;
+import com.intellij.database.dataSource.url.ui.BaseTypeDescriptor;
+import com.intellij.openapi.util.NlsContexts;
+import com.intellij.util.Consumer;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import static com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisAccountTypeFactory.CAPTION;
+import static com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisAccountTypeFactory.TYPE_NAME;
+
+public class RedisCacheTypeUIFactory implements TypesRegistryUi.TypeDescriptorUiFactory {
+    @Override
+    public void createTypeDescriptor(@NotNull Consumer<? super TypeDescriptorUi> consumer) {
+        consumer.consume(new BaseTypeDescriptor(TYPE_NAME) {
+            @Override
+            protected @NotNull ParamEditor createFieldImpl(@NlsContexts.Label @NotNull String s, @Nullable String s1, @NotNull DataInterchange dataInterchange) {
+                return new RedisCacheParamEditor(dataInterchange, FieldSize.LARGE, formatFieldCaption(CAPTION));
+            }
+        });
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
new file mode 100644
index 00000000000..8267a31936c
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
@@ -0,0 +1,10 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.microsoft.azure.toolkit.intellij.dbtools.TelemetryConnectionInterceptor;
+import static com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisCacheParamEditor.KEY_REDIS_CACHE;
+
+public class RedisConnectionInterceptor extends TelemetryConnectionInterceptor {
+    protected RedisConnectionInterceptor() {
+        super(KEY_REDIS_CACHE, "redis", "redis.connect_jdbc_from_dbtools");
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
new file mode 100644
index 00000000000..896df4a4bf3
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
@@ -0,0 +1,45 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.project.DumbAware;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.startup.ProjectActivity;
+import com.microsoft.azure.toolkit.intellij.dbtools.DbToolsWorkarounds;
+import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
+import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
+import kotlin.Unit;
+import kotlin.coroutines.Continuation;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.util.HashMap;
+
+public class RedisDbToolsWorkaround implements ProjectActivity, DumbAware {
+
+    @Nullable
+    @Override
+    public Object execute(@Nonnull Project project, @Nonnull Continuation<? super Unit> continuation) {
+        ApplicationManager.getApplication().executeOnPooledThread(() -> {
+            try {
+                loadRedisTemplates();
+                makeCacheShowAtTop();
+            } catch (final Throwable t) {
+                // swallow exception for preloading workarounds
+                AzureTelemeter.log(AzureTelemetry.Type.ERROR, new HashMap<>(), t);
+            }
+        });
+        return null;
+    }
+
+    private static void makeCacheShowAtTop() {
+        DbToolsWorkarounds.makeParameterShowAtTop(RedisAccountTypeFactory.PARAM_NAME);
+    }
+
+    private static void loadRedisTemplates() {
+        DbToolsWorkarounds.loadDriverTemplate(
+                "redis",
+                RedisAccountTypeFactory.TYPE_NAME,
+                "Azure",
+                "jdbc:redis://[[{user}:]{password}@]{host::localhost}[:{port::6379}][/{database:database/[^?]+:0}?][/{cache:redis_cache}?][\\?<&,{:identifier}={:param}>]");
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
new file mode 100644
index 00000000000..e367de85a4c
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
@@ -0,0 +1,10 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+
+import com.microsoft.azure.toolkit.redis.RedisCache;
+
+public class RedisJdbcUrl {
+    public static String from(RedisCache redisCache) {
+        return String.format("jdbc:redis://%s:%d?ssl=%b", redisCache.getHostName(), redisCache.getSSLPort(), true);
+    }
+}
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
new file mode 100644
index 00000000000..1ef820a0f75
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
@@ -0,0 +1,17 @@
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+
+
+import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
+
+public class RedisNotSignedInLabel extends NotSignedInTipLabel {
+    private static final String NOT_SIGNIN_TIPS = "<html><a href=\"\">Sign in</a> to select an existing Redis cache in Azure.</html>";
+
+    public RedisNotSignedInLabel() {
+        super(NOT_SIGNIN_TIPS);
+    }
+
+    @Override
+    protected void signIn() {
+        super.signIn();
+    }
+}
\ No newline at end of file
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis-dbtools.xml b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis-dbtools.xml
new file mode 100644
index 00000000000..2c8a2be6074
--- /dev/null
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis-dbtools.xml
@@ -0,0 +1,11 @@
+<idea-plugin>
+    <extensions defaultExtensionNs="com.intellij.database">
+        <connectionInterceptor implementation="com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisConnectionInterceptor"/>
+        <connectionInterceptor implementation="com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisAccessKeyAuthProvider"/>
+        <urlParamEditorProvider implementation="com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisAccountTypeFactory" order="first"/>
+        <urlParamEditorUiProvider implementation="com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisCacheTypeUIFactory" order="first"/>
+    </extensions>
+    <extensions defaultExtensionNs="com.intellij">
+        <postStartupActivity implementation="com.microsoft.azure.toolkit.intellij.redis.dbtools.RedisDbToolsWorkaround"/>
+    </extensions>
+</idea-plugin>
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis.xml b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis.xml
index 24fc4072572..41dbf7b839d 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis.xml
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/resources/META-INF/azure-intellij-plugin-redis.xml
@@ -1,4 +1,5 @@
 <idea-plugin>
+    <depends optional="true" config-file="azure-intellij-plugin-redis-dbtools.xml">com.intellij.database</depends>
     <extensions defaultExtensionNs="com.microsoft.tooling.msservices.intellij.azure">
         <explorerNodeProvider implementation="com.microsoft.azure.toolkit.ide.redis.RedisNodeProvider"/>
         <actions implementation="com.microsoft.azure.toolkit.ide.redis.RedisActionsContributor"/>
diff --git a/Utils/azure-toolkit-ide-libs/azure-toolkit-ide-common-lib/src/main/resources/bundles/com/microsoft/azure/toolkit/operation.properties b/Utils/azure-toolkit-ide-libs/azure-toolkit-ide-common-lib/src/main/resources/bundles/com/microsoft/azure/toolkit/operation.properties
index b32b9bd5f4a..93413add520 100644
--- a/Utils/azure-toolkit-ide-libs/azure-toolkit-ide-common-lib/src/main/resources/bundles/com/microsoft/azure/toolkit/operation.properties
+++ b/Utils/azure-toolkit-ide-libs/azure-toolkit-ide-common-lib/src/main/resources/bundles/com/microsoft/azure/toolkit/operation.properties
@@ -433,6 +433,7 @@ user/storage.open_azure_storage_explorer.storage=open storage ({0}) with Azure S
 ######################begin @AzureOperation&registered actions######################
 user/arm.update_deployment.deployment=update Azure Resource deployment ({0})
 user/cosmos.open_database_tools.account=open Azure Cosmos DB account ({0}) with "Database Tools and SQL" plugin
+user/redis.open_database_tools.name=open Redis Cache ({0}) with "Database Tools and SQL" plugin
 user/function.start_remote_debugging.app=start remote debugging for app ({0})
 user/function.enable_remote_debugging.app=enable remote debug of app ({0})
 user/function.disable_remote_debugging.app=disable remote debug of app ({0})

From 8c547fed8becc7ec3d14f0f38b057db3365bf931 Mon Sep 17 00:00:00 2001
From: Ilnar Agliamzianov <ilnar.agliamzianov@jetbrains.com>
Date: Mon, 19 Aug 2024 11:50:33 +0300
Subject: [PATCH 3/4] Add copyright to all added files

---
 .../intellij/cosmos/dbtools/CosmosDbAccountComboBox.java    | 5 +++++
 .../intellij/cosmos/dbtools/CosmosNotSignedInLabel.java     | 5 +++++
 .../toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java | 5 +++++
 .../intellij/database/dbtools/DatabaseNotSignedInLabel.java | 5 +++++
 .../intellij/database/dbtools/DatabaseServerClass.java      | 5 +++++
 .../intellij/database/dbtools/NoServersTipLabel.java        | 5 +++++
 .../intellij/database/dbtools/SqlDbServerComboBox.java      | 5 +++++
 .../toolkit/intellij/dbtools/AzureParamsEditorBase.java     | 5 +++++
 .../azure/toolkit/intellij/dbtools/DatabasePlugin.java      | 5 +++++
 .../azure/toolkit/intellij/dbtools/DatabaseTools.java       | 5 +++++
 .../azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java  | 5 +++++
 .../azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java | 5 +++++
 .../azure/toolkit/intellij/dbtools/NoResourceTipLabel.java  | 5 +++++
 .../azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java | 5 +++++
 .../intellij/dbtools/TelemetryConnectionInterceptor.java    | 5 +++++
 .../toolkit/intellij/redis/dbtools/NoCachesTipLabel.java    | 5 +++++
 .../intellij/redis/dbtools/RedisAccessKeyAuthProvider.java  | 5 +++++
 .../toolkit/intellij/redis/dbtools/RedisCacheComboBox.java  | 5 +++++
 .../intellij/redis/dbtools/RedisCacheParamEditor.java       | 5 +++++
 .../intellij/redis/dbtools/RedisConnectionInterceptor.java  | 5 +++++
 .../intellij/redis/dbtools/RedisDbToolsWorkaround.java      | 5 +++++
 .../azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java  | 6 +++++-
 .../intellij/redis/dbtools/RedisNotSignedInLabel.java       | 5 +++++
 23 files changed, 115 insertions(+), 1 deletion(-)

diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
index c0022cb83f9..63a20987164 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosDbAccountComboBox.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
index 043330a2647..32d1fa3f23f 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/CosmosNotSignedInLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.NotSignedInTipLabel;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
index d0de745f609..b5e0b58a8f2 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/NoAccountsTipLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.intellij.openapi.project.Project;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
index 2a262679aa9..c85d626154d 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseNotSignedInLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
 
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
index 8bc3de36c3e..a3ea615a858 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseServerClass.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
 import com.microsoft.azure.toolkit.lib.database.entity.IDatabaseServer;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
index 3fdf4229d8c..04476cd35be 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/NoServersTipLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
 import com.intellij.openapi.project.Project;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
index d6632d6c409..8d96eac4678 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/SqlDbServerComboBox.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.database.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
index baf8dc9ea2e..7a0a6927540 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/AzureParamsEditorBase.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.database.dataSource.DataSourceConfigurable;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
index ef4874b279c..ee99eb8d882 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabasePlugin.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.ide.plugins.PluginManagerCore;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
index 453e1e5c165..d710cc20f68 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DatabaseTools.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.openapi.project.Project;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
index 1f196ecb19e..94732196713 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbToolsWorkarounds.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.database.dataSource.DatabaseDriverImpl;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
index 819fdfa5372..d5214c3cec0 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/DbUtilsComboBoxBase.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.common.AzureComboBox;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
index df23421573b..84be0ea80b0 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NoResourceTipLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.icons.AllIcons;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
index 37610d88c65..839ebe684f6 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/NotSignedInTipLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.intellij.icons.AllIcons;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java
index 88c8edfe925..e80d76cdbc0 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-dbtools/src/main/java/com/microsoft/azure/toolkit/intellij/dbtools/TelemetryConnectionInterceptor.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.dbtools;
 
 import com.azure.resourcemanager.resources.fluentcore.arm.ResourceId;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
index 672d0de14b7..1c04020050b 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/NoCachesTipLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.intellij.openapi.project.Project;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
index dac6367be7d..ea266b03b6c 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisAccessKeyAuthProvider.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.intellij.credentialStore.Credentials;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
index 36088ee53f3..f10caf325e4 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheComboBox.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.DbUtilsComboBoxBase;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
index 886af326c15..f1ef0932fa1 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisCacheParamEditor.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.intellij.database.dataSource.LocalDataSource;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
index 8267a31936c..d5ae10bf506 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisConnectionInterceptor.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.TelemetryConnectionInterceptor;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
index 896df4a4bf3..58e7947a1fa 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisDbToolsWorkaround.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.intellij.openapi.application.ApplicationManager;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
index e367de85a4c..c9de25631fb 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisJdbcUrl.java
@@ -1,5 +1,9 @@
-package com.microsoft.azure.toolkit.intellij.redis.dbtools;
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
 
+package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 import com.microsoft.azure.toolkit.redis.RedisCache;
 
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
index 1ef820a0f75..5125f16f3e7 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-redis/src/main/java/com/microsoft/azure/toolkit/intellij/redis/dbtools/RedisNotSignedInLabel.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.redis.dbtools;
 
 

From c14e9058ad8cc6f0cdfb19b532fadddf0e4ac1bd Mon Sep 17 00:00:00 2001
From: Ilnar Agliamzianov <ilnar.agliamzianov@jetbrains.com>
Date: Mon, 2 Sep 2024 11:46:44 +0300
Subject: [PATCH 4/4] Add missing copyright to files

---
 .../dbtools/AzureCosmosDbAccountConnectionInterceptor.java   | 5 +++++
 .../cosmos/dbtools/AzureCosmosDbAccountParamEditor.java      | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
index 5d74960f338..60c44baac25 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountConnectionInterceptor.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.microsoft.azure.toolkit.intellij.dbtools.TelemetryConnectionInterceptor;
diff --git a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
index 69f53bf96c6..4a6b109472c 100644
--- a/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
+++ b/PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/AzureCosmosDbAccountParamEditor.java
@@ -1,3 +1,8 @@
+/*
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ */
+
 package com.microsoft.azure.toolkit.intellij.cosmos.dbtools;
 
 import com.intellij.credentialStore.OneTimeString;