DigitalState / Identities

api/src/Fixture/Staff.php

<?php

namespace App\Fixture;

use App\Entity\Staff as StaffEntity;
use Doctrine\Common\Persistence\ObjectManager;

The type Doctrine\Common\Persistence\ObjectManager was not found. Maybe you did not declare it correctly or list all dependencies?

use Ds\Component\Database\Fixture\Yaml;

The type Ds\Component\Database\Fixture\Yaml was not found. Maybe you did not declare it correctly or list all dependencies?

use LogicException;

/**
 * Trait Staff
 */
trait Staff
{
    use Yaml;

    /**
     * @var string
     */
    private $path;

    /**
     * {@inheritdoc}
     */
    public function load(ObjectManager $manager)
    {
        $objects = $this->parse($this->path);

        foreach ($objects as $object) {
            $staff = new StaffEntity;
            $staff
                ->setUuid($object->uuid)
                ->setOwner($object->owner)
                ->setOwnerUuid($object->owner_uuid)
                ->setTenant($object->tenant);

            foreach ($object->business_units as $uuid) {
                $businessUnit = $this->getReference($uuid);

                if (!$businessUnit) {
                    throw new LogicException('Business Unit "'.$uuid.'" does not exist.');
                }

                $staff->addBusinessUnit($businessUnit);
            }

            $manager->persist($staff);
            $this->setReference($object->uuid, $staff);
        }

        $manager->flush();
    }
}