mirror of
https://github.com/dani-garcia/vaultwarden
synced 2024-11-10 12:04:32 +01:00
More cipher optimization/cleanup
This commit is contained in:
parent
35868dd72c
commit
f9a73a9bbe
2 changed files with 22 additions and 33 deletions
|
@ -270,26 +270,13 @@ impl Cipher {
|
|||
|
||||
/// Returns whether this cipher is owned by an org in which the user has full access.
|
||||
pub fn is_in_full_access_org(&self, user_uuid: &str, conn: &DbConn) -> bool {
|
||||
if self.organization_uuid.is_none() {
|
||||
return false;
|
||||
if let Some(ref org_uuid) = self.organization_uuid {
|
||||
if let Some(user_org) = UserOrganization::find_by_user_and_org(&user_uuid, &org_uuid, &conn) {
|
||||
return user_org.has_full_access();
|
||||
}
|
||||
}
|
||||
let org_uuid = self.organization_uuid.as_ref().unwrap();
|
||||
let rows = users_organizations::table
|
||||
.filter(users_organizations::user_uuid.eq(user_uuid))
|
||||
.filter(users_organizations::org_uuid.eq(org_uuid))
|
||||
.filter(users_organizations::status.eq(UserOrgStatus::Confirmed as i32))
|
||||
.filter(
|
||||
// The user is an org admin or higher.
|
||||
users_organizations::atype.le(UserOrgType::Admin as i32)
|
||||
// The user was granted full access to the org by an org owner/admin.
|
||||
.or(users_organizations::access_all.eq(true))
|
||||
)
|
||||
.count()
|
||||
.first(&**conn)
|
||||
.ok()
|
||||
.unwrap_or(0);
|
||||
|
||||
rows != 0
|
||||
false
|
||||
}
|
||||
|
||||
/// Returns the user's access restrictions to this cipher. A return value
|
||||
|
@ -324,8 +311,7 @@ impl Cipher {
|
|||
// and `hide_passwords` columns. This could ideally be done as part
|
||||
// of the query, but Diesel doesn't support a max() or bool_or()
|
||||
// function on booleans and this behavior isn't portable anyway.
|
||||
match query.load::<(bool, bool)>(&**conn).ok() {
|
||||
Some(vec) => {
|
||||
if let Some(vec) = query.load::<(bool, bool)>(&**conn).ok() {
|
||||
let mut read_only = false;
|
||||
let mut hide_passwords = false;
|
||||
for (ro, hp) in vec.iter() {
|
||||
|
@ -334,13 +320,11 @@ impl Cipher {
|
|||
}
|
||||
|
||||
Some((read_only, hide_passwords))
|
||||
},
|
||||
None => {
|
||||
} else {
|
||||
// This cipher isn't in any collections accessible to the user.
|
||||
None
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn is_write_accessible_to_user(&self, user_uuid: &str, conn: &DbConn) -> bool {
|
||||
match self.get_access_restrictions(&user_uuid, &conn) {
|
||||
|
|
|
@ -393,8 +393,13 @@ impl UserOrganization {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
pub fn has_status(self, status: UserOrgStatus) -> bool {
|
||||
self.status == status as i32
|
||||
}
|
||||
|
||||
pub fn has_full_access(self) -> bool {
|
||||
self.access_all || self.atype >= UserOrgType::Admin
|
||||
(self.access_all || self.atype >= UserOrgType::Admin) &&
|
||||
self.has_status(UserOrgStatus::Confirmed)
|
||||
}
|
||||
|
||||
pub fn find_by_uuid(uuid: &str, conn: &DbConn) -> Option<Self> {
|
||||
|
|
Loading…
Reference in a new issue