mirror of
https://github.com/hanzo-fi/payments.git
synced 2026-10-30 20:31:20 +00:00
CrimsonandGitHub
4ae387a5d1
* Use formance-sdk-go/v3 to in connector uninstall e2e test * Compile API docs from openapi spec * Generate go SDK locally for use in integration tests * Get rid of unnecessary redirect in e2e tests now that SDK is generated * Omit generated SDK code from codecoverage checks * Add flake and envrc for nix * Run Dirty and Tests CI steps with nix * Run pre-commit using nix * Remove duplicated steps from Earthfile * Fixes for linter * Add go generate to Justfile pre-commit * Still run Deploy on ubuntu-latest * Address coderabbit feedback
107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package currencycloud
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/formancehq/payments/internal/connectors/plugins/public/currencycloud/client"
|
|
"github.com/formancehq/payments/internal/models"
|
|
)
|
|
|
|
type externalAccountsState struct {
|
|
LastPage int `json:"lastPage"`
|
|
LastCreatedAt time.Time `json:"lastCreatedAt"`
|
|
}
|
|
|
|
func (p *Plugin) fetchNextExternalAccounts(ctx context.Context, req models.FetchNextExternalAccountsRequest) (models.FetchNextExternalAccountsResponse, error) {
|
|
var oldState externalAccountsState
|
|
if req.State != nil {
|
|
if err := json.Unmarshal(req.State, &oldState); err != nil {
|
|
return models.FetchNextExternalAccountsResponse{}, err
|
|
}
|
|
}
|
|
|
|
if oldState.LastPage == 0 {
|
|
oldState.LastPage = 1
|
|
}
|
|
|
|
newState := externalAccountsState{
|
|
LastPage: oldState.LastPage,
|
|
LastCreatedAt: oldState.LastCreatedAt,
|
|
}
|
|
|
|
var accounts []models.PSPAccount
|
|
hasMore := false
|
|
page := oldState.LastPage
|
|
for {
|
|
pagedBeneficiaries, nextPage, err := p.client.GetBeneficiaries(ctx, page, req.PageSize)
|
|
if err != nil {
|
|
return models.FetchNextExternalAccountsResponse{}, err
|
|
}
|
|
|
|
if len(pagedBeneficiaries) == 0 {
|
|
break
|
|
}
|
|
|
|
accounts, err = fillBeneficiaries(accounts, pagedBeneficiaries, oldState)
|
|
if err != nil {
|
|
return models.FetchNextExternalAccountsResponse{}, err
|
|
}
|
|
|
|
var needMore bool
|
|
needMore, hasMore, accounts = shouldFetchMore(accounts, nextPage, req.PageSize)
|
|
if !needMore {
|
|
break
|
|
}
|
|
|
|
page = nextPage
|
|
}
|
|
|
|
newState.LastPage = page
|
|
if len(accounts) > 0 {
|
|
newState.LastCreatedAt = accounts[len(accounts)-1].CreatedAt
|
|
}
|
|
|
|
payload, err := json.Marshal(newState)
|
|
if err != nil {
|
|
return models.FetchNextExternalAccountsResponse{}, err
|
|
}
|
|
|
|
return models.FetchNextExternalAccountsResponse{
|
|
ExternalAccounts: accounts,
|
|
NewState: payload,
|
|
HasMore: hasMore,
|
|
}, nil
|
|
}
|
|
|
|
func fillBeneficiaries(
|
|
accounts []models.PSPAccount,
|
|
pagedBeneficiaries []*client.Beneficiary,
|
|
oldState externalAccountsState,
|
|
) ([]models.PSPAccount, error) {
|
|
for _, beneficiary := range pagedBeneficiaries {
|
|
switch beneficiary.CreatedAt.Compare(oldState.LastCreatedAt) {
|
|
case -1, 0:
|
|
// Account already ingested, skip
|
|
continue
|
|
default:
|
|
}
|
|
|
|
raw, err := json.Marshal(beneficiary)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
accounts = append(accounts, models.PSPAccount{
|
|
Reference: beneficiary.ID,
|
|
CreatedAt: beneficiary.CreatedAt,
|
|
Name: &beneficiary.Name,
|
|
DefaultAsset: &beneficiary.Currency,
|
|
Raw: raw,
|
|
})
|
|
}
|
|
|
|
return accounts, nil
|
|
}
|