bitwarden_rs/src/api/core/folders.rs

118 lines
3.1 KiB
Rust
Raw Normal View History

use rocket_contrib::json::Json;
use serde_json::Value;
2018-02-10 00:00:55 +00:00
2018-12-07 01:05:45 +00:00
use crate::db::models::*;
use crate::db::DbConn;
2018-02-10 00:00:55 +00:00
use crate::api::{EmptyResult, JsonResult, JsonUpcase, Notify, UpdateType};
2018-12-07 01:05:45 +00:00
use crate::auth::Headers;
2018-02-10 00:00:55 +00:00
use rocket::Route;
pub fn routes() -> Vec<Route> {
routes![
get_folders,
get_folder,
post_folders,
post_folder,
put_folder,
delete_folder_post,
delete_folder,
]
}
2018-02-10 00:00:55 +00:00
#[get("/folders")]
fn get_folders(headers: Headers, conn: DbConn) -> JsonResult {
2018-02-10 00:00:55 +00:00
let folders = Folder::find_by_user(&headers.user.uuid, &conn);
let folders_json: Vec<Value> = folders.iter().map(|c| c.to_json()).collect();
Ok(Json(json!({
"Data": folders_json,
"Object": "list",
"ContinuationToken": null,
2018-02-10 00:00:55 +00:00
})))
}
#[get("/folders/<uuid>")]
fn get_folder(uuid: String, headers: Headers, conn: DbConn) -> JsonResult {
2018-02-14 23:53:11 +00:00
let folder = match Folder::find_by_uuid(&uuid, &conn) {
2018-02-10 00:00:55 +00:00
Some(folder) => folder,
_ => err!("Invalid folder"),
2018-02-10 00:00:55 +00:00
};
if folder.user_uuid != headers.user.uuid {
err!("Folder belongs to another user")
}
Ok(Json(folder.to_json()))
}
#[derive(Deserialize)]
#[allow(non_snake_case)]
pub struct FolderData {
pub Name: String,
}
2018-02-10 00:00:55 +00:00
#[post("/folders", data = "<data>")]
fn post_folders(data: JsonUpcase<FolderData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult {
let data: FolderData = data.into_inner().data;
2018-02-10 00:00:55 +00:00
let mut folder = Folder::new(headers.user.uuid.clone(), data.Name);
2018-02-10 00:00:55 +00:00
folder.save(&conn)?;
nt.send_folder_update(UpdateType::FolderCreate, &folder);
2018-02-10 00:00:55 +00:00
Ok(Json(folder.to_json()))
}
#[post("/folders/<uuid>", data = "<data>")]
fn post_folder(uuid: String, data: JsonUpcase<FolderData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult {
put_folder(uuid, data, headers, conn, nt)
2018-02-10 00:00:55 +00:00
}
#[put("/folders/<uuid>", data = "<data>")]
fn put_folder(uuid: String, data: JsonUpcase<FolderData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult {
let data: FolderData = data.into_inner().data;
2018-02-10 00:00:55 +00:00
let mut folder = match Folder::find_by_uuid(&uuid, &conn) {
Some(folder) => folder,
_ => err!("Invalid folder"),
2018-02-10 00:00:55 +00:00
};
if folder.user_uuid != headers.user.uuid {
err!("Folder belongs to another user")
}
folder.name = data.Name;
2018-02-10 00:00:55 +00:00
folder.save(&conn)?;
nt.send_folder_update(UpdateType::FolderUpdate, &folder);
2018-02-10 00:00:55 +00:00
Ok(Json(folder.to_json()))
}
2018-02-17 22:38:55 +00:00
#[post("/folders/<uuid>/delete")]
fn delete_folder_post(uuid: String, headers: Headers, conn: DbConn, nt: Notify) -> EmptyResult {
delete_folder(uuid, headers, conn, nt)
2018-02-10 00:00:55 +00:00
}
#[delete("/folders/<uuid>")]
fn delete_folder(uuid: String, headers: Headers, conn: DbConn, nt: Notify) -> EmptyResult {
2018-02-10 00:00:55 +00:00
let folder = match Folder::find_by_uuid(&uuid, &conn) {
Some(folder) => folder,
_ => err!("Invalid folder"),
2018-02-10 00:00:55 +00:00
};
if folder.user_uuid != headers.user.uuid {
err!("Folder belongs to another user")
}
// Delete the actual folder entry
folder.delete(&conn)?;
nt.send_folder_update(UpdateType::FolderDelete, &folder);
Ok(())
2018-02-10 00:00:55 +00:00
}