From 41edb3da91ab9f461622b24ea1e492d1dc17b741 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=97=E5=89=91=E5=87=9B?= <2993354@qq.com> Date: Sat, 20 Jun 2026 22:22:04 +0800 Subject: [PATCH] feat(mongo): add database and collection deletion --- .../src/components/sidebar/TreeItem.vue | 87 ++++++++++++++++++- apps/desktop/src/i18n/locales/en.ts | 4 + apps/desktop/src/i18n/locales/es.ts | 4 + apps/desktop/src/i18n/locales/it.ts | 4 + apps/desktop/src/i18n/locales/ja.ts | 4 + apps/desktop/src/i18n/locales/pt-BR.ts | 4 + apps/desktop/src/i18n/locales/zh-CN.ts | 4 + apps/desktop/src/i18n/locales/zh-TW.ts | 4 + apps/desktop/src/lib/api.ts | 3 + apps/desktop/src/lib/http.ts | 12 +++ apps/desktop/src/lib/tauri.ts | 12 +++ crates/dbx-core/src/db/mongo_driver.rs | 28 ++++++ crates/dbx-core/src/mongo_ops.rs | 35 ++++++++ crates/dbx-web/src/main.rs | 3 + crates/dbx-web/src/routes/mongo.rs | 41 +++++++++ src-tauri/src/commands/mongo_cmd.rs | 31 +++++++ src-tauri/src/lib.rs | 3 + 17 files changed, 282 insertions(+), 1 deletion(-) diff --git a/apps/desktop/src/components/sidebar/TreeItem.vue b/apps/desktop/src/components/sidebar/TreeItem.vue index 14c10c5871..9becd85522 100644 --- a/apps/desktop/src/components/sidebar/TreeItem.vue +++ b/apps/desktop/src/components/sidebar/TreeItem.vue @@ -747,6 +747,14 @@ function requestDeleteSelectedNode(): boolean { dropDatabase(); return true; } + if (canDropMongoDatabase.value) { + dropDatabase(); + return true; + } + if (canDropMongoCollection.value) { + dropMongoCollection(); + return true; + } if (canDropSchema.value) { dropSchema(); return true; @@ -1276,6 +1284,8 @@ const createDatabaseCharset = ref("utf8mb4"); const createDatabaseCollation = ref("utf8mb4_unicode_ci"); const showDropDatabaseConfirm = ref(false); const dropDatabaseLoading = ref(false); +const showDropMongoCollectionConfirm = ref(false); +const dropMongoCollectionLoading = ref(false); const showFlushRedisDbConfirm = ref(false); const showCreateSchemaDialog = ref(false); const createSchemaName = ref(""); @@ -1745,7 +1755,7 @@ const canCreateTable = computed(() => { const canCreateDatabase = computed(() => { const config = props.node.connectionId ? connectionStore.getConfig(props.node.connectionId) : undefined; - return props.node.type === "connection" && (supportsDatabaseCreation(config?.db_type) || config?.db_type === "duckdb"); + return props.node.type === "connection" && (supportsDatabaseCreation(config?.db_type) || config?.db_type === "duckdb" || (config?.db_type === "mongodb" && config.driver_profile !== "mongodb-legacy")); }); const isDuckDbConnection = computed(() => { @@ -1763,6 +1773,16 @@ const canDropDatabase = computed(() => { return props.node.type === "database" && !isSqlServerLinkedNode(props.node) && supportsDatabaseCreation(config?.db_type); }); +const canDropMongoDatabase = computed(() => { + const config = props.node.connectionId ? connectionStore.getConfig(props.node.connectionId) : undefined; + return props.node.type === "mongo-db" && !!props.node.database && config?.driver_profile !== "mongodb-legacy"; +}); + +const canDropMongoCollection = computed(() => { + const config = props.node.connectionId ? connectionStore.getConfig(props.node.connectionId) : undefined; + return props.node.type === "mongo-collection" && !!props.node.database && config?.driver_profile !== "mongodb-legacy"; +}); + const canCreateSchema = computed(() => { const config = props.node.connectionId ? connectionStore.getConfig(props.node.connectionId) : undefined; return props.node.type === "database" && usesTreeSchemaMode(effectiveDatabaseTypeForConnection(config)) && !connectionUsesDatabaseObjectTreeMode(config); @@ -1857,6 +1877,10 @@ async function confirmTruncateTable() { } async function refreshDropDatabasePreviewSql() { + if (props.node.type === "mongo-db") { + dropDatabasePreviewSql.value = `db.getSiblingDB(${JSON.stringify(props.node.label)}).dropDatabase();`; + return; + } dropDatabasePreviewSql.value = ""; dropDatabasePreviewSql.value = await buildDropDatabaseSql({ databaseType: currentDatabaseType(), @@ -1939,6 +1963,12 @@ async function confirmCreateDatabase() { try { await connectionStore.ensureConnected(node.connectionId); const config = connectionStore.getConfig(node.connectionId); + if (config?.db_type === "mongodb") { + await api.mongoCreateDatabase(node.connectionId, name); + toast(t("contextMenu.createDatabaseSuccess", { name }), 3000); + await connectionStore.loadMongoDatabases(node.connectionId); + return; + } const sql = await buildCreateDatabaseSql({ databaseType: config?.db_type, driverProfile: config?.driver_profile, @@ -1960,6 +1990,11 @@ function dropDatabase() { showDropDatabaseConfirm.value = true; } +function dropMongoCollection() { + dropMongoCollectionLoading.value = false; + showDropMongoCollectionConfirm.value = true; +} + function flushRedisDb() { showFlushRedisDbConfirm.value = true; } @@ -1988,6 +2023,13 @@ async function confirmDropDatabase() { dropDatabaseLoading.value = true; try { await connectionStore.ensureConnected(node.connectionId); + if (node.type === "mongo-db" && node.database) { + await api.mongoDropDatabase(node.connectionId, node.database); + toast(t("contextMenu.dropDatabaseSuccess", { name: node.label }), 3000); + await connectionStore.loadMongoDatabases(node.connectionId); + showDropDatabaseConfirm.value = false; + return; + } const sql = dropDatabasePreviewSql.value || (await buildDropDatabaseSql({ @@ -2005,6 +2047,23 @@ async function confirmDropDatabase() { } } +async function confirmDropMongoCollection() { + const node = props.node; + if (node.type !== "mongo-collection" || !node.connectionId || !node.database || dropMongoCollectionLoading.value) return; + dropMongoCollectionLoading.value = true; + try { + await connectionStore.ensureConnected(node.connectionId); + await api.mongoDropCollection(node.connectionId, node.database, node.label); + toast(t("contextMenu.dropCollectionSuccess", { name: node.label }), 3000); + await connectionStore.loadMongoCollections(node.connectionId, node.database); + showDropMongoCollectionConfirm.value = false; + } catch (e: any) { + toast(t("contextMenu.tableOperationFailed", { message: e?.message || String(e) }), 5000); + } finally { + dropMongoCollectionLoading.value = false; + } +} + function openCreateSchemaDialog() { createSchemaName.value = ""; showCreateSchemaDialog.value = true; @@ -3263,6 +3322,22 @@ function treeItemMenuItems(): ContextMenuItem[] { items.push({ label: "", separator: true }); items.push({ label: t("redis.flushDb"), action: flushRedisDb, icon: Eraser, variant: "destructive" as const }); } + if (canDropMongoDatabase.value) { + items.push({ label: "", separator: true }); + items.push({ label: t("contextMenu.dropDatabase"), action: dropDatabase, icon: Trash2, shortcut: shortcutDelete, variant: "destructive" as const }); + } + return items; + } + + if (node.type === "mongo-collection") { + items.push({ label: t("contextMenu.copyName"), action: copyName, icon: Copy, shortcut: shortcutCopyName.value }); + items.push({ label: "", separator: true }); + items.push({ label: t("contextMenu.viewData"), action: toggle, icon: TableProperties }); + items.push({ label: t("contextMenu.newQuery"), action: newQuery, icon: TerminalSquare }); + if (canDropMongoCollection.value) { + items.push({ label: "", separator: true }); + items.push({ label: t("contextMenu.dropCollection"), action: dropMongoCollection, icon: Trash2, shortcut: shortcutDelete, variant: "destructive" as const }); + } return items; } @@ -3779,6 +3854,16 @@ function treeItemMenuItems(): ContextMenuItem[] { @confirm="confirmDropDatabase" /> + + diff --git a/apps/desktop/src/i18n/locales/en.ts b/apps/desktop/src/i18n/locales/en.ts index 6de672b4b1..e3ef68d008 100644 --- a/apps/desktop/src/i18n/locales/en.ts +++ b/apps/desktop/src/i18n/locales/en.ts @@ -1245,6 +1245,10 @@ export default { createDuckDbFileSuccess: 'DuckDB database file "{name}" created and attached', createDuckDbFileDesktopOnly: "Creating DuckDB database files is only available in the desktop app", dropDatabaseSuccess: 'Database "{name}" dropped', + dropCollection: "Drop Collection", + confirmDropCollectionTitle: "Drop Collection", + confirmDropCollectionMessage: 'Are you sure you want to drop collection "{name}"? This will permanently delete the collection and all its documents.', + dropCollectionSuccess: 'Collection "{name}" dropped', createDatabaseNamePlaceholder: "Database name", createDatabaseCharset: "Character set", createDatabaseCharsetPlaceholder: "e.g. utf8mb4", diff --git a/apps/desktop/src/i18n/locales/es.ts b/apps/desktop/src/i18n/locales/es.ts index d598983b0f..131c2b0d5e 100644 --- a/apps/desktop/src/i18n/locales/es.ts +++ b/apps/desktop/src/i18n/locales/es.ts @@ -1070,6 +1070,10 @@ export default { createDuckDbFileSuccess: 'Archivo de base de datos DuckDB "{name}" creado y adjuntado', createDuckDbFileDesktopOnly: "Crear archivos de base de datos DuckDB solo está disponible en la app de escritorio", dropDatabaseSuccess: 'Base de datos "{name}" eliminada', + dropCollection: "Eliminar colección", + confirmDropCollectionTitle: "Eliminar colección", + confirmDropCollectionMessage: '¿Estás seguro de que deseas eliminar la colección "{name}"? Esto borrará permanentemente la colección y todos sus documentos.', + dropCollectionSuccess: 'Colección "{name}" eliminada', createDatabaseNamePlaceholder: "Nombre de la base de datos", createSchema: "Crear esquema", dropSchema: "Eliminar esquema", diff --git a/apps/desktop/src/i18n/locales/it.ts b/apps/desktop/src/i18n/locales/it.ts index 848cc03bf4..ef0c3ab079 100644 --- a/apps/desktop/src/i18n/locales/it.ts +++ b/apps/desktop/src/i18n/locales/it.ts @@ -1182,6 +1182,10 @@ export default { createDuckDbFileSuccess: 'File di database DuckDB "{name}" creato e collegato', createDuckDbFileDesktopOnly: "La creazione di file di database DuckDB è disponibile solo nell'applicazione desktop", dropDatabaseSuccess: 'Database "{name}" eliminato', + dropCollection: "Elimina Collection", + confirmDropCollectionTitle: "Elimina Collection", + confirmDropCollectionMessage: 'Sei sicuro di voler eliminare la collection "{name}"? Questa operazione eliminerà permanentemente la collection e tutti i suoi documenti.', + dropCollectionSuccess: 'Collection "{name}" eliminata', createDatabaseNamePlaceholder: "Nome database", createDatabaseCharset: "Set di caratteri", createDatabaseCharsetPlaceholder: "es. utf8mb4", diff --git a/apps/desktop/src/i18n/locales/ja.ts b/apps/desktop/src/i18n/locales/ja.ts index 34e45643ca..f729ecd8db 100644 --- a/apps/desktop/src/i18n/locales/ja.ts +++ b/apps/desktop/src/i18n/locales/ja.ts @@ -1234,6 +1234,10 @@ export default { createDuckDbFileSuccess: "DuckDBデータベースファイル「{name}」を作成し、アタッチしました", createDuckDbFileDesktopOnly: "DuckDBデータベースファイルの作成はデスクトップアプリでのみ利用可能です", dropDatabaseSuccess: "データベース「{name}」を削除しました", + dropCollection: "コレクションを削除", + confirmDropCollectionTitle: "コレクションを削除", + confirmDropCollectionMessage: "本当にコレクション「{name}」を削除しますか？コレクションとすべてのドキュメントが永久に削除されます。", + dropCollectionSuccess: "コレクション「{name}」を削除しました", createDatabaseNamePlaceholder: "データベース名", createDatabaseCharset: "文字セット", createDatabaseCharsetPlaceholder: "例: utf8mb4", diff --git a/apps/desktop/src/i18n/locales/pt-BR.ts b/apps/desktop/src/i18n/locales/pt-BR.ts index 2521f3176b..c4fd22dfc8 100644 --- a/apps/desktop/src/i18n/locales/pt-BR.ts +++ b/apps/desktop/src/i18n/locales/pt-BR.ts @@ -1182,6 +1182,10 @@ export default { createDuckDbFileSuccess: 'Arquivo de banco de dados DuckDB "{name}" criado e anexado', createDuckDbFileDesktopOnly: "A criação de arquivos de banco de dados DuckDB só está disponível no aplicativo desktop", dropDatabaseSuccess: 'Banco de dados "{name}" removido', + dropCollection: "Remover coleção", + confirmDropCollectionTitle: "Remover coleção", + confirmDropCollectionMessage: 'Tem certeza de que deseja remover a coleção "{name}"? Isso excluirá permanentemente a coleção e todos os seus documentos.', + dropCollectionSuccess: 'Coleção "{name}" removida', createDatabaseNamePlaceholder: "Nome do banco de dados", createDatabaseCharset: "Conjunto de caracteres", createDatabaseCharsetPlaceholder: "por exemplo, utf8mb4", diff --git a/apps/desktop/src/i18n/locales/zh-CN.ts b/apps/desktop/src/i18n/locales/zh-CN.ts index ff5c5229bc..72e04aac9f 100644 --- a/apps/desktop/src/i18n/locales/zh-CN.ts +++ b/apps/desktop/src/i18n/locales/zh-CN.ts @@ -1244,6 +1244,10 @@ export default { createDuckDbFileSuccess: "DuckDB 数据库文件「{name}」已创建并附加", createDuckDbFileDesktopOnly: "新建 DuckDB 数据库文件仅支持桌面端", dropDatabaseSuccess: "数据库「{name}」已删除", + dropCollection: "删除集合", + confirmDropCollectionTitle: "删除集合", + confirmDropCollectionMessage: "确定要删除集合「{name}」吗？这将永久删除该集合及其所有文档。", + dropCollectionSuccess: "集合「{name}」已删除", createDatabaseNamePlaceholder: "数据库名称", createDatabaseCharset: "字符集", createDatabaseCharsetPlaceholder: "例如 utf8mb4", diff --git a/apps/desktop/src/i18n/locales/zh-TW.ts b/apps/desktop/src/i18n/locales/zh-TW.ts index 9b278c810d..83a08a5df4 100644 --- a/apps/desktop/src/i18n/locales/zh-TW.ts +++ b/apps/desktop/src/i18n/locales/zh-TW.ts @@ -1162,6 +1162,10 @@ export default { createDuckDbFileSuccess: "DuckDB 資料庫檔案「{name}」已建立並附加", createDuckDbFileDesktopOnly: "建立 DuckDB 資料庫檔案僅支援桌面端", dropDatabaseSuccess: "資料庫「{name}」已刪除", + dropCollection: "刪除集合", + confirmDropCollectionTitle: "刪除集合", + confirmDropCollectionMessage: "確定要刪除集合「{name}」嗎？這將永久刪除該集合及其所有文件。", + dropCollectionSuccess: "集合「{name}」已刪除", createDatabaseNamePlaceholder: "資料庫名稱", createDatabaseCharset: "字元集", createDatabaseCharsetPlaceholder: "例如 utf8mb4", diff --git a/apps/desktop/src/lib/api.ts b/apps/desktop/src/lib/api.ts index c2fcf61de2..cc02bf16c9 100644 --- a/apps/desktop/src/lib/api.ts +++ b/apps/desktop/src/lib/api.ts @@ -330,6 +330,9 @@ export const mqRawRequest = forward("mqRawRequest"); // MongoDB export const mongoListDatabases = forward("mongoListDatabases"); export const mongoListCollections = forward("mongoListCollections"); +export const mongoCreateDatabase = forward("mongoCreateDatabase"); +export const mongoDropDatabase = forward("mongoDropDatabase"); +export const mongoDropCollection = forward("mongoDropCollection"); export const documentFindDocuments = forward("documentFindDocuments"); export const mongoFindDocuments = forward("mongoFindDocuments"); export const mongoAggregateDocuments = forward("mongoAggregateDocuments"); diff --git a/apps/desktop/src/lib/http.ts b/apps/desktop/src/lib/http.ts index 3b23329959..564e4d9d26 100644 --- a/apps/desktop/src/lib/http.ts +++ b/apps/desktop/src/lib/http.ts @@ -1486,6 +1486,18 @@ export async function mongoListCollections(connectionId: string, database: strin return post("/api/mongo/list-collections", { connectionId, database }); } +export async function mongoCreateDatabase(connectionId: string, database: string): Promise { + await post("/api/mongo/create-database", { connectionId, database }); +} + +export async function mongoDropDatabase(connectionId: string, database: string): Promise { + await post("/api/mongo/drop-database", { connectionId, database }); +} + +export async function mongoDropCollection(connectionId: string, database: string, collection: string): Promise { + await post("/api/mongo/drop-collection", { connectionId, database, collection }); +} + export async function elasticsearchListIndices(connectionId: string): Promise { return mongoListCollections(connectionId, "default"); } diff --git a/apps/desktop/src/lib/tauri.ts b/apps/desktop/src/lib/tauri.ts index aee843a5c4..78fe028fa8 100644 --- a/apps/desktop/src/lib/tauri.ts +++ b/apps/desktop/src/lib/tauri.ts @@ -1323,6 +1323,18 @@ export async function mongoListCollections(connectionId: string, database: strin return invoke("mongo_list_collections", { connectionId, database }); } +export async function mongoCreateDatabase(connectionId: string, database: string): Promise { + return invoke("mongo_create_database", { connectionId, database }); +} + +export async function mongoDropDatabase(connectionId: string, database: string): Promise { + return invoke("mongo_drop_database", { connectionId, database }); +} + +export async function mongoDropCollection(connectionId: string, database: string, collection: string): Promise { + return invoke("mongo_drop_collection", { connectionId, database, collection }); +} + export async function elasticsearchListIndices(connectionId: string): Promise { return mongoListCollections(connectionId, "default"); } diff --git a/crates/dbx-core/src/db/mongo_driver.rs b/crates/dbx-core/src/db/mongo_driver.rs index abb9382882..225239122c 100644 --- a/crates/dbx-core/src/db/mongo_driver.rs +++ b/crates/dbx-core/src/db/mongo_driver.rs @@ -82,6 +82,34 @@ pub async fn list_collections(client: &Client, database: &str) -> Result Result<(), String> { + let database = database.trim(); + if database.is_empty() { + return Err("Database name is required".to_string()); + } + client.database(database).create_collection("dbx_init").await.map_err(|e| e.to_string()) +} + +pub async fn drop_database(client: &Client, database: &str) -> Result<(), String> { + let database = database.trim(); + if database.is_empty() { + return Err("Database name is required".to_string()); + } + client.database(database).drop().await.map_err(|e| e.to_string()) +} + +pub async fn drop_collection(client: &Client, database: &str, collection: &str) -> Result<(), String> { + let database = database.trim(); + let collection = collection.trim(); + if database.is_empty() { + return Err("Database name is required".to_string()); + } + if collection.is_empty() { + return Err("Collection name is required".to_string()); + } + client.database(database).collection::(collection).drop().await.map_err(|e| e.to_string()) +} + pub async fn list_indexes(client: &Client, database: &str, collection: &str) -> Result, String> { let col = client.database(database).collection::(collection); let mut cursor = col.list_indexes().await.map_err(|e| e.to_string())?; diff --git a/crates/dbx-core/src/mongo_ops.rs b/crates/dbx-core/src/mongo_ops.rs index 281336fb5d..44ade7e6e1 100644 --- a/crates/dbx-core/src/mongo_ops.rs +++ b/crates/dbx-core/src/mongo_ops.rs @@ -79,6 +79,41 @@ pub async fn mongo_list_collections_core( } } +pub async fn mongo_create_database_core(state: &AppState, connection_id: &str, database: &str) -> Result<(), String> { + ensure_document_pool(state, connection_id).await?; + let connections = state.connections.read().await; + match connections.get(connection_id).ok_or("Not found")? { + PoolKind::MongoDb(client) => mongo_driver::create_database(client, database).await, + PoolKind::Agent(_) => Err("MongoDB legacy agent does not support create database".to_string()), + _ => Err("Not a MongoDB connection".to_string()), + } +} + +pub async fn mongo_drop_database_core(state: &AppState, connection_id: &str, database: &str) -> Result<(), String> { + ensure_document_pool(state, connection_id).await?; + let connections = state.connections.read().await; + match connections.get(connection_id).ok_or("Not found")? { + PoolKind::MongoDb(client) => mongo_driver::drop_database(client, database).await, + PoolKind::Agent(_) => Err("MongoDB legacy agent does not support drop database".to_string()), + _ => Err("Not a MongoDB connection".to_string()), + } +} + +pub async fn mongo_drop_collection_core( + state: &AppState, + connection_id: &str, + database: &str, + collection: &str, +) -> Result<(), String> { + ensure_document_pool(state, connection_id).await?; + let connections = state.connections.read().await; + match connections.get(connection_id).ok_or("Not found")? { + PoolKind::MongoDb(client) => mongo_driver::drop_collection(client, database, collection).await, + PoolKind::Agent(_) => Err("MongoDB legacy agent does not support drop collection".to_string()), + _ => Err("Not a MongoDB connection".to_string()), + } +} + #[allow(clippy::too_many_arguments)] pub async fn document_find_documents_core( state: &AppState, diff --git a/crates/dbx-web/src/main.rs b/crates/dbx-web/src/main.rs index ce97a056f2..794cb10fd1 100644 --- a/crates/dbx-web/src/main.rs +++ b/crates/dbx-web/src/main.rs @@ -335,6 +335,9 @@ async fn main() { // MongoDB .route("/mongo/list-databases", post(routes::mongo::list_databases)) .route("/mongo/list-collections", post(routes::mongo::list_collections)) + .route("/mongo/create-database", post(routes::mongo::create_database)) + .route("/mongo/drop-database", post(routes::mongo::drop_database)) + .route("/mongo/drop-collection", post(routes::mongo::drop_collection)) .route("/document-store/find-documents", post(routes::mongo::document_find_documents)) .route("/mongo/find-documents", post(routes::mongo::find_documents)) .route("/mongo/aggregate-documents", post(routes::mongo::aggregate_documents)) diff --git a/crates/dbx-web/src/routes/mongo.rs b/crates/dbx-web/src/routes/mongo.rs index 0a7fede627..34d49531f3 100644 --- a/crates/dbx-web/src/routes/mongo.rs +++ b/crates/dbx-web/src/routes/mongo.rs @@ -56,6 +56,14 @@ pub struct MongoCollectionRequest { pub database: String, } +#[derive(Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct MongoCollectionNameRequest { + pub connection_id: String, + pub database: String, + pub collection: String, +} + #[derive(Deserialize)] #[serde(rename_all = "camelCase")] pub struct MongoFindRequest { @@ -157,6 +165,39 @@ pub async fn list_collections( Ok(Json(result)) } +pub async fn create_database( + State(state): State>, + Json(req): Json, +) -> Result, AppError> { + ensure_writable(&state.app, &req.connection_id, "Create database").await?; + dbx_core::mongo_ops::mongo_create_database_core(&state.app, &req.connection_id, &req.database) + .await + .map_err(AppError)?; + Ok(Json(serde_json::json!({ "ok": true }))) +} + +pub async fn drop_database( + State(state): State>, + Json(req): Json, +) -> Result, AppError> { + ensure_writable(&state.app, &req.connection_id, "Drop database").await?; + dbx_core::mongo_ops::mongo_drop_database_core(&state.app, &req.connection_id, &req.database) + .await + .map_err(AppError)?; + Ok(Json(serde_json::json!({ "ok": true }))) +} + +pub async fn drop_collection( + State(state): State>, + Json(req): Json, +) -> Result, AppError> { + ensure_writable(&state.app, &req.connection_id, "Drop collection").await?; + dbx_core::mongo_ops::mongo_drop_collection_core(&state.app, &req.connection_id, &req.database, &req.collection) + .await + .map_err(AppError)?; + Ok(Json(serde_json::json!({ "ok": true }))) +} + pub async fn find_documents( State(state): State>, Json(req): Json, diff --git a/src-tauri/src/commands/mongo_cmd.rs b/src-tauri/src/commands/mongo_cmd.rs index 89c490ba2c..d004eb61ca 100644 --- a/src-tauri/src/commands/mongo_cmd.rs +++ b/src-tauri/src/commands/mongo_cmd.rs @@ -40,6 +40,37 @@ pub async fn mongo_list_collections( dbx_core::mongo_ops::mongo_list_collections_core(&state, &connection_id, &database).await } +#[tauri::command] +pub async fn mongo_create_database( + state: State<'_, Arc>, + connection_id: String, + database: String, +) -> Result<(), String> { + ensure_connection_writable(&state, &connection_id, "Create database").await?; + dbx_core::mongo_ops::mongo_create_database_core(&state, &connection_id, &database).await +} + +#[tauri::command] +pub async fn mongo_drop_database( + state: State<'_, Arc>, + connection_id: String, + database: String, +) -> Result<(), String> { + ensure_connection_writable(&state, &connection_id, "Drop database").await?; + dbx_core::mongo_ops::mongo_drop_database_core(&state, &connection_id, &database).await +} + +#[tauri::command] +pub async fn mongo_drop_collection( + state: State<'_, Arc>, + connection_id: String, + database: String, + collection: String, +) -> Result<(), String> { + ensure_connection_writable(&state, &connection_id, "Drop collection").await?; + dbx_core::mongo_ops::mongo_drop_collection_core(&state, &connection_id, &database, &collection).await +} + #[tauri::command] #[allow(clippy::too_many_arguments)] pub async fn mongo_find_documents( diff --git a/src-tauri/src/lib.rs b/src-tauri/src/lib.rs index 46f2b2c1ac..cd3c767bb6 100644 --- a/src-tauri/src/lib.rs +++ b/src-tauri/src/lib.rs @@ -535,6 +535,9 @@ pub fn run() { commands::sqlite_backup::backup_sqlite_database, commands::mongo_cmd::mongo_list_databases, commands::mongo_cmd::mongo_list_collections, + commands::mongo_cmd::mongo_create_database, + commands::mongo_cmd::mongo_drop_database, + commands::mongo_cmd::mongo_drop_collection, commands::mongo_cmd::document_find_documents, commands::mongo_cmd::mongo_find_documents, commands::mongo_cmd::mongo_aggregate_documents,