AppFlowy/backend/src/service/ws_service/router.rs

use crate::service::ws_service::{entities::SessionId, WSClient, WSServer};
use actix::Addr;

use crate::service::user_service::LoggedUser;
use actix_web::{
    get,
    web::{Data, Path, Payload},
    Error,
    HttpRequest,
    HttpResponse,
};
use actix_web_actors::ws;

#[get("/{token}")]
pub async fn establish_ws_connection(
    request: HttpRequest,
    payload: Payload,
    token: Path<String>,
    server: Data<Addr<WSServer>>,
) -> Result<HttpResponse, Error> {
    match LoggedUser::from_token(token.clone()) {
        Ok(user) => {
            let client = WSClient::new(
                SessionId::new(user.user_id.clone()),
                server.get_ref().clone(),
            );
            let result = ws::start(client, &request, payload);
            match result {
                Ok(response) => Ok(response.into()),
                Err(e) => {
                    log::error!("ws connection error: {:?}", e);
                    Err(e)
                },
            }
        },
        Err(e) => {
            if e.is_unauthorized() {
                Ok(HttpResponse::Unauthorized().json(e))
            } else {
                Ok(HttpResponse::BadRequest().json(e))
            }
        },
    }
}