dbx/apps/desktop/src/lib/databaseCapabilitySets.ts

import type { DatabaseType } from "@/types/database";

export const SCHEMA_AWARE_TYPES = new Set<DatabaseType>([
  "postgres",
  "sqlserver",
  "oracle",
  "redshift",
  "dameng",
  "gaussdb",
  "kingbase",
  "highgo",
  "vastbase",
  "yashandb",
  "jdbc",
  "h2",
  "snowflake",
  "trino",
  "db2",
  "tdengine",
]);

export const SQL_FILE_UNSUPPORTED_TYPES = new Set<DatabaseType>(["redis", "mongodb", "elasticsearch"]);

export const DIAGRAM_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "sqlserver",
  "oracle",
  "redshift",
  "dameng",
  "gaussdb",
  "kingbase",
  "highgo",
  "vastbase",
  "goldendb",
  "yashandb",
  "access",
  "h2",
  "db2",
]);

export const DATABASE_SEARCH_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "sqlserver",
  "oracle",
  "redshift",
  "duckdb",
  "clickhouse",
  "dameng",
  "gaussdb",
  "kingbase",
  "highgo",
  "vastbase",
  "goldendb",
  "yashandb",
  "access",
  "h2",
  "snowflake",
  "trino",
  "hive",
  "db2",
  "informix",
  "neo4j",
  "cassandra",
  "bigquery",
  "kylin",
  "sundb",
  "tdengine",
]);

export const TABLE_IMPORT_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "duckdb",
  "clickhouse",
  "sqlserver",
  "oracle",
  "doris",
  "starrocks",
  "redshift",
  "dameng",
  "gaussdb",
  "kingbase",
  "highgo",
  "vastbase",
  "goldendb",
  "yashandb",
  "access",
]);

export const TABLE_STRUCTURE_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "duckdb",
  "sqlserver",
]);

export const CREATE_DATABASE_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlserver",
  "clickhouse",
  "oracle",
  "dameng",
  "gaussdb",
  "doris",
  "starrocks",
  "redshift",
]);

export const FIELD_LINEAGE_SUPPORTED_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "sqlserver",
  "oracle",
  "redshift",
  "dameng",
  "gaussdb",
]);

export const SINGLE_DATABASE_TYPES = new Set<DatabaseType>(["oracle", "dameng", "access"]);

export const FETCH_FIRST_TYPES = new Set<DatabaseType>(["oracle", "dameng"]);

export const TREE_SCHEMA_TYPES = new Set<DatabaseType>([
  "postgres",
  "redshift",
  "sqlserver",
  "gaussdb",
  "kingbase",
  "highgo",
  "vastbase",
  "yashandb",
  "jdbc",
  "trino",
  "h2",
  "tdengine",
]);

export const PG_LIKE_STRUCTURE_TYPES = new Set<DatabaseType>(["postgres", "redshift", "gaussdb"]);

export const AGENT_DRIVER_TYPES = new Set<DatabaseType>([
  "dameng",
  "kingbase",
  "highgo",
  "vastbase",
  "yashandb",
  "goldendb",
  "access",
  "oracle",
  "h2",
  "snowflake",
  "trino",
  "hive",
  "db2",
  "informix",
  "neo4j",
  "cassandra",
  "bigquery",
  "kylin",
  "sundb",
  "gaussdb",
  "tdengine",
]);

export const TRANSFER_SQL_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "sqlserver",
  "oracle",
  "clickhouse",
  "duckdb",
  "dameng",
  "gaussdb",
]);

export const DIAGRAM_SQL_TYPES = new Set<DatabaseType>([
  "mysql",
  "postgres",
  "sqlite",
  "sqlserver",
  "oracle",
  "redshift",
  "dameng",
  "gaussdb",
]);