feat(plugins): improve connection lifecycle and session tabs
This commit is contained in:
1 parent
6be8d5544e
commit
1321dc9b9d
25 files changed
+849
-44
No files matched your search
@@ -56,7 +56,16 @@ import DatabaseIcon from "@/components/icons/DatabaseIcon.vue";
|
||||
import PluginConnectionFields from "@/components/plugins/PluginConnectionFields.vue";
|
||||
import PluginIcon from "@/components/plugins/PluginIcon.vue";
|
||||
import * as api from "@/lib/backend/api";
|
||||
import { buildPluginConnectionConfig, createFrontendPluginRegistry, parsePluginConnectionProviderOptionValue, pluginConnectionActionsForDialog, pluginConnectionFormValues, pluginConnectionProviderIcon, pluginConnectionProviderOptionValue } from "@/lib/plugins/frontendPlugin";
|
||||
import {
|
||||
buildPluginConnectionConfig,
|
||||
createFrontendPluginRegistry,
|
||||
parsePluginConnectionProviderOptionValue,
|
||||
pluginConnectionActionsForDialog,
|
||||
pluginConnectionConnectTimeoutDefault,
|
||||
pluginConnectionFormValues,
|
||||
pluginConnectionProviderIcon,
|
||||
pluginConnectionProviderOptionValue,
|
||||
} from "@/lib/plugins/frontendPlugin";
|
||||
import type { PluginCenterFocus } from "@/lib/plugins/pluginCenterNavigation";
|
||||
import { isTauriRuntime } from "@/lib/backend/tauriRuntime";
|
||||
import { applyMeilisearchBasePathToExternalConfig, applyParsedConnectionUrl, normalizeMongoConnectionString, parseConnectionUrl } from "@/lib/connection/connectionUrl";
|
||||
@@ -4099,12 +4108,26 @@ function connectionConfigForSubmit(id: string, generatedName = "", validatePlugi
|
||||
}
|
||||
const existing = props.editConfig?.db_type === "plugin" && props.editConfig.plugin_id === entry.plugin.manifest.id && props.editConfig.plugin_connection_provider === entry.contribution.id ? props.editConfig : undefined;
|
||||
config = buildPluginConnectionConfig(entry.plugin.manifest.id, entry.contribution, values, existing) as LegacyConnectionConfig;
|
||||
// buildPluginConnectionConfig already mirrored the provider's resolved
|
||||
// connect_timeout_secs (declared default or advanced-form value) into the
|
||||
// typed field; capture it before the generic form overwrite below.
|
||||
const resolvedPluginConnectTimeout = pluginConnectionConnectTimeoutDefault(entry.contribution) === undefined ? undefined : config.connect_timeout_secs;
|
||||
config.id = id;
|
||||
config.name = form.value.name.trim() || config.name;
|
||||
config.note = form.value.note;
|
||||
config.color = form.value.color;
|
||||
config.transport_layers = form.value.transport_layers || [];
|
||||
config.connect_timeout_secs = form.value.connect_timeout_secs;
|
||||
if (resolvedPluginConnectTimeout !== undefined) {
|
||||
// A provider declaring its own connect_timeout_secs field makes it the
|
||||
// single source of truth (declared default or advanced-form value): the
|
||||
// typed timeout mirrors it, and the generic global/per-connection DBX
|
||||
// timeout radios do not apply. Otherwise the host RPC deadline and the
|
||||
// plugin's own handshake timeout could disagree and the host would kill
|
||||
// slow connects first.
|
||||
config.connect_timeout_secs = resolvedPluginConnectTimeout;
|
||||
config.connect_timeout_inherit = false;
|
||||
}
|
||||
config.query_timeout_secs = form.value.query_timeout_secs;
|
||||
config.idle_timeout_secs = form.value.idle_timeout_secs;
|
||||
config.keepalive_interval_secs = form.value.keepalive_interval_secs;
|
||||
|
||||
@@ -0,0 +1,135 @@
|
||||
// @vitest-environment happy-dom
|
||||
|
||||
import { createApp, nextTick, ref, type App } from "vue";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import type { InstalledPlugin, PluginWorkbenchContribution } from "@/types/database";
|
||||
|
||||
const mocks = vi.hoisted(() => ({
|
||||
readPluginUiEntry: vi.fn(),
|
||||
readPluginUiAsset: vi.fn(),
|
||||
subscribePluginEvents: vi.fn(),
|
||||
repushPluginConnection: vi.fn(),
|
||||
reopenPluginConnection: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock("@/lib/backend/api", () => ({
|
||||
invokePlugin: vi.fn(),
|
||||
notifyPlugin: vi.fn(),
|
||||
sendPluginBinary: vi.fn(),
|
||||
readPluginUiEntry: mocks.readPluginUiEntry,
|
||||
readPluginUiAsset: mocks.readPluginUiAsset,
|
||||
subscribePluginEvents: mocks.subscribePluginEvents,
|
||||
}));
|
||||
vi.mock("@/lib/backend/tauriRuntime", () => ({ isTauriRuntime: () => false }));
|
||||
vi.mock("@/lib/common/clipboard", () => ({ copyToClipboard: vi.fn() }));
|
||||
vi.mock("@/composables/useTheme", () => ({ useTheme: () => ({ isDark: ref(false), themeRevision: ref(0) }) }));
|
||||
vi.mock("@/stores/settingsStore", () => ({ useSettingsStore: () => ({ editorSettings: { uiFontFamily: "", fontFamily: "", fontSize: 14 } }) }));
|
||||
vi.mock("@/stores/connectionStore", () => ({
|
||||
useConnectionStore: () => ({
|
||||
repushPluginConnection: mocks.repushPluginConnection,
|
||||
reopenPluginConnection: mocks.reopenPluginConnection,
|
||||
}),
|
||||
}));
|
||||
vi.mock("vue-i18n", () => ({ useI18n: () => ({ locale: ref("en"), t: (key: string) => key }) }));
|
||||
|
||||
import PluginWorkbenchHost from "./PluginWorkbenchHost.vue";
|
||||
|
||||
const plugin: InstalledPlugin = {
|
||||
manifest: { id: "sample", name: "Sample", version: "1.0.0", permissions: [], drivers: [], contributions: [] },
|
||||
compatibility: { compatible: true },
|
||||
};
|
||||
const contribution: PluginWorkbenchContribution = { type: "workbench", id: "sample.main", label: "Sample" };
|
||||
|
||||
function deferred() {
|
||||
let resolve!: () => void;
|
||||
const promise = new Promise<void>((resolvePromise) => {
|
||||
resolve = resolvePromise;
|
||||
});
|
||||
return { promise, resolve };
|
||||
}
|
||||
|
||||
async function flushWorkbenchLoad() {
|
||||
await nextTick();
|
||||
await new Promise((resolve) => setTimeout(resolve, 0));
|
||||
await nextTick();
|
||||
}
|
||||
|
||||
describe("PluginWorkbenchHost initialization", () => {
|
||||
let app: App<Element> | undefined;
|
||||
let root: HTMLDivElement;
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
vi.stubGlobal("requestAnimationFrame", (callback: FrameRequestCallback) => window.setTimeout(() => callback(0), 0));
|
||||
vi.stubGlobal("getComputedStyle", () => Object.assign([], { getPropertyValue: () => "" }));
|
||||
mocks.readPluginUiEntry.mockResolvedValue({ dataBase64: btoa("<!doctype html><html><body></body></html>"), contentType: "text/html" });
|
||||
mocks.subscribePluginEvents.mockResolvedValue(vi.fn());
|
||||
mocks.repushPluginConnection.mockReset().mockResolvedValue(undefined);
|
||||
mocks.reopenPluginConnection.mockResolvedValue(undefined);
|
||||
root = document.createElement("div");
|
||||
document.body.appendChild(root);
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
app?.unmount();
|
||||
root.remove();
|
||||
vi.unstubAllGlobals();
|
||||
});
|
||||
|
||||
async function mountHost() {
|
||||
app = createApp(PluginWorkbenchHost, { plugin, contribution, context: { connectionId: "connection" } });
|
||||
app.mount(root);
|
||||
await flushWorkbenchLoad();
|
||||
const frame = root.querySelector("iframe");
|
||||
expect(frame).toBeInstanceOf(HTMLIFrameElement);
|
||||
const target = frame!.contentWindow!;
|
||||
const postMessage = vi.spyOn(target, "postMessage").mockImplementation(() => {});
|
||||
const ready = () => window.dispatchEvent(new MessageEvent("message", { source: target, data: { source: "dbx-plugin", version: 1, type: "ready" } }));
|
||||
ready();
|
||||
await vi.waitFor(() => expect(mocks.repushPluginConnection).toHaveBeenCalled());
|
||||
await new Promise((resolve) => setTimeout(resolve, 0));
|
||||
mocks.repushPluginConnection.mockClear();
|
||||
postMessage.mockClear();
|
||||
return { frame: frame!, postMessage, ready };
|
||||
}
|
||||
|
||||
it("runs reinit before one init when load precedes ready", async () => {
|
||||
const firstReinit = deferred();
|
||||
const { frame, postMessage, ready } = await mountHost();
|
||||
mocks.repushPluginConnection.mockReturnValue(firstReinit.promise);
|
||||
|
||||
frame.dispatchEvent(new Event("load"));
|
||||
ready();
|
||||
|
||||
expect(mocks.repushPluginConnection).toHaveBeenCalledTimes(1);
|
||||
expect(postMessage).not.toHaveBeenCalled();
|
||||
firstReinit.resolve();
|
||||
await vi.waitFor(() => expect(postMessage).toHaveBeenCalledTimes(1));
|
||||
expect(postMessage.mock.calls[0]?.[0]).toMatchObject({ type: "init" });
|
||||
|
||||
const secondReinit = deferred();
|
||||
mocks.repushPluginConnection.mockReturnValue(secondReinit.promise);
|
||||
frame.dispatchEvent(new Event("load"));
|
||||
ready();
|
||||
|
||||
expect(mocks.repushPluginConnection).toHaveBeenCalledTimes(2);
|
||||
expect(postMessage).toHaveBeenCalledTimes(1);
|
||||
secondReinit.resolve();
|
||||
await vi.waitFor(() => expect(postMessage).toHaveBeenCalledTimes(2));
|
||||
});
|
||||
|
||||
it("runs reinit before one init when ready precedes load", async () => {
|
||||
const reinit = deferred();
|
||||
const { frame, postMessage, ready } = await mountHost();
|
||||
mocks.repushPluginConnection.mockReturnValue(reinit.promise);
|
||||
|
||||
ready();
|
||||
frame.dispatchEvent(new Event("load"));
|
||||
|
||||
expect(mocks.repushPluginConnection).toHaveBeenCalledTimes(1);
|
||||
expect(postMessage).not.toHaveBeenCalled();
|
||||
reinit.resolve();
|
||||
await vi.waitFor(() => expect(postMessage).toHaveBeenCalledTimes(1));
|
||||
expect(postMessage.mock.calls[0]?.[0]).toMatchObject({ type: "init" });
|
||||
});
|
||||
});
|
||||
@@ -8,6 +8,8 @@ import { PluginHostBridge, pluginSandboxDocument, type PluginBridgeTheme, type P
|
||||
import type { InstalledPlugin, PluginWorkbenchContribution } from "@/types/database";
|
||||
import { useI18n } from "vue-i18n";
|
||||
import { useTheme } from "@/composables/useTheme";
|
||||
import { useSettingsStore } from "@/stores/settingsStore";
|
||||
import { useConnectionStore } from "@/stores/connectionStore";
|
||||
|
||||
const props = withDefaults(
|
||||
defineProps<{
|
||||
@@ -21,13 +23,14 @@ const props = withDefaults(
|
||||
const emit = defineEmits<{
|
||||
ready: [];
|
||||
error: [message: string];
|
||||
openWorkbench: [pluginId: string, contributionId: string, context?: PluginWorkbenchContext];
|
||||
openWorkbench: [pluginId: string, contributionId: string, context?: PluginWorkbenchContext, options?: { forceNew?: boolean }];
|
||||
openFilesystem: [pluginId: string, providerId: string, context?: PluginWorkbenchContext];
|
||||
closeTab: [];
|
||||
}>();
|
||||
|
||||
const { t, locale: appLocale } = useI18n();
|
||||
const { isDark, themeRevision } = useTheme();
|
||||
const settingsStore = useSettingsStore();
|
||||
const iframe = ref<HTMLIFrameElement>();
|
||||
const source = ref("");
|
||||
const loading = ref(true);
|
||||
@@ -70,8 +73,9 @@ function createBridge() {
|
||||
notify: api.notifyPlugin,
|
||||
sendBinary: api.sendPluginBinary,
|
||||
readAsset: api.readPluginUiAsset,
|
||||
openWorkbench: async (pluginId, contributionId, context) => emit("openWorkbench", pluginId, contributionId, context),
|
||||
openWorkbench: async (pluginId, contributionId, context, options) => emit("openWorkbench", pluginId, contributionId, context, options),
|
||||
openFilesystem: async (pluginId, providerId, context) => emit("openFilesystem", pluginId, providerId, context),
|
||||
reopenConnection: (pluginId, connectionId) => useConnectionStore().reopenPluginConnection(connectionId, pluginId),
|
||||
closeTab: () => emit("closeTab"),
|
||||
saveFile: (_pluginId, request, data) => savePluginFile(request, data),
|
||||
copyText: (_pluginId, text) => copyToClipboard(text),
|
||||
@@ -79,6 +83,16 @@ function createBridge() {
|
||||
appLocale.value,
|
||||
currentBridgeTheme(),
|
||||
);
|
||||
// An iframe reload (F5 / webview restart) drops the plugin sidecar's
|
||||
// in-memory connection registry while the host still holds the connection
|
||||
// open. Re-push the connection config through the same path as a sidebar
|
||||
// open before the plugin receives its fresh init, so it can reconnect
|
||||
// without the user reopening the connection from the sidebar.
|
||||
bridge.onReinit = async () => {
|
||||
const connectionId = props.context?.connectionId;
|
||||
if (!connectionId) return;
|
||||
await useConnectionStore().repushPluginConnection(connectionId);
|
||||
};
|
||||
}
|
||||
|
||||
/** Keep a plugin-supplied name from smuggling path separators or traversal into the save dialog. */
|
||||
@@ -161,6 +175,7 @@ async function inlineLocalUiAssets(html: string, pluginId: string): Promise<stri
|
||||
|
||||
async function loadWorkbench() {
|
||||
const generation = ++loadGeneration;
|
||||
bridge?.dispose();
|
||||
bridge = undefined;
|
||||
loading.value = true;
|
||||
frameReady.value = false;
|
||||
@@ -234,10 +249,18 @@ watch(appLocale, (locale) => bridge?.updateLocale(locale));
|
||||
// path — dark/light, palette switch, custom colors — re-pushes the resolved
|
||||
// tokens; watching isDark/custom colors alone misses palette-only switches.
|
||||
watch(themeRevision, () => bridge?.updateTheme(currentBridgeTheme()));
|
||||
// Font settings are applied outside applyTheme() (see App.vue applyUiFontFamily)
|
||||
// and therefore never bump themeRevision; watch them explicitly so font token
|
||||
// changes reach live plugin bridges without waiting for the next theme switch.
|
||||
watch(
|
||||
() => [settingsStore.editorSettings.uiFontFamily, settingsStore.editorSettings.fontFamily, settingsStore.editorSettings.fontSize],
|
||||
() => bridge?.updateTheme(currentBridgeTheme()),
|
||||
);
|
||||
|
||||
onBeforeUnmount(() => {
|
||||
disposed = true;
|
||||
loadGeneration += 1;
|
||||
bridge?.dispose();
|
||||
bridge = undefined;
|
||||
window.removeEventListener("message", onMessage);
|
||||
unsubscribeEvents?.();
|
||||
|
||||
@@ -30,6 +30,12 @@ const error = ref("");
|
||||
const deferred = ref(false);
|
||||
let loadGeneration = 0;
|
||||
const resolvedConnectionId = computed(() => props.connectionId || (typeof props.context?.connectionId === "string" ? props.context.connectionId : undefined));
|
||||
// The host's re-push-on-reinit and the plugin itself both key off
|
||||
// context.connectionId; filesystem tabs carry it only as a tab-level prop.
|
||||
const hostContext = computed(() => {
|
||||
if (!resolvedConnectionId.value || props.context?.connectionId) return props.context;
|
||||
return { ...props.context, connectionId: resolvedConnectionId.value };
|
||||
});
|
||||
const entry = computed(() => createFrontendPluginRegistry(plugins.value, appLocale.value).findWorkbench(props.pluginId, props.contributionId));
|
||||
|
||||
async function load() {
|
||||
@@ -54,7 +60,18 @@ async function refresh() {
|
||||
loading.value = true;
|
||||
error.value = "";
|
||||
try {
|
||||
if (resolvedConnectionId.value) await connectionStore.ensureConnected(resolvedConnectionId.value);
|
||||
const connectionId = resolvedConnectionId.value;
|
||||
if (connectionId) {
|
||||
if (connectionStore.connectedIds.has(connectionId)) {
|
||||
// The connection is open but the plugin sidecar's registry may have
|
||||
// been wiped (iframe reload, tab restore): ensureConnected's health
|
||||
// fast-path only probes the sidecar process and would not heal that.
|
||||
// Force a re-push instead; it no-ops when a push just happened.
|
||||
await connectionStore.repushPluginConnection(connectionId);
|
||||
} else {
|
||||
await connectionStore.ensureConnected(connectionId);
|
||||
}
|
||||
}
|
||||
await load();
|
||||
} catch (cause) {
|
||||
loading.value = false;
|
||||
@@ -73,9 +90,14 @@ function start() {
|
||||
void load();
|
||||
}
|
||||
|
||||
function openWorkbench(pluginId: string, contributionId: string, context?: PluginWorkbenchContext) {
|
||||
function openWorkbench(pluginId: string, contributionId: string, context?: PluginWorkbenchContext, options?: { forceNew?: boolean }) {
|
||||
const target = createFrontendPluginRegistry(plugins.value, appLocale.value).findWorkbench(pluginId, contributionId);
|
||||
queryStore.openPluginWorkbench(pluginId, contributionId, { title: target?.contribution.label || contributionId, context });
|
||||
// Session tabs are per-connection: title them after the connection (like the
|
||||
// sidebar-opened tab) so parallel sessions read "server", "server (2)", …
|
||||
// — the numbering itself is applied in queryStore.openPluginWorkbench.
|
||||
const contextConnectionId = typeof context?.connectionId === "string" ? context.connectionId : "";
|
||||
const connectionName = contextConnectionId ? connectionStore.getConfig(contextConnectionId)?.name : undefined;
|
||||
queryStore.openPluginWorkbench(pluginId, contributionId, { title: connectionName || target?.contribution.label || contributionId, context, forceNew: options?.forceNew === true });
|
||||
}
|
||||
|
||||
function openFilesystem(pluginId: string, providerId: string, context?: PluginWorkbenchContext) {
|
||||
@@ -105,6 +127,6 @@ defineExpose({ refresh });
|
||||
<AlertTriangle class="mt-0.5 size-4 shrink-0" />
|
||||
<span>{{ error || t("pluginPlatform.workbenchUnavailableFallback") }}</span>
|
||||
</div>
|
||||
<PluginWorkbenchHost v-else class="min-h-0 flex-1" :plugin="entry.plugin" :contribution="entry.contribution" :context="context" @open-workbench="openWorkbench" @open-filesystem="openFilesystem" @close-tab="emit('closeTab')" />
|
||||
<PluginWorkbenchHost v-else class="min-h-0 flex-1" :plugin="entry.plugin" :contribution="entry.contribution" :context="hostContext" @open-workbench="openWorkbench" @open-filesystem="openFilesystem" @close-tab="emit('closeTab')" />
|
||||
</div>
|
||||
</template>
|
||||
+33
-1
@@ -1,6 +1,16 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
import type { InstalledPlugin, PluginConnectionProviderContribution, PluginFormFieldValue } from "@/types/database";
|
||||
import { buildPluginConnectionConfig, createFrontendPluginRegistry, initialPluginFormValues, parsePluginConnectionProviderOptionValue, pluginConnectionActionsForDialog, pluginConnectionFormValues, pluginConnectionProviderIcon, pluginConnectionProviderOptionValue } from "./frontendPlugin";
|
||||
import {
|
||||
buildPluginConnectionConfig,
|
||||
createFrontendPluginRegistry,
|
||||
initialPluginFormValues,
|
||||
parsePluginConnectionProviderOptionValue,
|
||||
pluginConnectionActionsForDialog,
|
||||
pluginConnectionConnectTimeoutDefault,
|
||||
pluginConnectionFormValues,
|
||||
pluginConnectionProviderIcon,
|
||||
pluginConnectionProviderOptionValue,
|
||||
} from "./frontendPlugin";
|
||||
|
||||
function installedPlugin(id: string, contributions: InstalledPlugin["manifest"]["contributions"] = []): InstalledPlugin {
|
||||
return {
|
||||
@@ -70,6 +80,28 @@ describe("FrontendPluginRegistry", () => {
|
||||
expect(transient.save_password).toBe(false);
|
||||
});
|
||||
|
||||
it("materializes the provider's connect_timeout_secs default into the typed timeout", () => {
|
||||
const provider = connectionProvider({
|
||||
fields: [{ key: "connect_timeout_secs", label: "Connect timeout", type: "number", default: 30 }],
|
||||
});
|
||||
expect(pluginConnectionConnectTimeoutDefault(provider)).toBe(30);
|
||||
expect(buildPluginConnectionConfig("example.plugin", provider, {}).connect_timeout_secs).toBe(30);
|
||||
// An advanced-form value the user tuned is mirrored into the typed field too.
|
||||
expect(buildPluginConnectionConfig("example.plugin", provider, { connect_timeout_secs: 60 }).connect_timeout_secs).toBe(60);
|
||||
|
||||
// Once the provider declares the field, its resolved value is the single
|
||||
// source of truth: a stale typed value from before the declaration is
|
||||
// healed to the declared default.
|
||||
const existing = buildPluginConnectionConfig("example.plugin", connectionProvider({ fields: [] }), {});
|
||||
existing.connect_timeout_secs = 7;
|
||||
expect(buildPluginConnectionConfig("example.plugin", provider, {}, existing).connect_timeout_secs).toBe(30);
|
||||
|
||||
// Providers without the well-known field keep the generic 10s default.
|
||||
const generic = connectionProvider({ fields: [] });
|
||||
expect(pluginConnectionConnectTimeoutDefault(generic)).toBeUndefined();
|
||||
expect(buildPluginConnectionConfig("example.plugin", generic, {}).connect_timeout_secs).toBe(10);
|
||||
});
|
||||
|
||||
it("round-trips plugin connection provider picker values", () => {
|
||||
const value = pluginConnectionProviderOptionValue("example/plugin", "ssh:main");
|
||||
expect(parsePluginConnectionProviderOptionValue(value)).toEqual({ pluginId: "example/plugin", providerId: "ssh:main" });
|
||||
|
||||
@@ -105,6 +105,22 @@ export function pluginConnectionProviderIcon(entry: PluginContributionEntry<Plug
|
||||
return entry.contribution.icon || entry.plugin.manifest.icon;
|
||||
}
|
||||
|
||||
/**
|
||||
* Well-known provider field key whose declared default seeds the typed
|
||||
* `ConnectionConfig.connect_timeout_secs`. A plugin knows its own transport
|
||||
* (SSH handshakes on slow links need far more than the generic 10s), so a
|
||||
* declared default wins over the global timeout unless the user explicitly
|
||||
* picks a per-connection value in the dialog's Advanced tab.
|
||||
*/
|
||||
export const PLUGIN_CONNECT_TIMEOUT_FIELD_KEY = "connect_timeout_secs";
|
||||
|
||||
export function pluginConnectionConnectTimeoutDefault(contribution: PluginConnectionProviderContribution): number | undefined {
|
||||
const field = contribution.fields.find((candidate) => candidate.key === PLUGIN_CONNECT_TIMEOUT_FIELD_KEY && effectiveFieldBinding(candidate) === "config");
|
||||
const value = field?.default;
|
||||
if (typeof value !== "number" || !Number.isFinite(value) || value <= 0) return undefined;
|
||||
return Math.min(300, Math.max(1, Math.trunc(value)));
|
||||
}
|
||||
|
||||
export function pluginConnectionActionsForDialog(contribution: PluginConnectionProviderContribution, editing: boolean): PluginConnectionAction[] {
|
||||
const actions: PluginConnectionAction[] = [
|
||||
...(contribution.actions || []).map((action) => ({ ...action, kind: "custom" as const })),
|
||||
@@ -159,7 +175,7 @@ export function buildPluginConnectionConfig(pluginId: string, contribution: Plug
|
||||
plugin_connection_type: contribution.database_type,
|
||||
connection_secrets: connectionSecrets,
|
||||
transport_layers: existing?.transport_layers || [],
|
||||
connect_timeout_secs: existing?.connect_timeout_secs || 10,
|
||||
connect_timeout_secs: existing?.connect_timeout_secs || pluginConnectionConnectTimeoutDefault(contribution) || 10,
|
||||
query_timeout_secs: existing?.query_timeout_secs || 60,
|
||||
idle_timeout_secs: existing?.idle_timeout_secs || 60,
|
||||
keepalive_interval_secs: existing?.keepalive_interval_secs || 30,
|
||||
@@ -198,6 +214,17 @@ export function buildPluginConnectionConfig(pluginId: string, contribution: Plug
|
||||
config.database = value === undefined || value === "" ? undefined : String(value);
|
||||
}
|
||||
}
|
||||
// A config-bound connect_timeout_secs field is the plugin's own handshake
|
||||
// timeout (the SSH plugin lets advanced users tune it). Mirror the resolved
|
||||
// value into the typed field so the host RPC deadline never fires before the
|
||||
// plugin's own timeout. Only applies while the provider declares the field —
|
||||
// a stale external_config key from an older manifest must not leak through.
|
||||
if (pluginConnectionConnectTimeoutDefault(contribution) !== undefined) {
|
||||
const pluginConnectTimeout = externalConfig[PLUGIN_CONNECT_TIMEOUT_FIELD_KEY];
|
||||
if (typeof pluginConnectTimeout === "number" && Number.isFinite(pluginConnectTimeout) && pluginConnectTimeout > 0) {
|
||||
config.connect_timeout_secs = Math.min(300, Math.max(1, Math.trunc(pluginConnectTimeout)));
|
||||
}
|
||||
}
|
||||
return config;
|
||||
}
|
||||
|
||||
|
||||
@@ -58,6 +58,55 @@ describe("PluginHostBridge", () => {
|
||||
expect(messages[0]).toMatchObject({ source: "dbx-host", type: "init", locale: "zh-CN", context: { connectionId: "connection" } });
|
||||
});
|
||||
|
||||
it("runs onReinit before the init message when the iframe reloads", async () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
const bridge = new PluginHostBridge(plugin(), workbench, {}, () => target, {
|
||||
invoke: vi.fn(),
|
||||
notify: vi.fn(),
|
||||
sendBinary: vi.fn(),
|
||||
readAsset: vi.fn(),
|
||||
});
|
||||
const ready = () => bridge.handleWindowMessage({ source: target, data: { source: "dbx-plugin", version: 1, type: "ready" } } as MessageEvent);
|
||||
|
||||
const order: string[] = [];
|
||||
bridge.onReinit = () =>
|
||||
new Promise<void>((resolve) => {
|
||||
order.push("reinit");
|
||||
setTimeout(resolve, 5);
|
||||
});
|
||||
const originalPost = target.postMessage.bind(target);
|
||||
(target as { postMessage: (m: unknown) => void }).postMessage = (message: unknown) => {
|
||||
if ((message as { type?: string }).type === "init") order.push("init");
|
||||
originalPost(message);
|
||||
};
|
||||
|
||||
expect(ready()).toBe(true);
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(1));
|
||||
expect(order).toEqual(["reinit", "init"]);
|
||||
});
|
||||
|
||||
it("initializes only once when ready is repeated for the same bridge", async () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
const bridge = new PluginHostBridge(plugin(), workbench, {}, () => target, {
|
||||
invoke: vi.fn(),
|
||||
notify: vi.fn(),
|
||||
sendBinary: vi.fn(),
|
||||
readAsset: vi.fn(),
|
||||
});
|
||||
const onReinit = vi.fn();
|
||||
bridge.onReinit = onReinit;
|
||||
const ready = () => bridge.handleWindowMessage({ source: target, data: { source: "dbx-plugin", version: 1, type: "ready" } } as MessageEvent);
|
||||
|
||||
ready();
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(1));
|
||||
ready();
|
||||
|
||||
expect(onReinit).toHaveBeenCalledTimes(1);
|
||||
expect(messages).toHaveLength(1);
|
||||
});
|
||||
|
||||
it("snapshots nested Vue reactive context values before sending them to the plugin", () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
@@ -135,6 +184,69 @@ describe("PluginHostBridge", () => {
|
||||
expect(messages[0]).toMatchObject({ id: "2", error: "Plugin has not declared permission 'host.workbench'" });
|
||||
});
|
||||
|
||||
it("passes forceNew through host.openWorkbench and defaults it to false", async () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
const openWorkbench = vi.fn();
|
||||
const bridge = new PluginHostBridge(plugin(["host.workbench"]), workbench, {}, () => target, {
|
||||
invoke: vi.fn(),
|
||||
notify: vi.fn(),
|
||||
sendBinary: vi.fn(),
|
||||
readAsset: vi.fn(),
|
||||
openWorkbench,
|
||||
});
|
||||
const request = (id: string, params: unknown) =>
|
||||
bridge.handleWindowMessage({
|
||||
source: target,
|
||||
data: { source: "dbx-plugin", version: 1, type: "request", id, method: "host.openWorkbench", params },
|
||||
} as MessageEvent);
|
||||
|
||||
request("new", { contributionId: "sample.other", context: { connectionId: "c1" }, forceNew: true });
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(1));
|
||||
expect(openWorkbench).toHaveBeenNthCalledWith(1, "sample", "sample.other", { connectionId: "c1" }, { forceNew: true });
|
||||
expect(messages[0]).toMatchObject({ id: "new", result: null });
|
||||
|
||||
request("default", { contributionId: "sample.other" });
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(2));
|
||||
expect(openWorkbench).toHaveBeenNthCalledWith(2, "sample", "sample.other", undefined, { forceNew: false });
|
||||
});
|
||||
|
||||
it("reopens a plugin connection through host.reopenConnection", async () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
const reopenConnection = vi.fn().mockResolvedValue(undefined);
|
||||
const bridge = new PluginHostBridge(plugin(), workbench, {}, () => target, {
|
||||
invoke: vi.fn(),
|
||||
notify: vi.fn(),
|
||||
sendBinary: vi.fn(),
|
||||
readAsset: vi.fn(),
|
||||
reopenConnection,
|
||||
});
|
||||
const request = (id: string, params: unknown) =>
|
||||
bridge.handleWindowMessage({
|
||||
source: target,
|
||||
data: { source: "dbx-plugin", version: 1, type: "request", id, method: "host.reopenConnection", params },
|
||||
} as MessageEvent);
|
||||
|
||||
request("reopen", { connectionId: "9f1c2a34-0000-4000-8000-abcdef012345" });
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(1));
|
||||
expect(reopenConnection).toHaveBeenCalledWith("sample", "9f1c2a34-0000-4000-8000-abcdef012345");
|
||||
expect(messages[0]).toMatchObject({ id: "reopen", result: { ok: true } });
|
||||
|
||||
// Missing/invalid connectionId is rejected before reaching the host.
|
||||
request("missing", {});
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(2));
|
||||
expect(messages[1]).toMatchObject({ id: "missing" });
|
||||
expect((messages[1] as { error?: string }).error).toContain("connectionId");
|
||||
|
||||
// Host-side failures (connection missing, not plugin-backed, connect
|
||||
// failed) surface as the request error so the plugin can show them.
|
||||
reopenConnection.mockRejectedValueOnce(new Error("Connection is not plugin-backed"));
|
||||
request("not-plugin", { connectionId: "mysql-1" });
|
||||
await vi.waitFor(() => expect(messages).toHaveLength(3));
|
||||
expect(messages[2]).toMatchObject({ id: "not-plugin", error: "Connection is not plugin-backed" });
|
||||
});
|
||||
|
||||
it("opens only the owning plugin filesystem with explicit permission", async () => {
|
||||
const messages: unknown[] = [];
|
||||
const target = { postMessage: (message: unknown) => messages.push(message) } as unknown as Window;
|
||||
|
||||
@@ -38,8 +38,10 @@ export interface PluginHostBridgeApi {
|
||||
notify(pluginId: string, method: string, params?: unknown): Promise<void>;
|
||||
sendBinary(pluginId: string, channel: string, dataBase64: string): Promise<void>;
|
||||
readAsset(pluginId: string, path: string): Promise<PluginUiAssetPayload>;
|
||||
openWorkbench?(pluginId: string, contributionId: string, context?: PluginWorkbenchContext): Promise<void> | void;
|
||||
openWorkbench?(pluginId: string, contributionId: string, context?: PluginWorkbenchContext, options?: { forceNew?: boolean }): Promise<void> | void;
|
||||
openFilesystem?(pluginId: string, providerId: string, context?: PluginWorkbenchContext): Promise<void> | void;
|
||||
/** Explicit user-triggered reconnect of an owned plugin connection (full flow, interactive password prompt allowed). */
|
||||
reopenConnection?(pluginId: string, connectionId: string): Promise<void>;
|
||||
closeTab?(): Promise<void> | void;
|
||||
/** Persist plugin bytes through the host's native save dialog. Resolves null when the user cancels. */
|
||||
saveFile?(pluginId: string, request: PluginSaveFileRequest, data: Uint8Array): Promise<PluginSaveFileResult | null>;
|
||||
@@ -63,6 +65,9 @@ export class PluginHostBridge {
|
||||
private locale: string;
|
||||
private theme?: PluginBridgeTheme;
|
||||
|
||||
/** Invoked once before each iframe load generation sends its init message. */
|
||||
onReinit?: () => Promise<void> | void;
|
||||
|
||||
constructor(
|
||||
private readonly plugin: InstalledPlugin,
|
||||
private readonly workbench: PluginWorkbenchContribution,
|
||||
@@ -82,7 +87,7 @@ export class PluginHostBridge {
|
||||
if (!target || event.source !== target || !isRecord(event.data)) return false;
|
||||
if (event.data.source !== PLUGIN_MESSAGE_SOURCE || event.data.version !== BRIDGE_VERSION) return false;
|
||||
if (event.data.type === "ready") {
|
||||
this.sendInit();
|
||||
void this.handleReady();
|
||||
return true;
|
||||
}
|
||||
if (event.data.type === "shortcut" && event.data.shortcut === "closeTab") {
|
||||
@@ -94,7 +99,52 @@ export class PluginHostBridge {
|
||||
return true;
|
||||
}
|
||||
|
||||
private handleReady(): void {
|
||||
this.requestInit("ready");
|
||||
}
|
||||
|
||||
sendInit(): void {
|
||||
this.requestInit("load");
|
||||
}
|
||||
|
||||
private requestInit(signal: "load" | "ready"): void {
|
||||
if ((this.initSignals.load && this.initSignals.ready) || (signal === "load" && this.initSignals.load)) {
|
||||
this.initGeneration += 1;
|
||||
this.initSignals = { load: false, ready: false };
|
||||
this.initStarted = false;
|
||||
}
|
||||
if (this.initSignals[signal]) return;
|
||||
this.initSignals[signal] = true;
|
||||
if (this.initStarted) return;
|
||||
this.initStarted = true;
|
||||
const generation = this.initGeneration;
|
||||
const reinit = this.onReinit;
|
||||
if (!reinit) {
|
||||
if (!this.disposed && generation === this.initGeneration) this.postInit();
|
||||
return;
|
||||
}
|
||||
void (async () => {
|
||||
try {
|
||||
await reinit();
|
||||
} catch (error) {
|
||||
console.warn("[DBX][plugin-bridge:reinit]", error);
|
||||
}
|
||||
// A newer load generation supersedes this one; never post a stale init.
|
||||
if (!this.disposed && generation === this.initGeneration) this.postInit();
|
||||
})();
|
||||
}
|
||||
|
||||
/** Stop future posts (queued inits after an async reinit) for a torn-down bridge. */
|
||||
dispose(): void {
|
||||
this.disposed = true;
|
||||
}
|
||||
|
||||
private disposed = false;
|
||||
private initGeneration = 0;
|
||||
private initSignals = { load: false, ready: false };
|
||||
private initStarted = false;
|
||||
|
||||
private postInit(): void {
|
||||
this.post({
|
||||
source: HOST_MESSAGE_SOURCE,
|
||||
version: BRIDGE_VERSION,
|
||||
@@ -186,9 +236,15 @@ export class PluginHostBridge {
|
||||
this.requirePermission("host.workbench");
|
||||
if (!this.api.openWorkbench) throw new Error("Host workbench navigation is unavailable");
|
||||
const input = requireRecord(params, "host.openWorkbench params");
|
||||
await this.api.openWorkbench(this.plugin.manifest.id, requireProtocolName(input.contributionId, "workbench contribution"), isRecord(input.context) ? input.context : undefined);
|
||||
await this.api.openWorkbench(this.plugin.manifest.id, requireProtocolName(input.contributionId, "workbench contribution"), isRecord(input.context) ? input.context : undefined, { forceNew: input.forceNew === true });
|
||||
return null;
|
||||
}
|
||||
if (method === "host.reopenConnection") {
|
||||
if (!this.api.reopenConnection) throw new Error("Connection reopen is unavailable");
|
||||
const input = requireRecord(params, "host.reopenConnection params");
|
||||
await this.api.reopenConnection(this.plugin.manifest.id, requireProtocolName(input.connectionId, "connectionId"));
|
||||
return { ok: true };
|
||||
}
|
||||
if (method === "host.openFilesystem") {
|
||||
this.requirePermission("host.filesystem");
|
||||
if (!this.api.openFilesystem) throw new Error("Host filesystem navigation is unavailable");
|
||||
@@ -498,8 +554,9 @@ export function pluginSdkSource(initialTheme?: PluginBridgeTheme): string {
|
||||
const asset = await request('ui.readAsset', { path });
|
||||
return URL.createObjectURL(new Blob([decode(asset.dataBase64)], { type: asset.contentType }));
|
||||
},
|
||||
openWorkbench: (contributionId, childContext) => request('host.openWorkbench', { contributionId, context: childContext }),
|
||||
openWorkbench: (contributionId, childContext, options) => request('host.openWorkbench', { contributionId, context: childContext, forceNew: !!(options && options.forceNew) }),
|
||||
openFilesystem: (providerId, childContext) => request('host.openFilesystem', { providerId, context: childContext }),
|
||||
reopenConnection: (connectionId) => request('host.reopenConnection', { connectionId }),
|
||||
saveFile: (options = {}, data) => {
|
||||
if (data === undefined) return request('host.saveFile', options);
|
||||
if (typeof data === 'string') return request('host.saveFile', { ...(options || {}), dataBase64: data });
|
||||
|
||||
@@ -548,6 +548,35 @@ describe("connectionStore plugin password prompt", () => {
|
||||
expect(connectDb).toHaveBeenCalledWith(expect.objectContaining({ id: "ssh-1", external_config: expect.objectContaining({ authentication: "private-key" }) }), expect.any(Number));
|
||||
});
|
||||
|
||||
it("rejects reopening a connection owned by another plugin before prompting or connecting", async () => {
|
||||
const connectDb = vi.fn().mockResolvedValue("ssh-1");
|
||||
installApiMocks({ connectDb, listPlugins: vi.fn().mockResolvedValue([sshInstalledPlugin()]) });
|
||||
installPasswordPromptMock();
|
||||
requestPassword.mockResolvedValue({ password: PROMPT_TYPED_PASSWORD, rememberPassword: false });
|
||||
const { useConnectionStore } = await import("@/stores/connectionStore");
|
||||
const store = useConnectionStore();
|
||||
store.connections = [pluginConnection({ external_config: { authentication: "password" } })];
|
||||
|
||||
await expect(store.reopenPluginConnection("ssh-1", "io.dbx.other")).rejects.toThrow("Connection is owned by another plugin");
|
||||
|
||||
expect(requestPassword).not.toHaveBeenCalled();
|
||||
expect(connectDb).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("allows the owning plugin to reopen its connection", async () => {
|
||||
const connectDb = vi.fn().mockResolvedValue("ssh-1");
|
||||
installApiMocks({ connectDb, listPlugins: vi.fn().mockResolvedValue([sshInstalledPlugin()]) });
|
||||
installPasswordPromptMock();
|
||||
const { useConnectionStore } = await import("@/stores/connectionStore");
|
||||
const store = useConnectionStore();
|
||||
store.connections = [pluginConnection()];
|
||||
|
||||
await expect(store.reopenPluginConnection("ssh-1", "io.dbx.ssh")).resolves.toBeUndefined();
|
||||
|
||||
expect(requestPassword).not.toHaveBeenCalled();
|
||||
expect(connectDb).toHaveBeenCalledWith(expect.objectContaining({ id: "ssh-1" }), expect.any(Number));
|
||||
});
|
||||
|
||||
it("still prompts for a plugin connection whose manifest requires the login password", async () => {
|
||||
installApiMocks({ listPlugins: vi.fn().mockResolvedValue([sshInstalledPlugin()]) });
|
||||
installPasswordPromptMock();
|
||||
|
||||
@@ -95,4 +95,74 @@ describe("queryStore openPluginWorkbench reuse", () => {
|
||||
// Adoption must not replace the live context (same no-deep-reload rule).
|
||||
expect(queryStore.tabs.find((t) => t.id === id)?.pluginWorkbench?.context).toEqual({ connectionId: "conn-1" });
|
||||
});
|
||||
|
||||
it("numbers same-connection session tabs Termius-style and never backfills", async () => {
|
||||
const queryStore = useQueryStore();
|
||||
|
||||
// First tab keeps the bare connection name; numbering starts at (1) from
|
||||
// the second tab on.
|
||||
const firstId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", {
|
||||
title: "SSH server222",
|
||||
connectionId: "conn-1",
|
||||
context: { connectionId: "conn-1" },
|
||||
});
|
||||
// forceNew is orthogonal to numbering: it skips dedup, the count still grows.
|
||||
const secondId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", {
|
||||
title: "SSH server222",
|
||||
connectionId: "conn-1",
|
||||
context: { connectionId: "conn-1", workbenchId: "wb-2" },
|
||||
forceNew: true,
|
||||
});
|
||||
const thirdId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", {
|
||||
title: "SSH server222",
|
||||
connectionId: "conn-1",
|
||||
context: { connectionId: "conn-1", workbenchId: "wb-3" },
|
||||
forceNew: true,
|
||||
});
|
||||
|
||||
const titles = Object.fromEntries(queryStore.tabs.map((tab) => [tab.id, tab.title]));
|
||||
expect([titles[firstId], titles[secondId], titles[thirdId]]).toEqual(["SSH server222", "SSH server222 (1)", "SSH server222 (2)"]);
|
||||
|
||||
// Closing tab (1) must not renumber (2): existing titles stay stable (no
|
||||
// backfill, no drift). The next tab advances beyond the highest live suffix
|
||||
// so two open sessions never share a title. (Removed directly: the stubbed
|
||||
// window in this harness lacks the timers closeTab's cleanup needs.)
|
||||
queryStore.tabs = queryStore.tabs.filter((tab) => tab.id !== secondId);
|
||||
const fourthId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", {
|
||||
title: "SSH server222",
|
||||
connectionId: "conn-1",
|
||||
context: { connectionId: "conn-1", workbenchId: "wb-4" },
|
||||
forceNew: true,
|
||||
});
|
||||
expect(queryStore.tabs.find((tab) => tab.id === thirdId)?.title).toBe("SSH server222 (2)");
|
||||
expect(queryStore.tabs.find((tab) => tab.id === fourthId)?.title).toBe("SSH server222 (3)");
|
||||
|
||||
// Closing the highest suffix may reuse that now-free number, but must not
|
||||
// collide with any session that remains open.
|
||||
queryStore.tabs = queryStore.tabs.filter((tab) => tab.id !== fourthId);
|
||||
const fifthId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", {
|
||||
title: "SSH server222",
|
||||
connectionId: "conn-1",
|
||||
context: { connectionId: "conn-1", workbenchId: "wb-5" },
|
||||
forceNew: true,
|
||||
});
|
||||
expect(queryStore.tabs.find((tab) => tab.id === fifthId)?.title).toBe("SSH server222 (3)");
|
||||
});
|
||||
|
||||
it("numbers sessions independently per connection and reads bridge-style context-only connection ids", async () => {
|
||||
const queryStore = useQueryStore();
|
||||
|
||||
queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", { title: "A", connectionId: "conn-1", context: { connectionId: "conn-1" }, forceNew: true });
|
||||
const otherConnectionId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", { title: "B", connectionId: "conn-2", context: { connectionId: "conn-2" }, forceNew: true });
|
||||
// A different connection keeps its own bare title, unaffected by conn-1's count.
|
||||
expect(queryStore.tabs.find((tab) => tab.id === otherConnectionId)?.title).toBe("B");
|
||||
|
||||
// Bridge-path legacy tab: connection only inside context (tab-level "").
|
||||
// It must join conn-1's numbering (second conn-1 tab → (1)) and its own
|
||||
// tab-level connectionId gets normalized for reconnect/restore logic.
|
||||
const bridgedId = queryStore.openPluginWorkbench("io.dbx.ssh", "workbench", { title: "A", context: { connectionId: "conn-1" }, forceNew: true });
|
||||
const bridged = queryStore.tabs.find((tab) => tab.id === bridgedId);
|
||||
expect(bridged?.title).toBe("A (1)");
|
||||
expect(bridged?.connectionId).toBe("conn-1");
|
||||
});
|
||||
});
|
||||
@@ -4334,8 +4334,8 @@ export const useConnectionStore = defineStore("connection", () => {
|
||||
invalidateConnectionMetadataLifetime(connectionId, database);
|
||||
}
|
||||
|
||||
async function ensureConnected(connectionId: string, options: { activate?: boolean; verifyHealth?: boolean } = {}) {
|
||||
if (connectedIds.value.has(connectionId)) {
|
||||
async function ensureConnected(connectionId: string, options: { activate?: boolean; verifyHealth?: boolean; forceReconnect?: boolean } = {}) {
|
||||
if (!options.forceReconnect && connectedIds.value.has(connectionId)) {
|
||||
// Pure navigation can safely trust the existing connected state. Its
|
||||
// destination will perform the real API request, while blocking here on
|
||||
// a health probe makes an otherwise local tab switch take up to 5s.
|
||||
@@ -4450,6 +4450,49 @@ export const useConnectionStore = defineStore("connection", () => {
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Re-push an already-open plugin connection's config (credentials included)
|
||||
* to its sidecar through the same connection/connect path used when opening
|
||||
* from the sidebar. A plugin iframe reload can leave the sidecar's in-memory
|
||||
* connection registry empty while the host still considers the connection
|
||||
* open, and ensureConnected()'s health fast-path would not heal that. Silent
|
||||
* no-op when the connection is not open, is not plugin-backed, or its
|
||||
* credentials are no longer available (save_password=false without a live
|
||||
* session credential): the plugin then keeps its existing "reopen from the
|
||||
* sidebar" guidance instead of triggering an interactive prompt from a
|
||||
* background re-init.
|
||||
*/
|
||||
async function repushPluginConnection(connectionId: string): Promise<void> {
|
||||
const config = getConfig(connectionId);
|
||||
if (!config || config.db_type !== "plugin" || !connectedIds.value.has(connectionId)) return;
|
||||
// A successful connect/health probe within the TTL means the sidebar open
|
||||
// (or a fresh restore connect) pushed the config moments ago and the
|
||||
// sidecar registry cannot plausibly be empty yet — skipping here keeps the
|
||||
// first open from paying a redundant disconnect+connect cycle on the
|
||||
// plugin's first `ready`. The 2s in-memory TTL dies with the frontend, so
|
||||
// every realistic reload path still re-pushes.
|
||||
if (hasRecentConnectionHealthCheck(connectionId)) return;
|
||||
if (connectionNeedsPasswordPrompt(config) && (await pluginConnectionPasswordPromptNeeded(config)) && !(await hasSessionCredential(connectionId))) return;
|
||||
await ensureConnected(connectionId, { activate: false, forceReconnect: true });
|
||||
}
|
||||
|
||||
/**
|
||||
* Explicit, user-triggered reconnect of a plugin connection (the plugin's own
|
||||
* "reconnect" button). Unlike repushPluginConnection — the silent background
|
||||
* heal — this runs the full connect flow and MAY show the interactive
|
||||
* password prompt, which is appropriate for a deliberate user action. Editing
|
||||
* a connection drops it from connectedIds without notifying plugins, so the
|
||||
* sidecar's config goes stale; this is the plugin's way to request a fresh
|
||||
* connection/connect with the updated config.
|
||||
*/
|
||||
async function reopenPluginConnection(connectionId: string, pluginId: string): Promise<void> {
|
||||
const config = getConfig(connectionId);
|
||||
if (!config) throw new Error("Connection config not found");
|
||||
if (config.db_type !== "plugin") throw new Error("Connection is not plugin-backed");
|
||||
if (config.plugin_id !== pluginId) throw new Error("Connection is owned by another plugin");
|
||||
await ensureConnected(connectionId, { activate: false, forceReconnect: true });
|
||||
}
|
||||
|
||||
function setBeforeConnectHandler(handler: BeforeConnectHandler | null) {
|
||||
beforeConnectHandler = handler;
|
||||
}
|
||||
@@ -9334,6 +9377,8 @@ export const useConnectionStore = defineStore("connection", () => {
|
||||
startCreatingConnectionInGroup,
|
||||
stopCreatingConnectionInGroup,
|
||||
connect,
|
||||
repushPluginConnection,
|
||||
reopenPluginConnection,
|
||||
cancelConnecting,
|
||||
disconnect,
|
||||
hasDisconnectInFlight,
|
||||
|
||||
@@ -3465,9 +3465,23 @@ export const useQueryStore = defineStore("query", () => {
|
||||
tab.nacosTargetKeyword = undefined;
|
||||
}
|
||||
|
||||
/**
|
||||
* Effective connection binding of a plugin tab: bridge-created tabs carry
|
||||
* the connection only inside context (tab-level connectionId is ""), so
|
||||
* read both. New tabs normalize the context value up to the tab level at
|
||||
* creation, but restored/legacy tabs still need the fallback.
|
||||
*/
|
||||
function pluginTabConnectionId(tab: QueryTab): string {
|
||||
if (tab.connectionId) return tab.connectionId;
|
||||
const contextConnectionId = tab.pluginWorkbench?.context?.connectionId;
|
||||
return typeof contextConnectionId === "string" ? contextConnectionId : "";
|
||||
}
|
||||
|
||||
function openPluginWorkbench(pluginId: string, contributionId: string, options: { title?: string; connectionId?: string; database?: string; context?: Record<string, unknown>; forceNew?: boolean } = {}) {
|
||||
const contextConnectionId = typeof options.context?.connectionId === "string" ? options.context.connectionId : "";
|
||||
const connectionId = options.connectionId || contextConnectionId;
|
||||
if (!options.forceNew) {
|
||||
const existing = tabs.value.find((tab) => tab.mode === "plugin-workbench" && tab.pluginWorkbench?.pluginId === pluginId && tab.pluginWorkbench?.contributionId === contributionId && tab.connectionId === (options.connectionId || ""));
|
||||
const existing = tabs.value.find((tab) => tab.mode === "plugin-workbench" && tab.pluginWorkbench?.pluginId === pluginId && tab.pluginWorkbench?.contributionId === contributionId && pluginTabConnectionId(tab) === connectionId);
|
||||
if (existing) {
|
||||
// Reopening surfaces the existing tab as-is. Replacing the context
|
||||
// here (openPluginConnection mints a fresh workbenchId per click)
|
||||
@@ -3484,11 +3498,23 @@ export const useQueryStore = defineStore("query", () => {
|
||||
}
|
||||
}
|
||||
|
||||
// Termius-style session numbering: the first same-plugin + same-connection
|
||||
// tab keeps the bare connection name; each additional one gets " (n)"
|
||||
// where n advances beyond the highest live suffix (second tab → (1)).
|
||||
// Numbers are assigned at creation and never backfilled after a close —
|
||||
// stable titles beat dense numbering, while live titles stay unique.
|
||||
const siblingTabs = tabs.value.filter((tab) => tab.mode === "plugin-workbench" && tab.pluginWorkbench?.pluginId === pluginId && pluginTabConnectionId(tab) === connectionId);
|
||||
const nextSessionNumber =
|
||||
siblingTabs.reduce((highest, tab) => {
|
||||
const suffix = / \((\d+)\)$/.exec(tab.title);
|
||||
return suffix ? Math.max(highest, Number(suffix[1])) : highest;
|
||||
}, 0) + 1;
|
||||
const baseTitle = options.title || contributionId;
|
||||
const id = uuid();
|
||||
const tab: QueryTab = {
|
||||
id,
|
||||
title: options.title || contributionId,
|
||||
connectionId: options.connectionId || "",
|
||||
title: siblingTabs.length >= 1 ? `${baseTitle} (${nextSessionNumber})` : baseTitle,
|
||||
connectionId,
|
||||
database: options.database || "",
|
||||
sql: "",
|
||||
isExecuting: false,
|
||||
|
||||
Reference in new issue
Block a user