feat(transaction): #80 calculate account balances

handler/transaction.go

@@ -37,9 +37,26 @@ func (h TransactionImpl) Handle(r *http.ServeMux) {
 	r.Handle("GET /transaction", h.handleTransactionPage())
 	r.Handle("GET /transaction/{id}", h.handleTransactionItemComp())
 	r.Handle("POST /transaction/{id}", h.handleUpdateTransaction())
+	r.Handle("POST /transaction/recalculate", h.handleRecalculate())
 	r.Handle("DELETE /transaction/{id}", h.handleDeleteTransaction())
 }
 
+func (h TransactionImpl) handleRecalculate() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		user := middleware.GetUser(r)
+		if user == nil {
+			utils.DoRedirect(w, r, "/auth/signin")
+			return
+		}
+
+		err := h.s.RecalculateBalances(user)
+		if err != nil {
+			handleError(w, r, err)
+			return
+		}
+	}
+}
+
 func (h TransactionImpl) handleTransactionPage() http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		user := middleware.GetUser(r)

service/transaction.go

@@ -31,6 +31,8 @@ type Transaction interface {
 	Get(user *types.User, id string) (*types.Transaction, error)
 	GetAll(user *types.User) ([]*types.Transaction, error)
 	Delete(user *types.User, id string) error
+
+	RecalculateBalances(user *types.User) error
 }
 
 type TransactionImpl struct {
@@ -69,6 +71,15 @@ func (s TransactionImpl) Add(user *types.User, transactionInput types.Transactio
 		return nil, err
 	}
 
+	r, err = s.db.Exec(`
+		UPDATE account
+		SET current_balance = current_balance + ?
+		WHERE id = ? AND user_id = ?`, transaction.Value, transaction.AccountId, user.Id)
+	err = db.TransformAndLogDbError("transaction UpdateAccount", r, err)
+	if err != nil {
+		return nil, err
+	}
+
 	return transaction, nil
 }
 
@@ -93,11 +104,33 @@ func (s TransactionImpl) Update(user *types.User, input types.TransactionInput)
 		return nil, types.ErrInternal
 	}
 
+	if transaction.AccountId != nil {
+		r, err := s.db.Exec(`
+			UPDATE account
+			SET current_balance = current_balance - ?
+			WHERE id = ? AND user_id = ?`, transaction.Value, transaction.AccountId, user.Id)
+		err = db.TransformAndLogDbError("transaction UpdateAccount", r, err)
+		if err != nil {
+			return nil, err
+		}
+	}
+
 	transaction, err = s.validateAndEnrichTransaction(transaction, user.Id, input)
 	if err != nil {
 		return nil, err
 	}
 
+	if transaction.AccountId != nil {
+		r, err := s.db.Exec(`
+			UPDATE account
+			SET current_balance = current_balance + ?
+			WHERE id = ? AND user_id = ?`, transaction.Value, transaction.AccountId, user.Id)
+		err = db.TransformAndLogDbError("transaction UpdateAccount", r, err)
+		if err != nil {
+			return nil, err
+		}
+	}
+
 	r, err := s.db.NamedExec(`
 		UPDATE "transaction"
 		SET
@@ -169,7 +202,17 @@ func (s TransactionImpl) Delete(user *types.User, id string) error {
 		return fmt.Errorf("could not parse Id: %w", ErrBadRequest)
 	}
 
-	r, err := s.db.Exec("DELETE FROM \"transaction\" WHERE id = ? AND user_id = ?", uuid, user.Id)
+	r, err := s.db.Exec(`
+		UPDATE account
+		SET current_balance = current_balance - (SELECT value FROM "transaction" WHERE id = ? AND user_id = ?)
+		WHERE id = (select account_id from "transaction" where id = ? AND user_id = ?)
+		`, uuid, user.Id, uuid, user.Id)
+	err = db.TransformAndLogDbError("transaction Delete", r, err)
+	if err != nil {
+		return err
+	}
+
+	r, err = s.db.Exec("DELETE FROM \"transaction\" WHERE id = ? AND user_id = ?", uuid, user.Id)
 	err = db.TransformAndLogDbError("transaction Delete", r, err)
 	if err != nil {
 		return err
@@ -290,3 +333,67 @@ func (s TransactionImpl) validateAndEnrichTransaction(transaction *types.Transac
 		UpdatedBy: &updatedBy,
 	}, nil
 }
+
+func (s TransactionImpl) RecalculateBalances(user *types.User) error {
+	transactionMetric.WithLabelValues("recalculate").Inc()
+	if user == nil {
+		return ErrUnauthorized
+	}
+
+	r, err := s.db.Exec(`
+		UPDATE account
+		SET current_balance = 0
+		WHERE user_id = ?`, user.Id)
+	err = db.TransformAndLogDbError("transaction RecalculateBalances", r, err)
+	if err != nil && err != db.ErrNotFound {
+		return err
+	}
+
+	r, err = s.db.Exec(`
+		UPDATE treasure_chest
+		SET current_balance = 0
+		WHERE user_id = ?`, user.Id)
+	err = db.TransformAndLogDbError("transaction RecalculateBalances", r, err)
+	if err != nil && err != db.ErrNotFound {
+		return err
+	}
+
+	rows, err := s.db.Queryx(`SELECT account_id, treasure_chest_id, value FROM "transaction" WHERE user_id = ?`, user.Id)
+	err = db.TransformAndLogDbError("transaction RecalculateBalances", nil, err)
+	if err != nil && err != db.ErrNotFound {
+		return err
+	}
+	defer rows.Close()
+
+	transaction := &types.Transaction{}
+	for rows.Next() {
+		err = rows.StructScan(transaction)
+		err = db.TransformAndLogDbError("transaction RecalculateBalances", nil, err)
+		if err != nil {
+			return err
+		}
+	}
+
+	if transaction.AccountId != nil {
+		r, err = s.db.Exec(`
+			UPDATE account
+			SET current_balance = current_balance + ?
+			WHERE id = ? AND user_id = ?`, transaction.Value, transaction.AccountId, user.Id)
+		err = db.TransformAndLogDbError("transaction RecalculateBalances", r, err)
+		if err != nil {
+			return err
+		}
+	}
+	if transaction.TreasureChestId != nil {
+		r, err = s.db.Exec(`
+			UPDATE treasure_chest
+			SET current_balance = current_balance + ?
+			WHERE id = ? AND user_id = ?`, transaction.Value, transaction.TreasureChestId, user.Id)
+		err = db.TransformAndLogDbError("transaction RecalculateBalances", r, err)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

template/account/account.templ

@@ -80,7 +80,11 @@ templ AccountItem(account *types.Account) {
 	<div id="account" class="border-1 border-gray-300 w-full my-4 p-4 bg-gray-50 rounded-lg">
 		<div class="text-xl flex justify-end gap-4">
 			<p class="mr-auto">{ account.Name }</p>
-			<p class="mr-20 text-green-700">{ displayBalance(account.CurrentBalance) }</p>
+			if account.CurrentBalance < 0 {
+				<p class="mr-20 text-red-700">{ displayBalance(account.CurrentBalance) }</p>
+			} else {
+				<p class="mr-20 text-green-700">{ displayBalance(account.CurrentBalance) }</p>
+			}
 			<button
 				hx-get={ "/account/" + account.Id.String() + "?edit=true" }
 				hx-target="closest #account"

template/auth/user.templ

@@ -11,16 +11,24 @@ templ UserComp(user string) {
 						<path d="M9.293 12.95l.707.707L15.657 8l-1.414-1.414L10 10.828 5.757 6.586 4.343 8z"></path>
 					</svg>
 				</button>
-				<div class="absolute hidden group-has-hover:block w-full">
+				<div class="absolute hidden group-has-hover:block w-full z-2">
 					<ul class="w-fit float-right mr-4 p-3 border-2 border-gray-200 rounded-lg bg-white shadow-lg">
 						<li class="mb-1">
 							<a class="button w-full px-1 button-neglect block" hx-post="/api/auth/signout" hx-target="#user-info">Sign Out</a>
 						</li>
-						<li class="mb-1">
+						<li class="mb-4">
 							<a class="button w-full px-1 button-neglect block" href="/auth/change-password">Change Password</a>
 						</li>
+						<li class="mb-4">
+							<button
+								hx-post="/transaction/recalculate"
+								hx-swap="none"
+								type="button"
+								class="button text-left w-full px-1 button-neglect block mt-4"
+							>Recalculate</button>
+						</li>
 						<li>
-							<a class="button w-full px-1 button-neglect text-gray-400 mt-4 block" href="/auth/delete-account" class="">
+							<a class="button w-full px-1 button-neglect text-gray-400 block" href="/auth/delete-account" class="">
 								Delete
 								Account
 							</a>

template/treasurechest/treasure_chest.templ

@@ -104,7 +104,11 @@ templ TreasureChestItem(treasureChest *types.TreasureChest) {
 	<div id="treasurechest" class={ "border-1 border-gray-300 w-full p-4 bg-gray-50 rounded-lg" + identation }>
 		<div class="text-xl flex justify-end items-center gap-4">
 			<p class="mr-auto">{ treasureChest.Name }</p>
-			<p class="mr-20 text-green-700">{ displayBalance(treasureChest.CurrentBalance) }</p>
+			if treasureChest.CurrentBalance < 0 {
+				<p class="mr-20 text-red-700">{ displayBalance(treasureChest.CurrentBalance) }</p>
+			} else {
+				<p class="mr-20 text-green-700">{ displayBalance(treasureChest.CurrentBalance) }</p>
+			}
 			<button
 				hx-get={ "/treasurechest/" + treasureChest.Id.String() + "?edit=true" }
 				hx-target="closest #treasurechest"