1. /Users/gregor/PhpstormProjects/siesta/src/Siesta
  2. GeneratorPlugin
  3. Entity
  4. ReferenceGetterSetter.php
 
Code Coverage
 
Classes and Traits
Functions and Methods
Lines
Total
100.00% covered (success)
100.00%
1 / 1
100.00% covered (success)
100.00%
5 / 5
 CRAP
100.00% covered (success)
100.00%
46 / 46
ReferenceGetterSetter
100.00% covered (success)
100.00%
1 / 1
100.00% covered (success)
100.00%
6 / 6
 12
100.00% covered (success)
100.00%
46 / 46
 getUseClassNameList
100.00% covered (success)
100.00%
1 / 1
 4
100.00% covered (success)
100.00%
10 / 10
 getDependantPluginList
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
0 / 0
 generate
100.00% covered (success)
100.00%
1 / 1
 2
100.00% covered (success)
100.00%
5 / 5
 generateReferenceGetter
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
14 / 14
 getFindByPKParameter
100.00% covered (success)
100.00%
1 / 1
 2
100.00% covered (success)
100.00%
5 / 5
 generateReferenceSetter
100.00% covered (success)
100.00%
1 / 1
 2
100.00% covered (success)
100.00%
12 / 12
<?php
declare(strict_types = 1);
namespace Siesta\GeneratorPlugin\Entity;
use Siesta\CodeGenerator\CodeGenerator;
use Siesta\GeneratorPlugin\BasePlugin;
use Siesta\GeneratorPlugin\ServiceClass\GetEntityByIdPlugin;
use Siesta\Model\Entity;
use Siesta\Model\PHPType;
use Siesta\Model\Reference;
/**
 * @author Gregor Müller
 */
class ReferenceGetterSetter extends BasePlugin
{
    /**
     * @param Entity $entity
     *
     * @return string[]
     */
    public function getUseClassNameList(Entity $entity) : array
    {
        $useClassList = [];
        foreach ($entity->getReferenceList() as $reference) {
            $foreinEntity = $reference->getForeignEntity();
            $useClassList[] = $foreinEntity->getInstantiationClass();
        }
        foreach ($entity->getReferenceList() as $reference) {
            $foreignEntity = $reference->getForeignEntity();
            $serviceFactory = $foreignEntity->getServiceFactoryClass();
            if ($serviceFactory !== null) {
                $useClassList[] = $serviceFactory;
            }
        }
        return $useClassList;
    }
    /**
     * @return string[]
     */
    public function getDependantPluginList() : array
    {
        return ['Siesta\GeneratorPlugin\Entity\MemberPlugin'];
    }
    /**
     * @param Entity $entity
     * @param CodeGenerator $codeGenerator
     */
    public function generate(Entity $entity, CodeGenerator $codeGenerator)
    {
        $this->setup($entity, $codeGenerator);
        foreach ($entity->getReferenceList() as $reference) {
            $this->generateReferenceGetter($reference);
            $this->generateReferenceSetter($reference);
        }
    }
    /**
     * @param Reference $reference
     */
    protected function generateReferenceGetter(Reference $reference)
    {
        $foreignEntity = $reference->getForeignEntity();
        $methodName = 'get' . $reference->getMethodName();
        $memberName = '$this->' . $reference->getName();
        $method = $this->codeGenerator->newPublicMethod($methodName);
        $method->addParameter(PHPType::BOOL, 'forceReload', 'false');
        $method->setReturnType($foreignEntity->getInstantiationClassShortName(), true);
        // code to load the entity
        $method->addIfStart($memberName . ' === null || $forceReload');
        $parameter = $this->getFindByPKParameter($reference);
        $reload = $foreignEntity->getServiceAccess() . "->" . GetEntityByIdPlugin::METHOD_ENTITY_BY_PK . "($parameter);";
        $method->addLine($memberName . ' = ' . $reload);
        $method->addIfEnd();
        $method->addLine('return ' . $memberName . ';');
        $method->end();
    }
    /**
     * @param Reference $reference
     *
     * @return string
     */
    protected function getFindByPKParameter(Reference $reference)
    {
        $parameterList = [];
        foreach ($reference->getReferenceMappingList() as $mapping) {
            $localAttribute = $mapping->getLocalAttribute();
            $parameterList[] = '$this->' . $localAttribute->getPhpName();
        }
        return implode(", ", $parameterList);
    }
    /**
     * @param Reference $reference
     */
    protected function generateReferenceSetter(Reference $reference)
    {
        $name = $reference->getName();
        $methodName = 'set' . $reference->getMethodName();
        $foreignEntity = $reference->getForeignEntity();
        $method = $this->codeGenerator->newPublicMethod($methodName);
        $method->addParameter($foreignEntity->getClassShortName(), 'entity', 'null');
        $method->addLine('$this->' . $name . ' = $entity;');
        foreach ($reference->getReferenceMappingList() as $referenceMapping) {
            $localAttribute = $referenceMapping->getLocalAttribute();
            $foreignAttribute = $referenceMapping->getForeignAttribute();
            $method->addLine('$this->' . $localAttribute->getPhpName() . ' = ($entity !== null) ? $entity->get' . $foreignAttribute->getMethodName() . '(true) : null;');
        }
        $method->end();
    }
}