custom/plugins/Module23ImprovedTax/src/StoreFrontend/SalesChannel/ImprovedCountryTax.php line 52

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace Module23ImprovedTax\StoreFrontend\SalesChannel;
  7. use Module23ImprovedTax\Extension\Core\System\Country\ImprovedTaxEntity;
  8. use Shopware\Core\Checkout\Customer\CustomerEntity;
  9. use Shopware\Core\Framework\Api\Context\AdminApiSource;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityLoadedEvent;
  11. use Shopware\Core\System\Country\CountryEntity;
  12. use function array_filter;
  13. use function assert;
  15. class ImprovedCountryTax
  16. {
  17.     private ?CustomerEntity $customer = null;
  19.     public function __invoke(EntityLoadedEvent $event): void
  20.     {
  21.         if ($event->getContext()->getSource() instanceof AdminApiSource) {
  22.             return;
  23.         }
  25.         $country     = $this->country($event);
  26.         $improvedTax = $this->improvedTax($country);
  28.         if ($this->isCustomerCompany() && $improvedTax->isCompanyTaxFree()) {
  29.             $this->makeCompanyTaxFree($country);
  31.             return;
  32.         }
  34.         if ($this->isCustomerCompany() && $improvedTax->isCustomerTaxFree()) {
  35.             return;
  36.         }
  38.         $this->makeCustomerTaxFree($country);
  39.     }
  41.     private function makeCompanyTaxFree(CountryEntity $country): void
  42.     {
  43.         $country->setCompanyTaxFree(true);
  44.         $country->getCompanyTax()->setEnabled(true);
  46.         $country->setTaxFree(false);
  47.         $country->getCustomerTax()->setEnabled(false);
  48.     }
  50.     private function makeCustomerTaxFree(CountryEntity $country): void
  51.     {
  52.         $country->setTaxFree(true);
  53.         $country->getCustomerTax()->setEnabled(true);
  55.         $country->setCompanyTaxFree(false);
  56.         $country->getCompanyTax()->setEnabled(false);
  57.     }
  59.     private function isCustomerCompany(): bool
  60.     {
  61.         $isCompanyCustomer = false;
  62.         if ((null !== $this->customer)
  63.             && (null !== $this->customer->getDefaultBillingAddress())
  64.             && (false === empty($this->customer->getDefaultBillingAddress()->getCompany()))
  65.         ) {
  66.             $isCompanyCustomer = true;
  67.         }
  69.         return $isCompanyCustomer;
  70.     }
  72.     private function improvedTax(CountryEntity $country): ImprovedTaxEntity
  73.     {
  74.         $improvedTax = $country->getExtension('improvedTax');
  75.         assert($improvedTax instanceof ImprovedTaxEntity);
  77.         return $improvedTax;
  78.     }
  80.     private function country(EntityLoadedEvent $event): CountryEntity
  81.     {
  82.         return array_filter($event->getEntities())[0];
  83.     }
  85.     public function setCustomer(CustomerEntity $customer): void
  86.     {
  87.         $this->customer = $customer;
  88.     }
  89. }