opt: move api to plugins

src/plugin/api.rs

@@ -0,0 +1,84 @@
+use std::ffi::{c_char};
+
+use crate::{
+    flutter::{FlutterHandler, SESSIONS},
+    plugins::PLUGIN_REGISTRAR,
+    ui_session_interface::Session,
+};
+
+// API provided by RustDesk.
+pub type AddSessionFunc = fn(session_id: String) -> 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.
+#[derive(Clone)]
+pub enum SessionHook {
+    OnSessionRgba(fn(String, Vec<i8>) -> Vec<i8>),
+}
+
+// #[repr(C)]
+pub struct RustDeskApiTable {
+    pub add_session: AddSessionFunc,
+    pub remove_session: RemoveSessionFunc,
+    pub add_session_hook: AddSessionHookFunc,
+    pub remove_session_hook: RemoveSessionHookFunc,
+}
+
+fn load_plugin(path: *const c_char) -> i32 {
+    PLUGIN_REGISTRAR.load_plugin(path)
+}
+
+fn unload_plugin(path: *const c_char) -> i32 {
+    PLUGIN_REGISTRAR.unload_plugin(path)
+}
+
+fn add_session(session_id: String) -> bool {
+    let mut sessions = SESSIONS.write().unwrap();
+    // Create a dummy session in SESSIONS.
+    let mut session: Session<FlutterHandler> = Session::default();
+    session.id = session_id;
+    if sessions.contains_key(&session.id) {
+        return false;
+    }
+    let _ = sessions.insert(session.id.to_owned(), session);
+    // true
+    false
+}
+
+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 {
+            add_session,
+            remove_session,
+            add_session_hook,
+            remove_session_hook,
+        }
+    }
+}

src/plugin/mod.rs

@@ -8,6 +8,7 @@ mod errno;
 pub mod ipc;
 mod plog;
 mod plugins;
+pub mod api;
 
 pub use plugins::{
     handle_client_event, handle_listen_event, handle_server_event, handle_ui_event, load_plugin,