61 lines
1.7 KiB
Rust
61 lines
1.7 KiB
Rust
use crate::model::{
|
|
role::Role,
|
|
user::{SchnupperBetreuerUser, User, UserWithRolesAndNotificationCount},
|
|
};
|
|
use futures::future::join_all;
|
|
use rocket::{
|
|
get,
|
|
http::Status,
|
|
request::{FlashMessage, FromRequest, Outcome},
|
|
routes, Request, Route, State,
|
|
};
|
|
use rocket_dyn_templates::{tera::Context, Template};
|
|
use sqlx::SqlitePool;
|
|
|
|
// Custom request guard to extract the Referer header
|
|
struct Referer(String);
|
|
|
|
#[rocket::async_trait]
|
|
impl<'r> FromRequest<'r> for Referer {
|
|
type Error = ();
|
|
|
|
async fn from_request(request: &'r Request<'_>) -> Outcome<Self, Self::Error> {
|
|
match request.headers().get_one("Referer") {
|
|
Some(referer) => Outcome::Success(Referer(referer.to_string())),
|
|
None => Outcome::Error((Status::BadRequest, ())),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[get("/schnupper")]
|
|
async fn index(
|
|
db: &State<SqlitePool>,
|
|
user: SchnupperBetreuerUser,
|
|
flash: Option<FlashMessage<'_>>,
|
|
) -> Template {
|
|
let schnupperant = Role::find_by_name(db, "schnupperant").await.unwrap();
|
|
|
|
let user_futures: Vec<_> = User::all_with_role(db, &schnupperant)
|
|
.await
|
|
.into_iter()
|
|
.map(|u| async move { UserWithRolesAndNotificationCount::from_user(u, db).await })
|
|
.collect();
|
|
let users: Vec<UserWithRolesAndNotificationCount> = join_all(user_futures).await;
|
|
|
|
let mut context = Context::new();
|
|
if let Some(msg) = flash {
|
|
context.insert("flash", &msg.into_inner());
|
|
}
|
|
context.insert("schnupperanten", &users);
|
|
context.insert(
|
|
"loggedin_user",
|
|
&UserWithRolesAndNotificationCount::from_user(user.into(), db).await,
|
|
);
|
|
|
|
Template::render("admin/schnupper/index", context.into_json())
|
|
}
|
|
|
|
pub fn routes() -> Vec<Route> {
|
|
routes![index]
|
|
}
|