vw_small

Hardened fork of Vaultwarden (https://github.com/dani-garcia/vaultwarden) with fewer features.
git clone https://git.philomathiclife.com/repos/vw_small
Log | Files | Refs | README

commit 5272b465cc9255280dff5c493a92354513ba7222
parent b75f38033b7f18157c4db493087ef72fd36e7b01
Author: Miroslav Prasil <miroslav@prasil.info>
Date:   Wed,  6 Feb 2019 13:39:32 +0000

Update revision of affected users when deleting Collection

Diffstat:
Msrc/db/models/collection.rs | 16++++++++++------
1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/db/models/collection.rs b/src/db/models/collection.rs @@ -44,12 +44,7 @@ use crate::error::MapResult; /// Database methods impl Collection { pub fn save(&mut self, conn: &DbConn) -> EmptyResult { - // Update affected users revision - UserOrganization::find_by_collection_and_org(&self.uuid, &self.org_uuid, conn) - .iter() - .for_each(|user_org| { - User::update_uuid_revision(&user_org.user_uuid, conn); - }); + self.update_users_revision(conn); diesel::replace_into(collections::table) .values(&*self) @@ -58,6 +53,7 @@ impl Collection { } pub fn delete(self, conn: &DbConn) -> EmptyResult { + self.update_users_revision(conn); CollectionCipher::delete_all_by_collection(&self.uuid, &conn)?; CollectionUser::delete_all_by_collection(&self.uuid, &conn)?; @@ -73,6 +69,14 @@ impl Collection { Ok(()) } + pub fn update_users_revision(&self, conn: &DbConn) { + UserOrganization::find_by_collection_and_org(&self.uuid, &self.org_uuid, conn) + .iter() + .for_each(|user_org| { + User::update_uuid_revision(&user_org.user_uuid, conn); + }); + } + pub fn find_by_uuid(uuid: &str, conn: &DbConn) -> Option<Self> { collections::table .filter(collections::uuid.eq(uuid))