54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
use std::any::Any;
|
|
|
|
pub trait HasFileData: std::any::Any {
|
|
fn files(&self) -> Option<std::sync::Arc<dyn FileEngine>> {
|
|
None
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "serialize")]
|
|
/// A file engine that serializes files to bytes
|
|
#[derive(serde::Serialize, serde::Deserialize, Debug, PartialEq, Clone)]
|
|
pub struct SerializedFileEngine {
|
|
pub files: std::collections::HashMap<String, Vec<u8>>,
|
|
}
|
|
|
|
#[cfg(feature = "serialize")]
|
|
#[async_trait::async_trait(?Send)]
|
|
impl FileEngine for SerializedFileEngine {
|
|
fn files(&self) -> Vec<String> {
|
|
self.files.keys().cloned().collect()
|
|
}
|
|
|
|
async fn read_file(&self, file: &str) -> Option<Vec<u8>> {
|
|
self.files.get(file).cloned()
|
|
}
|
|
|
|
async fn read_file_to_string(&self, file: &str) -> Option<String> {
|
|
self.read_file(file)
|
|
.await
|
|
.map(|bytes| String::from_utf8_lossy(&bytes).to_string())
|
|
}
|
|
|
|
async fn get_native_file(&self, file: &str) -> Option<Box<dyn Any>> {
|
|
self.read_file(file)
|
|
.await
|
|
.map(|val| Box::new(val) as Box<dyn Any>)
|
|
}
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
pub trait FileEngine {
|
|
// get a list of file names
|
|
fn files(&self) -> Vec<String>;
|
|
|
|
// read a file to bytes
|
|
async fn read_file(&self, file: &str) -> Option<Vec<u8>>;
|
|
|
|
// read a file to string
|
|
async fn read_file_to_string(&self, file: &str) -> Option<String>;
|
|
|
|
// returns a file in platform's native representation
|
|
async fn get_native_file(&self, file: &str) -> Option<Box<dyn Any>>;
|
|
}
|