fix: incorrect payment update due to not updating wallet

AtomiCloud · Mar 1, 2024 · dca4011 · dca4011
1 parent 878fb40
commit dca4011
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/Domain/Payment/Service.cs b/Domain/Payment/Service.cs
@@ -1,11 +1,13 @@
 using CSharp_Result;
 using Domain.Transaction;
+using Domain.Wallet;
 
 namespace Domain.Payment;
 
 public class PaymentService(
   IPaymentRepository repo,
   IPaymentGateway gateway,
+  IWalletRepository walletRepo,
   ITransactionRepository transactionRepo,
   ITransactionGenerator generator,
   ITransactionManager transactionManager
@@ -49,8 +51,16 @@ public Task<Result<IEnumerable<PaymentPrincipal>>> Search(PaymentSearch search)
   public Task<Result<Payment>> CompleteById(Guid id, PaymentRecord record)
   {
     return transactionManager.Start(() =>
-      repo.UpdateById(id, record)
+      repo
+        // update payment
+        .UpdateById(id, record)
         .NullToError(id.ToString())
+        // update wallet
+        .DoAwait(DoType.MapErrors, w =>
+          walletRepo.Deposit(w.Wallet.Id, w.Principal.Record.CapturedAmount)
+            .NullToError(w.Wallet.Id.ToString())
+        )
+        // update transaction
         .DoAwait(DoType.MapErrors, x =>
           transactionRepo.Create(x.Wallet.Id, generator.Deposit(x.Principal))
             .Then(t => repo.Link(t.Id, x.Principal.Reference.Id), Errors.MapAll)
@@ -61,8 +71,16 @@ public Task<Result<Payment>> CompleteById(Guid id, PaymentRecord record)
   public Task<Result<Payment>> CompleteByRef(string reference, PaymentRecord record)
   {
     return transactionManager.Start(() =>
-      repo.UpdateByRef(reference, record)
+      repo
+        // update payment
+        .UpdateByRef(reference, record)
         .NullToError(reference)
+        // update wallet
+        .DoAwait(DoType.MapErrors, w =>
+          walletRepo.Deposit(w.Wallet.Id, w.Principal.Record.CapturedAmount)
+            .NullToError(w.Wallet.Id.ToString())
+        )
+        // update transaction
         .DoAwait(DoType.MapErrors, x =>
           transactionRepo.Create(x.Wallet.Id, generator.Deposit(x.Principal))
             .Then(t => repo.Link(t.Id, x.Principal.Reference.Id), Errors.MapAll)