From 6074d5892e79f3f65951c540ded7fb75430c947a Mon Sep 17 00:00:00 2001
From: pawelsa <pawelsa@stud.ntnu.no>
Date: Sat, 16 May 2026 19:56:09 +0200
Subject: [PATCH] chore: Update TransacitionComponent based on math changes.

---
 .../elements/TransactionComponent.java        | 28 +++++++++++++------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/main/java/edu/ntnu/idi/idatt/view/components/elements/TransactionComponent.java b/src/main/java/edu/ntnu/idi/idatt/view/components/elements/TransactionComponent.java
index 26fd7d1..4939ebd 100644
--- a/src/main/java/edu/ntnu/idi/idatt/view/components/elements/TransactionComponent.java
+++ b/src/main/java/edu/ntnu/idi/idatt/view/components/elements/TransactionComponent.java
@@ -6,6 +6,7 @@
 import edu.ntnu.idi.idatt.model.transaction.Purchase;
 import edu.ntnu.idi.idatt.model.transaction.Sale;
 import edu.ntnu.idi.idatt.model.transaction.Transaction;
+import edu.ntnu.idi.idatt.service.transaction.PurchaseCalculator;
 import edu.ntnu.idi.idatt.service.transaction.SaleCalculator;
 import edu.ntnu.idi.idatt.service.transaction.TransactionCalculator;
 import edu.ntnu.idi.idatt.view.components.ui.UICompositor;
@@ -20,7 +21,6 @@ public class TransactionComponent extends VBox {
   public TransactionComponent(Transaction transaction) {
 
     Stock stock = transaction.getShare().getStock();
-    TransactionCalculator calculator = transaction.getCalculator();
 
     Label title;
     Label name = new Label(stock.toString());
@@ -30,18 +30,20 @@ public TransactionComponent(Transaction transaction) {
     Label amountOfShares;
     Label saleProfit;
 
-    totalValue = new Label(String.format("Total: %.2f USD", calculator.calculateTotal()));
-
-    taxComissionValue = new Label(String.format(
-        "Tax & Comission: %.2f USD", calculator.calculateTax().add(calculator.calculateCommision())));
-
-    grossValue = new Label(String.format("Gross: %.2f USD", calculator.calculateGross()));
-
     amountOfShares = new Label(
         String.format("Shares: %.2f [%s]", transaction.getShare().getQuantity(), stock.getSymbol()));
 
     if (transaction instanceof Purchase) {
       title = new Label("PURCHASE");
+      PurchaseCalculator pCalc = (PurchaseCalculator) transaction.getCalculator();
+
+      totalValue = new Label(String.format("Total: %.2f USD", pCalc.calculateTotal()));
+
+      taxComissionValue = new Label(String.format(
+          "Tax & Comission: %.2f USD", pCalc.calculateTax().add(pCalc.calculateCommision())));
+
+      grossValue = new Label(String.format("Gross: %.2f USD", pCalc.calculateGross()));
+
       saleProfit = new Label();
       CssUtils.apply(title, CssUtils.GREEN);
 
@@ -50,9 +52,17 @@ public TransactionComponent(Transaction transaction) {
       CssUtils.apply(title, CssUtils.RED);
 
       SaleCalculator sCalc = (SaleCalculator) transaction.getCalculator();
+
+      totalValue = new Label(String.format("Total: %.2f USD", sCalc.calculateTotal()));
+
+      taxComissionValue = new Label(String.format(
+          "Tax & Comission: %.2f USD", sCalc.calculateTax().add(sCalc.calculateCommision())));
+
+      grossValue = new Label(String.format("Gross: %.2f USD", sCalc.calculateGross()));
+
       saleProfit = new Label(String.format("Profit: %.2f USD", sCalc.calculateProfit()));
 
-      CssUtils.apply(saleProfit, CssUtils.generateValueColors(sCalc.calculateProfit().doubleValue()));
+      CssUtils.apply(saleProfit, CssUtils.generateValueColors(sCalc.calculateProfit()));
     }
 
     else {