chill-bundles/src/Bundle/ChillPersonBundle/Form/ChoiceLoader/PersonChoiceLoader.php

<?php

/*
 * Copyright (C) 2018 Champs-Libres <info@champs-libres.coop>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

namespace Chill\PersonBundle\Form\ChoiceLoader;

use Symfony\Component\Form\ChoiceList\Loader\ChoiceLoaderInterface;
use Symfony\Component\Form\ChoiceList\ChoiceListInterface;
use Chill\PersonBundle\Entity\Person;
use Chill\PersonBundle\Repository\PersonRepository;

/**
 * Allow to load a list of person
 */
class PersonChoiceLoader implements ChoiceLoaderInterface
{
    protected PersonRepository $personRepository;
    
    protected array $lazyLoadedPersons = [];
    
    protected array $centers = [];
    
    /**
     * PersonChoiceLoader constructor.
     *
     * @param EntityRepository $personRepository
     * @param array|null $centers
     */
    public function __construct(
        PersonRepository $personRepository,
        array $centers = null
    ) {
        $this->personRepository = $personRepository;
        if (NULL !== $centers) {
            $this->centers = $centers;
        }
    }
    
    /**
     * @return bool
     */
    protected function hasCenterFilter()
    {
        return count($this->centers) > 0;
    }
    
    /**
     * @param null $value
     * @return ChoiceListInterface
     */
    public function loadChoiceList($value = null): ChoiceListInterface
    {
        $list = new \Symfony\Component\Form\ChoiceList\ArrayChoiceList(
            $this->lazyLoadedPersons, 
            function(Person $p) use ($value) {
                return \call_user_func($value, $p);
            });
        
        return $list;
    }
    
    /**
     * @param array $values
     * @param null $value
     * @return array
     */
    public function loadChoicesForValues(array $values, $value = null)
    {
        $choices = [];
        
        foreach($values as $value) {
            if (empty($value)) {
                continue;
            }
            
            $person = $this->personRepository->find($value);
            
            if ($this->hasCenterFilter() && 
                !\in_array($person->getCenter(), $this->centers)) {
                throw new \RuntimeException("chosen a person not in correct center");
            }
            
            $choices[] = $person;
        }

        return $choices;
    }
    
    /**
     * @param array $choices
     * @param null $value
     * @return array|string[]
     */
    public function loadValuesForChoices(array $choices, $value = null)
    {
        $values = [];
        
        foreach ($choices as $choice) {
            if (NULL === $choice) {
                $values[] = null;
                continue;
            }
            
            $id = \call_user_func($value, $choice);
            $values[] = $id;
            $this->lazyLoadedPersons[$id] = $choice;
        }
        
        return $values;
    }
}