package repository

import (
	"context"
	"errors"
	"fmt"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

// WalletMoveCustomer is a customer on either side of an exchange or a transfer.
type WalletMoveCustomer struct {
	ID             uuid.UUID
	OrganizationID uuid.UUID
	Name           string
	PhoneNumber    *string
	IsDefault      bool
	IsActive       bool
}

// WalletMoveRepository reads what exchanges and transfers check before moving a
// balance (docs/prd-point-coin.md F4, F5). The balance itself only moves through
// WalletProcessor.
type WalletMoveRepository interface {
	// GetCustomer returns ErrWalletNotFound when the customer does not exist.
	GetCustomer(ctx context.Context, customerID uuid.UUID) (*WalletMoveCustomer, error)
}

type walletMoveRepository struct {
	db *gorm.DB
}

func NewWalletMoveRepository(db *gorm.DB) WalletMoveRepository {
	return &walletMoveRepository{db: db}
}

func (r *walletMoveRepository) GetCustomer(ctx context.Context, customerID uuid.UUID) (*WalletMoveCustomer, error) {
	return r.customer(ctx, "id = ?", customerID)
}

func (r *walletMoveRepository) customer(ctx context.Context, where string, arg interface{}) (*WalletMoveCustomer, error) {
	var c WalletMoveCustomer
	err := DBFromContext(ctx, r.db).WithContext(ctx).
		Table("customers").
		Select("id, organization_id, name, phone_number, is_default, is_active").
		Where(where, arg).
		Take(&c).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrWalletNotFound
		}
		return nil, fmt.Errorf("failed to get customer: %w", err)
	}
	return &c, nil
}