Scrum 43 Plaid Frontend

backend/internal/controllers/investors.go

@@ -106,6 +106,25 @@ func (c *InvestorsController) GetPortfolio(ctx *fiber.Ctx) error {
 	return ctx.JSON(investors)
 }
 
+func (c *InvestorsController) GetCashBalance(ctx *fiber.Ctx) error {
+	userId, ok := ctx.Locals("userId").(string)
+	if !ok {
+		return &api_errors.INVALID_UUID
+	}
+
+	id, err := uuid.Parse(userId)
+	if err != nil {
+		return &api_errors.INVALID_UUID
+	}
+
+	investors, err := transactions.GetCashBalance(c.ServiceParams.DB, id)
+	if err != nil {
+		return err
+	}
+
+	return ctx.JSON(investors)
+}
+
 func (c *InvestorsController) GetHistory(ctx *fiber.Ctx) error {
 	paginationParams := new(types.PaginationParams)
 
@@ -165,11 +184,17 @@ func (c *InvestorsController) GetInvestor(ctx *fiber.Ctx) error {
 		return err
 	}
 
+	cashBalance, err := transactions.GetCashBalance(c.ServiceParams.DB, id)
+	if err != nil {
+		return err
+	}
+
 	investor := models.Investor{
 		ID:                    id,
 		Profile:               profile,
 		TotalInvestmentAmount: totalValue,
 		InvestmentBreakdown:   investments,
+		CashBalance:           cashBalance,
 	}
 
 	return ctx.Status(fiber.StatusOK).JSON(investor)

backend/internal/models/investor.go

@@ -7,6 +7,7 @@ type Investor struct {
 	Profile               InvestorProfile   `json:"profile"`
 	TotalInvestmentAmount int               `json:"total_investment_amount"`
 	InvestmentBreakdown   map[uuid.UUID]int `json:"investment_breakdown"`
+	CashBalance           int               `json:"cash_balance"`
 }
 
 type InvestorProfile struct {

backend/internal/transactions/investors.go

@@ -222,7 +222,7 @@ func GetHistory(db *pgxpool.Pool, investorID uuid.UUID, limit int, offset int) (
 	return history, nil
 }
 
-func GetCashBalance(pool *pgxpool.Pool, investorID string) (int, error) {
+func GetCashBalance(pool *pgxpool.Pool, investorID uuid.UUID) (int, error) {
 	query := "SELECT cash_balance_cents FROM investors WHERE supabase_id = $1"
 
 	var balance int