Merge pull request #55 from TheRipperoni/requestPlcOpSignature

PDS: Implemented Request PLC Signature API

Author: rudyfraser

rsky-pds/src/apis/com/atproto/identity/mod.rs

@@ -1,4 +1,5 @@
 pub mod get_recommended_did_credentials;
+pub mod request_plc_operation_signature;
 pub mod resolve_handle;
 pub mod sign_plc_operation;
 pub mod update_handle;

rsky-pds/src/apis/com/atproto/identity/request_plc_operation_signature.rs

@@ -0,0 +1,91 @@
+use crate::account_manager::helpers::account::{ActorAccount, AvailabilityFlags};
+use crate::account_manager::AccountManager;
+use crate::apis::ApiError;
+use crate::auth_verifier::AccessFull;
+use crate::mailer::{send_plc_operation, TokenParam};
+use crate::models::models::EmailTokenPurpose;
+
+#[tracing::instrument(skip_all)]
+async fn get_requester_did(auth: &AccessFull) -> Result<String, ApiError> {
+    match &auth.access.credentials {
+        None => {
+            tracing::error!("Failed to find access credentials");
+            Err(ApiError::RuntimeError)
+        }
+        Some(res) => match &res.did {
+            None => {
+                tracing::error!("Failed to find did");
+                Err(ApiError::RuntimeError)
+            }
+            Some(did) => Ok(did.clone()),
+        },
+    }
+}
+
+#[tracing::instrument(skip_all)]
+async fn get_account(requester_did: &str) -> Result<ActorAccount, ApiError> {
+    let availability_flags = AvailabilityFlags {
+        include_taken_down: Some(true),
+        include_deactivated: Some(true),
+    };
+    match AccountManager::get_account(&requester_did.to_string(), Some(availability_flags)).await {
+        Ok(account) => match account {
+            None => {
+                tracing::error!("Account not found despite valid credentials");
+                Err(ApiError::RuntimeError)
+            }
+            Some(account) => Ok(account),
+        },
+        Err(error) => {
+            tracing::error!("Error getting account\n{error}");
+            Err(ApiError::RuntimeError)
+        }
+    }
+}
+
+#[tracing::instrument(skip_all)]
+async fn create_email_token(requester: &str) -> Result<String, ApiError> {
+    match AccountManager::create_email_token(
+        &requester.to_string(),
+        EmailTokenPurpose::PlcOperation,
+    )
+    .await
+    {
+        Ok(res) => Ok(res),
+        Err(error) => {
+            tracing::error!("Failed to create plc operation token\n{error}");
+            Err(ApiError::RuntimeError)
+        }
+    }
+}
+
+#[tracing::instrument(skip_all)]
+async fn do_plc_operation(account: &ActorAccount, token: String) -> Result<(), ApiError> {
+    match &account.email {
+        None => {
+            tracing::error!("Failed to find email for account");
+            Err(ApiError::RuntimeError)
+        }
+        Some(email) => match send_plc_operation(email.clone(), TokenParam { token }).await {
+            Ok(_) => {
+                tracing::debug!("Successfully sent PLC Operation Email");
+                Ok(())
+            }
+            Err(error) => {
+                tracing::error!("Failed to send PLC Operation Token Email\n{error}");
+                Err(ApiError::RuntimeError)
+            }
+        },
+    }
+}
+
+#[rocket::post("/xrpc/com.atproto.identity.requestPlcOperationSignature")]
+#[tracing::instrument(skip_all)]
+pub async fn request_plc_operation_signature(auth: AccessFull) -> Result<(), ApiError> {
+    let requester = get_requester_did(&auth).await?;
+    let account = get_account(requester.as_str()).await?;
+    let token = create_email_token(requester.as_str()).await?;
+    do_plc_operation(&account, token).await?;
+
+    Ok(())
+}

rsky-pds/src/main.rs

@@ -246,6 +246,7 @@ async fn rocket() -> _ {
                 com::atproto::identity::update_handle::update_handle,
                 com::atproto::identity::get_recommended_did_credentials::get_recommended_did_credentials,
                 com::atproto::identity::sign_plc_operation::sign_plc_operation,
+                com::atproto::identity::request_plc_operation_signature::request_plc_operation_signature,
                 com::atproto::repo::apply_writes::apply_writes,
                 com::atproto::repo::create_record::create_record,
                 com::atproto::repo::delete_record::delete_record,