payments/internal/storage/payment_initiations.go

package storage

import (
	"context"
	"database/sql"
	"encoding/json"
	"fmt"
	"math/big"
	stdtime "time"

	"github.com/hanzo-fi/go-libs/v5/pkg/query"
	"github.com/hanzo-fi/go-libs/v5/pkg/storage/bun/paginate"
	"github.com/hanzo-fi/go-libs/v5/pkg/types/pointer"
	"github.com/hanzo-fi/go-libs/v5/pkg/types/time"
	internalEvents "github.com/hanzo-fi/payments/internal/events"
	"github.com/hanzo-fi/payments/pkg/domain/models"
	"github.com/hanzo-fi/payments/pkg/events"
	"github.com/pkg/errors"
	"github.com/uptrace/bun"
)

type paymentInitiation struct {
	bun.BaseModel `bun:"payment_initiations"`

	// Mandatory fields
	ID          models.PaymentInitiationID   `bun:"id,pk,type:character varying,notnull"`
	ConnectorID models.ConnectorID           `bun:"connector_id,type:character varying,notnull"`
	Reference   string                       `bun:"reference,type:text,notnull"`
	CreatedAt   time.Time                    `bun:"created_at,type:timestamp without time zone,notnull"`
	ScheduledAt time.Time                    `bun:"scheduled_at,type:timestamp without time zone,notnull"`
	Description string                       `bun:"description,type:text,notnull"`
	Type        models.PaymentInitiationType `bun:"type,type:text,notnull"`
	Amount      *big.Int                     `bun:"amount,type:numeric,notnull"`
	Asset       string                       `bun:"asset,type:text,notnull"`

	// Optional fields
	SourceAccountID      *models.AccountID `bun:"source_account_id,type:character varying"`
	DestinationAccountID *models.AccountID `bun:"destination_account_id,type:character varying,notnull"`

	// Optional fields with default
	// c.f. https://bun.uptrace.dev/guide/models.html#default
	Metadata map[string]string `bun:"metadata,type:jsonb,nullzero,notnull,default:'{}'"`
}

type paymentInitiationRelatedPayment struct {
	bun.BaseModel `bun:"payment_initiation_related_payments"`

	// Mandatory fields
	PaymentInitiationID models.PaymentInitiationID `bun:"payment_initiation_id,pk,type:character varying,notnull"`
	PaymentID           models.PaymentID           `bun:"payment_id,pk,type:character varying,notnull"`
	CreatedAt           time.Time                  `bun:"created_at,type:timestamp without time zone,notnull"`
}

type paymentInitiationAdjustment struct {
	bun.BaseModel `bun:"payment_initiation_adjustments"`

	// Mandatory fields
	ID                  models.PaymentInitiationAdjustmentID     `bun:"id,pk,type:character varying,notnull"`
	PaymentInitiationID models.PaymentInitiationID               `bun:"payment_initiation_id,type:character varying,notnull"`
	CreatedAt           time.Time                                `bun:"created_at,type:timestamp without time zone,notnull"`
	Status              models.PaymentInitiationAdjustmentStatus `bun:"status,type:text,notnull"`

	// Optional fields
	Error  *string  `bun:"error,type:text"`
	Amount *big.Int `bun:"amount,type:numeric"`
	Asset  *string  `bun:"asset,type:text"`

	// Optional fields with default
	// c.f. https://bun.uptrace.dev/guide/models.html#default
	Metadata map[string]string `bun:"metadata,type:jsonb,nullzero,notnull,default:'{}'"`
}

func (s *store) PaymentInitiationsInsert(ctx context.Context, pi models.PaymentInitiation, adjustments ...models.PaymentInitiationAdjustment) (err error) {
	var tx bun.Tx
	tx, err = s.db.BeginTx(ctx, nil)
	if err != nil {
		return e("upsert payment initiations", err)
	}
	defer func() {
		rollbackOnTxError(ctx, &tx, err)
	}()

	toInsert := fromPaymentInitiationModels(pi)
	adjustmentsToInsert := make([]paymentInitiationAdjustment, 0, len(adjustments))
	for _, adj := range adjustments {
		adjustmentsToInsert = append(adjustmentsToInsert, fromPaymentInitiationAdjustmentModels(adj))
	}

	_, err = tx.NewInsert().
		Model(&toInsert).
		Exec(ctx)
	if err != nil {
		return e("failed to insert payment initiations", err)
	}

	// Create outbox event for payment initiation
	outboxEvents := make([]models.OutboxEvent, 0, 1+len(adjustments))

	payload := internalEvents.PaymentInitiationMessagePayload{
		ID:          pi.ID.String(),
		ConnectorID: pi.ConnectorID.String(),
		Provider:    models.ToV3Provider(pi.ConnectorID.Provider),
		Reference:   pi.Reference,
		CreatedAt:   pi.CreatedAt,
		ScheduledAt: pi.ScheduledAt,
		Description: pi.Description,
		Type:        pi.Type.String(),
		Amount:      pi.Amount,
		Asset:       pi.Asset,
		Metadata:    pi.Metadata,
	}
	if pi.SourceAccountID != nil {
		payload.SourceAccountID = pi.SourceAccountID.String()
	}
	if pi.DestinationAccountID != nil {
		payload.DestinationAccountID = pi.DestinationAccountID.String()
	}

	var payloadBytes []byte
	payloadBytes, err = json.Marshal(&payload)
	if err != nil {
		return e("failed to marshal payment initiation event payload", err)
	}

	outboxEvents = append(outboxEvents, models.OutboxEvent{
		ID: models.EventID{
			EventIdempotencyKey: pi.IdempotencyKey(),
			ConnectorID:         &pi.ConnectorID,
		},
		EventType:   events.EventTypeSavedPaymentInitiation,
		EntityID:    pi.ID.String(),
		Payload:     payloadBytes,
		CreatedAt:   stdtime.Now().UTC(),
		Status:      models.OUTBOX_STATUS_PENDING,
		ConnectorID: &pi.ConnectorID,
	})

	if len(adjustmentsToInsert) > 0 {
		err = tx.NewInsert().
			Model(&adjustmentsToInsert).
			On("CONFLICT (id) DO NOTHING").
			Returning("*").
			Scan(ctx, &adjustmentsToInsert)
		if err != nil {
			return e("failed to insert payment initiation adjustments", err)
		}

		// Create outbox events for each inserted adjustment
		for _, adj := range adjustmentsToInsert {
			adjModel := toPaymentInitiationAdjustmentModels(adj)
			adjPayload := internalEvents.PaymentInitiationAdjustmentMessagePayload{
				ID:                  adjModel.ID.String(),
				PaymentInitiationID: adjModel.ID.PaymentInitiationID.String(),
				Status:              adjModel.Status.String(),
				Amount:              adjModel.Amount,
				Asset:               adjModel.Asset,
				Metadata:            adjModel.Metadata,
			}
			if adjModel.Error != nil {
				errorStr := adjModel.Error.Error()
				adjPayload.Error = &errorStr
			}

			var adjPayloadBytes []byte
			adjPayloadBytes, err = json.Marshal(&adjPayload)
			if err != nil {
				return e("failed to marshal payment initiation adjustment event payload", err)
			}

			outboxEvents = append(outboxEvents, models.OutboxEvent{
				ID: models.EventID{
					EventIdempotencyKey: adjModel.IdempotencyKey(),
					ConnectorID:         &adjModel.ID.PaymentInitiationID.ConnectorID,
				},
				EventType:   events.EventTypeSavedPaymentInitiationAdjustment,
				EntityID:    adjModel.ID.String(),
				Payload:     adjPayloadBytes,
				CreatedAt:   stdtime.Now().UTC(),
				Status:      models.OUTBOX_STATUS_PENDING,
				ConnectorID: &adjModel.ID.PaymentInitiationID.ConnectorID,
			})
		}
	}

	// Insert outbox events
	if len(outboxEvents) > 0 {
		if err = s.OutboxEventsInsert(ctx, tx, outboxEvents); err != nil {
			return err
		}
	}

	err = tx.Commit()
	if err != nil {
		return e("failed to commit transaction", err)
	}
	return nil
}

func (s *store) PaymentInitiationsUpdateMetadata(ctx context.Context, piID models.PaymentInitiationID, metadata map[string]string) error {
	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return e("update payment metadata", err)
	}
	defer func() {
		rollbackOnTxError(ctx, &tx, err)
	}()

	var pi paymentInitiation
	err = tx.NewSelect().
		Model(&pi).
		Column("id", "metadata").
		Where("id = ?", piID).
		Scan(ctx)
	if err != nil {
		return e("update payment initiation metadata", err)
	}

	if pi.Metadata == nil {
		pi.Metadata = make(map[string]string)
	}

	for k, v := range metadata {
		pi.Metadata[k] = v
	}

	_, err = tx.NewUpdate().
		Model(&pi).
		Column("metadata").
		Where("id = ?", piID).
		Exec(ctx)
	if err != nil {
		return e("update payment initiation metadata", err)
	}

	err = tx.Commit()
	if err != nil {
		return e("failed to commit transaction", err)
	}
	return nil
}

func (s *store) PaymentInitiationsGet(ctx context.Context, piID models.PaymentInitiationID) (*models.PaymentInitiation, error) {
	var pi paymentInitiation
	err := s.db.NewSelect().
		Model(&pi).
		Where("id = ?", piID).
		Scan(ctx)
	if err != nil {
		return nil, e("failed to get payment initiation", err)
	}

	res := toPaymentInitiationModels(pi)
	return &res, nil
}

func (s *store) PaymentInitiationsDeleteFromConnectorID(ctx context.Context, connectorID models.ConnectorID) error {
	_, err := s.db.NewDelete().
		Model((*paymentInitiation)(nil)).
		Where("connector_id = ?", connectorID).
		Exec(ctx)
	return e("failed to delete payment initiations", err)
}

func (s *store) PaymentInitiationsDelete(ctx context.Context, piID models.PaymentInitiationID) error {
	_, err := s.db.NewDelete().
		Model((*paymentInitiation)(nil)).
		Where("id = ?", piID).
		Exec(ctx)
	return e("failed to delete payment initiation", err)
}

type PaymentInitiationQuery struct{}

type ListPaymentInitiationsQuery paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationQuery]]

func NewListPaymentInitiationsQuery(opts paginate.PaginatedQueryOptions[PaymentInitiationQuery]) ListPaymentInitiationsQuery {
	return ListPaymentInitiationsQuery{
		Order:    paginate.OrderAsc,
		PageSize: opts.PageSize,
		Options:  opts,
	}
}

func (s *store) paymentsInitiationQueryContext(qb query.Builder) (string, string, []any, error) {
	join := ""
	where, args, err := qb.Build(query.ContextFn(func(key, operator string, value any) (string, []any, error) {
		switch {
		case key == "reference",
			key == "id",
			key == "connector_id",
			key == "type",
			key == "asset",
			key == "source_account_id",
			key == "destination_account_id":
			if operator != "$match" {
				return "", nil, e(fmt.Sprintf("'%s' column can only be used with $match", key), ErrValidation)
			}
			return fmt.Sprintf("%s = ?", key), []any{value}, nil

		case key == "status":
			if operator != "$match" {
				return "", nil, e(fmt.Sprintf("'%s' column can only be used with $match", key), ErrValidation)
			}

			// we only care about the latest adjustment, so we need to sort the adjustments
			join = `JOIN payment_initiation_adjustments AS current_adj
ON (current_adj.payment_initiation_id = payment_initiation.id)
LEFT OUTER JOIN payment_initiation_adjustments newer_adj
ON (newer_adj.payment_initiation_id = payment_initiation.id AND current_adj.sort_id < newer_adj.sort_id)`

			return fmt.Sprintf("current_adj.%s = ? AND newer_adj.id IS NULL", key), []any{value}, nil
		case key == "amount":
			return fmt.Sprintf("%s %s ?", key, query.DefaultComparisonOperatorsMapping[operator]), []any{value}, nil
		case metadataRegex.Match([]byte(key)):
			if operator != "$match" {
				return "", nil, e(fmt.Sprintf("'%s' column can only be used with $match", key), ErrValidation)
			}
			match := metadataRegex.FindAllStringSubmatch(key, 3)

			key := "metadata"
			return key + " @> ?", []any{map[string]any{
				match[0][1]: value,
			}}, nil
		}
		return "", nil, e(fmt.Sprintf("unknown key '%s' when building query", key), ErrValidation)
	}))

	return join, where, args, err
}

func (s *store) PaymentInitiationsList(ctx context.Context, q ListPaymentInitiationsQuery) (*paginate.Cursor[models.PaymentInitiation], error) {
	var (
		join  string
		where string
		args  []any
		err   error
	)
	if q.Options.QueryBuilder != nil {
		join, where, args, err = s.paymentsInitiationQueryContext(q.Options.QueryBuilder)
		if err != nil {
			return nil, err
		}
	}

	cursor, err := paginateWithOffset[paginate.PaginatedQueryOptions[PaymentInitiationQuery], paymentInitiation](s, ctx,
		(*paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationQuery]])(&q),
		func(query *bun.SelectQuery) *bun.SelectQuery {
			if join != "" {
				query = query.Join(join)
			}

			if where != "" {
				query = query.Where(where, args...)
			}

			// TODO(polo): sorter ?
			query = query.Order("payment_initiation.created_at DESC", "payment_initiation.sort_id DESC")

			return query
		},
	)
	if err != nil {
		return nil, e("failed to fetch payment initiations", err)
	}

	pis := make([]models.PaymentInitiation, 0, len(cursor.Data))
	for _, pi := range cursor.Data {
		pis = append(pis, toPaymentInitiationModels(pi))
	}

	return &paginate.Cursor[models.PaymentInitiation]{
		PageSize: cursor.PageSize,
		HasMore:  cursor.HasMore,
		Previous: cursor.Previous,
		Next:     cursor.Next,
		Data:     pis,
	}, nil
}

func (s *store) PaymentInitiationRelatedPaymentsUpsert(ctx context.Context, piID models.PaymentInitiationID, pID models.PaymentID, createdAt stdtime.Time) error {
	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return e("failed to begin transaction", err)
	}
	defer func() {
		rollbackOnTxError(ctx, &tx, err)
	}()

	toInsert := paymentInitiationRelatedPayment{
		PaymentInitiationID: piID,
		PaymentID:           pID,
		CreatedAt:           time.New(createdAt),
	}

	var res sql.Result
	res, err = tx.NewInsert().
		Model(&toInsert).
		On("CONFLICT (payment_initiation_id, payment_id) DO NOTHING").
		Exec(ctx)
	if err != nil {
		return e("failed to insert payment initiation related payments", err)
	}

	var rowsAffected int64
	rowsAffected, err = res.RowsAffected()
	if err != nil {
		return e("failed to get rows affected", err)
	}

	// Create outbox event only if related payment was actually inserted
	if rowsAffected > 0 {
		relatedPayment := models.PaymentInitiationRelatedPayments{
			PaymentInitiationID: piID,
			PaymentID:           pID,
		}

		payload := internalEvents.PaymentInitiationRelatedPaymentMessagePayload{
			PaymentInitiationID: piID.String(),
			PaymentID:           pID.String(),
		}

		var payloadBytes []byte
		payloadBytes, err = json.Marshal(&payload)
		if err != nil {
			return e("failed to marshal payment initiation related payment event payload", err)
		}

		outboxEvent := models.OutboxEvent{
			ID: models.EventID{
				EventIdempotencyKey: relatedPayment.IdempotencyKey(),
				ConnectorID:         &piID.ConnectorID,
			},
			EventType:   events.EventTypeSavedPaymentInitiationRelatedPayment,
			EntityID:    fmt.Sprintf("%s:%s", piID.String(), pID.String()),
			Payload:     payloadBytes,
			CreatedAt:   stdtime.Now().UTC(),
			Status:      models.OUTBOX_STATUS_PENDING,
			ConnectorID: &piID.ConnectorID,
		}

		if err = s.OutboxEventsInsert(ctx, tx, []models.OutboxEvent{outboxEvent}); err != nil {
			return err
		}
	}

	if err = tx.Commit(); err != nil {
		return e("failed to commit transaction", err)
	}
	return nil
}

func (s *store) PaymentInitiationIDsListFromPaymentID(ctx context.Context, id models.PaymentID) ([]models.PaymentInitiationID, error) {
	var paymentInitiationRelatedPayments []paymentInitiationRelatedPayment
	err := s.db.NewSelect().
		Model(&paymentInitiationRelatedPayments).
		Column("payment_initiation_id").
		Where("payment_id = ?", id).
		Scan(ctx)
	if err != nil {
		return nil, e("failed to get payment initiation related payments", err)
	}

	ids := make([]models.PaymentInitiationID, 0, len(paymentInitiationRelatedPayments))
	for _, pi := range paymentInitiationRelatedPayments {
		ids = append(ids, pi.PaymentInitiationID)
	}

	return ids, nil
}

type PaymentInitiationRelatedPaymentsQuery struct{}

type ListPaymentInitiationRelatedPaymentsQuery paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationRelatedPaymentsQuery]]

func NewListPaymentInitiationRelatedPaymentsQuery(opts paginate.PaginatedQueryOptions[PaymentInitiationRelatedPaymentsQuery]) ListPaymentInitiationRelatedPaymentsQuery {
	return ListPaymentInitiationRelatedPaymentsQuery{
		Order:    paginate.OrderAsc,
		PageSize: opts.PageSize,
		Options:  opts,
	}
}

func (s *store) PaymentInitiationRelatedPaymentsList(ctx context.Context, piID models.PaymentInitiationID, q ListPaymentInitiationRelatedPaymentsQuery) (*paginate.Cursor[models.Payment], error) {
	cursor, err := paginateWithOffset[paginate.PaginatedQueryOptions[PaymentInitiationRelatedPaymentsQuery], paymentInitiationRelatedPayment](s, ctx,
		(*paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationRelatedPaymentsQuery]])(&q),
		func(query *bun.SelectQuery) *bun.SelectQuery {
			// TODO(polo): sorter ?
			query = query.Order("created_at DESC", "sort_id DESC")
			query.Where("payment_initiation_id = ?", piID)

			return query
		},
	)
	if err != nil {
		return nil, e("failed to fetch accounts", err)
	}

	pis := make([]models.Payment, 0, len(cursor.Data))
	for _, pi := range cursor.Data {
		p, err := s.PaymentsGet(ctx, pi.PaymentID)
		if err != nil {
			return nil, e("failed to get payment", err)
		}

		pis = append(pis, *p)
	}

	return &paginate.Cursor[models.Payment]{
		PageSize: cursor.PageSize,
		HasMore:  cursor.HasMore,
		Previous: cursor.Previous,
		Next:     cursor.Next,
		Data:     pis,
	}, nil
}

func (s *store) PaymentInitiationAdjustmentsUpsert(ctx context.Context, adj models.PaymentInitiationAdjustment) error {
	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return e("failed to begin transaction", err)
	}
	defer func() {
		rollbackOnTxError(ctx, &tx, err)
	}()

	toInsert := fromPaymentInitiationAdjustmentModels(adj)

	var res sql.Result
	res, err = tx.NewInsert().
		Model(&toInsert).
		On("CONFLICT (id) DO NOTHING").
		Exec(ctx)
	if err != nil {
		return e("failed to insert payment initiation adjustments", err)
	}

	rowsAffected, err := res.RowsAffected()
	if err != nil {
		return e("failed to get rows affected", err)
	}

	// Create outbox event only if adjustment was actually inserted
	if rowsAffected > 0 {
		adjPayload := internalEvents.PaymentInitiationAdjustmentMessagePayload{
			ID:                  adj.ID.String(),
			PaymentInitiationID: adj.ID.PaymentInitiationID.String(),
			Status:              adj.Status.String(),
			Amount:              adj.Amount,
			Asset:               adj.Asset,
			Metadata:            adj.Metadata,
		}
		if adj.Error != nil {
			errorStr := adj.Error.Error()
			adjPayload.Error = &errorStr
		}

		var adjPayloadBytes []byte
		adjPayloadBytes, err = json.Marshal(&adjPayload)
		if err != nil {
			return e("failed to marshal payment initiation adjustment event payload", err)
		}

		outboxEvent := models.OutboxEvent{
			ID: models.EventID{
				EventIdempotencyKey: adj.IdempotencyKey(),
				ConnectorID:         &adj.ID.PaymentInitiationID.ConnectorID,
			},
			EventType:   events.EventTypeSavedPaymentInitiationAdjustment,
			EntityID:    adj.ID.String(),
			Payload:     adjPayloadBytes,
			CreatedAt:   stdtime.Now().UTC(),
			Status:      models.OUTBOX_STATUS_PENDING,
			ConnectorID: &adj.ID.PaymentInitiationID.ConnectorID,
		}

		if err = s.OutboxEventsInsert(ctx, tx, []models.OutboxEvent{outboxEvent}); err != nil {
			return err
		}
	}

	if err = tx.Commit(); err != nil {
		return e("failed to commit transaction", err)
	}
	return nil
}

func (s *store) PaymentInitiationAdjustmentsUpsertIfPredicate(
	ctx context.Context,
	adj models.PaymentInitiationAdjustment,
	predicate func(models.PaymentInitiationAdjustment) bool,
) (bool, error) {
	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return false, e("upsert payment initiations", err)
	}
	defer func() {
		rollbackOnTxError(ctx, &tx, err)
	}()

	var previousAdj paymentInitiationAdjustment
	err = tx.NewSelect().
		Model(&previousAdj).
		Where("payment_initiation_id = ?", adj.ID.PaymentInitiationID).
		Order("created_at DESC", "sort_id DESC").
		For("UPDATE"). // Prevent another transaction to select/insert a new adjustment while this one is not committed
		Limit(1).
		Scan(ctx)
	if err != nil {
		return false, e("failed to get previous payment initiation adjustment", err)
	}

	if !predicate(toPaymentInitiationAdjustmentModels(previousAdj)) {
		// Explicitly rollback to release the FOR UPDATE lock
		if rollbackErr := tx.Rollback(); rollbackErr != nil {
			return false, e("failed to rollback transaction", rollbackErr)
		}
		return false, nil
	}

	toInsert := fromPaymentInitiationAdjustmentModels(adj)
	var res sql.Result
	res, err = tx.NewInsert().
		Model(&toInsert).
		On("CONFLICT (id) DO NOTHING").
		Exec(ctx)
	if err != nil {
		return false, e("failed to insert payment initiation adjustments", err)
	}

	var rowsAffected int64
	rowsAffected, err = res.RowsAffected()
	if err != nil {
		return false, e("failed to get rows affected", err)
	}

	// Create outbox event only if adjustment was actually inserted
	if rowsAffected > 0 {
		adjPayload := internalEvents.PaymentInitiationAdjustmentMessagePayload{
			ID:                  adj.ID.String(),
			PaymentInitiationID: adj.ID.PaymentInitiationID.String(),
			Status:              adj.Status.String(),
			Amount:              adj.Amount,
			Asset:               adj.Asset,
			Metadata:            adj.Metadata,
		}
		if adj.Error != nil {
			errorStr := adj.Error.Error()
			adjPayload.Error = &errorStr
		}

		var adjPayloadBytes []byte
		adjPayloadBytes, err = json.Marshal(&adjPayload)
		if err != nil {
			return false, e("failed to marshal payment initiation adjustment event payload", err)
		}

		outboxEvent := models.OutboxEvent{
			ID: models.EventID{
				EventIdempotencyKey: adj.IdempotencyKey(),
				ConnectorID:         &adj.ID.PaymentInitiationID.ConnectorID,
			},
			EventType:   events.EventTypeSavedPaymentInitiationAdjustment,
			EntityID:    adj.ID.String(),
			Payload:     adjPayloadBytes,
			CreatedAt:   stdtime.Now().UTC(),
			Status:      models.OUTBOX_STATUS_PENDING,
			ConnectorID: &adj.ID.PaymentInitiationID.ConnectorID,
		}

		if err = s.OutboxEventsInsert(ctx, tx, []models.OutboxEvent{outboxEvent}); err != nil {
			return false, err
		}
	}

	err = tx.Commit()
	if err != nil {
		return false, e("failed to commit transaction", err)
	}
	return rowsAffected > 0, nil
}

func (s *store) PaymentInitiationAdjustmentsGet(ctx context.Context, id models.PaymentInitiationAdjustmentID) (*models.PaymentInitiationAdjustment, error) {
	var adj paymentInitiationAdjustment
	err := s.db.NewSelect().
		Model(&adj).
		Where("id = ?", id).
		Scan(ctx)
	if err != nil {
		return nil, e("failed to get payment initiation adjustment", err)
	}

	res := toPaymentInitiationAdjustmentModels(adj)
	return &res, nil
}

type PaymentInitiationAdjustmentsQuery struct{}

type ListPaymentInitiationAdjustmentsQuery paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationAdjustmentsQuery]]

func NewListPaymentInitiationAdjustmentsQuery(opts paginate.PaginatedQueryOptions[PaymentInitiationAdjustmentsQuery]) ListPaymentInitiationAdjustmentsQuery {
	return ListPaymentInitiationAdjustmentsQuery{
		Order:    paginate.OrderAsc,
		PageSize: opts.PageSize,
		Options:  opts,
	}
}

func (s *store) paymentsInitiationAdjustmentsQueryContext(qb query.Builder) (string, []any, error) {
	where, args, err := qb.Build(query.ContextFn(func(key, operator string, value any) (string, []any, error) {
		switch {
		case key == "status":
			if operator != "$match" {
				return "", nil, e("'status' column can only be used with $match", ErrValidation)
			}
			return fmt.Sprintf("%s = ?", key), []any{value}, nil
		case metadataRegex.Match([]byte(key)):
			if operator != "$match" {
				return "", nil, e("'metadata' column can only be used with $match", ErrValidation)
			}
			match := metadataRegex.FindAllStringSubmatch(key, 3)
			key := "metadata"
			return key + " @> ?", []any{map[string]any{
				match[0][1]: value,
			}}, nil
		default:
			return "", nil, e(fmt.Sprintf("unknown key '%s' when building query", key), ErrValidation)
		}
	}))
	return where, args, err
}

func (s *store) PaymentInitiationAdjustmentsList(ctx context.Context, piID models.PaymentInitiationID, q ListPaymentInitiationAdjustmentsQuery) (*paginate.Cursor[models.PaymentInitiationAdjustment], error) {
	var (
		where string
		args  []any
		err   error
	)
	if q.Options.QueryBuilder != nil {
		where, args, err = s.paymentsInitiationAdjustmentsQueryContext(q.Options.QueryBuilder)
		if err != nil {
			return nil, err
		}
	}

	cursor, err := paginateWithOffset[paginate.PaginatedQueryOptions[PaymentInitiationAdjustmentsQuery], paymentInitiationAdjustment](s, ctx,
		(*paginate.OffsetPaginatedQuery[paginate.PaginatedQueryOptions[PaymentInitiationAdjustmentsQuery]])(&q),
		func(query *bun.SelectQuery) *bun.SelectQuery {
			if where != "" {
				query = query.Where(where, args...)
			}

			// TODO(polo): sorter ?
			query = query.Order("created_at DESC", "sort_id DESC")
			query.Where("payment_initiation_id = ?", piID)

			return query
		},
	)
	if err != nil {
		return nil, e("failed to fetch accounts", err)
	}

	pis := make([]models.PaymentInitiationAdjustment, 0, len(cursor.Data))
	for _, pi := range cursor.Data {
		pis = append(pis, toPaymentInitiationAdjustmentModels(pi))
	}

	return &paginate.Cursor[models.PaymentInitiationAdjustment]{
		PageSize: cursor.PageSize,
		HasMore:  cursor.HasMore,
		Previous: cursor.Previous,
		Next:     cursor.Next,
		Data:     pis,
	}, nil
}

func fromPaymentInitiationModels(from models.PaymentInitiation) paymentInitiation {
	return paymentInitiation{
		ID:                   from.ID,
		ConnectorID:          from.ConnectorID,
		Reference:            from.Reference,
		CreatedAt:            time.New(from.CreatedAt),
		ScheduledAt:          time.New(from.ScheduledAt),
		Description:          from.Description,
		Type:                 from.Type,
		Amount:               from.Amount,
		Asset:                from.Asset,
		DestinationAccountID: from.DestinationAccountID,
		SourceAccountID:      from.SourceAccountID,
		Metadata:             from.Metadata,
	}
}

func toPaymentInitiationModels(from paymentInitiation) models.PaymentInitiation {
	return models.PaymentInitiation{
		ID:                   from.ID,
		ConnectorID:          from.ConnectorID,
		Reference:            from.Reference,
		CreatedAt:            from.CreatedAt.Time,
		ScheduledAt:          from.ScheduledAt.Time,
		Description:          from.Description,
		Type:                 from.Type,
		SourceAccountID:      from.SourceAccountID,
		DestinationAccountID: from.DestinationAccountID,
		Amount:               from.Amount,
		Asset:                from.Asset,
		Metadata:             from.Metadata,
	}
}

func fromPaymentInitiationAdjustmentModels(from models.PaymentInitiationAdjustment) paymentInitiationAdjustment {
	return paymentInitiationAdjustment{
		ID:                  from.ID,
		PaymentInitiationID: from.ID.PaymentInitiationID,
		CreatedAt:           time.New(from.CreatedAt),
		Status:              from.Status,
		Amount:              from.Amount,
		Asset:               from.Asset,
		Error: func() *string {
			if from.Error == nil {
				return nil
			}
			return pointer.For(from.Error.Error())
		}(),
		Metadata: from.Metadata,
	}
}

func toPaymentInitiationAdjustmentModels(from paymentInitiationAdjustment) models.PaymentInitiationAdjustment {
	return models.PaymentInitiationAdjustment{
		ID:        from.ID,
		CreatedAt: from.CreatedAt.Time,
		Status:    from.Status,
		Amount:    from.Amount,
		Asset:     from.Asset,
		Error: func() error {
			if from.Error == nil {
				return nil
			}

			return errors.New(*from.Error)
		}(),
		Metadata: from.Metadata,
	}
}