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        libs.push(load_lib(path)?);
56
57        if let Some(parent) = path.parent() {
58            let file_name = path.file_name().and_then(|s| s.to_str()).unwrap_or("");
59
60            let (prefix, suffix) = if file_name.ends_with(".dll") {
61                ("", ".dll")
62            } else if file_name.starts_with("lib") && file_name.ends_with(".dylib") {
63                ("lib", ".dylib")
64            } else if file_name.starts_with("lib") && file_name.ends_with(".so") {
65                ("lib", ".so")
66            } else {
67                ("", "")
68            };
69
70            let mut try_load = |name: &str| {
71                let p = parent.join(format!("{}{}{}", prefix, name, suffix));
72                if p.exists() {
73                    if let Ok(lib) = load_lib(&p) {
74                        libs.push(lib);
75                    }
76                }
77            };
78
79            try_load("MaaToolkit");
80
81            let is_agent_server = file_name.contains("MaaAgentServer");
82            if !is_agent_server {
83                try_load("MaaAgentClient");
84            }
85        }
86
87        Ok(Self { libs })
88    }
89
90    pub unsafe fn get<T>(
91        &self,
92        symbol: &[u8],
93    ) -> Result<libloading::Symbol<'_, T>, libloading::Error> {
94        for lib in &self.libs {
95            if let Ok(s) = unsafe { lib.get(symbol) } {
96                return Ok(s);
97            }
98        }
99        unsafe { self.libs[0].get(symbol) }
100    }
101}
102
103#[cfg(feature = "dynamic")]
104pub unsafe fn load_library(path: &std::path::Path) -> Result<(), String> {
105    if INSTANCE.get().is_some() {
106        return Err("Library already loaded".to_string());
107    }
108    let lib = unsafe { MaaFramework::new(path).map_err(|e| e.to_string())? };
109    INSTANCE
110        .set(lib)
111        .map_err(|_| "Library already loaded".to_string())
112}
113
114#[cfg(feature = "dynamic")]
115macro_rules! shim {
116    ($name:ident ( $($arg:ident : $type:ty),* $(,)? ) -> $ret:ty) => {
117        pub unsafe fn $name($($arg : $type),*) -> $ret {
118            let lib = INSTANCE.get().expect("MaaFramework library not loaded!");
119            unsafe { (lib.$name)($($arg),*) }
120        }
121    }
122}
123
124#[cfg(feature = "dynamic")]
125include!(concat!(env!("OUT_DIR"), "/shims.rs"));
maa_framework_sys/lib.rs

maa_framework_sys/
lib.rs
