katbin/src/api/routes/paste.rs

use rocket::{Rocket, response::status, http::{Status, Cookies, Cookie}};
use rocket_contrib::json::Json;
use crate::core::paste::{entity::Paste, diesel::create_paste};
use serde_json::Value;
use crate::utils::{db, phonetic_key};
use std::ops::{DerefMut};
use rocket::response::status::Custom;
use crate::utils::phonetic_key::get_random_id;


#[post("/", data = "<paste>")]
fn create(mut paste: Json<Paste>, conn: db::DbConn, mut ck: Cookies) -> Custom<Json<Value>> {
    // Check if frontend sent a session cookie
    let session = match ck.get_private("session") {
        Some(c) => c.value().to_string(),
        None => {
            let user_id = get_random_id();
            ck.add_private(Cookie::new("session", user_id.clone()));
            user_id
        }
    };

    let new_paste = paste.deref_mut();
    if new_paste.id.is_none() {
        new_paste.id = Some(phonetic_key::get_random_id());
    }

    match create_paste(new_paste, &conn) {
        Ok(_) => {
            status::Custom(Status::Created, Json(json!({
                    "msg": "Successfully created paste",
                    "paste_id": new_paste.id
                })))
        }
        Err(e) => {
            status::Custom(Status::InternalServerError, Json(json!({
                "err": e.to_string(),
                "msg": "Failed to create paste"
            })))
        }
    }
}

pub fn fuel(rocket: Rocket) -> Rocket {
    rocket.mount("/api/paste", routes![create])
}