mirror of
https://github.com/kirjavascript/trueLMAO.git
synced 2024-06-02 03:27:21 -04:00
153 lines
4.3 KiB
Rust
153 lines
4.3 KiB
Rust
type FileData = Vec<u8>;
|
|
|
|
// wasm
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen::prelude::*;
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen::JsCast;
|
|
#[cfg(target_arch = "wasm32")]
|
|
use web_sys::{window, Url, File, HtmlInputElement, FileReader};
|
|
#[cfg(target_arch = "wasm32")]
|
|
use js_sys::{Uint8Array, ArrayBuffer};
|
|
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
pub struct FileDialog {
|
|
tx: std::sync::mpsc::Sender<FileData>,
|
|
rx: std::sync::mpsc::Receiver<FileData>,
|
|
input: HtmlInputElement,
|
|
closure: Option<Closure<dyn FnMut()>>,
|
|
}
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
impl Default for FileDialog {
|
|
fn default() -> Self {
|
|
let (tx, rx) = std::sync::mpsc::channel();
|
|
|
|
let document = window().unwrap().document().unwrap();
|
|
let body = document.body().unwrap();
|
|
let input = document.create_element("input").unwrap().dyn_into::<HtmlInputElement>().unwrap();
|
|
input.set_attribute("type", "file").unwrap();
|
|
input.style().set_property("display", "none").unwrap();
|
|
body.append_child(&input).unwrap();
|
|
|
|
Self {
|
|
rx,
|
|
tx,
|
|
input,
|
|
closure: None,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
impl Drop for FileDialog {
|
|
fn drop(&mut self) {
|
|
self.input.remove();
|
|
if self.closure.is_some() {
|
|
std::mem::replace(&mut self.closure, None).unwrap().forget();
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
impl FileDialog {
|
|
pub fn open(&mut self) {
|
|
if let Some(closure) = &self.closure {
|
|
self.input.remove_event_listener_with_callback("change", closure.as_ref().unchecked_ref()).unwrap();
|
|
std::mem::replace(&mut self.closure, None).unwrap().forget();
|
|
}
|
|
|
|
let tx = self.tx.clone();
|
|
let input_clone = self.input.clone();
|
|
|
|
let closure = Closure::once(move || {
|
|
if let Some(file) = input_clone.files().and_then(|files| files.get(0)) {
|
|
let reader = FileReader::new().unwrap();
|
|
let reader_clone = reader.clone();
|
|
let onload_closure = Closure::once(Box::new(move || {
|
|
let array_buffer = reader_clone.result().unwrap().dyn_into::<ArrayBuffer>().unwrap();
|
|
let buffer = Uint8Array::new(&array_buffer).to_vec();
|
|
tx.send(buffer).ok();
|
|
}));
|
|
|
|
reader.set_onload(Some(onload_closure.as_ref().unchecked_ref()));
|
|
reader.read_as_array_buffer(&file).unwrap();
|
|
onload_closure.forget();
|
|
}
|
|
});
|
|
|
|
self.input.add_event_listener_with_callback("change", closure.as_ref().unchecked_ref()).unwrap();
|
|
self.closure = Some(closure);
|
|
self.input.click();
|
|
}
|
|
|
|
pub fn get(&self) -> Option<Vec<u8>> {
|
|
if let Ok(file) = self.rx.try_recv() {
|
|
Some(file)
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
pub fn save(&self, filename: &str, filedata: FileData) {
|
|
let array = js_sys::Uint8Array::from(filedata.as_slice());
|
|
let blob_parts = js_sys::Array::new();
|
|
blob_parts.push(&array.buffer());
|
|
|
|
let file = File::new_with_blob_sequence_and_options(
|
|
&blob_parts.into(),
|
|
filename,
|
|
web_sys::FilePropertyBag::new().type_("application/octet-stream")
|
|
).unwrap();
|
|
let url = Url::create_object_url_with_blob(&file);
|
|
if let Some(window) = web_sys::window() {
|
|
window.location().set_href(&url.unwrap()).ok();
|
|
}
|
|
}
|
|
}
|
|
|
|
// native
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
use rfd;
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
pub struct FileDialog {
|
|
file: Option<FileData>,
|
|
}
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
impl Default for FileDialog {
|
|
fn default() -> Self {
|
|
Self {
|
|
file: None,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
impl FileDialog {
|
|
pub fn open(&mut self) {
|
|
let path = rfd::FileDialog::new().pick_file();
|
|
if let Some(path) = path {
|
|
self.file = std::fs::read(path).ok();
|
|
}
|
|
}
|
|
|
|
pub fn get(&mut self) -> Option<FileData> {
|
|
std::mem::replace(&mut self.file, None)
|
|
}
|
|
|
|
pub fn save(&self, filename: &str, file: FileData) {
|
|
let path = rfd::FileDialog::new()
|
|
.set_file_name(filename)
|
|
.save_file();
|
|
|
|
if let Some(path) = path {
|
|
std::fs::write(path, file).ok();
|
|
}
|
|
}
|
|
}
|