token-in-address

src/main.rs

@@ -28,10 +28,18 @@ async fn connect_handler(
     req: HttpRequest,
     state: web::Data<AppState>,
 ) -> Result<HttpResponse, actix_web::Error> {
+    /*
     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()