You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.9 KiB
Rust
48 lines
1.9 KiB
Rust
use crate::{Prefix, ResolverError, Webfinger};
|
|
use async_trait::async_trait;
|
|
|
|
/// A trait to easily generate a WebFinger endpoint for any resource repository.
|
|
///
|
|
/// The `R` type is your resource repository (a database for instance) that will be passed to the
|
|
/// [`find`](Resolver::find) and [`endpoint`](Resolver::endpoint) functions.
|
|
#[async_trait]
|
|
pub trait AsyncResolver {
|
|
type Repo: Send;
|
|
/// Returns the domain name of the current instance.
|
|
async fn instance_domain<'a>(&self) -> &'a str;
|
|
|
|
/// Tries to find a resource, `acct`, in the repository `resource_repo`.
|
|
///
|
|
/// `acct` is not a complete `acct:` URI, it only contains the identifier of the requested resource
|
|
/// (e.g. `test` for `acct:test@example.org`)
|
|
///
|
|
/// If the resource couldn't be found, you may probably want to return a [`ResolverError::NotFound`].
|
|
async fn find(
|
|
&self,
|
|
prefix: Prefix,
|
|
acct: String,
|
|
resource_repo: Self::Repo,
|
|
) -> Result<Webfinger, ResolverError>;
|
|
|
|
/// Returns a WebFinger result for a requested resource.
|
|
async fn endpoint<R: Into<String> + Send>(
|
|
&self,
|
|
resource: R,
|
|
resource_repo: Self::Repo,
|
|
) -> Result<Webfinger, ResolverError> {
|
|
let resource = resource.into();
|
|
let mut parsed_query = resource.splitn(2, ':');
|
|
let res_prefix = Prefix::from(parsed_query.next().ok_or(ResolverError::InvalidResource)?);
|
|
let res = parsed_query.next().ok_or(ResolverError::InvalidResource)?;
|
|
|
|
let mut parsed_res = res.splitn(2, '@');
|
|
let user = parsed_res.next().ok_or(ResolverError::InvalidResource)?;
|
|
let domain = parsed_res.next().ok_or(ResolverError::InvalidResource)?;
|
|
if domain == self.instance_domain().await {
|
|
self.find(res_prefix, user.to_string(), resource_repo).await
|
|
} else {
|
|
Err(ResolverError::WrongDomain)
|
|
}
|
|
}
|
|
}
|