Remove vault folder from wasm-internal

Author: Hinton

crates/bitwarden-vault/src/cipher/attachment_client.rs

@@ -10,6 +10,10 @@ use crate::{
     Cipher, DecryptError, EncryptError, VaultClient,
 };
 
+#[cfg(feature = "wasm")]
+use wasm_bindgen::prelude::*;
+
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
 pub struct AttachmentsClient {
     pub(crate) client: Client,
 }
@@ -34,6 +38,24 @@ pub enum DecryptFileError {
     Io(#[from] std::io::Error),
 }
 
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
+impl AttachmentsClient {
+    pub fn decrypt_buffer(
+        &self,
+        cipher: Cipher,
+        attachment: AttachmentView,
+        encrypted_buffer: &[u8],
+    ) -> Result<Vec<u8>, DecryptError> {
+        let key_store = self.client.internal.get_key_store();
+
+        Ok(key_store.decrypt(&AttachmentFile {
+            cipher,
+            attachment,
+            contents: EncString::from_buffer(encrypted_buffer)?,
+        })?)
+    }
+}
+
 impl AttachmentsClient {
     pub fn encrypt_buffer(
         &self,
@@ -49,6 +71,7 @@ impl AttachmentsClient {
             contents: buffer,
         })?)
     }
+
     pub fn encrypt_file(
         &self,
         cipher: Cipher,
@@ -65,20 +88,6 @@ impl AttachmentsClient {
         Ok(attachment)
     }
 
-    pub fn decrypt_buffer(
-        &self,
-        cipher: Cipher,
-        attachment: AttachmentView,
-        encrypted_buffer: &[u8],
-    ) -> Result<Vec<u8>, DecryptError> {
-        let key_store = self.client.internal.get_key_store();
-
-        Ok(key_store.decrypt(&AttachmentFile {
-            cipher,
-            attachment,
-            contents: EncString::from_buffer(encrypted_buffer)?,
-        })?)
-    }
     pub fn decrypt_file(
         &self,
         cipher: Cipher,
@@ -93,6 +102,7 @@ impl AttachmentsClient {
     }
 }
 
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
 impl VaultClient {
     pub fn attachments(&self) -> AttachmentsClient {
         AttachmentsClient {

crates/bitwarden-vault/src/cipher/cipher_client.rs

@@ -85,6 +85,7 @@ impl CiphersClient {
     }
 }
 
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
 impl VaultClient {
     pub fn ciphers(&self) -> CiphersClient {
         CiphersClient {

crates/bitwarden-vault/src/folder_client.rs

@@ -33,6 +33,7 @@ impl FoldersClient {
     }
 }
 
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
 impl VaultClient {
     pub fn folders(&self) -> FoldersClient {
         FoldersClient {

crates/bitwarden-vault/src/lib.rs

@@ -31,5 +31,7 @@ mod vault_client;
 pub use vault_client::{VaultClient, VaultClientExt};
 
 mod sync;
-mod totp_client;
 pub use sync::{SyncRequest, SyncResponse};
+
+mod totp_client;
+pub use totp_client::TotpClient;

crates/bitwarden-vault/src/totp_client.rs

@@ -1,10 +1,42 @@
+use bitwarden_core::Client;
 use chrono::{DateTime, Utc};
 
 use crate::{
     generate_totp, generate_totp_cipher_view, CipherListView, TotpError, TotpResponse, VaultClient,
 };
 
-impl VaultClient {
+#[cfg(feature = "wasm")]
+use wasm_bindgen::prelude::*;
+
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
+pub struct TotpClient {
+    pub(crate) client: Client,
+}
+
+#[cfg(feature = "wasm")]
+#[wasm_bindgen]
+impl TotpClient {
+    /// Generates a TOTP code from a provided key
+    ///
+    /// # Arguments
+    /// - `key` - Can be:
+    ///     - A base32 encoded string
+    ///     - OTP Auth URI
+    ///     - Steam URI
+    /// - `time_ms` - Optional timestamp in milliseconds
+    #[wasm_bindgen(js_name = "generate_totp")]
+    pub fn generate_totp_wasm(
+        &self,
+        key: String,
+        time_ms: Option<f64>,
+    ) -> Result<TotpResponse, TotpError> {
+        let datetime = time_ms.and_then(|time| DateTime::<Utc>::from_timestamp_millis(time as i64));
+
+        self.generate_totp(key, datetime)
+    }
+}
+
+impl TotpClient {
     /// Generate a TOTP code from a provided key.
     ///
     /// Key can be either:
@@ -30,3 +62,12 @@ impl VaultClient {
         generate_totp_cipher_view(&mut key_store.context(), view, time)
     }
 }
+
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
+impl VaultClient {
+    pub fn totp(&self) -> TotpClient {
+        TotpClient {
+            client: self.client.clone(),
+        }
+    }
+}

crates/bitwarden-vault/src/vault_client.rs

@@ -5,7 +5,11 @@ use crate::{
     SyncRequest, SyncResponse,
 };
 
+#[cfg(feature = "wasm")]
+use wasm_bindgen::prelude::*;
+
 #[derive(Clone)]
+#[cfg_attr(feature = "wasm", wasm_bindgen)]
 pub struct VaultClient {
     pub(crate) client: Client,
 }

crates/bitwarden-wasm-internal/src/client.rs

@@ -5,10 +5,10 @@ use bitwarden_core::{Client, ClientSettings};
 use bitwarden_error::bitwarden_error;
 use bitwarden_exporters::ExporterClientExt;
 use bitwarden_generators::GeneratorClientsExt;
-use bitwarden_vault::VaultClientExt;
+use bitwarden_vault::{VaultClient, VaultClientExt};
 use wasm_bindgen::prelude::*;
 
-use crate::{CryptoClient, VaultClient};
+use crate::CryptoClient;
 
 #[wasm_bindgen]
 pub struct BitwardenClient(pub(crate) Client);
@@ -46,7 +46,7 @@ impl BitwardenClient {
     }
 
     pub fn vault(&self) -> VaultClient {
-        VaultClient::new(self.0.vault())
+        self.0.vault()
     }
 
     /// Constructs a specific client for generating passwords and passphrases

crates/bitwarden-wasm-internal/src/lib.rs

@@ -6,10 +6,8 @@ mod custom_types;
 mod init;
 mod pure_crypto;
 mod ssh;
-mod vault;
 
 pub use bitwarden_ipc::wasm::*;
 pub use client::BitwardenClient;
 pub use crypto::CryptoClient;
 pub use init::init_sdk;
-pub use vault::VaultClient;