miniquad/
fs.rs

1#[cfg(target_os = "ios")]
2use crate::native::ios;
3
4#[derive(Debug)]
5pub enum Error {
6    IOError(std::io::Error),
7    DownloadFailed,
8    AndroidAssetLoadingError,
9    /// MainBundle pathForResource returned null
10    IOSAssetNoSuchFile,
11    /// NSData dataWithContentsOfFile or data.bytes are null
12    IOSAssetNoData,
13}
14
15impl From<std::io::Error> for Error {
16    fn from(e: std::io::Error) -> Error {
17        Error::IOError(e)
18    }
19}
20
21impl std::fmt::Display for Error {
22    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
23        match self {
24            Self::IOError(e) => write!(f, "I/O error: {e}"),
25            Self::DownloadFailed => write!(f, "Download failed"),
26            Self::AndroidAssetLoadingError => write!(f, "[android] Failed to load asset"),
27            Self::IOSAssetNoSuchFile => write!(f, "[ios] No such asset file"),
28            Self::IOSAssetNoData => write!(f, "[ios] No data in asset file"),
29        }
30    }
31}
32
33impl std::error::Error for Error {}
34
35pub type Response = Result<Vec<u8>, Error>;
36
37/// Filesystem path on desktops or HTTP URL in WASM
38pub fn load_file<F: Fn(Response) + 'static>(path: &str, on_loaded: F) {
39    #[cfg(target_arch = "wasm32")]
40    wasm::load_file(path, on_loaded);
41
42    #[cfg(target_os = "android")]
43    load_file_android(path, on_loaded);
44
45    #[cfg(target_os = "ios")]
46    ios::load_file(path, on_loaded);
47
48    #[cfg(not(any(target_arch = "wasm32", target_os = "android", target_os = "ios")))]
49    load_file_desktop(path, on_loaded);
50}
51
52#[cfg(target_os = "android")]
53fn load_file_android<F: Fn(Response)>(path: &str, on_loaded: F) {
54    fn load_file_sync(path: &str) -> Response {
55        use crate::native;
56
57        let filename = std::ffi::CString::new(path).unwrap();
58
59        let mut data: native::android_asset = unsafe { std::mem::zeroed() };
60
61        unsafe { native::android::load_asset(filename.as_ptr(), &mut data as _) };
62
63        if data.content.is_null() == false {
64            let slice =
65                unsafe { std::slice::from_raw_parts(data.content, data.content_length as _) };
66            let response = slice.iter().map(|c| *c as _).collect::<Vec<_>>();
67            Ok(response)
68        } else {
69            Err(Error::AndroidAssetLoadingError)
70        }
71    }
72
73    let response = load_file_sync(path);
74
75    on_loaded(response);
76}
77
78#[cfg(target_arch = "wasm32")]
79mod wasm {
80    use super::Response;
81    use crate::native;
82
83    use std::{cell::RefCell, collections::HashMap, thread_local};
84
85    thread_local! {
86        #[allow(clippy::type_complexity)]
87        static FILES: RefCell<HashMap<u32, Box<dyn Fn(Response)>>> = RefCell::new(HashMap::new());
88    }
89
90    #[no_mangle]
91    pub extern "C" fn file_loaded(file_id: u32) {
92        use super::Error;
93        use native::wasm::fs;
94
95        FILES.with(|files| {
96            let mut files = files.borrow_mut();
97            let callback = files
98                .remove(&file_id)
99                .unwrap_or_else(|| panic!("Unknown file loaded!"));
100            let file_len = unsafe { fs::fs_get_buffer_size(file_id) };
101            if file_len == -1 {
102                callback(Err(Error::DownloadFailed));
103            } else {
104                let mut buffer = vec![0; file_len as usize];
105                unsafe { fs::fs_take_buffer(file_id, buffer.as_mut_ptr(), file_len as u32) };
106
107                callback(Ok(buffer));
108            }
109        })
110    }
111
112    pub fn load_file<F: Fn(Response) + 'static>(path: &str, on_loaded: F) {
113        use native::wasm::fs;
114        use std::ffi::CString;
115
116        let url = CString::new(path).unwrap();
117        let file_id = unsafe { fs::fs_load_file(url.as_ptr(), url.as_bytes().len() as u32) };
118        FILES.with(|files| {
119            let mut files = files.borrow_mut();
120            files.insert(file_id, Box::new(on_loaded));
121        });
122    }
123}
124
125#[cfg(not(any(target_arch = "wasm32", target_os = "android", target_os = "ios")))]
126fn load_file_desktop<F: Fn(Response)>(path: &str, on_loaded: F) {
127    fn load_file_sync(path: &str) -> Response {
128        use std::fs::File;
129        use std::io::Read;
130
131        let mut response = vec![];
132        let mut file = File::open(path)?;
133        file.read_to_end(&mut response)?;
134        Ok(response)
135    }
136
137    let response = load_file_sync(path);
138
139    on_loaded(response);
140}