chill-bundles/src/Bundle/ChillReportBundle/DataFixtures/ORM/LoadReports.php

<?php

declare(strict_types=1);

/*
 * Chill is a software for social workers
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

namespace Chill\ReportBundle\DataFixtures\ORM;

use Chill\CustomFieldsBundle\Entity\CustomField;
use Chill\MainBundle\DataFixtures\ORM\LoadScopes;
use Chill\MainBundle\DataFixtures\ORM\LoadUsers;
use Chill\PersonBundle\Entity\Person;
use Chill\ReportBundle\Entity\Report;
use DateTime;
use Doctrine\Common\DataFixtures\AbstractFixture;
use Doctrine\Common\DataFixtures\OrderedFixtureInterface;
use Doctrine\Persistence\ObjectManager;
use Faker\Factory as FakerFactory;
use Symfony\Component\DependencyInjection\ContainerAwareInterface;

use function count;
use function in_array;

/**
 * Load reports into DB.
 */
class LoadReports extends AbstractFixture implements ContainerAwareInterface, OrderedFixtureInterface
{
    use \Symfony\Component\DependencyInjection\ContainerAwareTrait;

    /**
     * @var \Faker\Generator
     */
    private $faker;

    public function __construct()
    {
        $this->faker = FakerFactory::create('fr_FR');
    }

    public function getOrder()
    {
        return 15002;
    }

    public function load(ObjectManager $manager)
    {
        $this->createExpected($manager);

        //create random 2 times, to allow multiple report on some people
        $this->createRandom($manager, 90);
        $this->createRandom($manager, 30);

        $manager->flush();
    }

    private function createExpected(ObjectManager $manager)
    {
        $charline = $this->container->get('doctrine.orm.entity_manager')
            ->getRepository(Person::class)
            ->findOneBy(['firstName' => 'Charline', 'lastName' => 'DEPARDIEU']);

        if (null !== $charline) {
            $report = (new Report())
                ->setPerson($charline)
                ->setCFGroup($this->getReference('cf_group_report_logement'))
                ->setDate(new DateTime('2015-01-05'))
                ->setScope($this->getReference('scope_social'));
            $this->fillReport($report);

            $manager->persist($report);
        } else {
            echo 'WARNING: Charline DEPARDIEU not found in database';
        }
    }

    private function createRandom(ObjectManager $manager, $percentage)
    {
        $people = $this->getPeopleRandom($percentage);

        foreach ($people as $person) {
            //create a report, set logement or education report
            $report = (new Report())
                ->setPerson($person)
                ->setCFGroup(
                    random_int(0, 10) > 5 ?
                            $this->getReference('cf_group_report_logement') :
                            $this->getReference('cf_group_report_education')
                )
                ->setScope($this->getScopeRandom());
            $this->fillReport($report);
            $manager->persist($report);
        }
    }

    private function fillReport(Report $report)
    {
        //setUser
        $usernameRef = array_rand(LoadUsers::$refs);
        $report->setUser(
            $this->getReference($usernameRef)
        );

        //set date if null
        if ($report->getDate() === null) {
            //set date. 30% of the dates are 2015-05-01
            $expectedDate = new DateTime('2015-01-05');

            if (random_int(0, 100) < 30) {
                $report->setDate($expectedDate);
            } else {
                $report->setDate($this->faker->dateTimeBetween('-1 year', 'now')
                    ->setTime(0, 0, 0));
            }
        }

        //fill data
        $datas = [];

        foreach ($report->getCFGroup()->getCustomFields() as $field) {
            switch ($field->getType()) {
                case 'title':
                    $datas[$field->getSlug()] = [];

                    break;

                case 'choice':
                    $datas[$field->getSlug()] = $this->getRandomChoice($field);

                    break;

                case 'text':
                    $datas[$field->getSlug()] = $this->faker->realText($field->getOptions()['maxLength']);

                    break;
            }
        }
        $report->setCFData($datas);

        return $report;
    }

    private function getPeopleRandom($percentage)
    {
        $people = $this->container->get('doctrine.orm.entity_manager')
            ->getRepository(Person::class)
            ->findAll();

        //keep only a part ($percentage) of the people
        $selectedPeople = [];

        foreach ($people as $person) {
            if (random_int(0, 100) < $percentage) {
                $selectedPeople[] = $person;
            }
        }

        return $selectedPeople;
    }

    /**
     * pick a random choice.
     *
     * @return string|string[]
     */
    private function getRandomChoice(CustomField $field): string|array
    {
        $choices = $field->getOptions()['choices'];
        $multiple = $field->getOptions()['multiple'];
        $other = $field->getOptions()['other'];

        //add other if allowed
        if ($other) {
            $choices[] = ['slug' => '_other'];
        }

        //initialize results
        $picked = [];

        if ($multiple) {
            $numberSelected = random_int(1, count($choices) - 1);

            for ($i = 0; $i < $numberSelected; ++$i) {
                $picked[] = $this->pickChoice($choices);
            }

            if ($other) {
                $result = ['_other' => null, '_choices' => $picked];

                if (in_array('_other', $picked, true)) {
                    $result['_other'] = $this->faker->realText(70);
                }

                return $result;
            }
        } else {
            $picked = $this->pickChoice($choices);

            if ($other) {
                $result = ['_other' => null, '_choices' => $picked];

                if ('_other' === $picked) {
                    $result['_other'] = $this->faker->realText(70);
                }

                return $result;
            }
        }

        return $picked;
    }

    /**
     * @return \Chill\MainBundle\Entity\Scope
     */
    private function getScopeRandom()
    {
        $ref = LoadScopes::$references[array_rand(LoadScopes::$references)];

        return $this->getReference($ref);
    }

    /**
     * pick a choice within a 'choices' options (for choice type).
     *
     * @return string the slug of the selected choice
     */
    private function pickChoice(array $choices)
    {
        return $choices[array_rand($choices)]['slug'];
    }
}