Something went wrong. Try again.
PDS software with bells & whistles you didn’t even know you needed. will move this to its own account when ready.
Something went wrong. Try again.
6.7 kB · 227 lines
Rust
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228use crate::api::error::ApiError;use crate::state::AppState;use crate::sync::util::{AccountStatus, assert_repo_availability, get_account_with_status};use axum::{ Json, extract::{Query, State}, http::StatusCode, response::{IntoResponse, Response},};use cid::Cid;use jacquard_repo::commit::Commit;use jacquard_repo::storage::BlockStore;use serde::{Deserialize, Serialize};use std::str::FromStr;use tracing::error;
async fn get_rev_from_commit(state: &AppState, cid_str: &str) -> Option<String> { let cid = Cid::from_str(cid_str).ok()?; let block = state.block_store.get(&cid).await.ok()??; let commit = Commit::from_cbor(&block).ok()?; Some(commit.rev().to_string())}
#[derive(Deserialize)]pub struct GetLatestCommitParams { pub did: String,}
#[derive(Serialize)]pub struct GetLatestCommitOutput { pub cid: String, pub rev: String,}
pub async fn get_latest_commit( State(state): State<AppState>, Query(params): Query<GetLatestCommitParams>,) -> Response { let did = params.did.trim(); if did.is_empty() { return ApiError::InvalidRequest("did is required".into()).into_response(); }
let account = match assert_repo_availability(&state.db, did, false).await { Ok(a) => a, Err(e) => return e.into_response(), };
let Some(repo_root_cid) = account.repo_root_cid else { return ApiError::RepoNotFound(Some("Repo not initialized".into())).into_response(); };
let Some(rev) = get_rev_from_commit(&state, &repo_root_cid).await else { error!( "Failed to parse commit for DID {}: CID {}", did, repo_root_cid ); return ApiError::InternalError(Some("Failed to read repo commit".into())).into_response(); };
( StatusCode::OK, Json(GetLatestCommitOutput { cid: repo_root_cid, rev, }), ) .into_response()}
#[derive(Deserialize)]pub struct ListReposParams { pub limit: Option<i64>, pub cursor: Option<String>,}
#[derive(Serialize)]#[serde(rename_all = "camelCase")]pub struct RepoInfo { pub did: String, pub head: String, pub rev: String, pub active: bool, #[serde(skip_serializing_if = "Option::is_none")] pub status: Option<String>,}
#[derive(Serialize)]pub struct ListReposOutput { #[serde(skip_serializing_if = "Option::is_none")] pub cursor: Option<String>, pub repos: Vec<RepoInfo>,}
pub async fn list_repos( State(state): State<AppState>, Query(params): Query<ListReposParams>,) -> Response { let limit = params.limit.unwrap_or(50).clamp(1, 1000); let cursor_did = params.cursor.as_deref().unwrap_or(""); let result = sqlx::query!( r#" SELECT u.did, u.deactivated_at, u.takedown_ref, r.repo_root_cid, r.repo_rev FROM repos r JOIN users u ON r.user_id = u.id WHERE u.did > $1 ORDER BY u.did ASC LIMIT $2 "#, cursor_did, limit + 1 ) .fetch_all(&state.db) .await; match result { Ok(rows) => { let has_more = rows.len() as i64 > limit; let mut repos: Vec<RepoInfo> = Vec::new(); for row in rows.iter().take(limit as usize) { let rev = match get_rev_from_commit(&state, &row.repo_root_cid).await { Some(r) => r, None => { if let Some(ref stored_rev) = row.repo_rev { stored_rev.clone() } else { tracing::warn!( "Failed to parse commit for DID {} in list_repos: CID {}", row.did, row.repo_root_cid ); continue; } } }; let status = if row.takedown_ref.is_some() { AccountStatus::Takendown } else if row.deactivated_at.is_some() { AccountStatus::Deactivated } else { AccountStatus::Active }; repos.push(RepoInfo { did: row.did.clone(), head: row.repo_root_cid.clone(), rev, active: status.is_active(), status: status.as_str().map(String::from), }); } let next_cursor = if has_more { repos.last().map(|r| r.did.clone()) } else { None }; ( StatusCode::OK, Json(ListReposOutput { cursor: next_cursor, repos, }), ) .into_response() } Err(e) => { error!("DB error in list_repos: {:?}", e); ApiError::InternalError(Some("Database error".into())).into_response() } }}
#[derive(Deserialize)]pub struct GetRepoStatusParams { pub did: String,}
#[derive(Serialize)]pub struct GetRepoStatusOutput { pub did: String, pub active: bool, #[serde(skip_serializing_if = "Option::is_none")] pub status: Option<String>, #[serde(skip_serializing_if = "Option::is_none")] pub rev: Option<String>,}
pub async fn get_repo_status( State(state): State<AppState>, Query(params): Query<GetRepoStatusParams>,) -> Response { let did = params.did.trim(); if did.is_empty() { return ApiError::InvalidRequest("did is required".into()).into_response(); }
let account = match get_account_with_status(&state.db, did).await { Ok(Some(a)) => a, Ok(None) => { return ApiError::RepoNotFound(Some(format!("Could not find repo for DID: {}", did))) .into_response(); } Err(e) => { error!("DB error in get_repo_status: {:?}", e); return ApiError::InternalError(Some("Database error".into())).into_response(); } };
let rev = if account.status.is_active() { if let Some(ref cid) = account.repo_root_cid { get_rev_from_commit(&state, cid).await } else { None } } else { None };
( StatusCode::OK, Json(GetRepoStatusOutput { did: account.did, active: account.status.is_active(), status: account.status.as_str().map(String::from), rev, }), ) .into_response()}