use futures::Stream;
use futures::StreamExt;
use js_sys::{ArrayBuffer, AsyncIterator, Uint8Array};
use wasm_bindgen::{JsCast, JsValue};
use wasm_bindgen_futures::{JsFuture, stream::JsStream};
use web_sys::{
FileSystemCreateWritableOptions, FileSystemDirectoryHandle, FileSystemFileHandle,
FileSystemGetFileOptions, FileSystemRemoveOptions, FileSystemWritableFileStream,
};
type DirectoryEntry = crate::DirectoryEntry<DirectoryHandle, FileHandle>;
#[derive(Debug, Clone)]
pub struct DirectoryHandle(FileSystemDirectoryHandle);
#[derive(Debug, Clone)]
pub struct FileHandle(FileSystemFileHandle);
#[derive(Debug, Clone)]
pub struct WritableFileStream(FileSystemWritableFileStream);
#[derive(Debug, Clone)]
pub struct File(web_sys::File);
impl From<FileSystemDirectoryHandle> for DirectoryHandle {
fn from(handle: FileSystemDirectoryHandle) -> Self {
Self(handle)
}
}
impl From<FileSystemFileHandle> for FileHandle {
fn from(handle: FileSystemFileHandle) -> Self {
Self(handle)
}
}
impl From<FileSystemWritableFileStream> for WritableFileStream {
fn from(handle: FileSystemWritableFileStream) -> Self {
Self(handle)
}
}
impl From<web_sys::File> for File {
fn from(handle: web_sys::File) -> Self {
Self(handle)
}
}
impl crate::private::Sealed for DirectoryHandle {}
impl crate::private::Sealed for FileHandle {}
impl crate::private::Sealed for WritableFileStream {}
impl crate::DirectoryHandle for DirectoryHandle {
type Error = JsValue;
type FileHandleT = FileHandle;
async fn get_file_handle_with_options(
&self,
name: &str,
options: &crate::GetFileHandleOptions,
) -> Result<Self::FileHandleT, Self::Error> {
let fs_options = FileSystemGetFileOptions::new();
fs_options.set_create(options.create);
let file_system_file_handle = FileSystemFileHandle::from(
JsFuture::from(self.0.get_file_handle_with_options(name, &fs_options)).await?,
);
Ok(FileHandle(file_system_file_handle))
}
async fn get_directory_handle_with_options(
&self,
name: &str,
options: &crate::GetDirectoryHandleOptions,
) -> Result<Self, Self::Error> {
use web_sys::FileSystemGetDirectoryOptions;
let fs_options = FileSystemGetDirectoryOptions::new();
fs_options.set_create(options.create);
let file_system_directory_handle = FileSystemDirectoryHandle::from(
JsFuture::from(self.0.get_directory_handle_with_options(name, &fs_options)).await?,
);
Ok(DirectoryHandle(file_system_directory_handle))
}
async fn remove_entry(&mut self, name: &str) -> Result<(), Self::Error> {
JsFuture::from(self.0.remove_entry(name)).await?;
Ok(())
}
async fn remove_entry_with_options(
&mut self,
name: &str,
options: &crate::FileSystemRemoveOptions,
) -> Result<(), Self::Error> {
let fs_options = FileSystemRemoveOptions::new();
fs_options.set_recursive(options.recursive);
JsFuture::from(self.0.remove_entry_with_options(name, &fs_options)).await?;
Ok(())
}
async fn entries(
&self,
) -> Result<impl Stream<Item = Result<(String, DirectoryEntry), Self::Error>>, Self::Error>
{
let entries_iterator = self.0.entries();
let async_iterator = AsyncIterator::from(entries_iterator);
let js_stream: JsStream = JsStream::from(async_iterator);
let stream = js_stream.map(|item| {
match item {
Ok(js_array) => {
let array = js_sys::Array::from(&js_array);
let filename = array
.get(0)
.as_string()
.ok_or_else(|| JsValue::from_str("Invalid filename"))?;
let handle = array.get(1);
let entry = if handle.has_type::<FileSystemFileHandle>() {
DirectoryEntry::File(FileHandle(FileSystemFileHandle::from(handle)))
} else if handle.has_type::<FileSystemDirectoryHandle>() {
DirectoryEntry::Directory(DirectoryHandle(FileSystemDirectoryHandle::from(
handle,
)))
} else {
return Err(JsValue::from_str("Unknown handle type"));
};
Ok((filename, entry))
}
Err(e) => Err(e),
}
});
Ok(stream)
}
}
impl crate::FileHandle for FileHandle {
type Error = JsValue;
type WritableFileStreamT = WritableFileStream;
async fn create_writable_with_options(
&mut self,
options: &crate::CreateWritableOptions,
) -> Result<Self::WritableFileStreamT, Self::Error> {
let fs_options = FileSystemCreateWritableOptions::new();
fs_options.set_keep_existing_data(options.keep_existing_data);
let file_system_writable_file_stream = FileSystemWritableFileStream::unchecked_from_js(
JsFuture::from(self.0.create_writable_with_options(&fs_options)).await?,
);
Ok(WritableFileStream(file_system_writable_file_stream))
}
async fn read(&self) -> Result<Vec<u8>, Self::Error> {
self.get_file().await?.read().await
}
async fn read_range<R: std::ops::RangeBounds<usize> + Send>(
&self,
range: R,
) -> Result<Vec<u8>, Self::Error> {
let file = self.get_file().await?;
file.read_range(range).await
}
async fn size(&self) -> Result<usize, Self::Error> {
let size = self.get_file().await?.size();
Ok(size)
}
}
impl FileHandle {
pub async fn get_file(&self) -> Result<File, JsValue> {
let file: web_sys::File = JsFuture::from(self.0.get_file()).await?.into();
Ok(File(file))
}
}
impl crate::WritableFileStream for WritableFileStream {
type Error = JsValue;
async fn write_at_cursor_pos(&mut self, data: &[u8]) -> Result<(), Self::Error> {
let uint8_array = js_sys::Uint8Array::from(data);
let array = js_sys::Array::new();
array.push(&uint8_array);
let file = web_sys::File::new_with_u8_array_sequence(&array, "filename")?;
JsFuture::from(self.0.write_with_blob(&file)?).await?;
Ok(())
}
async fn write_with_params(&mut self, params: &crate::WriteParams) -> Result<(), Self::Error> {
use crate::WriteCommandType;
use web_sys::{WriteCommandType as WebWriteCommandType, WriteParams as WebWriteParams};
let web_params = WebWriteParams::new(match params.command_type {
WriteCommandType::Write => WebWriteCommandType::Write,
WriteCommandType::Seek => WebWriteCommandType::Seek,
WriteCommandType::Truncate => WebWriteCommandType::Truncate,
});
if let Some(data) = ¶ms.data {
let uint8_array = js_sys::Uint8Array::from(data.as_slice());
let array = js_sys::Array::new();
array.push(&uint8_array);
let file = web_sys::File::new_with_u8_array_sequence(&array, "filename")?;
web_params.set_data(&file.into());
}
if let Some(position) = params.position {
web_params.set_position(Some(position as f64));
}
if let Some(size) = params.size {
web_params.set_size(Some(size as f64));
}
JsFuture::from(self.0.write_with_write_params(&web_params)?).await?;
Ok(())
}
async fn truncate(&mut self, size: usize) -> Result<(), Self::Error> {
JsFuture::from(self.0.truncate_with_u32(size as u32)?).await?;
Ok(())
}
async fn close(&mut self) -> Result<(), Self::Error> {
JsFuture::from(self.0.close()).await?;
Ok(())
}
async fn seek(&mut self, offset: usize) -> Result<(), Self::Error> {
JsFuture::from(self.0.seek_with_u32(offset as u32)?).await?;
Ok(())
}
}
impl File {
fn size(&self) -> usize {
self.0.size() as usize
}
async fn read(&self) -> Result<Vec<u8>, JsValue> {
let buffer = ArrayBuffer::unchecked_from_js(JsFuture::from(self.0.array_buffer()).await?);
let uint8_array = Uint8Array::new(&buffer);
let mut vec = vec![0; self.size()];
uint8_array.copy_to(&mut vec);
Ok(vec)
}
async fn read_range<R: std::ops::RangeBounds<usize>>(
&self,
range: R,
) -> Result<Vec<u8>, JsValue> {
use std::ops::Bound;
use web_sys::Blob;
let size = self.size();
let start = match range.start_bound() {
Bound::Included(&n) => n,
Bound::Excluded(&n) => n + 1,
Bound::Unbounded => 0,
};
let end = match range.end_bound() {
Bound::Included(&n) => n + 1,
Bound::Excluded(&n) => n,
Bound::Unbounded => size,
};
if start >= size {
return Ok(Vec::new());
}
let actual_end = end.min(size);
if start >= actual_end {
return Ok(Vec::new());
}
let blob: Blob = self
.0
.slice_with_f64_and_f64(start as f64, actual_end as f64)?;
let buffer = ArrayBuffer::unchecked_from_js(JsFuture::from(blob.array_buffer()).await?);
let uint8_array = Uint8Array::new(&buffer);
let mut vec = vec![0; actual_end - start];
uint8_array.copy_to(&mut vec);
Ok(vec)
}
#[allow(dead_code)]
pub async fn text(&self) -> Result<String, JsValue> {
JsFuture::from(self.0.text())
.await?
.as_string()
.ok_or(JsValue::NULL)
}
}