fix: manual adjustments did not work

- latest balance was not fetched properly

Author: nootey

internal/repositories/account_repository.go

@@ -846,3 +846,32 @@ func (r *AccountRepository) DeleteAccountSnapshots(tx *gorm.DB, accountID int64)
 func (r *AccountRepository) DeleteAllBalancesForAccount(tx *gorm.DB, accountID int64) error {
 	return tx.Where("account_id = ?", accountID).Delete(&models.Balance{}).Error
 }
+
+func (r *AccountRepository) GetLatestBalance(tx *gorm.DB, accountID int64) (decimal.Decimal, error) {
+	var endBalance sql.NullString
+
+	err := tx.
+		Raw(`
+          SELECT end_balance::text
+          FROM balances
+          WHERE account_id = ?
+          ORDER BY as_of DESC
+          LIMIT 1
+       `, accountID).
+		Scan(&endBalance).Error
+	if err != nil {
+		return decimal.Zero, err
+	}
+
+	if !endBalance.Valid {
+		// no row found - treat as 0
+		return decimal.Zero, nil
+	}
+
+	d, err := decimal.NewFromString(endBalance.String)
+	if err != nil {
+		return decimal.Zero, err
+	}
+
+	return d, nil
+}

internal/services/account_service.go

@@ -423,9 +423,7 @@ func (s *AccountService) UpdateAccount(userID int64, id int64, req *models.Accou
 			return fmt.Errorf("invalid balance value: %w", err)
 		}
 
-		// Current end balance from snapshot
-		asOf := time.Now().UTC()
-		current, err := utils.GetEndBalanceAsOf(tx, exAcc.ID, asOf)
+		current, err := s.Repo.GetLatestBalance(tx, exAcc.ID)
 		if err != nil {
 			tx.Rollback()
 			return err

internal/services/statistics_service.go

@@ -274,7 +274,6 @@ func (s *StatisticsService) GetCurrentMonthStats(userID int64, accountID *int64)
 		if tr.TransactionInflow.Account.AccountType.Classification == "liability" {
 			debtRepaymentTotal = debtRepaymentTotal.Add(tr.Amount)
 		}
-		fmt.Println(debtRepaymentTotal)
 
 		takeHome = takeHome.Sub(tr.Amount)
 	}

pkg/utils/ledger_helper.go

@@ -1,42 +1,11 @@
 package utils
 
 import (
-	"database/sql"
 	"fmt"
 	"time"
 	"wealth-warden/internal/models"
-
-	"github.com/shopspring/decimal"
-	"gorm.io/gorm"
 )
 
-func GetEndBalanceAsOf(tx *gorm.DB, accountID int64, t time.Time) (decimal.Decimal, error) {
-	var endBalance sql.NullString
-
-	err := tx.
-		Raw(`
-			SELECT end_balance::text
-			FROM balances
-			WHERE account_id = ? AND as_of = ?
-		`, accountID, t.UTC().Format("2006-01-02")).
-		Scan(&endBalance).Error
-	if err != nil {
-		return decimal.Zero, err
-	}
-
-	if !endBalance.Valid {
-		// no row found → treat as 0
-		return decimal.Zero, nil
-	}
-
-	d, err := decimal.NewFromString(endBalance.String)
-	if err != nil {
-		return decimal.Zero, err
-	}
-
-	return d, nil
-}
-
 func ValidateAccount(acc *models.Account, role string) error {
 	if acc.ClosedAt != nil {
 		return fmt.Errorf("%s account (ID=%d) is closed and cannot be used", role, acc.ID)