67 lines
3.7 KiB
TypeScript
67 lines
3.7 KiB
TypeScript
import type { ConnectionConfig, DatabaseType } from "@/types/database";
|
|
import { filterDatabaseNamesForVisiblePicker, normalizeVisibleDatabaseSelection } from "@/lib/database/visibleDatabases";
|
|
|
|
const DRAFT_VISIBLE_DATABASES_PREFIX = "__visible_draft_";
|
|
|
|
// Turso and Cloudflare D1 target one fixed SQLite-compatible `main` namespace;
|
|
// non-database services expose their own root objects rather than database namespaces.
|
|
const UNSUPPORTED_VISIBLE_DATABASE_TYPES = new Set<DatabaseType>(["turso", "cloudflare-d1", "elasticsearch", "easysearch", "qdrant", "milvus", "weaviate", "chromadb", "etcd", "zookeeper", "mq", "nacos"]);
|
|
|
|
type VisibleDatabaseConnectionFields = Pick<
|
|
ConnectionConfig,
|
|
"db_type" | "driver_profile" | "host" | "port" | "username" | "database" | "connection_string" | "url_params" | "redis_connection_mode" | "redis_sentinel_master" | "redis_sentinel_nodes" | "redis_cluster_nodes" | "etcd_endpoints" | "jdbc_driver_class"
|
|
>;
|
|
|
|
type VisibleObjectFilterConnectionFields = VisibleDatabaseConnectionFields & Pick<ConnectionConfig, "visible_databases" | "visible_schemas">;
|
|
|
|
export function buildDraftVisibleDatabasesConnectionId(seed: string): string {
|
|
return `${DRAFT_VISIBLE_DATABASES_PREFIX}${seed}`;
|
|
}
|
|
|
|
export function connectionCanChooseVisibleDatabases(connection: Pick<ConnectionConfig, "db_type"> | undefined): boolean {
|
|
return !!connection?.db_type && !UNSUPPORTED_VISIBLE_DATABASE_TYPES.has(connection.db_type);
|
|
}
|
|
|
|
export function initialVisibleDatabaseSelection(databaseNames: string[], visibleDatabases: string[] | undefined, connection?: Pick<ConnectionConfig, "db_type" | "driver_profile" | "visible_databases">): string[] {
|
|
if (Array.isArray(visibleDatabases)) {
|
|
return normalizeVisibleDatabaseSelection(visibleDatabases, databaseNames);
|
|
}
|
|
return filterDatabaseNamesForVisiblePicker(databaseNames, connection);
|
|
}
|
|
|
|
export function appendVisibleDatabaseSelection(visibleDatabases: string[] | undefined, databaseName: string): string[] | undefined {
|
|
if (!Array.isArray(visibleDatabases)) return undefined;
|
|
const name = databaseName.trim();
|
|
if (!name || visibleDatabases.includes(name)) return visibleDatabases;
|
|
return [...visibleDatabases, name];
|
|
}
|
|
|
|
export function visibleDatabaseSelectionIsStale(previous: VisibleDatabaseConnectionFields, current: VisibleDatabaseConnectionFields): boolean {
|
|
return visibleDatabaseFingerprint(previous) !== visibleDatabaseFingerprint(current);
|
|
}
|
|
|
|
export function visibleObjectFiltersNeedReset(previous: VisibleObjectFilterConnectionFields, current: VisibleObjectFilterConnectionFields): boolean {
|
|
const hasVisibleDatabaseFilter = Array.isArray(current.visible_databases);
|
|
const hasVisibleSchemaFilter = !!current.visible_schemas && Object.keys(current.visible_schemas).length > 0;
|
|
return (hasVisibleDatabaseFilter || hasVisibleSchemaFilter) && visibleDatabaseSelectionIsStale(previous, current);
|
|
}
|
|
|
|
function visibleDatabaseFingerprint(connection: VisibleDatabaseConnectionFields): string {
|
|
return JSON.stringify({
|
|
db_type: connection.db_type,
|
|
driver_profile: connection.driver_profile || "",
|
|
host: connection.host || "",
|
|
port: Number(connection.port) || 0,
|
|
username: connection.username || "",
|
|
database: connection.database || "",
|
|
connection_string: connection.connection_string || "",
|
|
url_params: connection.url_params || "",
|
|
redis_connection_mode: connection.redis_connection_mode || "",
|
|
redis_sentinel_master: connection.redis_sentinel_master || "",
|
|
redis_sentinel_nodes: connection.redis_sentinel_nodes || "",
|
|
redis_cluster_nodes: connection.redis_cluster_nodes || "",
|
|
etcd_endpoints: connection.etcd_endpoints || "",
|
|
jdbc_driver_class: connection.jdbc_driver_class || "",
|
|
});
|
|
}
|