fix(trino): support schema-aware table browsing and inserts

src/lib/databaseCapabilities.ts

@@ -12,6 +12,7 @@ export const SCHEMA_AWARE_TYPES = new Set<DatabaseType>([
   "jdbc",
   "h2",
   "snowflake",
+  "trino",
   "db2",
   "informix",
 ]);
@@ -109,7 +110,14 @@ export const SINGLE_DATABASE_TYPES = new Set<DatabaseType>(["oracle", "dameng"])
 
 export const FETCH_FIRST_TYPES = new Set<DatabaseType>(["oracle", "dameng"]);
 
-export const TREE_SCHEMA_TYPES = new Set<DatabaseType>(["postgres", "redshift", "sqlserver", "gaussdb", "jdbc"]);
+export const TREE_SCHEMA_TYPES = new Set<DatabaseType>([
+  "postgres",
+  "redshift",
+  "sqlserver",
+  "gaussdb",
+  "jdbc",
+  "trino",
+]);
 
 export const PG_LIKE_STRUCTURE_TYPES = new Set<DatabaseType>(["postgres", "redshift", "gaussdb"]);
 

src/lib/tableEditing.ts

@@ -12,18 +12,21 @@ export function editablePrimaryKeys(databaseType: DatabaseType | undefined, colu
 
 export function isTableDataEditable(databaseType: DatabaseType | undefined, primaryKeys: string[]): boolean {
   if (databaseType === "hive") return true;
+  if (databaseType === "trino") return true;
   return primaryKeys.length > 0;
 }
 
 export function supportsDataGridTransaction(databaseType: DatabaseType | undefined): boolean {
-  return databaseType !== "hive";
+  return databaseType !== "hive" && databaseType !== "trino";
 }
 
 export function canEditExistingTableRows(
   databaseType: DatabaseType | undefined,
   hiveTableTransactional?: boolean,
+  primaryKeys: string[] = [],
 ): boolean {
   if (databaseType === "hive") return hiveTableTransactional === true;
+  if (databaseType === "trino") return primaryKeys.length > 0;
   return true;
 }