refactor(frontend): centralize database feature capabilities

src/lib/databaseFeatureSupport.ts

@@ -0,0 +1,81 @@
+import type { DatabaseType } from "@/types/database";
+import {
+  AGENT_DRIVER_TYPES,
+  CREATE_DATABASE_SUPPORTED_TYPES,
+  DATABASE_SEARCH_SUPPORTED_TYPES,
+  DIAGRAM_SUPPORTED_TYPES,
+  FETCH_FIRST_TYPES,
+  FIELD_LINEAGE_SUPPORTED_TYPES,
+  PG_LIKE_STRUCTURE_TYPES,
+  SCHEMA_AWARE_TYPES,
+  SINGLE_DATABASE_TYPES,
+  SQL_FILE_UNSUPPORTED_TYPES,
+  TABLE_IMPORT_SUPPORTED_TYPES,
+  TABLE_STRUCTURE_SUPPORTED_TYPES,
+  TRANSFER_SQL_TYPES,
+  TREE_SCHEMA_TYPES,
+} from "./databaseCapabilitySets";
+
+export function isSchemaAware(dbType?: DatabaseType): boolean {
+  return !!dbType && SCHEMA_AWARE_TYPES.has(dbType);
+}
+
+export function usesTreeSchemaMode(dbType?: DatabaseType): boolean {
+  return !!dbType && TREE_SCHEMA_TYPES.has(dbType);
+}
+
+export function isSingleDatabase(dbType?: DatabaseType): boolean {
+  return !!dbType && SINGLE_DATABASE_TYPES.has(dbType);
+}
+
+export function usesFetchFirst(dbType?: DatabaseType): boolean {
+  return !!dbType && FETCH_FIRST_TYPES.has(dbType);
+}
+
+export function supportsSqlFileExecution(dbType?: DatabaseType): boolean {
+  return !!dbType && !SQL_FILE_UNSUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsSchemaDiagram(dbType?: DatabaseType): boolean {
+  return !!dbType && DIAGRAM_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsDatabaseSearch(dbType?: DatabaseType): boolean {
+  return !!dbType && DATABASE_SEARCH_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsTableImport(dbType?: DatabaseType): boolean {
+  return !!dbType && TABLE_IMPORT_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsTableStructureEditing(dbType?: DatabaseType): boolean {
+  return !!dbType && TABLE_STRUCTURE_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsDatabaseCreation(dbType?: DatabaseType): boolean {
+  return !!dbType && CREATE_DATABASE_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsFieldLineage(dbType?: DatabaseType): boolean {
+  return !!dbType && FIELD_LINEAGE_SUPPORTED_TYPES.has(dbType);
+}
+
+export function supportsTransfer(dbType?: DatabaseType): boolean {
+  return !!dbType && TRANSFER_SQL_TYPES.has(dbType);
+}
+
+export function supportsDriverManagement(dbType?: DatabaseType): boolean {
+  return !!dbType && AGENT_DRIVER_TYPES.has(dbType);
+}
+
+export function supportsObjectBrowser(dbType?: DatabaseType): boolean {
+  return !!dbType && !["redis", "mongodb", "elasticsearch"].includes(dbType);
+}
+
+export function supportsTableTruncate(dbType?: DatabaseType): boolean {
+  return !!dbType && dbType !== "sqlite" && dbType !== "duckdb";
+}
+
+export function usesPostgresLikeStructureCopy(dbType?: DatabaseType): boolean {
+  return !!dbType && PG_LIKE_STRUCTURE_TYPES.has(dbType);
+}