feat: add hooks for session

src/api.rs

@@ -1,32 +1,87 @@
 use std::ffi::c_char;
 
-use crate::plugins::PLUGIN_REGISTRAR;
+use crate::{
+    flutter::{FlutterHandler, SESSIONS},
+    plugins::PLUGIN_REGISTRAR,
+    ui_session_interface::Session,
+};
 
 // API provided by RustDesk.
 pub type LoadPluginFunc = fn(*const c_char) -> i32;
 pub type UnloadPluginFunc = fn(*const c_char) -> i32;
+pub type AddSessionFunc = fn(session: Session<FlutterHandler>) -> bool;
+pub type RemoveSessionFunc = fn(session_id: &String) -> bool;
+pub type AddSessionHookFunc = fn(session_id: String, key: String, hook: SessionHook) -> bool;
+pub type RemoveSessionHookFunc = fn(session_id: String, key: &String) -> bool;
+
+/// Hooks for session.
+#[repr(usize)]
+#[derive(Clone)]
+pub enum SessionHook {
+    OnSessionRgba(fn(&Session<FlutterHandler>, Vec<i8>) -> Vec<i8>) = 1,
+}
 
 #[repr(C)]
 pub struct RustDeskApiTable {
-    pub(crate) register_plugin: LoadPluginFunc,
+    pub(crate) load_plugin: LoadPluginFunc,
     pub(crate) unload_plugin: UnloadPluginFunc,
+    pub add_session: AddSessionFunc,
+    pub remove_session: RemoveSessionFunc,
+    pub add_session_hook: AddSessionHookFunc,
+    pub remove_session_hook: RemoveSessionHookFunc,
 }
 
-#[no_mangle]
 fn load_plugin(path: *const c_char) -> i32 {
     PLUGIN_REGISTRAR.load_plugin(path)
 }
 
-#[no_mangle]
 fn unload_plugin(path: *const c_char) -> i32 {
     PLUGIN_REGISTRAR.unload_plugin(path)
 }
 
+fn add_session(session: Session<FlutterHandler>) -> bool {
+    let mut sessions = SESSIONS.write().unwrap();
+    if sessions.contains_key(&session.id) {
+        return false;
+    }
+    let _ = sessions.insert(session.id.to_owned(), session);
+    true
+}
+
+fn remove_session(session_id: &String) -> bool {
+    let mut sessions = SESSIONS.write().unwrap();
+    if !sessions.contains_key(session_id) {
+        return false;
+    }
+    let _ = sessions.remove(session_id);
+    true
+}
+
+fn add_session_hook(session_id: String, key: String, hook: SessionHook) -> bool {
+    let sessions = SESSIONS.read().unwrap();
+    if let Some(session) = sessions.get(&session_id) {
+        return session.add_session_hook(key, hook);
+    }
+    false
+}
+
+fn remove_session_hook(session_id: String, key: &String) -> bool {
+    let sessions = SESSIONS.read().unwrap();
+    if let Some(session) = sessions.get(&session_id) {
+        return session.remove_session_hook(key);
+    }
+    false
+}
+
 impl Default for RustDeskApiTable {
     fn default() -> Self {
         Self {
-            register_plugin: load_plugin,
-            unload_plugin: unload_plugin,
+            load_plugin,
+            unload_plugin,
+            add_session,
+            remove_session,
+            add_session_hook,
+            remove_session_hook,
         }
     }
 }