diff --git a/src/main.rs b/src/main.rs index d5d2e57..8302759 100644 --- a/src/main.rs +++ b/src/main.rs @@ -28,10 +28,18 @@ async fn connect_handler( req: HttpRequest, state: web::Data, ) -> Result { + /* let token = match req.headers().get("Authorization") { Some(val) => val.to_str().unwrap_or("").split(" ").last().unwrap_or(""), None => return Err(ErrorUnauthorized("Unauthorized")), }; + */ + + let token = match req.match_info().get("token") { + Some(val) => val, + None => return Err(ErrorUnauthorized("Unauthorized")), + }; + let listener_id = data::get_auth_id(&token).await.map_err(|e| { eprintln!("TOKEN check failed: {}", e); ErrorUnauthorized("Unauthorized") @@ -143,7 +151,7 @@ async fn main() -> std::io::Result<()> { App::new() .wrap(Logger::default()) .app_data(web::Data::new(state.clone())) - .route("/", web::get().to(connect_handler)) + .route("/{token}", web::get().to(connect_handler)) }) .bind("0.0.0.0:8080")? .run()