add balance api

Author: fupelaqu

common/src/main/scala/app/softnetwork/payment/spi/BalanceApi.scala

@@ -0,0 +1,14 @@
+package app.softnetwork.payment.spi
+
+trait BalanceApi { _: PaymentContext =>
+
+  /** @param currency
+    *   - currency
+    * @param creditedUserId
+    *   - optional credited user id
+    * @return
+    *   balance
+    */
+  def loadBalance(currency: String, creditedUserId: Option[String]): Option[Int]
+
+}

common/src/main/scala/app/softnetwork/payment/spi/PaymentProvider.scala

@@ -16,7 +16,8 @@ private[payment] trait PaymentProvider
     with PayOutApi
     with TransferApi
     with RefundApi
-    with RecurringPaymentApi {
+    with RecurringPaymentApi
+    with BalanceApi {
 
   protected lazy val mlog: Logger = Logger(LoggerFactory.getLogger(getClass.getName))
 

mangopay/src/main/scala/app/softnetwork/payment/spi/MangoPayProvider.scala

@@ -2650,6 +2650,15 @@ trait MangoPayProvider extends PaymentProvider {
     }
   }
 
+  /** @param currency
+   *   - currency
+   * @param creditedUserId
+   *   - optional credited user id
+   * @return
+   *   balance
+   */
+  override def loadBalance(currency: String, creditedUserId: Option[String]): Option[Int] =
+    clientFees().map(fees => (fees * 100).toInt)
 }
 
 class MangoPayProviderFactory extends PaymentProviderSpi {

stripe/src/main/scala/app/softnetwork/payment/spi/StripeBalanceApi.scala

@@ -0,0 +1,53 @@
+package app.softnetwork.payment.spi
+
+import app.softnetwork.payment.config.StripeApi
+import com.google.gson.Gson
+import com.stripe.model.Balance
+import collection.JavaConverters._
+
+trait StripeBalanceApi extends BalanceApi { self: StripeContext =>
+
+  /** @param currency
+    *   - currency
+    * @param creditedUserId
+    *   - optional credited user id
+    * @return
+    *   balance
+    */
+  override def loadBalance(currency: String, creditedUserId: Option[String]): Option[Int] = {
+    var requestOptions = StripeApi().requestOptionsBuilder
+
+    creditedUserId match {
+      case Some(value) =>
+        requestOptions = requestOptions.setStripeAccount(value)
+      case _ =>
+    }
+
+    // load balance
+    val balances: Seq[Balance.Available] =
+      Balance
+        .retrieve(requestOptions.build())
+        .getAvailable
+        .asScala
+
+    val availableAmount =
+      balances.find(
+        _.getCurrency.toLowerCase() == currency.toLowerCase
+      ) match {
+        case Some(balance) =>
+          balance.getAmount.intValue()
+        case None =>
+          mlog.info(
+            s"balances for ${creditedUserId.getOrElse(self.config.clientId)} -> ${new Gson().toJson(balances)}"
+          )
+          0
+      }
+
+    mlog.info(
+      s"balance available amount for ${creditedUserId.getOrElse(self.config.clientId)} is $availableAmount"
+    )
+
+    Option(availableAmount)
+  }
+
+}

stripe/src/main/scala/app/softnetwork/payment/spi/StripePayOutApi.scala

@@ -3,16 +3,15 @@ package app.softnetwork.payment.spi
 import app.softnetwork.payment.config.StripeApi
 import app.softnetwork.payment.model.{PayOutTransaction, Transaction, TransferTransaction}
 import app.softnetwork.serialization.asJson
-import com.stripe.model.Balance
 //import com.stripe.param.PaymentIntentCaptureParams
-import com.google.gson.Gson
 import com.stripe.model.{PaymentIntent, Payout, Transfer}
 import com.stripe.param.PayoutCreateParams
 
 import scala.util.{Failure, Success, Try}
 import collection.JavaConverters._
 
-trait StripePayOutApi extends PayOutApi { _: StripeContext with StripeTransferApi =>
+trait StripePayOutApi extends PayOutApi {
+  _: StripeContext with StripeTransferApi with StripeBalanceApi =>
 
   /** @param maybePayOutTransaction
     *   - pay out transaction
@@ -172,23 +171,9 @@ trait StripePayOutApi extends PayOutApi { _: StripeContext with StripeTransferAp
                 requestOptions = requestOptions.setStripeAccount(payOutTransaction.creditedUserId)
 
                 // load balance
-                val balances: Seq[Balance.Available] =
-                  Balance
-                    .retrieve(requestOptions.build())
-                    .getAvailable
-                    .asScala
-                    .toSeq
-
                 val availableAmount =
-                  balances.find(
-                    _.getCurrency.toLowerCase() == payOutTransaction.currency.toLowerCase
-                  ) match {
-                    case Some(balance) =>
-                      balance.getAmount.intValue()
-                    case None =>
-                      mlog.info(s"balances -> ${new Gson().toJson(balances)}")
-                      0
-                  }
+                  loadBalance(payOutTransaction.currency, Option(payOutTransaction.creditedUserId))
+                    .getOrElse(0)
 
                 mlog.info(
                   s"balance available amount for ${payOutTransaction.creditedUserId} is $availableAmount"
@@ -207,23 +192,7 @@ trait StripePayOutApi extends PayOutApi { _: StripeContext with StripeTransferAp
               } else {
                 // we receive funds from Stripe
                 // load balance
-                val balances: Seq[Balance.Available] =
-                  Balance
-                    .retrieve(requestOptions.build())
-                    .getAvailable
-                    .asScala
-                    .toSeq
-
-                val availableAmount =
-                  balances.find(
-                    _.getCurrency.toLowerCase() == payOutTransaction.currency.toLowerCase
-                  ) match {
-                    case Some(balance) =>
-                      balance.getAmount.intValue()
-                    case None =>
-                      mlog.info(s"balances -> ${new Gson().toJson(balances)}")
-                      0
-                  }
+                val availableAmount = loadBalance(payOutTransaction.currency, None).getOrElse(0)
 
                 mlog.info(s"balance available amount for our stripe account is $availableAmount")
 

stripe/src/main/scala/app/softnetwork/payment/spi/StripeProvider.scala

@@ -36,7 +36,8 @@ trait StripeProvider
     with StripePayOutApi
     with StripeRecurringPaymentApi
     with StripeRefundApi
-    with StripeTransferApi {
+    with StripeTransferApi
+    with StripeBalanceApi {
 
   /** @return
     *   client