xevion/dynamic-preauth
1
0
Fork 0
mirror of https://github.com/Xevion/dynamic-preauth.git synced 2025-12-05 23:14:53 -06:00
Code Issues Packages Projects Releases Wiki Activity

Apply Session middleware to all but /notify with separate router, improve /notify key error handling

Browse Source
This commit is contained in:
Xevion
2024-12-23 19:53:26 -06:00
parent de3dacda77
commit bc39909f3c
1 changed files with 27 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/main.rs
View File

@@ -203,7 +203,7 @@ pub async fn download(req: &mut Request, res: &mut Response, depot: &mut Depot)
}
#[handler]
pub async fn notify(req: &mut Request, res: &mut Response, depot: &mut Depot) {
pub async fn notify(req: &mut Request, res: &mut Response) {
let key = req.query::<String>("key");
match key {
@@ -219,14 +219,23 @@ pub async fn notify(req: &mut Request, res: &mut Response, depot: &mut Depot) {
};
let store = &mut *STORE.lock().await;
let session = store.sessions.get_mut(&key).expect("Session not found");
let session = store.sessions.get_mut(&key);
let message = OutgoingMessage::TokenAlert { token: key };
session
.send_message(message)
.expect("Failed to buffer token alert message");
match session {
Some(session) => {
let message = OutgoingMessage::TokenAlert { token: key };
session
.send_message(message)
.expect("Failed to buffer token alert message");
res.render("Notification sent");
res.render("Notification sent");
}
None => {
tracing::warn!("Session not found for key while attempting notify: {}", key);
res.status_code(StatusCode::UNAUTHORIZED);
return;
}
}
}
None => {
res.status_code(StatusCode::BAD_REQUEST);
@@ -337,12 +346,18 @@ async fn main() {
let router = Router::new()
.hoop(CatchPanic::new())
.hoop(cors)
.hoop(session_middleware)
.push(Router::with_path("download/<id>").get(download))
.push(Router::with_path("session").get(get_session))
.push(Router::with_path("ws").goal(connect))
// /notify does not need a session, nor should it have one
.push(Router::with_path("notify").post(notify))
.push(Router::with_path("<**path>").get(static_dir));
.push(
Router::new()
.hoop(session_middleware)
.push(Router::with_path("download/<id>").get(download))
.push(Router::with_path("session").get(get_session))
// websocket /ws
.push(Router::with_path("ws").goal(connect))
// static files
.push(Router::with_path("<**path>").get(static_dir)),
);
let service = Service::new(router).hoop(Logger::new());