feat(milvus): add database and collection deletion

apps/desktop/src/composables/__tests__/useSidebarConnectionMutationRuntime.selection.spec.ts

@@ -40,6 +40,18 @@ function connectionGroupNode(groupId: string): TreeNode {
   };
 }
 
+function vectorDatabaseNode(database: string): TreeNode {
+  return {
+    id: `conn-1:${database}`,
+    label: database,
+    type: "vector-database",
+    connectionId: "conn-1",
+    database,
+    isExpanded: false,
+    children: [],
+  };
+}
+
 function connectionStore(selectedTreeNodeIds: string[]) {
   const lastSelectedTreeNodeId = selectedTreeNodeIds[selectedTreeNodeIds.length - 1] ?? null;
   return {
@@ -57,6 +69,7 @@ function connectionStore(selectedTreeNodeIds: string[]) {
     disconnect: vi.fn().mockResolvedValue(undefined),
     isTreeNodeChildrenLoaded: vi.fn(() => false),
     getConfig: vi.fn(() => undefined),
+    isDefaultDatabase: vi.fn(() => false),
   };
 }
 
@@ -73,6 +86,22 @@ function runtime(activeNode: TreeNode, store: ReturnType<typeof connectionStore>
   });
 }
 
+describe("sidebar default database state", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    sidebarFormTarget.value = null;
+  });
+
+  it("recognizes the configured Milvus vector database as default", () => {
+    const store = connectionStore([]);
+    store.isDefaultDatabase.mockReturnValue(true);
+    const { isNodeDefaultDatabase } = runtime(vectorDatabaseNode("analytics"), store);
+
+    expect(isNodeDefaultDatabase.value).toBe(true);
+    expect(store.isDefaultDatabase).toHaveBeenCalledWith("conn-1", "analytics");
+  });
+});
+
 describe("sidebar connection move selection", () => {
   beforeEach(() => {
     vi.clearAllMocks();

apps/desktop/src/composables/__tests__/useSidebarDatabaseSpecificMutationRuntime.mongo.spec.ts

@@ -89,6 +89,31 @@ function mongoConfig(driverProfile?: string, production = false) {
   };
 }
 
+function milvusConfig(database?: string) {
+  return {
+    id: "conn-1",
+    name: "Milvus",
+    db_type: "milvus" as const,
+    host: "localhost",
+    port: 19530,
+    username: "",
+    password: "",
+    database,
+  };
+}
+
+function milvusDatabaseNode(database: string): TreeNode {
+  return {
+    id: `conn-1:${database}`,
+    label: database,
+    type: "vector-database",
+    connectionId: "conn-1",
+    database,
+    isExpanded: false,
+    children: [],
+  };
+}
+
 function mongoDatabaseNode(): TreeNode {
   return {
     id: "conn-1:app",
@@ -209,6 +234,29 @@ describe("MongoDB sidebar mutation runtime", () => {
     cloneMongoCollectionLoading.value = false;
   });
 
+  it("protects the built-in and configured Milvus databases from deletion", () => {
+    mocks.getConfig.mockReturnValue(milvusConfig("analytics"));
+    const activeNode = shallowRef(milvusDatabaseNode("default"));
+    const feature = useSidebarDatabaseSpecificMutationRuntime({
+      activeNode,
+      connectionStore: {
+        getConfig: mocks.getConfig,
+      } as any,
+    });
+
+    expect(feature.canDropMilvusDatabase.value).toBe(false);
+
+    activeNode.value = milvusDatabaseNode("analytics");
+    expect(feature.canDropMilvusDatabase.value).toBe(false);
+
+    activeNode.value = milvusDatabaseNode("archive");
+    expect(feature.canDropMilvusDatabase.value).toBe(true);
+
+    mocks.getConfig.mockReturnValue(mongoConfig());
+    activeNode.value = milvusDatabaseNode("archive-2");
+    expect(feature.canDropMilvusDatabase.value).toBe(false);
+  });
+
   it("keeps Legacy MongoDB mutations available while limiting index actions to the Indexes group", () => {
     mocks.getConfig.mockReturnValue(mongoConfig("mongodb-legacy"));
     const activeNode = shallowRef(mongoDatabaseNode());

apps/desktop/src/composables/useSidebarConnectionMutationRuntime.ts

@@ -290,7 +290,11 @@ export function useSidebarConnectionMutationRuntime(options: SidebarConnectionMu
 
   const isPinned = computed(() => activeNode.value.pinned || connectionStore.isTreeNodePinned(activeNode.value));
   const isNodeDefaultDatabase = computed(
-    () => (activeNode.value.type === "database" || activeNode.value.type === "redis-db" || activeNode.value.type === "mongo-db") && !!activeNode.value.connectionId && !!activeNode.value.database && connectionStore.isDefaultDatabase(activeNode.value.connectionId, activeNode.value.database),
+    () =>
+      (activeNode.value.type === "database" || activeNode.value.type === "redis-db" || activeNode.value.type === "mongo-db" || activeNode.value.type === "vector-database") &&
+      !!activeNode.value.connectionId &&
+      !!activeNode.value.database &&
+      connectionStore.isDefaultDatabase(activeNode.value.connectionId, activeNode.value.database),
   );
   const isNodeDefaultSchema = computed(() => activeNode.value.type === "schema" && !!activeNode.value.connectionId && !!activeNode.value.schema && connectionStore.isDefaultSchema(activeNode.value.connectionId, activeNode.value.schema));
   const isConnected = computed(() => activeNode.value.type === "connection" && !!activeNode.value.connectionId && connectionStore.connectedIds.has(activeNode.value.connectionId));

apps/desktop/src/composables/useSidebarDatabaseSpecificMutationRuntime.ts

@@ -117,6 +117,15 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
   }
 
   const canDropMongoDatabase = computed(() => activeNode.value.type === "mongo-db" && !!activeNode.value.database && usesAnyMongoDriver(activeNode.value));
+  function canDropMilvusDatabaseNode(node: TreeNode): boolean {
+    const connectionId = node.connectionId;
+    const database = node.database;
+    if (node.type !== "vector-database" || !connectionId || !database || database === "default") return false;
+    const config = connectionStore.getConfig(connectionId);
+    return config?.db_type === "milvus" && config.database !== database;
+  }
+
+  const canDropMilvusDatabase = computed(() => canDropMilvusDatabaseNode(activeNode.value));
 
   function canMutateMongoCollectionNode(node: TreeNode): boolean {
     if (node.type !== "mongo-collection" || !node.connectionId || !node.database) return false;
@@ -132,6 +141,7 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
   }
 
   const canDropMongoCollection = computed(() => canMutateMongoCollectionNode(activeNode.value));
+  const canDropMilvusCollection = computed(() => activeNode.value.type === "vector-collection" && !!activeNode.value.connectionId && !!activeNode.value.database && connectionStore.getConfig(activeNode.value.connectionId)?.db_type === "milvus");
   const canRenameMongoCollection = computed(() => canRenameMongoCollectionNode(activeNode.value));
   const canCloneMongoCollection = computed(() => canCloneMongoCollectionNode(activeNode.value));
 
@@ -732,6 +742,10 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
     showDropMongoCollectionConfirm.value = true;
   }
 
+  function dropMilvusCollection() {
+    dropMongoCollection();
+  }
+
   function dropMongoIndex() {
     dropMongoIndexLoading.value = false;
     showDropMongoIndexConfirm.value = true;
@@ -836,8 +850,6 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
         toast(t("contextMenu.dropCollectionSuccess", { name: collectionName }), 3000);
         showDropMongoCollectionConfirm.value = false;
         await refreshMongoTreeAfterDrop(node, async () => {
-          // The final collection can remove its database; if the database
-          // remains, refresh its preserved expanded children as well.
           await connectionStore.loadMongoDatabases(connectionId);
           await connectionStore.loadMongoCollections(connectionId, database);
         });
@@ -846,6 +858,54 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
     });
   }
 
+  async function confirmDropMilvusDatabase() {
+    const node = sidebarDangerTarget.value ?? activeNode.value;
+    const connectionId = node.connectionId;
+    const database = node.database;
+    if (!connectionId || !database || !canDropMilvusDatabaseNode(node)) return;
+    await runMongoSidebarMutation({
+      connection: connectionStore.getConfig(connectionId),
+      database,
+      reviewText: `POST /v2/vectordb/databases/drop\n{"dbName":${JSON.stringify(database)}}`,
+      source: t("production.sourceSidebar"),
+      loading: dropDatabaseLoading,
+      beforeExecute: () => connectionStore.ensureConnected(connectionId),
+      execute: () => api.vectorDropDatabase(connectionId, database),
+      onSuccess: async () => {
+        toast(t("contextMenu.dropDatabaseSuccess", { name: node.label }), 3000);
+        showDropDatabaseConfirm.value = false;
+        await refreshMongoTreeAfterDrop(node, () => connectionStore.loadMilvusDatabases(connectionId));
+      },
+      onError: toastMutationError,
+    });
+  }
+
+  async function confirmDropMilvusCollection() {
+    const node = sidebarDangerTarget.value ?? activeNode.value;
+    const connectionId = node.connectionId;
+    const database = node.database;
+    if (node.type !== "vector-collection" || !connectionId || !database || connectionStore.getConfig(connectionId)?.db_type !== "milvus") return;
+    const collectionName = node.label;
+    await runMongoSidebarMutation({
+      connection: connectionStore.getConfig(connectionId),
+      database,
+      reviewText: `POST /v2/vectordb/collections/drop\n{"dbName":${JSON.stringify(database)},"collectionName":${JSON.stringify(collectionName)}}`,
+      source: t("production.sourceSidebar"),
+      loading: dropMongoCollectionLoading,
+      beforeExecute: () => connectionStore.ensureConnected(connectionId),
+      execute: () => api.vectorDropCollection(connectionId, database, collectionName),
+      onSuccess: async () => {
+        toast(t("contextMenu.dropCollectionSuccess", { name: collectionName }), 3000);
+        showDropMongoCollectionConfirm.value = false;
+        await refreshMongoTreeAfterDrop(node, async () => {
+          await connectionStore.loadMilvusDatabases(connectionId);
+          await connectionStore.loadVectorCollections(connectionId, database);
+        });
+      },
+      onError: toastMutationError,
+    });
+  }
+
   async function refreshMongoTreeAfterDrop(node: TreeNode, refresh: () => Promise<void>) {
     try {
       await refresh();
@@ -937,7 +997,9 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
 
   return {
     canDropMongoDatabase,
+    canDropMilvusDatabase,
     canDropMongoCollection,
+    canDropMilvusCollection,
     canRenameMongoCollection,
     canCloneMongoCollection,
     prepareRenameMongoCollectionDialog,
@@ -998,6 +1060,9 @@ export function useSidebarDatabaseSpecificMutationRuntime(options: SidebarDataba
     confirmFlushRedisDb,
     confirmDropMongoDatabase,
     confirmDropMongoCollection,
+    confirmDropMilvusDatabase,
+    confirmDropMilvusCollection,
+    dropMilvusCollection,
     confirmDropMongoIndex,
     confirmDropAllMongoIndexes,
   };