feat(connections): fetch server connections

apps/desktop/src/entities/database/dialects/clickhouse/schema/index.ts

@@ -1,4 +1,7 @@
 import type { WithSchema } from '../../../utils/types'
 import type { InformationSchema } from './information'
 
+import type { System } from './system'
+
 export type Database = WithSchema<InformationSchema, 'information_schema'>
+  & WithSchema<System, 'system'>

apps/desktop/src/entities/database/dialects/clickhouse/schema/system.ts

@@ -0,0 +1,20 @@
+/**
+ * @name system
+ * @type schema
+ */
+export interface System {
+  databases: Databases
+}
+
+/**
+ * @name databases
+ * @type table
+ */
+interface Databases {
+  name: string
+  engine: string
+  data_path: string
+  metadata_path: string
+  uuid: string
+  comment: string
+}

apps/desktop/src/entities/database/dialects/mssql/schema/index.ts

@@ -1,4 +1,7 @@
 import type { WithSchema } from '../../../utils/types'
 import type { InformationSchema } from './information'
 
+import type { Sys } from './sys'
+
 export type Database = WithSchema<InformationSchema, 'information_schema'>
+  & WithSchema<Sys, 'sys'>

apps/desktop/src/entities/database/dialects/mssql/schema/sys.ts

@@ -0,0 +1,88 @@
+/**
+ * @name sys
+ * @type schema
+ */
+export interface Sys {
+  databases: Databases
+}
+
+/**
+ * @name databases
+ * @type table
+ */
+interface Databases {
+  name: string
+  database_id: number
+  source_database_id: number | null
+  owner_sid: string
+  create_date: Date
+  compatibility_level: number
+  collation_name: string | null
+  user_access: number
+  user_access_desc: string | null
+  is_read_only: boolean
+  is_auto_shrink_on: boolean
+  state: number
+  state_desc: string | null
+  is_in_standby: boolean
+  is_cleanly_shutdown: boolean
+  is_supplemental_logging_enabled: boolean
+  snapshot_isolation_state: number
+  snapshot_isolation_state_desc: string | null
+  recovery_model: number
+  recovery_model_desc: string | null
+  page_verify_option: number
+  page_verify_option_desc: string | null
+  is_auto_create_stats_on: boolean
+  is_auto_update_stats_on: boolean
+  is_auto_update_stats_async_on: boolean
+  is_ansi_null_default_on: boolean
+  is_ansi_nulls_on: boolean
+  is_ansi_padding_on: boolean
+  is_ansi_warnings_on: boolean
+  is_arithabort_on: boolean
+  is_concat_null_yields_null_on: boolean
+  is_numeric_roundabort_on: boolean
+  is_quoted_identifier_on: boolean
+  is_recursive_triggers_on: boolean
+  is_cursor_close_on_commit_on: boolean
+  is_local_cursor_default: boolean
+  is_fulltext_enabled: boolean
+  is_trustworthy_on: boolean
+  is_db_chaining_on: boolean
+  is_parameterization_forced: boolean
+  is_master_key_encrypted_by_server: boolean
+  is_query_store_on: boolean
+  is_published: boolean
+  is_subscribed: boolean
+  is_merge_published: boolean
+  is_distributor: boolean
+  is_sync_with_backup: boolean
+  service_broker_guid: string
+  is_broker_enabled: boolean
+  log_reuse_wait: number
+  log_reuse_wait_desc: string | null
+  is_date_correlation_on: boolean
+  is_cdc_enabled: boolean
+  is_encrypted: boolean
+  is_honor_broker_priority_on: boolean
+  replica_id: string | null
+  group_database_id: string | null
+  resource_pool_id: number | null
+  default_language_lcid: number | null
+  default_language_name: string | null
+  default_fulltext_language_lcid: number | null
+  default_fulltext_language_name: string | null
+  is_nested_triggers_on: boolean | null
+  is_transform_noise_words_on: boolean | null
+  two_digit_year_cutoff: number | null
+  containment: number | null
+  containment_desc: string | null
+  target_recovery_time_in_seconds: number | null
+  delayed_durability: number | null
+  delayed_durability_desc: string | null
+  is_memory_optimized_elevate_to_snapshot_on: boolean | null
+  is_federation_member: boolean | null
+  is_remote_data_archive_enabled: boolean | null
+  is_mixed_page_allocation_on: boolean | null
+}

apps/desktop/src/entities/database/dialects/mysql/schema/information.ts

@@ -8,6 +8,19 @@ export interface InformationSchema {
   VIEWS: Views
   TABLE_CONSTRAINTS: TableConstraints
   KEY_COLUMN_USAGE: KeyColumnUsage
+  SCHEMATA: Schemata
+}
+
+/**
+ * @name SCHEMATA
+ * @type table
+ */
+interface Schemata {
+  CATALOG_NAME: string
+  SCHEMA_NAME: string
+  DEFAULT_CHARACTER_SET_NAME: string
+  DEFAULT_COLLATION_NAME: string
+  SQL_PATH: string | null
 }
 
 /**

apps/desktop/src/entities/database/dialects/postgres/schema/catalog.ts

@@ -4,6 +4,16 @@
  */
 export interface PgCatalog {
   pg_namespace: PgNamespace
+  pg_database: PgDatabase
+}
+
+/**
+ * @name pg_database
+ * @type table
+ */
+interface PgDatabase {
+  datname: string
+  datistemplate: boolean
 }
 
 /**

apps/desktop/src/entities/database/hooks/use-server-databases.ts

@@ -0,0 +1,66 @@
+import type { databases } from '~/drizzle'
+import { DatabaseType } from '@conar/shared/enums/database-type'
+import { SafeURL } from '@conar/shared/utils/safe-url'
+import { useState } from 'react'
+import { listDatabasesQuery } from '../sql/list-databases'
+
+export function useServerDatabases(database: typeof databases.$inferSelect) {
+  const [isExpanded, setIsExpanded] = useState(false)
+  const [databasesList, setDatabasesList] = useState<string[]>([])
+  const [isLoading, setIsLoading] = useState(false)
+
+  const toggleExpand = async () => {
+    if (isExpanded) {
+      setIsExpanded(false)
+      return
+    }
+
+    setIsExpanded(true)
+
+    if (databasesList.length === 0) {
+      setIsLoading(true)
+      const url = new SafeURL(database.connectionString)
+
+      switch (database.type) {
+        case DatabaseType.Postgres:
+          url.pathname = 'postgres'
+          break
+        case DatabaseType.MySQL:
+          url.pathname = 'mysql'
+          break
+        case DatabaseType.MSSQL:
+          url.pathname = 'master'
+          break
+        case DatabaseType.ClickHouse:
+          url.pathname = 'default'
+          break
+      }
+
+      const systemDatabase = {
+        ...database,
+        connectionString: url.toString(),
+      }
+
+      try {
+        const dbNames = await listDatabasesQuery.run(systemDatabase)
+        const currentDbName = new SafeURL(database.connectionString).pathname.replace(/^\//, '')
+        const otherDbs = dbNames.filter(name => name !== currentDbName)
+        setDatabasesList(otherDbs)
+      }
+      catch (err) {
+        console.error('Failed to list databases', err)
+      }
+      finally {
+        setIsLoading(false)
+      }
+    }
+  }
+
+  return {
+    isExpanded,
+    isLoading,
+    databasesList,
+    toggleExpand,
+    setIsExpanded,
+  }
+}

apps/desktop/src/entities/database/index.ts

@@ -1,6 +1,7 @@
 export * from './components/database-icon'
 export * from './components/react-flow-node'
 export * from './hooks/use-database-link-params'
+export * from './hooks/use-server-databases'
 export * from './lib/fetching'
 export * from './lib/last-opened-databases'
 export * from './queries/columns'
@@ -22,6 +23,7 @@ export * from './sql/set'
 export * from './sql/tables-and-schemas'
 export * from './sql/total'
 export * from './store'
+export * from './utils/clone-connection'
 export * from './utils/helpers'
 export * from './utils/sync'
 export * from './utils/table'

apps/desktop/src/entities/database/sql/list-databases.ts

@@ -0,0 +1,38 @@
+import { type } from 'arktype'
+import { createQuery } from '../query'
+
+export const listDatabasesType = type('string[]')
+
+export const listDatabasesQuery = createQuery({
+  type: listDatabasesType,
+  query: () => ({
+    postgres: db => db
+      .selectFrom('pg_catalog.pg_database')
+      .select('datname')
+      .where('datistemplate', '=', false)
+      .orderBy('datname')
+      .execute()
+      .then(rows => rows.map(r => r.datname)),
+
+    mysql: db => db
+      .selectFrom('information_schema.SCHEMATA')
+      .select('SCHEMA_NAME')
+      .orderBy('SCHEMA_NAME')
+      .execute()
+      .then(rows => rows.map(r => r.SCHEMA_NAME)),
+
+    mssql: db => db
+      .selectFrom('sys.databases')
+      .select('name')
+      .orderBy('name')
+      .execute()
+      .then(rows => rows.map(r => r.name)),
+
+    clickhouse: db => db
+      .selectFrom('system.databases')
+      .select('name')
+      .orderBy('name')
+      .execute()
+      .then(rows => rows.map(r => r.name)),
+  }),
+})

apps/desktop/src/entities/database/utils/clone-connection.ts

@@ -0,0 +1,20 @@
+import type { databases } from '~/drizzle'
+import { SafeURL } from '@conar/shared/utils/safe-url'
+import { v7 } from 'uuid'
+
+export function cloneConnectionForDatabase(
+  originalDatabase: typeof databases.$inferSelect,
+  newDatabaseName: string,
+) {
+  const url = new SafeURL(originalDatabase.connectionString)
+  url.pathname = newDatabaseName
+
+  return {
+    ...originalDatabase,
+    id: v7(),
+    name: newDatabaseName,
+    connectionString: url.toString(),
+    createdAt: new Date(),
+    updatedAt: new Date(),
+  } satisfies typeof databases.$inferSelect
+}