mirror of
https://github.com/LemmyNet/activitypub-federation-rust.git
synced 2024-05-09 20:52:39 +00:00
105 lines
2.9 KiB
Rust
105 lines
2.9 KiB
Rust
use crate::{config::Data, error::Error, fetch::fetch_object_http, traits::Collection};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::{
|
|
fmt::{Debug, Display, Formatter},
|
|
marker::PhantomData,
|
|
};
|
|
use url::Url;
|
|
|
|
/// Typed wrapper for Activitypub Collection ID which helps with dereferencing.
|
|
#[derive(Serialize, Deserialize)]
|
|
#[serde(transparent)]
|
|
pub struct CollectionId<Kind>(Box<Url>, PhantomData<Kind>)
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: Deserialize<'de2>;
|
|
|
|
impl<Kind> CollectionId<Kind>
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: Deserialize<'de2>,
|
|
{
|
|
/// Construct a new CollectionId instance
|
|
pub fn parse(url: &str) -> Result<Self, url::ParseError> {
|
|
Ok(Self(Box::new(Url::parse(url)?), PhantomData::<Kind>))
|
|
}
|
|
|
|
/// Fetches collection over HTTP
|
|
///
|
|
/// Unlike [ObjectId::dereference](crate::fetch::object_id::ObjectId::dereference) this method doesn't do
|
|
/// any caching.
|
|
pub async fn dereference(
|
|
&self,
|
|
owner: &<Kind as Collection>::Owner,
|
|
data: &Data<<Kind as Collection>::DataType>,
|
|
) -> Result<Kind, <Kind as Collection>::Error>
|
|
where
|
|
<Kind as Collection>::Error: From<Error>,
|
|
{
|
|
let res = fetch_object_http(&self.0, data).await?;
|
|
let redirect_url = &res.url;
|
|
Kind::verify(&res.object, redirect_url, data).await?;
|
|
Kind::from_json(res.object, owner, data).await
|
|
}
|
|
}
|
|
|
|
/// Need to implement clone manually, to avoid requiring Kind to be Clone
|
|
impl<Kind> Clone for CollectionId<Kind>
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn clone(&self) -> Self {
|
|
CollectionId(self.0.clone(), self.1)
|
|
}
|
|
}
|
|
|
|
impl<Kind> Display for CollectionId<Kind>
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
|
write!(f, "{}", self.0.as_str())
|
|
}
|
|
}
|
|
|
|
impl<Kind> Debug for CollectionId<Kind>
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
|
write!(f, "{}", self.0.as_str())
|
|
}
|
|
}
|
|
impl<Kind> From<CollectionId<Kind>> for Url
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn from(id: CollectionId<Kind>) -> Self {
|
|
*id.0
|
|
}
|
|
}
|
|
|
|
impl<Kind> From<Url> for CollectionId<Kind>
|
|
where
|
|
Kind: Collection + Send + 'static,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn from(url: Url) -> Self {
|
|
CollectionId(Box::new(url), PhantomData::<Kind>)
|
|
}
|
|
}
|
|
|
|
impl<Kind> PartialEq for CollectionId<Kind>
|
|
where
|
|
Kind: Collection,
|
|
for<'de2> <Kind as Collection>::Kind: serde::Deserialize<'de2>,
|
|
{
|
|
fn eq(&self, other: &Self) -> bool {
|
|
self.0.eq(&other.0) && self.1 == other.1
|
|
}
|
|
}
|