this repo has no description
lewis.moe
1use crate::api::repo::record::utils::{commit_and_log, RecordOp};
2use crate::api::repo::record::write::prepare_repo_write;
3use crate::repo::tracking::TrackingBlockStore;
4use crate::state::AppState;
5use axum::{
6 extract::State,
7 http::{HeaderMap, StatusCode},
8 response::{IntoResponse, Response},
9 Json,
10};
11use cid::Cid;
12use jacquard::types::string::Nsid;
13use jacquard_repo::{commit::Commit, mst::Mst, storage::BlockStore};
14use serde::Deserialize;
15use serde_json::json;
16use std::str::FromStr;
17use std::sync::Arc;
18use tracing::error;
19
20#[derive(Deserialize)]
21pub struct DeleteRecordInput {
22 pub repo: String,
23 pub collection: String,
24 pub rkey: String,
25 #[serde(rename = "swapRecord")]
26 pub swap_record: Option<String>,
27 #[serde(rename = "swapCommit")]
28 pub swap_commit: Option<String>,
29}
30
31pub async fn delete_record(
32 State(state): State<AppState>,
33 headers: HeaderMap,
34 Json(input): Json<DeleteRecordInput>,
35) -> Response {
36 let (did, user_id, current_root_cid) =
37 match prepare_repo_write(&state, &headers, &input.repo).await {
38 Ok(res) => res,
39 Err(err_res) => return err_res,
40 };
41
42 if let Some(swap_commit) = &input.swap_commit {
43 if Cid::from_str(swap_commit).ok() != Some(current_root_cid) {
44 return (
45 StatusCode::CONFLICT,
46 Json(json!({"error": "InvalidSwap", "message": "Repo has been modified"})),
47 )
48 .into_response();
49 }
50 }
51
52 let tracking_store = TrackingBlockStore::new(state.block_store.clone());
53
54 let commit_bytes = match tracking_store.get(¤t_root_cid).await {
55 Ok(Some(b)) => b,
56 _ => return (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({"error": "InternalError", "message": "Commit block not found"}))).into_response(),
57 };
58 let commit = match Commit::from_cbor(&commit_bytes) {
59 Ok(c) => c,
60 _ => return (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({"error": "InternalError", "message": "Failed to parse commit"}))).into_response(),
61 };
62
63 let mst = Mst::load(
64 Arc::new(tracking_store.clone()),
65 commit.data,
66 None,
67 );
68 let collection_nsid = match input.collection.parse::<Nsid>() {
69 Ok(n) => n,
70 Err(_) => return (StatusCode::BAD_REQUEST, Json(json!({"error": "InvalidCollection"}))).into_response(),
71 };
72 let key = format!("{}/{}", collection_nsid, input.rkey);
73
74 if let Some(swap_record_str) = &input.swap_record {
75 let expected_cid = Cid::from_str(swap_record_str).ok();
76 let actual_cid = mst.get(&key).await.ok().flatten();
77 if expected_cid != actual_cid {
78 return (StatusCode::CONFLICT, Json(json!({"error": "InvalidSwap", "message": "Record has been modified or does not exist"}))).into_response();
79 }
80 }
81
82 if mst.get(&key).await.ok().flatten().is_none() {
83 return (StatusCode::OK, Json(json!({}))).into_response();
84 }
85
86 let new_mst = match mst.delete(&key).await {
87 Ok(m) => m,
88 Err(e) => {
89 error!("Failed to delete from MST: {:?}", e);
90 return (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({"error": "InternalError", "message": format!("Failed to delete from MST: {:?}", e)}))).into_response();
91 }
92 };
93
94 let new_mst_root = match new_mst.persist().await {
95 Ok(c) => c,
96 Err(e) => {
97 error!("Failed to persist MST: {:?}", e);
98 return (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({"error": "InternalError", "message": "Failed to persist MST"}))).into_response();
99 }
100 };
101
102 let op = RecordOp::Delete { collection: input.collection, rkey: input.rkey };
103 let written_cids = tracking_store.get_written_cids();
104 let written_cids_str = written_cids.iter().map(|c| c.to_string()).collect::<Vec<_>>();
105
106 if let Err(e) = commit_and_log(&state, &did, user_id, Some(current_root_cid), new_mst_root, vec![op], &written_cids_str).await {
107 return (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({"error": "InternalError", "message": e}))).into_response();
108 };
109
110 (StatusCode::OK, Json(json!({}))).into_response()
111}