hongshaorou/rustdesk
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add callable trait

Browse Source
This commit is contained in:
Kingtous 2023-04-27 21:19:56 +08:00
parent ed64813aa0
commit 4acc3052cc
1 changed files with 6 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
src/plugin/api.rs
View File

@ -1,84 +1,14 @@
use std::ffi::{c_char}; use std::ffi::{c_char};
use lazy_static::lazy_static;
use crate::{ use crate::{
flutter::{FlutterHandler, SESSIONS}, flutter::{FlutterHandler, SESSIONS},
plugins::PLUGIN_REGISTRAR, plugins::PLUGIN_REGISTRAR,
ui_session_interface::Session, ui_session_interface::Session,
}; };
// API provided by RustDesk. pub trait Callable {
pub type AddSessionFunc = fn(session_id: String) -> bool; // Call
pub type RemoveSessionFunc = fn(session_id: &String) -> bool; fn onCall(method_name: 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,
}
}
} }