atrium-identity 0.1.9

use super::HandleResolver;
use crate::error::{Error, Result};
use atrium_api::types::string::{Did, Handle};
use atrium_common::resolver::Resolver;
use atrium_xrpc::HttpClient;
use atrium_xrpc::http::Request;
use std::sync::Arc;

const WELL_KNWON_PATH: &str = "/.well-known/atproto-did";

#[derive(Clone, Debug)]
pub struct WellKnownHandleResolverConfig<T> {
    pub http_client: Arc<T>,
}

pub struct WellKnownHandleResolver<T> {
    http_client: Arc<T>,
}

impl<T> WellKnownHandleResolver<T> {
    pub fn new(config: WellKnownHandleResolverConfig<T>) -> Self {
        Self { http_client: config.http_client }
    }
}

impl<T> Resolver for WellKnownHandleResolver<T>
where
    T: HttpClient + Send + Sync + 'static,
{
    type Input = Handle;
    type Output = Did;
    type Error = Error;

    async fn resolve(&self, handle: &Self::Input) -> Result<Self::Output> {
        let url = format!("https://{}{WELL_KNWON_PATH}", handle.as_str());
        // TODO: no-cache?
        let res = self
            .http_client
            .send_http(Request::builder().uri(url).body(Vec::new())?)
            .await
            .map_err(Error::HttpClient)?;
        if res.status().is_success() {
            String::from_utf8_lossy(res.body())
                .trim() // strip whitespace for best compatibility. https://atproto.com/specs/handle#https-well-known-method
                .parse::<Did>()
                .map_err(|e| Error::Did(e.to_string()))
        } else {
            Err(Error::HttpStatus(res.status()))
        }
    }
}

impl<T> HandleResolver for WellKnownHandleResolver<T> where T: HttpClient + Send + Sync + 'static {}