maa_framework_sys/
lib.rs

1//! Raw FFI bindings to the C API (auto-generated).
2
3#![allow(non_upper_case_globals)]
4#![allow(non_camel_case_types)]
5#![allow(non_snake_case)]
6#![allow(dead_code)]
7#![allow(rustdoc::broken_intra_doc_links)]
8
9#[cfg(not(feature = "dynamic"))]
10include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
11
12#[cfg(feature = "dynamic")]
13mod dynamic {
14    #![allow(non_upper_case_globals)]
15    #![allow(non_camel_case_types)]
16    #![allow(non_snake_case)]
17    #![allow(dead_code)]
18    #![allow(rustdoc::broken_intra_doc_links)]
19    use super::CompositeLibrary;
20    include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
21}
22
23#[cfg(feature = "dynamic")]
24pub use dynamic::*;
25
26#[cfg(feature = "dynamic")]
27static INSTANCE: once_cell::sync::OnceCell<MaaFramework> = once_cell::sync::OnceCell::new();
28
29#[cfg(feature = "dynamic")]
30pub struct CompositeLibrary {
31    libs: Vec<libloading::Library>,
32}
33
34#[cfg(feature = "dynamic")]
35impl CompositeLibrary {
36    pub unsafe fn new<P: AsRef<std::ffi::OsStr>>(path: P) -> Result<Self, libloading::Error> {
37        let path = std::path::Path::new(path.as_ref());
38        let mut libs = Vec::new();
39
40        let load_lib = |p: &std::path::Path| -> Result<libloading::Library, libloading::Error> {
41            #[cfg(target_os = "windows")]
42            {
43                //LOAD_WITH_ALTERED_SEARCH_PATH (0x00000008)
44                let p = p.canonicalize().unwrap_or(p.to_path_buf());
45                let lib = unsafe { libloading::os::windows::Library::load_with_flags(&p, 0x00000008)? };
46                Ok(lib.into())
47            }
48            #[cfg(not(target_os = "windows"))]
49            {
50                let p = p.canonicalize().unwrap_or(p.to_path_buf());
51                unsafe { libloading::Library::new(&p) }
52            }
53        };
54
55        let file_name = path.file_name().unwrap_or_default().to_string_lossy();
56        let dir = path.parent().unwrap_or(std::path::Path::new(""));
57
58        let prefix = if cfg!(target_os = "windows") {
59            ""
60        } else {
61            "lib"
62        };
63        let ext = if cfg!(target_os = "windows") {
64            ".dll"
65        } else if cfg!(target_os = "macos") {
66            ".dylib"
67        } else {
68            ".so"
69        };
70
71        let try_load = |name: &str| {
72            let p = dir.join(format!("{}{}{}", prefix, name, ext));
73            load_lib(&p).ok()
74        };
75
76        let is_agent_server = file_name.contains("MaaAgentServer");
77
78        if let Some(lib) = try_load("MaaToolkit") {
79            libs.push(lib);
80        }
81
82        let main_lib = load_lib(path)?;
83        libs.insert(0, main_lib);
84
85        if !is_agent_server {
86            if let Some(lib) = try_load("MaaAgentClient") {
87                libs.push(lib);
88            }
89        }
90
91        Ok(Self { libs })
92    }
93
94    pub unsafe fn get<T>(
95        &self,
96        symbol: &[u8],
97    ) -> Result<libloading::Symbol<'_, T>, libloading::Error> {
98        for lib in &self.libs {
99            if let Ok(s) = unsafe { lib.get(symbol) } {
100                return Ok(s);
101            }
102        }
103        unsafe { self.libs[0].get(symbol) }
104    }
105}
106
107#[cfg(feature = "dynamic")]
108pub unsafe fn load_library(path: &std::path::Path) -> Result<(), String> {
109    if INSTANCE.get().is_some() {
110        return Err("Library already loaded".to_string());
111    }
112    let lib = unsafe { MaaFramework::new(path).map_err(|e| e.to_string())? };
113    INSTANCE
114        .set(lib)
115        .map_err(|_| "Library already loaded".to_string())
116}
117
118#[cfg(feature = "dynamic")]
119macro_rules! shim {
120    ($name:ident ( $($arg:ident : $type:ty),* $(,)? ) -> $ret:ty) => {
121        pub unsafe fn $name($($arg : $type),*) -> $ret {
122            let lib = INSTANCE.get().expect("MaaFramework library not loaded!");
123            let func = match &lib.$name {
124                Ok(f) => f,
125                Err(e) => panic!("Function {} not loaded: {}", stringify!($name), e),
126            };
127
128            unsafe { func($($arg),*) }
129        }
130    }
131}
132
133#[cfg(feature = "dynamic")]
134include!(concat!(env!("OUT_DIR"), "/shims.rs"));
maa_framework_sys/lib.rs

maa_framework_sys/
lib.rs
