core/internal/adapter/repository_bun_tribe_change.go

package adapter

import (
	"context"
	"database/sql"
	"errors"
	"fmt"
	"time"

	"gitea.dwysokinski.me/twhelp/corev3/internal/bun/bunmodel"
	"gitea.dwysokinski.me/twhelp/corev3/internal/domain"
	"github.com/uptrace/bun"
)

type TribeChangeBunRepository struct {
	db bun.IDB
}

func NewTribeChangeBunRepository(db bun.IDB) *TribeChangeBunRepository {
	return &TribeChangeBunRepository{db: db}
}

func (repo *TribeChangeBunRepository) Create(ctx context.Context, params ...domain.CreateTribeChangeParams) error {
	if len(params) == 0 {
		return nil
	}

	now := time.Now()
	tribeChanges := make(bunmodel.TribeChanges, 0, len(params))

	for _, p := range params {
		tribeChanges = append(tribeChanges, bunmodel.TribeChange{
			ServerKey:  p.ServerKey(),
			PlayerID:   p.PlayerID(),
			NewTribeID: p.NewTribeID(),
			OldTribeID: p.OldTribeID(),
			CreatedAt:  now,
		})
	}

	if _, err := repo.db.NewInsert().
		Model(&tribeChanges).
		Ignore().
		Returning("").
		Exec(ctx); err != nil {
		return fmt.Errorf("something went wrong while inserting tribe changes into the db: %w", err)
	}

	return nil
}

func (repo *TribeChangeBunRepository) List(
	ctx context.Context,
	params domain.ListTribeChangesParams,
) (domain.ListTribeChangesResult, error) {
	var tribeChanges bunmodel.TribeChanges

	if err := repo.db.NewSelect().
		Model(&tribeChanges).
		Apply(listTribeChangesParamsApplier{params: params}.apply).
		Scan(ctx); err != nil && !errors.Is(err, sql.ErrNoRows) {
		return domain.ListTribeChangesResult{}, fmt.Errorf("couldn't select tribe changes from the db: %w", err)
	}

	converted, err := tribeChanges.ToDomain()
	if err != nil {
		return domain.ListTribeChangesResult{}, err
	}

	return domain.NewListTribeChangesResult(separateListResultAndNext(converted, params.Limit()))
}

type listTribeChangesParamsApplier struct {
	params domain.ListTribeChangesParams
}

//nolint:gocyclo
func (a listTribeChangesParamsApplier) apply(q *bun.SelectQuery) *bun.SelectQuery {
	if serverKeys := a.params.ServerKeys(); len(serverKeys) > 0 {
		q = q.Where("tc.server_key IN (?)", bun.In(serverKeys))
	}

	for _, s := range a.params.Sort() {
		switch s {
		case domain.TribeChangeSortCreatedAtASC:
			q = q.Order("tc.created_at ASC")
		case domain.TribeChangeSortCreatedAtDESC:
			q = q.Order("tc.created_at DESC")
		case domain.TribeChangeSortIDASC:
			q = q.Order("tc.id ASC")
		case domain.TribeChangeSortIDDESC:
			q = q.Order("tc.id DESC")
		case domain.TribeChangeSortServerKeyASC:
			q = q.Order("tc.server_key ASC")
		case domain.TribeChangeSortServerKeyDESC:
			q = q.Order("tc.server_key DESC")
		default:
			return q.Err(fmt.Errorf("%s: %w", s.String(), errInvalidSortValue))
		}
	}

	return q.Limit(a.params.Limit() + 1).Apply(a.applyCursor)
}

func (a listTribeChangesParamsApplier) applyCursor(q *bun.SelectQuery) *bun.SelectQuery {
	cursor := a.params.Cursor()

	if cursor.IsZero() {
		return q
	}

	sort := a.params.Sort()
	cursorApplier := cursorPaginationApplier{
		data: make([]cursorPaginationApplierDataElement, 0, len(sort)),
	}

	for _, s := range sort {
		var el cursorPaginationApplierDataElement

		switch s {
		case domain.TribeChangeSortIDASC:
			el.value = cursor.ID()
			el.unique = true
			el.column = "tc.id"
			el.direction = sortDirectionASC
		case domain.TribeChangeSortIDDESC:
			el.value = cursor.ID()
			el.unique = true
			el.column = "tc.id"
			el.direction = sortDirectionDESC
		case domain.TribeChangeSortServerKeyASC:
			el.value = cursor.ServerKey()
			el.column = "tc.server_key"
			el.direction = sortDirectionASC
		case domain.TribeChangeSortServerKeyDESC:
			el.value = cursor.ServerKey()
			el.column = "tc.server_key"
			el.direction = sortDirectionDESC
		case domain.TribeChangeSortCreatedAtASC:
			el.value = cursor.CreatedAt()
			el.column = "tc.created_at"
			el.direction = sortDirectionASC
		case domain.TribeChangeSortCreatedAtDESC:
			el.value = cursor.CreatedAt()
			el.column = "tc.created_at"
			el.direction = sortDirectionDESC
		default:
			return q.Err(fmt.Errorf("%s: %w", s.String(), errInvalidSortValue))
		}

		cursorApplier.data = append(cursorApplier.data, el)
	}

	return q.Apply(cursorApplier.apply)
}