Added not wasm config for get_ancestors

crates/bitwarden-uniffi/src/vault/collections.rs

@@ -1,10 +1,10 @@
-use std::{collections::HashMap, sync::Arc};
+use std::sync::Arc;
 
 use bitwarden_collections::{
-    collection::{Collection, CollectionView},
+    collection::{Collection, CollectionId, CollectionView},
     tree::{NodeItem, Tree},
 };
-use uuid::Uuid;
+use bitwarden_vault::collection_client::AncestorMap;
 
 use crate::{error::Error, Result};
 
@@ -40,25 +40,34 @@ pub struct CollectionViewTree {
 }
 
 #[derive(uniffi::Object)]
-#[allow(unused)]
 pub struct CollectionViewNodeItem {
     node_item: NodeItem<CollectionView>,
 }
 
 #[uniffi::export]
 impl CollectionViewTree {
-    pub fn get_item_by_id(&self, collection_id: Uuid) -> Option<Arc<CollectionViewNodeItem>> {
+    pub fn get_item_by_id(
+        &self,
+        collection_view: CollectionView,
+    ) -> Option<Arc<CollectionViewNodeItem>> {
         self.tree
-            .get_item_by_id(collection_id)
+            .get_item_by_id(collection_view.id.unwrap_or_default().into())
             .map(|n| Arc::new(CollectionViewNodeItem { node_item: n }))
     }
 
     pub fn get_root_items(&self) -> Vec<Arc<CollectionViewNodeItem>> {
         self.tree
-            .nodes
-            .iter()
-            .filter(|n| n.parent_idx.is_none())
-            .filter_map(|n| self.get_item_by_id(n.item_id))
+            .get_root_items()
+            .into_iter()
+            .map(|n| Arc::new(CollectionViewNodeItem { node_item: n }))
+            .collect()
+    }
+
+    pub fn get_flat_items(&self) -> Vec<Arc<CollectionViewNodeItem>> {
+        self.tree
+            .get_flat_items()
+            .into_iter()
+            .map(|n| Arc::new(CollectionViewNodeItem { node_item: n }))
             .collect()
     }
 }
@@ -77,7 +86,14 @@ impl CollectionViewNodeItem {
         self.node_item.children.clone()
     }
 
-    pub fn get_ancestors(&self) -> HashMap<Uuid, String> {
-        self.node_item.ancestors.clone()
+    pub fn get_ancestors(&self) -> AncestorMap {
+        AncestorMap {
+            ancestors: self
+                .node_item
+                .ancestors
+                .iter()
+                .map(|(&uuid, name)| (CollectionId::new(uuid), name.clone()))
+                .collect(),
+        }
     }
 }

crates/bitwarden-vault/src/collection_client.rs

@@ -1,10 +1,13 @@
+use std::collections::HashMap;
+
 use bitwarden_collections::{
-    collection::{Collection, CollectionView},
+    collection::{Collection, CollectionId, CollectionView},
     tree::{NodeItem, Tree},
 };
 use bitwarden_core::Client;
+use serde::{Deserialize, Serialize};
 #[cfg(feature = "wasm")]
-use bitwarden_error::js_sys::Map;
+use tsify::Tsify;
 #[cfg(feature = "wasm")]
 use wasm_bindgen::prelude::wasm_bindgen;
 
@@ -56,6 +59,16 @@ pub struct CollectionViewNodeItem {
     node_item: NodeItem<CollectionView>,
 }
 
+#[cfg_attr(
+    feature = "wasm",
+    derive(Tsify, Serialize, Deserialize),
+    tsify(into_wasm_abi, from_wasm_abi)
+)]
+#[cfg_attr(feature = "uniffi", derive(uniffi::Record))]
+pub struct AncestorMap {
+    pub ancestors: HashMap<CollectionId, String>,
+}
+
 #[cfg_attr(feature = "wasm", wasm_bindgen)]
 impl CollectionViewNodeItem {
     pub fn get_item(&self) -> CollectionView {
@@ -70,15 +83,15 @@ impl CollectionViewNodeItem {
         self.node_item.children.clone()
     }
 
-    #[cfg(feature = "wasm")]
-    pub fn get_ancestors(&self) -> Map {
-        self.node_item
-            .ancestors
-            .iter()
-            .fold(Map::new(), |map, (id, name)| {
-                map.set(&id.to_string().into(), &name.into());
-                map
-            })
+    pub fn get_ancestors(&self) -> AncestorMap {
+        AncestorMap {
+            ancestors: self
+                .node_item
+                .ancestors
+                .iter()
+                .map(|(&uuid, name)| (CollectionId::new(uuid), name.clone()))
+                .collect(),
+        }
     }
 }