130 lines
4.2 KiB
Rust
130 lines
4.2 KiB
Rust
pub mod filesystem;
|
|
|
|
#[cfg(feature = "storage_examples")]
|
|
pub mod examples;
|
|
|
|
#[cfg(feature = "storage_middleware")]
|
|
pub mod middleware;
|
|
|
|
use std::{
|
|
any::{Any, TypeId},
|
|
path::Path,
|
|
};
|
|
|
|
use crate::torrent_state::ManagedTorrentInfo;
|
|
|
|
pub trait StorageFactory: Send + Sync + Any {
|
|
type Storage: TorrentStorage;
|
|
|
|
fn create(&self, info: &ManagedTorrentInfo) -> anyhow::Result<Self::Storage>;
|
|
fn create_and_init(&self, info: &ManagedTorrentInfo) -> anyhow::Result<Self::Storage> {
|
|
let mut storage = self.create(info)?;
|
|
storage.init(info)?;
|
|
Ok(storage)
|
|
}
|
|
|
|
fn is_type_id(&self, type_id: TypeId) -> bool {
|
|
Self::type_id(self) == type_id
|
|
}
|
|
fn clone_box(&self) -> BoxStorageFactory;
|
|
}
|
|
|
|
pub type BoxStorageFactory = Box<dyn StorageFactory<Storage = Box<dyn TorrentStorage>>>;
|
|
|
|
pub trait StorageFactoryExt {
|
|
fn boxed(self) -> BoxStorageFactory;
|
|
}
|
|
|
|
impl<SF: StorageFactory> StorageFactoryExt for SF {
|
|
fn boxed(self) -> BoxStorageFactory {
|
|
struct Wrapper<SF> {
|
|
sf: SF,
|
|
}
|
|
|
|
impl<SF: StorageFactory> StorageFactory for Wrapper<SF> {
|
|
type Storage = Box<dyn TorrentStorage>;
|
|
|
|
fn create(&self, info: &ManagedTorrentInfo) -> anyhow::Result<Self::Storage> {
|
|
let s = self.sf.create(info)?;
|
|
Ok(Box::new(s))
|
|
}
|
|
|
|
fn is_type_id(&self, type_id: TypeId) -> bool {
|
|
self.sf.type_id() == type_id
|
|
}
|
|
|
|
fn clone_box(&self) -> BoxStorageFactory {
|
|
self.sf.clone_box()
|
|
}
|
|
}
|
|
|
|
Box::new(Wrapper { sf: self })
|
|
}
|
|
}
|
|
|
|
impl<U: StorageFactory + ?Sized> StorageFactory for Box<U> {
|
|
type Storage = U::Storage;
|
|
|
|
fn create(&self, info: &ManagedTorrentInfo) -> anyhow::Result<U::Storage> {
|
|
(**self).create(info)
|
|
}
|
|
|
|
fn clone_box(&self) -> BoxStorageFactory {
|
|
(**self).clone_box()
|
|
}
|
|
}
|
|
|
|
pub trait TorrentStorage: Send + Sync {
|
|
// Create/open files etc.
|
|
fn init(&mut self, meta: &ManagedTorrentInfo) -> anyhow::Result<()>;
|
|
|
|
/// Given a file_id (which you can get more info from in init_storage() through torrent info)
|
|
/// read buf.len() bytes into buf at offset.
|
|
fn pread_exact(&self, file_id: usize, offset: u64, buf: &mut [u8]) -> anyhow::Result<()>;
|
|
|
|
/// Given a file_id (which you can get more info from in init_storage() through torrent info)
|
|
/// write buf.len() bytes into the file at offset.
|
|
fn pwrite_all(&self, file_id: usize, offset: u64, buf: &[u8]) -> anyhow::Result<()>;
|
|
|
|
/// Remove a file from the storage. If not supported, or it doesn't matter, just return Ok(())
|
|
fn remove_file(&self, file_id: usize, filename: &Path) -> anyhow::Result<()>;
|
|
|
|
fn remove_directory_if_empty(&self, path: &Path) -> anyhow::Result<()>;
|
|
|
|
/// E.g. for filesystem backend ensure that the file has a certain length, and grow/shrink as needed.
|
|
fn ensure_file_length(&self, file_id: usize, length: u64) -> anyhow::Result<()>;
|
|
|
|
/// Replace the current storage with a dummy, and return a new one that should be used instead.
|
|
/// This is used to make the underlying object useless when e.g. pausing the torrent.
|
|
fn take(&self) -> anyhow::Result<Box<dyn TorrentStorage>>;
|
|
}
|
|
|
|
impl<U: TorrentStorage + ?Sized> TorrentStorage for Box<U> {
|
|
fn pread_exact(&self, file_id: usize, offset: u64, buf: &mut [u8]) -> anyhow::Result<()> {
|
|
(**self).pread_exact(file_id, offset, buf)
|
|
}
|
|
|
|
fn pwrite_all(&self, file_id: usize, offset: u64, buf: &[u8]) -> anyhow::Result<()> {
|
|
(**self).pwrite_all(file_id, offset, buf)
|
|
}
|
|
|
|
fn remove_file(&self, file_id: usize, filename: &Path) -> anyhow::Result<()> {
|
|
(**self).remove_file(file_id, filename)
|
|
}
|
|
|
|
fn ensure_file_length(&self, file_id: usize, length: u64) -> anyhow::Result<()> {
|
|
(**self).ensure_file_length(file_id, length)
|
|
}
|
|
|
|
fn take(&self) -> anyhow::Result<Box<dyn TorrentStorage>> {
|
|
(**self).take()
|
|
}
|
|
|
|
fn remove_directory_if_empty(&self, path: &Path) -> anyhow::Result<()> {
|
|
(**self).remove_directory_if_empty(path)
|
|
}
|
|
|
|
fn init(&mut self, meta: &ManagedTorrentInfo) -> anyhow::Result<()> {
|
|
(**self).init(meta)
|
|
}
|
|
}
|