comunicapiv3/src/controllers/user_ws_routes.rs

//! # WebSocket routes
//!
//! @author Pierre Hubert

use crate::controllers::{likes_controller, user_ws_actions};
use crate::data::error::Res;
use crate::data::user_ws_request_handler::UserWsRequestHandler;

pub type WsRequestProcess = Box<dyn Fn(&mut UserWsRequestHandler) -> Res>;

/// WebSocket route
pub struct UserWsRoute {
    pub route: String,
    pub handler: WsRequestProcess,
}

impl UserWsRoute {
    pub fn new<H>(route: &str, handler: H) -> UserWsRoute
        where H: 'static + Fn(&mut UserWsRequestHandler) -> Res {
        UserWsRoute {
            route: route.to_string(),
            handler: Box::new(handler),
        }
    }
}

/// Get the list of available WebSocket routes
pub fn get_user_ws_routes() -> Vec<UserWsRoute> {
    vec![
        // Main controller
        UserWsRoute::new("$main/set_incognito", user_ws_actions::set_incognito),
        UserWsRoute::new("$main/register_conv", user_ws_actions::register_conv),
        UserWsRoute::new("$main/unregister_conv", user_ws_actions::unregister_conv),
        UserWsRoute::new("$main/register_post", user_ws_actions::register_post),
        UserWsRoute::new("$main/unregister_post", user_ws_actions::unregister_post),

        // Likes controller
        UserWsRoute::new("likes/update", likes_controller::update)
    ]
}

/// Search for a route
pub fn find_user_ws_route(uri: &str) -> Option<UserWsRoute> {
    get_user_ws_routes().into_iter().find(|r| r.route == uri)
}