CaracoleBundle/Twig/StoreTwigExtension.php

<?php

namespace Lc\CaracoleBundle\Twig;

use Lc\CaracoleBundle\Repository\Config\TaxRateStore;
use Lc\CaracoleBundle\Repository\Config\UnitStore;
use Lc\CaracoleBundle\Repository\Merchant\MerchantRepositoryQuery;
use Lc\CaracoleBundle\Repository\Reminder\ReminderStore;
use Lc\CaracoleBundle\Repository\Section\SectionRepository;
use Lc\CaracoleBundle\Repository\Section\SectionRepositoryInterface;
use Lc\CaracoleBundle\Repository\Section\SectionRepositoryQuery;
use Lc\CaracoleBundle\Resolver\MerchantResolver;
use Lc\CaracoleBundle\Resolver\SectionResolver;
use Lc\ShopBundle\Context\UnitInterface;
use Twig\Extension\AbstractExtension;
use Twig\TwigFunction;

class StoreTwigExtension extends AbstractExtension
{
    protected MerchantRepositoryQuery $merchantRepositoryQuery;
    protected SectionRepositoryQuery $sectionRepositoryQuery;
    protected ReminderStore $reminderStore;
    protected MerchantResolver $merchantResolver;
    protected SectionResolver $sectionResolver;
    protected UnitStore $unitStore;
    protected TaxRateStore $taxRateStore;

    public function __construct(
            MerchantResolver $merchantResolver,
            SectionResolver $sectionResolver,
            MerchantRepositoryQuery $merchantRepositoryQuery,
            SectionRepositoryQuery $sectionRepositoryQuery,
            ReminderStore $reminderStore,
            UnitStore $unitStore,
            TaxRateStore $taxRateStore
    ) {
        $this->merchantResolver = $merchantResolver;
        $this->sectionResolver = $sectionResolver;
        $this->merchantRepositoryQuery = $merchantRepositoryQuery;
        $this->sectionRepositoryQuery = $sectionRepositoryQuery;
        $this->reminderStore = $reminderStore;
        $this->unitStore = $unitStore;
        $this->taxRateStore = $taxRateStore;
    }

    public function getFunctions()
    {
        return array(
                new TwigFunction('carac_sections', [$this, 'getSections']),
                new TwigFunction('carac_merchants', [$this, 'getMerchants']),
                new TwigFunction('carac_reminders', [$this, 'getReminders']),
                new TwigFunction('carac_units', [$this, 'getUnits']),
                new TwigFunction('carac_tax_rates', [$this, 'getTaxRates']),
        );
    }

    public function getSections()
    {
        return $this->sectionRepositoryQuery
                ->filterByMerchant($this->merchantResolver->getCurrent())
                ->find();
    }

    public function getMerchants()
    {
        return $this->merchantRepositoryQuery->find();
    }

    public function getReminders($params = [])
    {
        return $this->reminderStore
                ->setMerchant($this->merchantResolver->getCurrent())
                ->setSection($this->sectionResolver->getCurrent())
                ->get($params);
    }

    public function getUnits(){

        return $this->unitStore->getAsArray();
    }

    public function getTaxRates(){

        return $this->taxRateStore->getAsArray();
    }

}