Can get WebSocket access token

src/controllers/mod.rs

@@ -2,6 +2,7 @@ pub mod routes;
 pub mod server;
 
 pub mod server_controller;
+pub mod user_web_socket_controller;
 pub mod account_controller;
 pub mod user_controller;
 pub mod settings_controller;

src/controllers/routes.rs

@@ -1,6 +1,6 @@
 use std::error::Error;
 
-use crate::controllers::{account_controller, calls_controller, comments_controller, conversations_controller, friends_controller, groups_controller, likes_controller, movies_controller, notifications_controller, posts_controller, search_controller, server_controller, settings_controller, surveys_controller, user_controller, virtual_directory_controller, web_app_controller};
+use crate::controllers::{account_controller, calls_controller, comments_controller, conversations_controller, friends_controller, groups_controller, likes_controller, movies_controller, notifications_controller, posts_controller, search_controller, server_controller, settings_controller, surveys_controller, user_controller, user_web_socket_controller, virtual_directory_controller, web_app_controller};
 use crate::controllers::routes::Method::{GET, POST};
 use crate::data::http_request_handler::HttpRequestHandler;
 
@@ -128,6 +128,9 @@ pub fn get_routes() -> Vec<Route> {
         // Server meta routes
         Route::get_without_login("/", Box::new(server_controller::main_index)),
 
+        // Main user WebSocket
+        Route::post("/ws/token", Box::new(user_web_socket_controller::get_token)),
+
         // Account controller
         Route::limited_post_without_login("/account/create", Box::new(account_controller::create), LimitPolicy::SUCCESS(10)),
         Route::limited_post_without_login("/account/login", Box::new(account_controller::login_user), LimitPolicy::FAILURE(10)),

src/controllers/user_web_socket_controller.rs

@@ -0,0 +1,92 @@
+//! # User Web Socket controller
+//!
+//! Handles the WebSocket offered to the users
+
+use crate::api_data::res_get_ws_token::ResGetWsToken;
+use crate::constants::WS_ACCESS_TOKEN_LENGTH;
+use crate::data::base_request_handler::BaseRequestHandler;
+use crate::data::error::ResultBoxError;
+use crate::data::http_request_handler::HttpRequestHandler;
+use crate::utils::crypt_utils::rand_str;
+use crate::utils::date_utils::time;
+
+/// WebSocket access tokens list
+mod ws_tokens_list {
+    use std::sync::Arc;
+    use std::sync::Mutex;
+
+    use crate::constants::WS_ACCESS_TOKEN_LIFETIME;
+    use crate::data::user::UserID;
+    use crate::utils::date_utils::time;
+
+    #[derive(Debug)]
+    pub struct WsToken {
+        pub time: u64,
+        pub client_id: u32,
+        pub user_id: UserID,
+        pub token: String,
+        pub incognito: bool,
+    }
+
+    lazy_static! {
+        static ref WS_TOKENS: Arc<Mutex<Vec<WsToken>>> = {
+            Arc::new(Mutex::new(Vec::new()))
+        };
+    }
+
+    /// Get the list of WebSocket tokens
+    fn get_list() -> Arc<Mutex<Vec<WsToken>>> {
+        (*WS_TOKENS).clone()
+    }
+
+    /// Remove old entries from the list
+    pub fn clean_list() {
+        let list = get_list();
+        let mut list = list.lock().unwrap();
+
+        while let Some(first) = list.first() {
+            if first.time < time() - WS_ACCESS_TOKEN_LIFETIME {
+                list.remove(0);
+            } else {
+                break;
+            }
+        }
+    }
+
+    /// Add a new token to the list
+    pub fn add_token(t: WsToken) {
+        get_list().lock().unwrap().push(t)
+    }
+
+    /// Remove a specific access token from the list & return it
+    pub fn take_access_token(t: String) -> Option<WsToken> {
+        let list = get_list();
+        let mut list = list.lock().unwrap();
+        for i in 0..list.len() {
+            if list[i].token == t {
+                return Some(list.remove(i));
+            }
+        }
+
+        None
+    }
+}
+
+/// Get a WebSocket access token
+pub fn get_token(r: &mut HttpRequestHandler) -> ResultBoxError {
+    ws_tokens_list::clean_list();
+
+    let access_token = rand_str(WS_ACCESS_TOKEN_LENGTH);
+
+    let token = ws_tokens_list::WsToken {
+        user_id: r.user_id()?,
+        client_id: r.api_client().id,
+        time: time(),
+        token: access_token.to_string(),
+        incognito: r.post_bool_opt("incognito", false),
+    };
+
+    ws_tokens_list::add_token(token);
+
+    r.set_response(ResGetWsToken::new(access_token))
+}