bitwarden_vault/cipher/cipher_client/
move_many.rs

1use bitwarden_api_api::models::CipherBulkMoveRequestModel;
2use bitwarden_core::ApiError;
3use bitwarden_error::bitwarden_error;
4use bitwarden_state::repository::{RepositoryError, RepositoryOption};
5use thiserror::Error;
6#[cfg(feature = "wasm")]
7use wasm_bindgen::prelude::wasm_bindgen;
8
9use crate::{CipherId, CiphersClient, FolderId};
10
11#[allow(missing_docs)]
12#[bitwarden_error(flat)]
13#[derive(Debug, Error)]
14pub enum MoveCipherError {
15    #[error(transparent)]
16    Api(#[from] ApiError),
17    #[error(transparent)]
18    Repository(#[from] RepositoryError),
19}
20
21#[cfg_attr(feature = "wasm", wasm_bindgen)]
22impl CiphersClient {
23    /// Moves multiple [`Cipher`](crate::Cipher) objects to a folder, or clears their folder when
24    /// `folder_id` is `None`.
25    pub async fn move_many(
26        &self,
27        cipher_ids: Vec<CipherId>,
28        folder_id: Option<FolderId>,
29    ) -> Result<(), MoveCipherError> {
30        self.api_configurations
31            .api_client
32            .ciphers_api()
33            .move_many(Some(CipherBulkMoveRequestModel {
34                ids: cipher_ids.iter().map(|id| id.to_string()).collect(),
35                folder_id: folder_id.map(|id| id.to_string()),
36            }))
37            .await?;
38
39        let repository = self.repository.require()?;
40
41        let mut updated_ciphers = Vec::new();
42        for cipher_id in cipher_ids {
43            if let Some(mut cipher) = repository.get(cipher_id).await? {
44                cipher.folder_id = folder_id;
45                updated_ciphers.push((cipher_id, cipher));
46            }
47        }
48        repository.set_bulk(updated_ciphers).await?;
49        Ok(())
50    }
51}
52
53#[cfg(test)]
54mod tests {
55    use std::sync::Arc;
56
57    use bitwarden_api_api::apis::ApiClient;
58    use bitwarden_core::{
59        client::ApiConfigurations, key_management::create_test_crypto_with_user_key,
60    };
61    use bitwarden_crypto::{SymmetricCryptoKey, SymmetricKeyAlgorithm};
62    use bitwarden_state::repository::Repository;
63    use bitwarden_test::MemoryRepository;
64
65    use crate::{Cipher, CipherId, CiphersClient, FolderId};
66
67    const TEST_CIPHER_ID: &str = "5faa9684-c793-4a2d-8a12-b33900187097";
68    const TEST_CIPHER_ID_2: &str = "6faa9684-c793-4a2d-8a12-b33900187098";
69    const TEST_FOLDER_ID: &str = "7faa9684-c793-4a2d-8a12-b33900187099";
70
71    fn generate_test_cipher() -> Cipher {
72        Cipher {
73            id: TEST_CIPHER_ID.parse().ok(),
74            name: Some("2.pMS6/icTQABtulw52pq2lg==|XXbxKxDTh+mWiN1HjH2N1w==|Q6PkuT+KX/axrgN9ubD5Ajk2YNwxQkgs3WJM0S0wtG8=".parse().unwrap()),
75            r#type: crate::CipherType::Login,
76            notes: Default::default(),
77            organization_id: Default::default(),
78            folder_id: Default::default(),
79            favorite: Default::default(),
80            reprompt: Default::default(),
81            fields: Default::default(),
82            collection_ids: Default::default(),
83            key: Default::default(),
84            login: Default::default(),
85            identity: Default::default(),
86            card: Default::default(),
87            secure_note: Default::default(),
88            ssh_key: Default::default(),
89            bank_account: Default::default(),
90            drivers_license: Default::default(),
91            passport: Default::default(),
92            organization_use_totp: Default::default(),
93            edit: Default::default(),
94            permissions: Default::default(),
95            view_password: Default::default(),
96            local_data: Default::default(),
97            attachments: Default::default(),
98            password_history: Default::default(),
99            creation_date: Default::default(),
100            deleted_date: Default::default(),
101            revision_date: Default::default(),
102            archived_date: Default::default(),
103            data: Default::default(),
104        }
105    }
106
107    fn create_test_client(api_client: ApiClient) -> (CiphersClient, Arc<MemoryRepository<Cipher>>) {
108        let repository = Arc::new(MemoryRepository::<Cipher>::default());
109        #[allow(deprecated)]
110        let client = CiphersClient {
111            key_store: create_test_crypto_with_user_key(SymmetricCryptoKey::make(
112                SymmetricKeyAlgorithm::Aes256CbcHmac,
113            )),
114            api_configurations: Arc::new(ApiConfigurations::from_api_client(api_client)),
115            repository: Some(repository.clone() as Arc<dyn Repository<Cipher>>),
116            client: bitwarden_core::Client::new_test(None),
117        };
118        (client, repository)
119    }
120
121    #[tokio::test]
122    async fn test_move_many_updates_folder_id() {
123        let api_client = ApiClient::new_mocked(|mock| {
124            mock.ciphers_api.expect_move_many().returning(|_| Ok(()));
125        });
126
127        let (client, repository) = create_test_client(api_client);
128
129        let cipher_id: CipherId = TEST_CIPHER_ID.parse().unwrap();
130        let cipher_id_2: CipherId = TEST_CIPHER_ID_2.parse().unwrap();
131        let folder_id: FolderId = TEST_FOLDER_ID.parse().unwrap();
132
133        repository
134            .set(cipher_id, generate_test_cipher())
135            .await
136            .unwrap();
137        let mut cipher_2 = generate_test_cipher();
138        cipher_2.id = Some(cipher_id_2);
139        repository.set(cipher_id_2, cipher_2).await.unwrap();
140
141        client
142            .move_many(vec![cipher_id, cipher_id_2], Some(folder_id))
143            .await
144            .unwrap();
145
146        let c1: Cipher = repository.get(cipher_id).await.unwrap().unwrap();
147        let c2: Cipher = repository.get(cipher_id_2).await.unwrap().unwrap();
148        assert_eq!(c1.folder_id, Some(folder_id));
149        assert_eq!(c2.folder_id, Some(folder_id));
150    }
151
152    #[tokio::test]
153    async fn test_move_many_clears_folder_id() {
154        let api_client = ApiClient::new_mocked(|mock| {
155            mock.ciphers_api.expect_move_many().returning(|_| Ok(()));
156        });
157
158        let (client, repository) = create_test_client(api_client);
159
160        let cipher_id: CipherId = TEST_CIPHER_ID.parse().unwrap();
161        let folder_id: FolderId = TEST_FOLDER_ID.parse().unwrap();
162
163        let mut cipher = generate_test_cipher();
164        cipher.folder_id = Some(folder_id);
165        repository.set(cipher_id, cipher).await.unwrap();
166
167        client.move_many(vec![cipher_id], None).await.unwrap();
168
169        let c: Cipher = repository.get(cipher_id).await.unwrap().unwrap();
170        assert!(c.folder_id.is_none());
171    }
172
173    #[tokio::test]
174    async fn test_move_many_skips_missing_ciphers() {
175        let api_client = ApiClient::new_mocked(|mock| {
176            mock.ciphers_api.expect_move_many().returning(|_| Ok(()));
177        });
178
179        let (client, _repository) = create_test_client(api_client);
180        let cipher_id: CipherId = TEST_CIPHER_ID.parse().unwrap();
181        let folder_id: FolderId = TEST_FOLDER_ID.parse().unwrap();
182
183        let result = client.move_many(vec![cipher_id], Some(folder_id)).await;
184        assert!(result.is_ok());
185    }
186}
bitwarden_vault/cipher/cipher_client/move_many.rs

bitwarden_vault/cipher/cipher_client/
move_many.rs
