mirror of
https://git.asonix.dog/asonix/pict-rs
synced 2024-11-12 23:43:57 +00:00
111 lines
3.5 KiB
Rust
111 lines
3.5 KiB
Rust
use crate::{details::Details, store::Identifier};
|
|
use futures_util::Stream;
|
|
use uuid::Uuid;
|
|
|
|
pub(crate) mod sled;
|
|
|
|
pub(crate) struct Alias {
|
|
id: Uuid,
|
|
extension: String,
|
|
}
|
|
pub(crate) struct DeleteToken {
|
|
id: Uuid,
|
|
}
|
|
pub(crate) struct AlreadyExists;
|
|
|
|
impl Alias {
|
|
fn to_bytes(&self) -> Vec<u8> {
|
|
let mut v = self.id.as_bytes().to_vec();
|
|
|
|
v.extend_from_slice(self.extension.as_bytes());
|
|
|
|
v
|
|
}
|
|
|
|
fn from_slice(bytes: &[u8]) -> Option<Self> {
|
|
if bytes.len() > 16 {
|
|
let id = Uuid::from_slice(&bytes[0..16]).expect("Already checked length");
|
|
let extension = String::from_utf8_lossy(&bytes[16..]).to_string();
|
|
|
|
Some(Self { id, extension })
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
impl DeleteToken {
|
|
fn to_bytes(&self) -> Vec<u8> {
|
|
self.id.as_bytes().to_vec()
|
|
}
|
|
|
|
fn from_slice(bytes: &[u8]) -> Option<Self> {
|
|
Some(DeleteToken {
|
|
id: Uuid::from_slice(bytes).ok()?,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
pub(crate) trait SettingsRepo {
|
|
type Bytes: AsRef<[u8]> + From<Vec<u8>>;
|
|
type Error: std::error::Error;
|
|
|
|
async fn set(&self, key: &'static [u8], value: Self::Bytes) -> Result<(), Self::Error>;
|
|
async fn get(&self, key: &'static [u8]) -> Result<Option<Self::Bytes>, Self::Error>;
|
|
async fn remove(&self, key: &'static [u8]) -> Result<(), Self::Error>;
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
pub(crate) trait IdentifierRepo<I: Identifier> {
|
|
type Bytes: AsRef<[u8]> + From<Vec<u8>>;
|
|
type Error: std::error::Error;
|
|
|
|
async fn relate_details(&self, identifier: I, details: Details) -> Result<(), Self::Error>;
|
|
async fn details(&self, identifier: I) -> Result<Option<Details>, Self::Error>;
|
|
|
|
async fn relate_hash(&self, identifier: I, hash: Self::Bytes) -> Result<(), Self::Error>;
|
|
async fn hash(&self, identifier: I) -> Result<Self::Bytes, Self::Error>;
|
|
|
|
async fn cleanup(&self, identifier: I) -> Result<(), Self::Error>;
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
pub(crate) trait HashRepo<I: Identifier> {
|
|
type Bytes: AsRef<[u8]> + From<Vec<u8>>;
|
|
type Error: std::error::Error;
|
|
type Stream: Stream<Item = Result<Self::Bytes, Self::Error>>;
|
|
|
|
async fn hashes(&self) -> Self::Stream;
|
|
|
|
async fn create(&self, hash: Self::Bytes) -> Result<Result<(), AlreadyExists>, Self::Error>;
|
|
|
|
async fn relate_alias(&self, hash: Self::Bytes, alias: Alias) -> Result<(), Self::Error>;
|
|
async fn remove_alias(&self, hash: Self::Bytes, alias: Alias) -> Result<(), Self::Error>;
|
|
async fn aliases(&self, hash: Self::Bytes) -> Result<Vec<Alias>, Self::Error>;
|
|
|
|
async fn relate_identifier(&self, hash: Self::Bytes, identifier: I) -> Result<(), Self::Error>;
|
|
async fn identifier(&self, hash: Self::Bytes) -> Result<I, Self::Error>;
|
|
|
|
async fn cleanup(&self, hash: Self::Bytes) -> Result<(), Self::Error>;
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
pub(crate) trait AliasRepo {
|
|
type Bytes: AsRef<[u8]> + From<Vec<u8>>;
|
|
type Error: std::error::Error;
|
|
|
|
async fn create(&self, alias: Alias) -> Result<Result<(), AlreadyExists>, Self::Error>;
|
|
|
|
async fn relate_delete_token(
|
|
&self,
|
|
alias: Alias,
|
|
delete_token: DeleteToken,
|
|
) -> Result<Result<(), AlreadyExists>, Self::Error>;
|
|
async fn delete_token(&self, alias: Alias) -> Result<DeleteToken, Self::Error>;
|
|
|
|
async fn relate_hash(&self, alias: Alias, hash: Self::Bytes) -> Result<(), Self::Error>;
|
|
async fn hash(&self, alias: Alias) -> Result<Self::Bytes, Self::Error>;
|
|
|
|
async fn cleanup(&self, alias: Alias) -> Result<(), Self::Error>;
|
|
}
|