WIP: Clean-up domain.pure

Ivy-Apps · Apr 24, 2022 · 6c600b7 · 6c600b7
1 parent d784c5d
commit 6c600b7
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 60 deletions.
diff --git a/app/src/main/java/com/ivy/wallet/domain/pure/core/IvyCore.kt b/app/src/main/java/com/ivy/wallet/domain/pure/core/IvyCore.kt
@@ -3,17 +3,64 @@ package com.ivy.wallet.domain.pure.core
 import arrow.core.NonEmptyList
 import com.ivy.fp.Pure
 import com.ivy.wallet.domain.data.core.Transaction
+import com.ivy.wallet.domain.pure.util.mapIndexedNel
+import com.ivy.wallet.domain.pure.util.mapIndexedNelSuspend
+import com.ivy.wallet.domain.pure.util.nonEmptyListOfZeros
 import java.math.BigDecimal
 
 @Pure
 fun <Arg> calcValues(
     transactions: List<Transaction>,
     valueFunctions: NonEmptyList<ValueFunction<Arg>>,
     arg: Arg
+): NonEmptyList<BigDecimal> = calculateValueFunctionsSum(
+    valueFunctionArgument = arg,
+    transactions = transactions,
+    valueFunctions = valueFunctions
+)
+
+
+typealias ValueFunction<A> = (Transaction, A) -> BigDecimal
+typealias SuspendValueFunction<A> = suspend (Transaction, A) -> BigDecimal
+
+@Pure
+internal tailrec fun <A> calculateValueFunctionsSum(
+    valueFunctionArgument: A,
+    transactions: List<Transaction>,
+    valueFunctions: NonEmptyList<ValueFunction<A>>,
+    sum: NonEmptyList<BigDecimal> = nonEmptyListOfZeros(n = valueFunctions.size)
+): NonEmptyList<BigDecimal> {
+    return if (transactions.isEmpty())
+        sum
+    else
+        calculateValueFunctionsSum(
+            valueFunctionArgument = valueFunctionArgument,
+            transactions = transactions.drop(1),
+            valueFunctions = valueFunctions,
+            sum = sum.mapIndexedNel { index, sumValue ->
+                val valueFunction = valueFunctions[index]
+                sumValue + valueFunction(transactions.first(), valueFunctionArgument)
+            }
+        )
+}
+
+@Pure
+internal tailrec suspend fun <A> calculateValueFunctionsSumSuspend(
+    valueFunctionArgument: A,
+    transactions: List<Transaction>,
+    valueFunctions: NonEmptyList<SuspendValueFunction<A>>,
+    sum: NonEmptyList<BigDecimal> = nonEmptyListOfZeros(n = valueFunctions.size)
 ): NonEmptyList<BigDecimal> {
-    return calculateValueFunctionsSum(
-        valueFunctionArgument = arg,
-        transactions = transactions,
-        valueFunctions = valueFunctions
-    )
+    return if (transactions.isEmpty())
+        sum
+    else
+        calculateValueFunctionsSumSuspend(
+            valueFunctionArgument = valueFunctionArgument,
+            transactions = transactions.drop(1),
+            valueFunctions = valueFunctions,
+            sum = sum.mapIndexedNelSuspend { index, sumValue ->
+                val valueFunction = valueFunctions[index]
+                sumValue + valueFunction(transactions.first(), valueFunctionArgument)
+            }
+        )
 }
diff --git a/app/src/main/java/com/ivy/wallet/domain/pure/core/ValueFunction.kt b/app/src/main/java/com/ivy/wallet/domain/pure/core/ValueFunction.kt
diff --git a/app/src/main/java/com/ivy/wallet/domain/pure/data/CurrencyConvError.kt b/app/src/main/java/com/ivy/wallet/domain/pure/data/CurrencyConvError.kt
diff --git a/.../java/com/ivy/wallet/domain/pure/Utils.kt → .../com/ivy/wallet/domain/pure/util/Utils.kt b/.../java/com/ivy/wallet/domain/pure/Utils.kt → .../com/ivy/wallet/domain/pure/util/Utils.kt
@@ -1,4 +1,4 @@
-package com.ivy.wallet.domain.pure
+package com.ivy.wallet.domain.pure.util
 
 import arrow.core.NonEmptyList
 import java.math.BigDecimal