v0.12.0-preety_code

src/api/websocket.rs

@@ -1,17 +1,33 @@
 #[derive(Debug, Deserialize)]
-struct WsQuery { token: Option<String> }
-async fn websocket(State(state): State<AppState>, Query(query): Query<WsQuery>, ws: WebSocketUpgrade) -> Result<Response, AppError> {
+struct WsQuery {
+    token: Option<String>,
+}
+async fn websocket(
+    State(state): State<AppState>,
+    Query(query): Query<WsQuery>,
+    ws: WebSocketUpgrade,
+) -> Result<Response, AppError> {
     let expected = state.config.app_token.trim();
-    if !expected.is_empty() && query.token.as_deref() != Some(expected) { return Err(AppError::Unauthorized); }
+    if !expected.is_empty() && query.token.as_deref() != Some(expected) {
+        return Err(AppError::Unauthorized);
+    }
     Ok(ws.on_upgrade(move |socket| websocket_loop(state, socket)))
 }
 
 async fn websocket_loop(state: AppState, mut socket: WebSocket) {
     let initial = match build_bootstrap(&state).await {
         Ok(data) => json!({"event":"bootstrap","timestamp":Utc::now(),"data":data}),
-        Err(err) => json!({"event":"error","timestamp":Utc::now(),"data":{"message":err.to_string()}}),
+        Err(err) => {
+            json!({"event":"error","timestamp":Utc::now(),"data":{"message":err.to_string()}})
+        }
     };
-    if socket.send(Message::Text(initial.to_string())).await.is_err() { return; }
+    if socket
+        .send(Message::Text(initial.to_string()))
+        .await
+        .is_err()
+    {
+        return;
+    }
     let mut receiver = state.events.subscribe();
     loop {
         tokio::select! {
@@ -37,4 +53,3 @@ async fn websocket_loop(state: AppState, mut socket: WebSocket) {
         }
     }
 }
-