feat: Endpoint to returns user whitelisted users

Signed-off-by: Awiteb <a@4rs.nl>
This commit is contained in:
Awiteb 2024-07-16 15:46:53 +03:00
parent d065d7ea95
commit 411f7e37f0
Signed by: awiteb
GPG key ID: 3F6B55640AA6682F

View file

@ -20,7 +20,9 @@ use oxidetalis_core::types::{PublicKey, Signature};
use salvo::{ use salvo::{
http::StatusCode, http::StatusCode,
oapi::{endpoint, extract::JsonBody}, oapi::{endpoint, extract::JsonBody},
writing::Json,
Depot, Depot,
Extractible,
Request, Request,
Router, Router,
Writer, Writer,
@ -28,10 +30,11 @@ use salvo::{
use super::{ApiError, ApiResult}; use super::{ApiError, ApiResult};
use crate::{ use crate::{
database::UserTableExt, database::{UserTableExt, WhiteListExt},
extensions::DepotExt, extensions::DepotExt,
middlewares, middlewares,
schemas::{EmptySchema, MessageSchema, RegisterUserBody}, parameters::Pagination,
schemas::{EmptySchema, MessageSchema, RegisterUserBody, WhiteListedUser},
utils, utils,
}; };
@ -79,10 +82,50 @@ pub async fn register(
Ok(EmptySchema::new(StatusCode::CREATED)) Ok(EmptySchema::new(StatusCode::CREATED))
} }
/// (🔐) Get whitelisted users
#[endpoint(
operation_id = "whitelist",
tags("User"),
responses(
(status_code = 200, description = "Returns whitelisted users", content_type = "application/json", body = Vec<WhiteListedUser>),
(status_code = 403, description = "Not registered user, must register first", content_type = "application/json", body = MessageSchema),
(status_code = 401, description = "The entered signature or public key is invalid", content_type = "application/json", body = MessageSchema),
(status_code = 429, description = "Too many requests", content_type = "application/json", body = MessageSchema),
(status_code = 500, description = "Internal server error", content_type = "application/json", body = MessageSchema),
),
parameters(
Pagination,
("X-OTMP-PUBLIC" = PublicKey, Header, description = "Public key of the sender"),
("X-OTMP-SIGNATURE" = Signature, Header, description = "Signature of the request"),
),
)]
async fn user_whitelist(
req: &mut Request,
depot: &mut Depot,
) -> ApiResult<Json<Vec<WhiteListedUser>>> {
let pagination = Pagination::extract(req).await?;
let conn = depot.db_conn();
let user = conn
.get_user_by_pubk(
&utils::extract_public_key(req)
.expect("Public key should be checked in the middleware"),
)
.await?
.ok_or(ApiError::NotRegisteredUser)?;
Ok(Json(
conn.user_whitelist(&user, pagination.page, pagination.page_size)
.await?
.into_iter()
.map(Into::into)
.collect(),
))
}
/// The route of the endpoints of this module /// The route of the endpoints of this module
pub fn route() -> Router { pub fn route() -> Router {
Router::new() Router::new()
.push(Router::with_path("register").post(register)) .push(Router::with_path("register").post(register))
.push(Router::with_path("whitelist").get(user_whitelist))
.hoop(middlewares::public_key_check) .hoop(middlewares::public_key_check)
.hoop(middlewares::signature_check) .hoop(middlewares::signature_check)
} }