107 lines
3.0 KiB
Rust
107 lines
3.0 KiB
Rust
use rocket_contrib::json::Json;
|
|
use serde_json::Value;
|
|
|
|
use crate::{
|
|
api::{EmptyResult, JsonResult, JsonUpcase, Notify, UpdateType},
|
|
auth::Headers,
|
|
db::{models::*, DbConn},
|
|
};
|
|
|
|
pub fn routes() -> Vec<rocket::Route> {
|
|
routes![get_folders, get_folder, post_folders, post_folder, put_folder, delete_folder_post, delete_folder,]
|
|
}
|
|
|
|
#[get("/folders")]
|
|
fn get_folders(headers: Headers, conn: DbConn) -> Json<Value> {
|
|
let folders = Folder::find_by_user(&headers.user.uuid, &conn);
|
|
|
|
let folders_json: Vec<Value> = folders.iter().map(Folder::to_json).collect();
|
|
|
|
Json(json!({
|
|
"Data": folders_json,
|
|
"Object": "list",
|
|
"ContinuationToken": null,
|
|
}))
|
|
}
|
|
|
|
#[get("/folders/<uuid>")]
|
|
fn get_folder(uuid: String, headers: Headers, conn: DbConn) -> JsonResult {
|
|
let folder = match Folder::find_by_uuid(&uuid, &conn) {
|
|
Some(folder) => folder,
|
|
_ => err!("Invalid folder"),
|
|
};
|
|
|
|
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,
|
|
}
|
|
|
|
#[post("/folders", data = "<data>")]
|
|
fn post_folders(data: JsonUpcase<FolderData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult {
|
|
let data: FolderData = data.into_inner().data;
|
|
|
|
let mut folder = Folder::new(headers.user.uuid, data.Name);
|
|
|
|
folder.save(&conn)?;
|
|
nt.send_folder_update(UpdateType::FolderCreate, &folder);
|
|
|
|
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)
|
|
}
|
|
|
|
#[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;
|
|
|
|
let mut folder = match Folder::find_by_uuid(&uuid, &conn) {
|
|
Some(folder) => folder,
|
|
_ => err!("Invalid folder"),
|
|
};
|
|
|
|
if folder.user_uuid != headers.user.uuid {
|
|
err!("Folder belongs to another user")
|
|
}
|
|
|
|
folder.name = data.Name;
|
|
|
|
folder.save(&conn)?;
|
|
nt.send_folder_update(UpdateType::FolderUpdate, &folder);
|
|
|
|
Ok(Json(folder.to_json()))
|
|
}
|
|
|
|
#[post("/folders/<uuid>/delete")]
|
|
fn delete_folder_post(uuid: String, headers: Headers, conn: DbConn, nt: Notify) -> EmptyResult {
|
|
delete_folder(uuid, headers, conn, nt)
|
|
}
|
|
|
|
#[delete("/folders/<uuid>")]
|
|
fn delete_folder(uuid: String, headers: Headers, conn: DbConn, nt: Notify) -> EmptyResult {
|
|
let folder = match Folder::find_by_uuid(&uuid, &conn) {
|
|
Some(folder) => folder,
|
|
_ => err!("Invalid folder"),
|
|
};
|
|
|
|
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(())
|
|
}
|