payments/internal/connectors/plugins/public/generic/balances.go

package generic

import (
	"context"
	"encoding/json"
	"fmt"
	"math/big"

	"github.com/formancehq/go-libs/v3/currency"
	"github.com/formancehq/payments/internal/models"
	errorsutils "github.com/formancehq/payments/internal/utils/errors"
)

func (p *Plugin) fetchNextBalances(ctx context.Context, req models.FetchNextBalancesRequest) (models.FetchNextBalancesResponse, error) {
	var from models.PSPAccount
	if req.FromPayload == nil {
		return models.FetchNextBalancesResponse{}, errorsutils.NewWrappedError(
			fmt.Errorf("from payload is required"),
			models.ErrInvalidRequest,
		)
	}
	if err := json.Unmarshal(req.FromPayload, &from); err != nil {
		return models.FetchNextBalancesResponse{}, err
	}

	balances, err := p.client.GetBalances(ctx, from.Reference)
	if err != nil {
		return models.FetchNextBalancesResponse{}, err
	}

	var res []models.PSPBalance
	for _, balance := range balances.Balances {
		var amount big.Int
		_, ok := amount.SetString(balance.Amount, 10)
		if !ok {
			return models.FetchNextBalancesResponse{}, fmt.Errorf("failed to parse amount: %s", balance.Amount)
		}

		res = append(res, models.PSPBalance{
			AccountReference: balances.AccountID,
			CreatedAt:        balances.At,
			Amount:           &amount,
			Asset:            currency.FormatAsset(supportedCurrenciesWithDecimal, balance.Currency),
		})
	}

	return models.FetchNextBalancesResponse{
		Balances: res,
		HasMore:  false,
	}, nil
}