1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
use std::path::Path;
use thiserror::Error;
use wasm_bindgen::{JsCast, JsValue};
use wasm_bindgen_futures::JsFuture;
use web_sys::{
File, FileSystemDirectoryHandle, FileSystemFileHandle, FileSystemGetDirectoryOptions,
FileSystemGetFileOptions, FileSystemWritableFileStream, Url, js_sys, window,
};
use crate::FileStore;
#[derive(Clone)]
pub struct FilesOPFS {
directory: String,
}
impl FilesOPFS {
pub async fn new(directory: impl AsRef<str>) -> Result<Self, OPFSError> {
let directory = directory.as_ref();
let directory_string = directory.to_string();
let promise = window().unwrap().navigator().storage().get_directory();
let opfs_root: FileSystemDirectoryHandle = JsFuture::from(promise).await?.into();
let options = FileSystemGetDirectoryOptions::new();
options.set_create(true);
let directory: FileSystemDirectoryHandle =
JsFuture::from(opfs_root.get_directory_handle_with_options(directory, &options))
.await?
.into();
Ok(Self {
directory: directory_string,
})
}
pub async fn get_src(&self, file_name: impl AsRef<str>) -> Result<String, OPFSError> {
let promise = window().unwrap().navigator().storage().get_directory();
let opfs_root: FileSystemDirectoryHandle = JsFuture::from(promise).await?.into();
let directory: FileSystemDirectoryHandle =
JsFuture::from(opfs_root.get_directory_handle(&self.directory))
.await?
.into();
let handle: FileSystemFileHandle =
JsFuture::from(directory.get_file_handle(file_name.as_ref()))
.await?
.into();
let file: File = JsFuture::from(handle.get_file()).await?.into();
let src = Url::create_object_url_with_blob(&file)?;
Ok(src)
}
}
impl FileStore for FilesOPFS {
type Err = OPFSError;
async fn is_stored(&self, name: &str) -> Result<bool, Self::Err> {
let promise = window().unwrap().navigator().storage().get_directory();
let opfs_root: FileSystemDirectoryHandle = JsFuture::from(promise).await?.into();
let directory: FileSystemDirectoryHandle =
JsFuture::from(opfs_root.get_directory_handle(&self.directory))
.await?
.into();
let stored = JsFuture::from(directory.get_file_handle(name))
.await
.map(|_| true)
// TODO: distinguish between other errors and notfound
.unwrap_or(false);
Ok(stored)
}
async fn store(&self, name: &str, data: &[u8]) -> Result<(), Self::Err> {
let promise = window().unwrap().navigator().storage().get_directory();
let opfs_root: FileSystemDirectoryHandle = JsFuture::from(promise).await?.into();
let directory: FileSystemDirectoryHandle =
JsFuture::from(opfs_root.get_directory_handle(&self.directory))
.await?
.into();
let options = FileSystemGetFileOptions::new();
options.set_create(true);
let handle: FileSystemFileHandle =
JsFuture::from(directory.get_file_handle_with_options(name, &options))
.await?
.into();
let write_handle: FileSystemWritableFileStream =
JsFuture::from(handle.create_writable()).await?.into();
let buffer = js_sys::ArrayBuffer::new(data.len() as u32);
let u8arr = js_sys::Uint8Array::new(&buffer);
for (idx, v) in data.iter().enumerate() {
u8arr.set_index(idx as u32, *v);
}
let write_promise = write_handle.write_with_js_u8_array(&u8arr).unwrap();
// let write_promise = write_handle.write_with_u8_array(data)?;
let _ = JsFuture::from(write_promise).await?;
let _ = JsFuture::from(write_handle.close()).await?;
Ok(())
}
async fn delete(&self, name: &str) -> Result<(), Self::Err> {
let promise = window().unwrap().navigator().storage().get_directory();
let opfs_root: FileSystemDirectoryHandle = JsFuture::from(promise).await?.into();
let directory: FileSystemDirectoryHandle =
JsFuture::from(opfs_root.get_directory_handle(&self.directory))
.await?
.into();
let _ = JsFuture::from(directory.remove_entry(name)).await?;
Ok(())
}
}
#[derive(Error, Clone, Debug)]
pub enum OPFSError {
#[error("js opfs error: {0}")]
Error(String),
}
// TODO: better errors
impl From<JsValue> for OPFSError {
fn from(value: JsValue) -> Self {
Self::Error(
value
.dyn_into::<js_sys::Error>()
.ok()
.and_then(|err| err.message().as_string())
.unwrap_or(String::from("<no string>")),
)
}
}
|
