Page MenuHomePhorge
Files F32368268

D6788.1765324706.diff
No OneTemporary
Actions

Size
4 KB
Referenced Files
None
Subscribers
None

D6788.1765324706.diff
View Options

diff --git a/services/identity/src/database.rs b/services/identity/src/database.rs
--- a/services/identity/src/database.rs
+++ b/services/identity/src/database.rs
@@ -5,14 +5,14 @@
use aws_sdk_dynamodb::error::GetItemError;
use aws_sdk_dynamodb::model::AttributeValue;
use aws_sdk_dynamodb::output::{
- GetItemOutput, PutItemOutput, QueryOutput, UpdateItemOutput,
+ DeleteItemOutput, GetItemOutput, PutItemOutput, QueryOutput, UpdateItemOutput,
};
use aws_sdk_dynamodb::types::{Blob, SdkError};
use aws_sdk_dynamodb::{Client, Error as DynamoDBError};
use aws_types::sdk_config::SdkConfig;
use chrono::{DateTime, Utc};
use opaque_ke::{errors::ProtocolError, ServerRegistration};
-use tracing::{error, info, warn};
+use tracing::{debug, error, info, warn};
use crate::constants::{
ACCESS_TOKEN_SORT_KEY, ACCESS_TOKEN_TABLE,
@@ -181,6 +181,34 @@
.map_err(|e| Error::AwsSdk(e.into()))
}
+ pub async fn delete_user(
+ &self,
+ user_id: String,
+ ) -> Result<DeleteItemOutput, Error> {
+ debug!("Attempting to delete user: {}", user_id);
+
+ match self
+ .client
+ .delete_item()
+ .table_name(USERS_TABLE)
+ .key(
+ USERS_TABLE_PARTITION_KEY,
+ AttributeValue::S(user_id.clone()),
+ )
+ .send()
+ .await
+ {
+ Ok(out) => {
+ info!("User has been deleted {}", user_id);
+ Ok(out)
+ }
+ Err(e) => {
+ error!("DynamoDB client failed to delete user {}", user_id);
+ Err(Error::AwsSdk(e.into()))
+ }
+ }
+ }
+
pub async fn get_access_token_data(
&self,
user_id: String,
diff --git a/services/identity/src/service.rs b/services/identity/src/service.rs
--- a/services/identity/src/service.rs
+++ b/services/identity/src/service.rs
@@ -25,6 +25,7 @@
pub use proto::identity_service_server::IdentityServiceServer;
use proto::{
+ delete_user_response::DeleteResult as ProtoDeleteResult,
get_user_id_request::AuthType as ProtoAuthType,
identity_service_server::IdentityService,
login_request::Data::PakeLoginRequest,
@@ -39,9 +40,10 @@
registration_request::Data::PakeRegistrationRequestAndUserId,
registration_request::Data::PakeRegistrationUploadAndCredentialRequest,
registration_response::Data::PakeLoginResponse as PakeRegistrationLoginResponse,
- registration_response::Data::PakeRegistrationResponse, GetUserIdRequest,
- GetUserIdResponse, GetUserPublicKeyRequest, GetUserPublicKeyResponse,
- LoginRequest, LoginResponse, PakeLoginRequest as PakeLoginRequestStruct,
+ registration_response::Data::PakeRegistrationResponse, DeleteUserRequest,
+ DeleteUserResponse, GetUserIdRequest, GetUserIdResponse,
+ GetUserPublicKeyRequest, GetUserPublicKeyResponse, LoginRequest,
+ LoginResponse, PakeLoginRequest as PakeLoginRequestStruct,
PakeLoginResponse as PakeLoginResponseStruct, RegistrationRequest,
RegistrationResponse, VerifyUserTokenRequest, VerifyUserTokenResponse,
WalletLoginRequest as WalletLoginRequestStruct,
@@ -396,6 +398,20 @@
let response = Response::new(GetUserPublicKeyResponse { public_key });
Ok(response)
}
+
+ #[instrument(skip(self))]
+ async fn delete_user(
+ &self,
+ request: tonic::Request<DeleteUserRequest>,
+ ) -> Result<tonic::Response<DeleteUserResponse>, tonic::Status> {
+ let message = request.into_inner();
+ match self.client.delete_user(message.user_id).await {
+ Ok(_) => Ok(Response::new(DeleteUserResponse {
+ delete_result: ProtoDeleteResult::Success as i32,
+ })),
+ Err(e) => Err(handle_db_error(e)),
+ }
+ }
}
async fn put_token_helper(
diff --git a/shared/protos/identity.proto b/shared/protos/identity.proto
--- a/shared/protos/identity.proto
+++ b/shared/protos/identity.proto
@@ -18,6 +18,8 @@
// and device ID
rpc GetUserPublicKey(GetUserPublicKeyRequest) returns
(GetUserPublicKeyResponse) {}
+
+ rpc DeleteUser(DeleteUserRequest) returns (DeleteUserResponse) {}
}
// Helper types
@@ -153,3 +155,17 @@
message GetUserPublicKeyResponse {
string publicKey = 1;
}
+
+// DeleteUser
+message DeleteUserRequest {
+ string userID = 1;
+}
+
+message DeleteUserResponse {
+ enum DeleteResult {
+ SUCCESS = 0;
+ FAILURE = 1;
+ }
+
+ DeleteResult deleteResult = 1;
+}

File Metadata

Mime Type
text/plain
Expires
Tue, Dec 9, 11:58 PM (16 h, 13 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
5858955
Default Alt Text
D6788.1765324706.diff (4 KB)

Event Timeline