Improve transaction investment

3 years ago · 1a2243cfb7
1 changed files with 6 additions and 3 deletions
--- a/apps/api/src/app/portfolio/portfolio-calculator.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator.ts
@ -1027,9 +1027,12 @@ export class PortfolioCalculator {
        valueAtStartDate = valueOfInvestmentBeforeTransaction;
      }

-      const transactionInvestment = order.quantity
-        .mul(order.unitPrice)
-        .mul(this.getFactor(order.type));
+      const transactionInvestment =
+        order.type === 'BUY'
+          ? order.quantity.mul(order.unitPrice).mul(this.getFactor(order.type))
+          : order.quantity
+              .mul(totalInvestment.div(totalUnits))
+              .mul(this.getFactor(order.type));

      totalInvestment = totalInvestment.plus(transactionInvestment);