138 lines
3.0 KiB
Rust
138 lines
3.0 KiB
Rust
use actix_web::{delete, get, post, web, HttpResponse, Responder};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::time::SystemTime;
|
|
|
|
use crate::note::{generate_id, Note, NoteInfo, NotePublic};
|
|
use crate::size::LIMIT;
|
|
use crate::store;
|
|
|
|
pub fn now() -> u32 {
|
|
SystemTime::now()
|
|
.duration_since(SystemTime::UNIX_EPOCH)
|
|
.unwrap()
|
|
.as_secs() as u32
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct NotePath {
|
|
id: String,
|
|
}
|
|
|
|
#[get("/{id}")]
|
|
async fn one(path: web::Path<NotePath>) -> impl Responder {
|
|
let p = path.into_inner();
|
|
let note = store::get(&p.id);
|
|
match note {
|
|
None => return HttpResponse::NotFound().finish(),
|
|
Some(_) => return HttpResponse::Ok().json(NoteInfo {}),
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct CreateResponse {
|
|
id: String,
|
|
}
|
|
|
|
#[post("/")]
|
|
async fn create(note: web::Json<Note>) -> impl Responder {
|
|
let mut n = note.into_inner();
|
|
let id = generate_id();
|
|
let bad_req = HttpResponse::BadRequest().finish();
|
|
if n.views == None && n.expiration == None {
|
|
return bad_req;
|
|
}
|
|
match n.views {
|
|
Some(v) => {
|
|
if v > 100 {
|
|
return bad_req;
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
match n.expiration {
|
|
Some(e) => {
|
|
if e > 360 {
|
|
return bad_req;
|
|
}
|
|
let expiration = now() + (e * 60);
|
|
n.expiration = Some(expiration);
|
|
}
|
|
_ => {}
|
|
}
|
|
store::set(&id.clone(), &n.clone());
|
|
return HttpResponse::Ok().json(CreateResponse { id: id });
|
|
}
|
|
|
|
#[delete("/{id}")]
|
|
async fn delete(path: web::Path<NotePath>) -> impl Responder {
|
|
let p = path.into_inner();
|
|
let note = store::get(&p.id);
|
|
match note {
|
|
None => return HttpResponse::NotFound().finish(),
|
|
Some(note) => {
|
|
let mut changed = note.clone();
|
|
if changed.views == None && changed.expiration == None {
|
|
return HttpResponse::BadRequest().finish();
|
|
}
|
|
match changed.views {
|
|
Some(v) => {
|
|
changed.views = Some(v - 1);
|
|
let id = p.id.clone();
|
|
if v <= 1 {
|
|
store::del(&id);
|
|
} else {
|
|
store::set(&id, &changed.clone());
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
let n = now();
|
|
match changed.expiration {
|
|
Some(e) => {
|
|
if e < n {
|
|
store::del(&p.id.clone());
|
|
return HttpResponse::BadRequest().finish();
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
return HttpResponse::Ok().json(NotePublic {
|
|
contents: changed.contents,
|
|
meta: changed.meta,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct Status {
|
|
version: String,
|
|
max_size: usize,
|
|
}
|
|
|
|
#[get("/status")]
|
|
async fn status() -> impl Responder {
|
|
println!("Limit: {}", *LIMIT);
|
|
return HttpResponse::Ok().json(Status {
|
|
version: option_env!("CARGO_PKG_VERSION")
|
|
.unwrap_or("Unknown")
|
|
.to_string(),
|
|
max_size: *LIMIT,
|
|
});
|
|
}
|
|
|
|
pub fn init(cfg: &mut web::ServiceConfig) {
|
|
cfg.service(
|
|
web::scope("/api")
|
|
.service(
|
|
web::scope("/notes")
|
|
.service(one)
|
|
.service(create)
|
|
.service(delete)
|
|
.service(status),
|
|
)
|
|
.service(status),
|
|
);
|
|
}
|