bitwarden_send/
remove_password.rs

1use bitwarden_core::{ApiError, MissingFieldError, key_management::KeySlotIds, require};
2use bitwarden_crypto::{CryptoError, KeyStore};
3use bitwarden_error::bitwarden_error;
4use bitwarden_state::repository::{Repository, RepositoryError};
5use thiserror::Error;
6#[cfg(feature = "wasm")]
7use wasm_bindgen::prelude::*;
8
9use crate::{Send, SendId, SendView, error::SendParseError, send_client::SendClient};
10
11#[allow(missing_docs)]
12#[bitwarden_error(flat)]
13#[derive(Debug, Error)]
14pub enum RemoveSendPasswordError {
15    #[error(transparent)]
16    Api(#[from] ApiError),
17    #[error(transparent)]
18    Crypto(#[from] CryptoError),
19    #[error(transparent)]
20    MissingField(#[from] MissingFieldError),
21    #[error(transparent)]
22    Repository(#[from] RepositoryError),
23    #[error(transparent)]
24    SendParse(#[from] SendParseError),
25}
26
27async fn remove_send_password<R: Repository<Send> + ?Sized>(
28    key_store: &KeyStore<KeySlotIds>,
29    api_client: &bitwarden_api_api::apis::ApiClient,
30    repository: &R,
31    send_id: SendId,
32) -> Result<SendView, RemoveSendPasswordError> {
33    let resp = api_client
34        .sends_api()
35        .put_remove_password(&send_id.to_string())
36        .await?;
37
38    let send: Send = resp.try_into()?;
39
40    repository.set(require!(send.id), send.clone()).await?;
41
42    Ok(key_store.decrypt(&send)?)
43}
44
45#[cfg_attr(feature = "wasm", wasm_bindgen)]
46impl SendClient {
47    /// Remove the password from a [Send], saving the updated state to the server and local state.
48    pub async fn remove_password(
49        &self,
50        send_id: SendId,
51    ) -> Result<SendView, RemoveSendPasswordError> {
52        let key_store = self.client.internal.get_key_store();
53        let config = self.client.internal.get_api_configurations();
54        let repository = self.get_repository()?;
55
56        remove_send_password(key_store, &config.api_client, repository.as_ref(), send_id).await
57    }
58}
59
60#[cfg(test)]
61mod tests {
62    use bitwarden_api_api::{apis::ApiClient, models::SendResponseModel};
63    use bitwarden_core::key_management::{KeySlotIds, SymmetricKeySlotId};
64    use bitwarden_crypto::{KeyStore, SymmetricKeyAlgorithm};
65    use bitwarden_test::MemoryRepository;
66    use uuid::uuid;
67
68    use super::*;
69    use crate::{AuthType, Send, SendId, SendTextView, SendType, SendView};
70
71    async fn make_store_with_send(
72        send_id: uuid::Uuid,
73    ) -> (KeyStore<KeySlotIds>, MemoryRepository<Send>) {
74        let store: KeyStore<KeySlotIds> = KeyStore::default();
75        {
76            let mut ctx = store.context_mut();
77            let local_key_id = ctx.make_symmetric_key(SymmetricKeyAlgorithm::Aes256CbcHmac);
78            ctx.persist_symmetric_key(local_key_id, SymmetricKeySlotId::User)
79                .unwrap();
80        }
81
82        let repository = MemoryRepository::<Send>::default();
83        let send_view = SendView {
84            id: None,
85            access_id: None,
86            name: "Test Send".to_string(),
87            notes: None,
88            key: None,
89            new_password: None,
90            has_password: false,
91            r#type: SendType::Text,
92            file: None,
93            text: Some(SendTextView {
94                text: Some("Secret text".to_string()),
95                hidden: false,
96            }),
97            max_access_count: None,
98            access_count: 0,
99            disabled: false,
100            hide_email: false,
101            revision_date: "2025-01-01T00:00:00Z".parse().unwrap(),
102            deletion_date: "2025-01-10T00:00:00Z".parse().unwrap(),
103            expiration_date: None,
104            emails: Vec::new(),
105            auth_type: AuthType::None,
106        };
107        let mut send = store.encrypt(send_view).unwrap();
108        send.id = Some(SendId::new(send_id));
109        repository.set(SendId::new(send_id), send).await.unwrap();
110
111        (store, repository)
112    }
113
114    #[tokio::test]
115    async fn test_remove_send_password() {
116        let send_id = uuid!("25afb11c-9c95-4db5-8bac-c21cb204a3f1");
117        let (store, repository) = make_store_with_send(send_id).await;
118
119        // Get the encrypted fields from the stored send so the mock can return a realistic response
120        let stored = repository.get(SendId::new(send_id)).await.unwrap().unwrap();
121        let stored_name = stored.name.to_string();
122        let stored_key = stored.key.to_string();
123        let stored_deletion_date = stored.deletion_date.to_rfc3339();
124
125        let api_client = ApiClient::new_mocked(move |mock| {
126            let name = stored_name.clone();
127            let key = stored_key.clone();
128            let deletion_date = stored_deletion_date.clone();
129            mock.sends_api
130                .expect_put_remove_password()
131                .returning(move |_id| {
132                    Ok(SendResponseModel {
133                        id: Some(send_id),
134                        name: Some(name.clone()),
135                        revision_date: Some("2025-01-02T00:00:00Z".to_string()),
136                        object: Some("send".to_string()),
137                        access_id: None,
138                        r#type: Some(bitwarden_api_api::models::SendType::Text),
139                        auth_type: Some(bitwarden_api_api::models::AuthType::None),
140                        notes: None,
141                        file: None,
142                        text: None,
143                        data: None,
144                        key: Some(key.clone()),
145                        max_access_count: None,
146                        access_count: Some(0),
147                        password: None,
148                        emails: None,
149                        disabled: Some(false),
150                        expiration_date: None,
151                        deletion_date: Some(deletion_date.clone()),
152                        hide_email: Some(false),
153                    })
154                })
155                .once();
156        });
157
158        let result =
159            remove_send_password(&store, &api_client, &repository, SendId::new(send_id)).await;
160
161        assert!(result.is_ok());
162        let view = result.unwrap();
163        assert_eq!(view.id, Some(SendId::new(send_id)));
164        assert!(!view.has_password);
165        assert_eq!(view.auth_type, AuthType::None);
166    }
167
168    #[tokio::test]
169    async fn test_remove_send_password_http_error() {
170        let send_id = uuid!("25afb11c-9c95-4db5-8bac-c21cb204a3f1");
171        let (store, repository) = make_store_with_send(send_id).await;
172
173        let api_client = ApiClient::new_mocked(move |mock| {
174            mock.sends_api
175                .expect_put_remove_password()
176                .returning(move |_id| {
177                    Err(bitwarden_api_api::ApiError::Io(std::io::Error::other(
178                        "Simulated error",
179                    )))
180                })
181                .once();
182        });
183
184        let result =
185            remove_send_password(&store, &api_client, &repository, SendId::new(send_id)).await;
186
187        assert!(result.is_err());
188        assert!(matches!(
189            result.unwrap_err(),
190            RemoveSendPasswordError::Api(_)
191        ));
192    }
193}
bitwarden_send/remove_password.rs

bitwarden_send/
remove_password.rs
