neveko/neveko-message/src/controller.rs

#![allow(non_snake_case)]

use rocket::{
    delete,
    get,
    http::Status,
    post,
    response::status::Custom,
    serde::json::Json,
};

use neveko_core::{
    auth,
    message,
    models::*,
    proof,
    reqres,
};

/// Send message
#[post("/<r_type>", data = "<m_req>")]
pub async fn send_message(
    m_req: Json<Message>,
    r_type: String,
    token: proof::PaymentProof,
) -> Custom<Json<Message>> {
    let m_type: message::MessageType = if r_type == "multisig" {
        message::MessageType::Multisig
    } else {
        message::MessageType::Normal
    };
    let res = message::create(m_req, token.get_jwp(), m_type).await;
    Custom(Status::Ok, Json(res.unwrap_or_default()))
}

/// Return all messages
#[get("/")]
pub async fn get_messages(_token: auth::BearerToken) -> Custom<Json<Vec<Message>>> {
    let messages = message::find_all();
    Custom(Status::Ok, Json(messages.unwrap_or_default()))
}

/// Delete a message by mid
#[delete("/<mid>")]
pub async fn remove_message(mid: String, _token: auth::BearerToken) -> Custom<Json<Vec<Message>>> {
    let _ = message::delete(&mid);
    let messages = message::find_all();
    Custom(Status::Ok, Json(messages.unwrap_or_default()))
}

/// decipher a message body
#[get("/<mid>")]
pub async fn decipher(
    mid: String,
    _token: auth::BearerToken,
) -> Custom<Json<reqres::DecipheredMessageBody>> {
    let d_message = message::decipher_body(mid).await;
    Custom(Status::Ok, Json(d_message.unwrap_or_default()))
}