inklabs/kommerce-core

<?php
namespace inklabs\kommerce\EntityRepository;

use inklabs\kommerce\Entity\TaxRate;
use inklabs\kommerce\Exception\InvalidArgumentException;

class TaxRateRepository extends AbstractRepository implements TaxRateRepositoryInterface
{
    public function findByZip5AndState(string $zip5 = null, string $state = null): ?TaxRate
    {
        if ($zip5 === null and $state === null) {
            throw new InvalidArgumentException('Missing zip5 or state');
        }

        $query = $this->getQueryBuilder()
            ->select('TaxRate')
            ->from(TaxRate::class, 'TaxRate');

        if ($state !== null) {
            $query
                ->where('TaxRate.state = :state')
                ->setParameter('state', $state);
        }

        if ($zip5 !== null) {
            $query
                ->orWhere('TaxRate.zip5 = :zip5')
                ->orWhere('TaxRate.zip5From <= :zip5 AND TaxRate.zip5To >= :zip5')
                ->setParameter('zip5', $zip5);
        }

        $taxRates = $query
            ->orderBy('TaxRate.id')
            ->getQuery()
            ->getResult();

        return $this->getZip5OrRateOrStateTaxRate($taxRates);
    }

    /**
     * @param TaxRate[] $taxRates
     * @return TaxRate
     */
    protected function getZip5OrRateOrStateTaxRate(array $taxRates): ?TaxRate
    {
        $stateTaxRates = $zip5TaxRates = $rangeTaxRates = [];

        foreach ($taxRates as $taxRate) {
            if ($taxRate->getState() !== null) {
                $stateTaxRates[] = $taxRate;
            } elseif ($taxRate->getZip5() !== null) {
                $zip5TaxRates[] = $taxRate;
            } elseif ($taxRate->getZip5From() !== null) {
                $rangeTaxRates[] = $taxRate;
            }
        }
        unset($taxRates);

        if (! empty($zip5TaxRates)) {
            $taxRate = $zip5TaxRates[0];
        } elseif (! empty($rangeTaxRates)) {
            $taxRate = $rangeTaxRates[0];
        } elseif (! empty($stateTaxRates)) {
            $taxRate = $stateTaxRates[0];
        } else {
            // TODO: Investigate whether we should throw exception
            return null;
        }

        return $taxRate;
    }
}