mirror of
https://github.com/ViViDboarder/bitwarden_rs.git
synced 2024-11-22 13:16:39 +00:00
Removed some duplicated code in the delete cipher functions
This commit is contained in:
parent
e6b6d7f3a0
commit
96e20a66a0
@ -361,27 +361,12 @@ fn delete_attachment(uuid: String, attachment_id: String, headers: Headers, conn
|
|||||||
|
|
||||||
#[post("/ciphers/<uuid>/delete")]
|
#[post("/ciphers/<uuid>/delete")]
|
||||||
fn delete_cipher_post(uuid: String, headers: Headers, conn: DbConn) -> EmptyResult {
|
fn delete_cipher_post(uuid: String, headers: Headers, conn: DbConn) -> EmptyResult {
|
||||||
delete_cipher(uuid, headers, conn)
|
_delete_cipher_by_uuid(&uuid, &headers, &conn)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[delete("/ciphers/<uuid>")]
|
#[delete("/ciphers/<uuid>")]
|
||||||
fn delete_cipher(uuid: String, headers: Headers, conn: DbConn) -> EmptyResult {
|
fn delete_cipher(uuid: String, headers: Headers, conn: DbConn) -> EmptyResult {
|
||||||
let cipher = match Cipher::find_by_uuid(&uuid, &conn) {
|
_delete_cipher_by_uuid(&uuid, &headers, &conn)
|
||||||
Some(cipher) => cipher,
|
|
||||||
None => err!("Cipher doesn't exist")
|
|
||||||
};
|
|
||||||
|
|
||||||
if cipher.user_uuid != headers.user.uuid {
|
|
||||||
err!("Cipher is not owned by user")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Delete attachments
|
|
||||||
for a in Attachment::find_by_cipher(&cipher.uuid, &conn) { a.delete(&conn); }
|
|
||||||
|
|
||||||
// Delete cipher
|
|
||||||
cipher.delete(&conn);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[post("/ciphers/delete", data = "<data>")]
|
#[post("/ciphers/delete", data = "<data>")]
|
||||||
@ -397,20 +382,9 @@ fn delete_cipher_selected(data: Json<Value>, headers: Headers, conn: DbConn) ->
|
|||||||
};
|
};
|
||||||
|
|
||||||
for uuid in uuids {
|
for uuid in uuids {
|
||||||
let cipher = match Cipher::find_by_uuid(uuid, &conn) {
|
if let error @ Err(_) = _delete_cipher_by_uuid(uuid, &headers, &conn) {
|
||||||
Some(cipher) => cipher,
|
return error;
|
||||||
None => err!("Cipher doesn't exist")
|
|
||||||
};
|
};
|
||||||
|
|
||||||
if cipher.user_uuid != headers.user.uuid {
|
|
||||||
err!("Cipher is not owned by user")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Delete attachments
|
|
||||||
for a in Attachment::find_by_cipher(&cipher.uuid, &conn) { a.delete(&conn); }
|
|
||||||
|
|
||||||
// Delete cipher
|
|
||||||
cipher.delete(&conn);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@ -477,9 +451,7 @@ fn delete_all(data: Json<PasswordData>, headers: Headers, conn: DbConn) -> Empty
|
|||||||
|
|
||||||
// Delete ciphers and their attachments
|
// Delete ciphers and their attachments
|
||||||
for cipher in Cipher::find_by_user(&user.uuid, &conn) {
|
for cipher in Cipher::find_by_user(&user.uuid, &conn) {
|
||||||
for a in Attachment::find_by_cipher(&cipher.uuid, &conn) { a.delete(&conn); }
|
_delete_cipher(cipher, &conn);
|
||||||
|
|
||||||
cipher.delete(&conn);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete folders
|
// Delete folders
|
||||||
@ -487,3 +459,26 @@ fn delete_all(data: Json<PasswordData>, headers: Headers, conn: DbConn) -> Empty
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn _delete_cipher_by_uuid(uuid: &str, headers: &Headers, conn: &DbConn) -> EmptyResult {
|
||||||
|
let cipher = match Cipher::find_by_uuid(uuid, conn) {
|
||||||
|
Some(cipher) => cipher,
|
||||||
|
None => err!("Cipher doesn't exist"),
|
||||||
|
};
|
||||||
|
|
||||||
|
if cipher.user_uuid != headers.user.uuid {
|
||||||
|
err!("Cipher is not owned by user")
|
||||||
|
}
|
||||||
|
|
||||||
|
_delete_cipher(cipher, conn);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn _delete_cipher(cipher: Cipher, conn: &DbConn) {
|
||||||
|
// Delete the attachments
|
||||||
|
for a in Attachment::find_by_cipher(&cipher.uuid, &conn) { a.delete(&conn); }
|
||||||
|
|
||||||
|
// Delete the cipher
|
||||||
|
cipher.delete(conn);
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user