fix(db2): use schema tree mode for sidebar object loading

apps/desktop/src/lib/databaseCapabilitySets.ts

@@ -183,6 +183,7 @@ export const TREE_SCHEMA_TYPES = new Set<DatabaseType>([
   "postgres",
   "redshift",
   "sqlserver",
+  "db2",
   "gaussdb",
   "kingbase",
   "highgo",

crates/dbx-core/src/lib.rs

@@ -23,6 +23,7 @@ pub mod table_import;
 pub mod transfer;
 pub mod types;
 pub mod update;
+pub mod xlsx_export;
 
 pub const R2_CDN_BASE: &str = "https://dl.dbxio.com/";
 

packages/app-tests/databaseCapabilities.test.ts

@@ -24,6 +24,11 @@ test("treats Trino catalogs as schema tree roots", () => {
   assert.equal(TREE_SCHEMA_TYPES.has("trino"), true);
 });
 
+test("treats DB2 databases as schema tree roots", () => {
+  assert.equal(TREE_SCHEMA_TYPES.has("db2"), true);
+  assert.equal(usesTreeSchemaMode("db2"), true);
+});
+
 test("treats TDengine databases as schema tree roots and agent driver databases", () => {
   assert.equal(TREE_SCHEMA_TYPES.has("tdengine"), true);
   assert.equal(SCHEMA_AWARE_TYPES.has("tdengine"), true);

src-tauri/src/commands/mod.rs

@@ -22,3 +22,4 @@ pub mod system_fonts;
 pub mod table_import;
 pub mod transfer;
 pub mod update;
+pub mod xlsx_export;

src-tauri/src/lib.rs

@@ -337,6 +337,7 @@ pub fn run() {
             commands::transfer::cancel_transfer,
             commands::database_export::export_database_sql,
             commands::database_export::cancel_database_export,
+            commands::xlsx_export::export_query_result_xlsx,
             commands::agents::list_installed_agents,
             commands::agents::list_installed_agents_local,
             commands::agents::install_agent,