Skip to content

Remove duplicated Cipher and Folder client #225

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions crates/bitwarden-uniffi/src/vault/ciphers.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use bitwarden_vault::{Cipher, CipherListView, CipherView, Fido2CredentialView};
use bitwarden_vault::{Cipher, CipherListView, CipherView, Fido2CredentialView, OrganizationId};
use uuid::Uuid;

use crate::{error::Error, Result};
Expand Down Expand Up @@ -41,7 +41,7 @@
) -> Result<CipherView> {
Ok(self
.0
.move_to_organization(cipher, organization_id)
.move_to_organization(cipher, OrganizationId(organization_id))

Check warning on line 44 in crates/bitwarden-uniffi/src/vault/ciphers.rs

View check run for this annotation

Codecov / codecov/patch

crates/bitwarden-uniffi/src/vault/ciphers.rs#L44

Added line #L44 was not covered by tests
.map_err(Error::Cipher)?)
}
}
22 changes: 17 additions & 5 deletions crates/bitwarden-vault/src/cipher/cipher_client.rs
Original file line number Diff line number Diff line change
@@ -1,15 +1,27 @@
use bitwarden_core::Client;
use bitwarden_crypto::IdentifyKey;
use uuid::Uuid;
#[cfg(feature = "wasm")]
use wasm_bindgen::prelude::*;

use crate::{
Cipher, CipherError, CipherListView, CipherView, DecryptError, EncryptError, VaultClient,
};

/// NewType representing an Organization ID. This ensures type safety.
#[cfg_attr(
feature = "wasm",
derive(serde::Serialize, serde::Deserialize, tsify_next::Tsify),
tsify(into_wasm_abi, from_wasm_abi)
)]
#[repr(transparent)]
pub struct OrganizationId(pub uuid::Uuid);

#[cfg_attr(feature = "wasm", wasm_bindgen)]
pub struct CiphersClient {
pub(crate) client: Client,
}

#[cfg_attr(feature = "wasm", wasm_bindgen)]
impl CiphersClient {
pub fn encrypt(&self, mut cipher_view: CipherView) -> Result<Cipher, EncryptError> {
let key_store = self.client.internal.get_key_store();
Expand Down Expand Up @@ -55,10 +67,10 @@ impl CiphersClient {
pub fn move_to_organization(
&self,
mut cipher_view: CipherView,
organization_id: Uuid,
organization_id: OrganizationId,
) -> Result<CipherView, CipherError> {
let key_store = self.client.internal.get_key_store();
cipher_view.move_to_organization(&mut key_store.context(), organization_id)?;
cipher_view.move_to_organization(&mut key_store.context(), organization_id.0)?;
Ok(cipher_view)
}

Expand Down Expand Up @@ -206,7 +218,7 @@ mod tests {
// Move cipher to organization
let res = client.vault().ciphers().move_to_organization(
view,
"1bc9ac1e-f5aa-45f2-94bf-b181009709b8".parse().unwrap(),
OrganizationId("1bc9ac1e-f5aa-45f2-94bf-b181009709b8".parse().unwrap()),
);

assert!(res.is_err());
Expand Down Expand Up @@ -301,7 +313,7 @@ mod tests {
.ciphers()
.move_to_organization(
view,
"1bc9ac1e-f5aa-45f2-94bf-b181009709b8".parse().unwrap(),
OrganizationId("1bc9ac1e-f5aa-45f2-94bf-b181009709b8".parse().unwrap()),
)
.unwrap();
let new_cipher = client.vault().ciphers().encrypt(new_view).unwrap();
Expand Down
2 changes: 1 addition & 1 deletion crates/bitwarden-vault/src/cipher/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ pub use cipher::{
Cipher, CipherError, CipherListView, CipherListViewType, CipherRepromptType, CipherType,
CipherView,
};
pub use cipher_client::CiphersClient;
pub use cipher_client::{CiphersClient, OrganizationId};
pub use field::FieldView;
pub use identity::IdentityView;
pub use login::{
Expand Down
4 changes: 4 additions & 0 deletions crates/bitwarden-vault/src/folder_client.rs
Original file line number Diff line number Diff line change
@@ -1,14 +1,18 @@
use bitwarden_core::Client;
#[cfg(feature = "wasm")]
use wasm_bindgen::prelude::*;

use crate::{
error::{DecryptError, EncryptError},
Folder, FolderView, VaultClient,
};

#[cfg_attr(feature = "wasm", wasm_bindgen)]
pub struct FoldersClient {
pub(crate) client: Client,
}

#[cfg_attr(feature = "wasm", wasm_bindgen)]
impl FoldersClient {
pub fn encrypt(&self, folder_view: FolderView) -> Result<Folder, EncryptError> {
let key_store = self.client.internal.get_key_store();
Expand Down
2 changes: 1 addition & 1 deletion crates/bitwarden-wasm-internal/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -14,4 +14,4 @@ pub use client::BitwardenClient;
pub use crypto::CryptoClient;
pub use generators::GeneratorClient;
pub use init::init_sdk;
pub use vault::{folders::FoldersClient, VaultClient};
pub use vault::VaultClient;
86 changes: 0 additions & 86 deletions crates/bitwarden-wasm-internal/src/vault/ciphers.rs

This file was deleted.

19 changes: 0 additions & 19 deletions crates/bitwarden-wasm-internal/src/vault/folders.rs

This file was deleted.

10 changes: 3 additions & 7 deletions crates/bitwarden-wasm-internal/src/vault/mod.rs
Original file line number Diff line number Diff line change
@@ -1,15 +1,11 @@
pub mod attachments;
pub mod ciphers;
pub mod folders;
pub mod totp;

use attachments::AttachmentsClient;
use ciphers::CiphersClient;
use bitwarden_vault::{CiphersClient, FoldersClient};
use totp::TotpClient;
use wasm_bindgen::prelude::*;

use crate::FoldersClient;

#[wasm_bindgen]
pub struct VaultClient(bitwarden_vault::VaultClient);

Expand All @@ -26,11 +22,11 @@
}

pub fn ciphers(&self) -> CiphersClient {
CiphersClient::new(self.0.ciphers())
self.0.ciphers()

Check warning on line 25 in crates/bitwarden-wasm-internal/src/vault/mod.rs

View check run for this annotation

Codecov / codecov/patch

crates/bitwarden-wasm-internal/src/vault/mod.rs#L25

Added line #L25 was not covered by tests
}

pub fn folders(&self) -> FoldersClient {
FoldersClient::new(self.0.folders())
self.0.folders()

Check warning on line 29 in crates/bitwarden-wasm-internal/src/vault/mod.rs

View check run for this annotation

Codecov / codecov/patch

crates/bitwarden-wasm-internal/src/vault/mod.rs#L29

Added line #L29 was not covered by tests
}

pub fn totp(&self) -> TotpClient {
Expand Down