Something went wrong. Try again.
A lexicon-driven AppView for ATProto.
Something went wrong. Try again.
Rust
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264use serde::{Deserialize, Serialize};
use crate::error::AppError;use crate::http_retry::parse_retry_after;
#[derive(Serialize)]pub struct Profile { pub did: String, pub handle: String, #[serde(rename = "displayName", skip_serializing_if = "Option::is_none")] pub display_name: Option<String>, #[serde(skip_serializing_if = "Option::is_none")] pub description: Option<String>, #[serde(rename = "avatarURL", skip_serializing_if = "Option::is_none")] pub avatar_url: Option<String>, #[serde(skip_serializing_if = "Option::is_none")] pub avatar: Option<AvatarBlob>,}
#[derive(Serialize)]pub struct AvatarBlob { #[serde(rename = "$link")] pub link: String, #[serde(rename = "mimeType")] pub mime_type: String, pub size: Option<u64>,}
#[derive(Deserialize)]#[serde(rename_all = "camelCase")]pub struct DidDocument { #[serde(default)] pub also_known_as: Vec<String>, #[serde(default)] pub verification_method: Vec<DidVerificationMethod>, #[serde(default)] pub service: Vec<DidService>,}
#[derive(Deserialize)]#[serde(rename_all = "camelCase")]pub struct DidVerificationMethod { pub id: String, #[serde(rename = "type")] pub method_type: String, #[serde(default)] pub public_key_multibase: Option<String>,}
#[derive(Deserialize)]#[serde(rename_all = "camelCase")]pub struct DidService { pub id: String, pub service_endpoint: String,}
#[derive(Deserialize)]struct GetRecordResponse { value: serde_json::Value,}
/// Resolve a full profile for the given DID: DID document -> handle + PDS -> profile record.pub async fn resolve_profile( http: &reqwest::Client, plc_url: &str, did: &str,) -> Result<Profile, AppError> { let did_doc = resolve_did_document(http, plc_url, did).await?;
let handle = did_doc .also_known_as .iter() .find_map(|uri| uri.strip_prefix("at://")) .map(|h| h.to_string());
let pds_endpoint = did_doc .service .iter() .find(|s| s.id == "#atproto_pds") .map(|s| s.service_endpoint.clone()) .ok_or_else(|| AppError::NotFound("no PDS endpoint in DID document".into()))?;
let (display_name, description, avatar_url, avatar) = fetch_profile_from_pds(http, &pds_endpoint, did) .await .unwrap_or((None, None, None, None));
Ok(Profile { did: did.to_string(), handle: handle.unwrap_or_else(|| did.to_string()), display_name, description, avatar_url, avatar, })}
/// Resolve the PDS endpoint for a DID by fetching its DID document.pub async fn resolve_pds_endpoint( http: &reqwest::Client, plc_url: &str, did: &str,) -> Result<String, AppError> { let did_doc = resolve_did_document(http, plc_url, did).await?;
did_doc .service .iter() .find(|s| s.id == "#atproto_pds") .map(|s| s.service_endpoint.clone()) .ok_or_else(|| AppError::NotFound("no PDS endpoint in DID document".into()))}
/// Resolve the labeler service endpoint for a DID./// Tries `#atproto_labeler` first, falls back to `#atproto_pds`.pub async fn resolve_labeler_endpoint( http: &reqwest::Client, plc_url: &str, did: &str,) -> Result<String, AppError> { let did_doc = resolve_did_document(http, plc_url, did).await?;
did_doc .service .iter() .find(|s| s.id == "#atproto_labeler") .or_else(|| did_doc.service.iter().find(|s| s.id == "#atproto_pds")) .map(|s| s.service_endpoint.clone()) .ok_or_else(|| AppError::NotFound("no labeler or PDS endpoint in DID document".into()))}
/// Fetch a DID document from the PLC directory or via `did:web` resolution.pub async fn resolve_did_document( http: &reqwest::Client, plc_url: &str, did: &str,) -> Result<DidDocument, AppError> { let url = if let Some(domain) = did.strip_prefix("did:web:") { format!("https://{}/.well-known/did.json", domain) } else { format!("{}/{did}", plc_url.trim_end_matches('/')) };
let resp = { let max_retries = 5; let mut attempts = 0; loop { let r = http .get(&url) .send() .await .map_err(|e| AppError::Internal(format!("DID resolution failed: {e}")))?;
if r.status() == reqwest::StatusCode::TOO_MANY_REQUESTS { attempts += 1; if attempts >= max_retries { return Err(AppError::Internal(format!( "DID resolution for {did} rate-limited after {max_retries} retries" ))); } let wait = parse_retry_after(r.headers()); tracing::warn!( did, wait, attempts, max_retries, "rate limited during DID resolution, sleeping" ); tokio::time::sleep(tokio::time::Duration::from_secs(wait)).await; continue; }
break r; } };
if !resp.status().is_success() { return Err(AppError::NotFound(format!( "DID document not found for {did}" ))); }
resp.json() .await .map_err(|e| AppError::Internal(format!("invalid DID document: {e}")))}
/// Fetch the `app.bsky.actor.profile` record from the user's PDS and extract/// displayName, description, and avatar URL.async fn fetch_profile_from_pds( http: &reqwest::Client, pds_endpoint: &str, did: &str,) -> Result< ( Option<String>, Option<String>, Option<String>, Option<AvatarBlob>, ), AppError,> { let url = format!( "{}/xrpc/com.atproto.repo.getRecord?repo={}&collection=app.bsky.actor.profile&rkey=self", pds_endpoint.trim_end_matches('/'), did, );
let resp = http .get(&url) .send() .await .map_err(|e| AppError::Internal(format!("PDS request failed: {e}")))?;
if !resp.status().is_success() { return Ok((None, None, None, None)); }
let record: GetRecordResponse = resp .json() .await .map_err(|e| AppError::Internal(format!("invalid PDS response: {e}")))?;
let value = &record.value;
let display_name = value .get("displayName") .and_then(|v| v.as_str()) .map(|s| s.to_string());
let description = value .get("description") .and_then(|v| v.as_str()) .map(|s| s.to_string());
let avatar_value = value.get("avatar");
let avatar_url = avatar_value .and_then(|avatar| avatar.get("ref")) .and_then(|r| r.get("$link")) .and_then(|link| link.as_str()) .map(|cid| { format!( "{}/xrpc/com.atproto.sync.getBlob?did={}&cid={}", pds_endpoint.trim_end_matches('/'), did, cid, ) });
let avatar = avatar_value.and_then(|av| { let link = av.get("ref")?.get("$link")?.as_str()?.to_string(); let mime_type = av.get("mimeType")?.as_str()?.to_string(); let size = av.get("size").and_then(|s| s.as_u64()); Some(AvatarBlob { link, mime_type, size, }) });
Ok((display_name, description, avatar_url, avatar))}