chore: add update-profile use case

Author: bodymindarts

proto/api/bria.proto

@@ -8,6 +8,7 @@ option go_package = "github.com/GaloyMoney/terraform-provider-bria/client/proto/
 
 service BriaService {
   rpc CreateProfile (CreateProfileRequest) returns (CreateProfileResponse) {}
+  rpc UpdateProfile (UpdateProfileRequest) returns (UpdateProfileResponse) {}
   rpc ListProfiles (ListProfilesRequest) returns (ListProfilesResponse) {}
   rpc CreateProfileApiKey (CreateProfileApiKeyRequest) returns (CreateProfileApiKeyResponse) {}
 
@@ -60,6 +61,13 @@ message CreateProfileResponse {
   string id = 1;
 }
 
+message UpdateProfileRequest {
+  string id = 1;
+  optional SpendingPolicy spending_policy = 2;
+}
+
+message UpdateProfileResponse {}
+
 message CreateProfileApiKeyRequest {
   string profile_name = 1;
 }

src/api/server/convert.rs

@@ -644,6 +644,9 @@ impl From<ApplicationError> for tonic::Status {
             ApplicationError::ProfileError(ProfileError::ProfileNameNotFound(_)) => {
                 tonic::Status::not_found(err.to_string())
             }
+            ApplicationError::ProfileError(ProfileError::ProfileIdNotFound(_)) => {
+                tonic::Status::not_found(err.to_string())
+            }
             ApplicationError::PayoutError(PayoutError::PayoutIdNotFound(_)) => {
                 tonic::Status::not_found(err.to_string())
             }

src/api/server/mod.rs

@@ -57,6 +57,36 @@ impl BriaService for Bria {
         .await
     }
 
+    #[instrument(name = "bria.update_profile", skip_all, fields(error, error.level, error.message), err)]
+    async fn update_profile(
+        &self,
+        request: Request<UpdateProfileRequest>,
+    ) -> Result<Response<UpdateProfileResponse>, Status> {
+        crate::tracing::record_error(|| async move {
+            extract_tracing(&request);
+
+            let key = extract_api_token(&request)?;
+            let profile = self.app.authenticate(key).await?;
+            let request = request.into_inner();
+            let spending_policy = request
+                .spending_policy
+                .map(|policy| profile::SpendingPolicy::try_from((policy, self.app.network())))
+                .transpose()?;
+            self.app
+                .update_profile(
+                    &profile,
+                    request
+                        .id
+                        .parse()
+                        .map_err(ApplicationError::CouldNotParseIncomingUuid)?,
+                    spending_policy,
+                )
+                .await?;
+            Ok(Response::new(UpdateProfileResponse {}))
+        })
+        .await
+    }
+
     #[instrument(name = "bria.list_profiles", skip_all, fields(error, error.level, error.message), err)]
     async fn list_profiles(
         &self,

src/app/mod.rs

@@ -153,6 +153,24 @@ impl App {
         Ok(new_profile)
     }
 
+    #[instrument(name = "app.update_profile", skip(self), err)]
+    pub async fn update_profile(
+        &self,
+        profile: &Profile,
+        profile_id: ProfileId,
+        spending_policy: Option<SpendingPolicy>,
+    ) -> Result<(), ApplicationError> {
+        let mut target_profile = self
+            .profiles
+            .find_by_id(profile.account_id, profile_id)
+            .await?;
+        target_profile.update_spending_policy(spending_policy);
+        let mut tx = self.pool.begin().await?;
+        self.profiles.update(&mut tx, target_profile).await?;
+        tx.commit().await?;
+        Ok(())
+    }
+
     #[instrument(name = "app.list_profiles", skip(self), err)]
     pub async fn list_profiles(&self, profile: &Profile) -> Result<Vec<Profile>, ApplicationError> {
         let profiles = self.profiles.list_for_account(profile.account_id).await?;

src/profile/entity.rs

@@ -16,6 +16,7 @@ pub enum ProfileEvent {
     SpendingPolicyUpdated {
         spending_policy: SpendingPolicy,
     },
+    SpendingPolicyRemoved {},
 }
 
 #[derive(Debug, Builder)]
@@ -24,11 +25,26 @@ pub struct Profile {
     pub id: ProfileId,
     pub account_id: AccountId,
     pub name: String,
-    #[builder(default, setter(strip_option))]
+    #[builder(default)]
     pub spending_policy: Option<SpendingPolicy>,
+
+    pub(super) events: EntityEvents<ProfileEvent>,
 }
 
 impl Profile {
+    pub fn update_spending_policy(&mut self, policy: Option<SpendingPolicy>) {
+        if self.spending_policy != policy {
+            self.spending_policy = policy.clone();
+            if let Some(policy) = policy {
+                self.events.push(ProfileEvent::SpendingPolicyUpdated {
+                    spending_policy: policy,
+                });
+            } else {
+                self.events.push(ProfileEvent::SpendingPolicyRemoved {});
+            }
+        }
+    }
+
     pub fn is_destination_allowed(&self, destination: &PayoutDestination) -> bool {
         self.spending_policy
             .as_ref()
@@ -44,7 +60,7 @@ impl Profile {
     }
 }
 
-#[derive(Debug, Clone, Serialize, Deserialize)]
+#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
 pub struct SpendingPolicy {
     pub allowed_payout_addresses: Vec<Address>,
     pub maximum_payout: Option<Satoshis>,
@@ -109,19 +125,20 @@ impl TryFrom<EntityEvents<ProfileEvent>> for Profile {
 
     fn try_from(events: EntityEvents<ProfileEvent>) -> Result<Self, Self::Error> {
         let mut builder = ProfileBuilder::default();
-        for event in events.into_iter() {
+        for event in events.iter() {
             match event {
                 ProfileEvent::Initialized { id, account_id } => {
-                    builder = builder.id(id).account_id(account_id)
+                    builder = builder.id(*id).account_id(*account_id)
                 }
                 ProfileEvent::NameUpdated { name } => {
-                    builder = builder.name(name);
+                    builder = builder.name(name.clone());
                 }
                 ProfileEvent::SpendingPolicyUpdated { spending_policy } => {
-                    builder = builder.spending_policy(spending_policy);
+                    builder = builder.spending_policy(Some(spending_policy.clone()));
                 }
+                ProfileEvent::SpendingPolicyRemoved {} => builder = builder.spending_policy(None),
             }
         }
-        builder.build()
+        builder.events(events).build()
     }
 }

src/profile/error.rs

@@ -1,11 +1,15 @@
 use thiserror::Error;
 
+use crate::primitives::ProfileId;
+
 #[derive(Error, Debug)]
 pub enum ProfileError {
     #[error("ProfileError - Api key does not exist")]
     ProfileKeyNotFound,
     #[error("ProfileError - Could not find profile with name: {0}")]
     ProfileNameNotFound(String),
+    #[error("ProfileError - Could not find profile with id: {0}")]
+    ProfileIdNotFound(ProfileId),
     #[error("ProfileError - Sqlx: {0}")]
     Sqlx(#[from] sqlx::Error),
     #[error("ProfileError - EntityError: {0}")]

src/profile/repo.rs

@@ -70,6 +70,34 @@ impl Profiles {
         Ok(profiles)
     }
 
+    pub async fn find_by_id(
+        &self,
+        account_id: AccountId,
+        id: ProfileId,
+    ) -> Result<Profile, ProfileError> {
+        let rows = sqlx::query!(
+            r#"SELECT p.id, e.sequence, e.event_type, e.event
+               FROM bria_profiles p
+               JOIN bria_profile_events e ON p.id = e.id
+               WHERE p.account_id = $1 AND p.id = $2
+               ORDER BY p.id, sequence"#,
+            account_id as AccountId,
+            id as ProfileId
+        )
+        .fetch_all(&self.pool)
+        .await?;
+
+        if !rows.is_empty() {
+            let mut events = EntityEvents::new();
+            for row in rows {
+                events.load_event(row.sequence as usize, row.event)?;
+            }
+            Ok(Profile::try_from(events)?)
+        } else {
+            Err(ProfileError::ProfileIdNotFound(id))
+        }
+    }
+
     pub async fn find_by_name(
         &self,
         account_id: AccountId,
@@ -157,4 +185,21 @@ impl Profiles {
             Err(ProfileError::ProfileKeyNotFound)
         }
     }
+
+    pub async fn update(
+        &self,
+        tx: &mut Transaction<'_, Postgres>,
+        profile: Profile,
+    ) -> Result<(), ProfileError> {
+        if !profile.events.is_dirty() {
+            return Ok(());
+        }
+        EntityEvents::<ProfileEvent>::persist(
+            "bria_profile_events",
+            tx,
+            profile.events.new_serialized_events(profile.id),
+        )
+        .await?;
+        Ok(())
+    }
 }

tests/helpers.rs

@@ -35,12 +35,7 @@ pub async fn create_test_account(pool: &sqlx::PgPool) -> anyhow::Result<Profile>
     let app = AdminApp::new(pool.clone(), bitcoin::Network::Regtest);
 
     let profile_key = app.create_account(name.clone()).await?;
-    Ok(Profile {
-        id: profile_key.profile_id,
-        account_id: profile_key.account_id,
-        name,
-        spending_policy: None,
-    })
+    Ok(Profiles::new(pool).find_by_key(&profile_key.key).await?)
 }
 
 pub async fn bitcoind_client() -> anyhow::Result<bitcoincore_rpc::Client> {