1. /Users/gregor/PhpstormProjects/siesta/src/Siesta
  2. GeneratorPlugin
  3. Entity
  4. CollectorGetterSetter.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%
7 / 7
 CRAP
100.00% covered (success)
100.00%
65 / 65
CollectorGetterSetter
100.00% covered (success)
100.00%
1 / 1
100.00% covered (success)
100.00%
8 / 8
 13
100.00% covered (success)
100.00%
65 / 65
 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%
6 / 6
 generateCollectionGetter
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
13 / 13
 generateLoadCall
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
6 / 6
 generateDeleteAll
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
12 / 12
 generateAddToCollection
100.00% covered (success)
100.00%
1 / 1
 1
100.00% covered (success)
100.00%
14 / 14
 getServiceClassInvocationSignature
100.00% covered (success)
100.00%
1 / 1
 2
100.00% covered (success)
100.00%
4 / 4
<?php
declare(strict_types = 1);
namespace Siesta\GeneratorPlugin\Entity;
use Siesta\CodeGenerator\CodeGenerator;
use Siesta\GeneratorPlugin\BasePlugin;
use Siesta\GeneratorPlugin\ServiceClass\CollectionAccessPlugin;
use Siesta\Model\Collection;
use Siesta\Model\Entity;
use Siesta\Model\PHPType;
class CollectorGetterSetter extends BasePlugin
{
    const METHOD_ADD_TO_PREFIX = "addTo";
    /**
     * @param Entity $entity
     *
     * @return array
     */
    public function getUseClassNameList(Entity $entity) : array
    {
        $useList = [];
        foreach ($entity->getCollectionList() as $collection) {
            $foreignEntity = $collection->getForeignEntity();
            $useList[] = $foreignEntity->getInstantiationClass();
        }
        foreach ($entity->getReferenceList() as $reference) {
            $foreignEntity = $reference->getForeignEntity();
            $serviceFactory = $foreignEntity->getServiceFactoryClass();
            if ($serviceFactory !== null) {
                $useList[] = $serviceFactory;
            }
        }
        return $useList;
    }
    /**
     * @return array
     */
    public function getDependantPluginList() : array
    {
        return [];
    }
    /**
     * @param Entity $entity
     * @param CodeGenerator $codeGenerator
     */
    public function generate(Entity $entity, CodeGenerator $codeGenerator)
    {
        $this->setup($entity, $codeGenerator);
        foreach ($this->entity->getCollectionList() as $collection) {
            $this->generateCollectionGetter($collection);
            $this->generateDeleteAll($collection);
            $this->generateAddToCollection($collection);
        }
    }
    /**
     * @param Collection $collection
     */
    protected function generateCollectionGetter(Collection $collection)
    {
        $methodName = 'get' . $collection->getMethodName();
        $name = $collection->getName();
        $foreignEntity = $collection->getForeignEntity();
        $method = $this->codeGenerator->newPublicMethod($methodName);
        $method->addParameter(PHPType::BOOL, 'forceReload', 'false');
        $method->addConnectionNameParameter();
        $method->setReturnType($foreignEntity->getInstantiationClassShortName() . '[]');
        // check if (re)load is needed
        $method->addIfStart('$this->' . $name . ' === null || $forceReload');
        $method->addLine($this->generateLoadCall($collection));
        $method->addIfEnd();
        // return collection
        $method->addLine('return $this->' . $name . ';');
        $method->end();
    }
    /**
     * @param Collection $collection
     *
     * @return string
     */
    protected function generateLoadCall(Collection $collection)
    {
        $foreignEntity = $collection->getForeignEntity();
        $serviceAccess = $foreignEntity->getServiceAccess();
        $memberName = $collection->getName();
        $methodName = CollectionAccessPlugin::getSelectByReferenceName($collection->getForeignReference());
        $invocationSignature = $this->getServiceClassInvocationSignature();
        return '$this->' . $memberName . ' = ' . $serviceAccess . '->' . $methodName . '(' . $invocationSignature . ', $connectionName);';
    }
    /**
     * @param Collection $collection
     */
    protected function generateDeleteAll(Collection $collection)
    {
        $methodName = "deleteAll" . $collection->getMethodName();
        $method = $this->codeGenerator->newPublicMethod($methodName);
        $method->addConnectionNameParameter();
        // invoke delete call on service class
        $foreignEntity = $collection->getForeignEntity();
        $serviceAccess = $foreignEntity->getServiceAccess();
        $deleteMethodName = CollectionAccessPlugin::getDeleteByReferenceName($collection->getForeignReference());
        $invocationSignature = $this->getServiceClassInvocationSignature();
        $deleteCall = $serviceAccess . '->' . $deleteMethodName . '(' . $invocationSignature . ', $connectionName);';
        $method->addLine($deleteCall);
        // reset collection to null
        $method->addLine('$this->' . $collection->getName() . ' = null;');
        $method->end();
    }
    /**
     * @param Collection $collection
     */
    protected function generateAddToCollection(Collection $collection)
    {
        $foreignEntity = $collection->getForeignEntity();
        $foreignClass = $foreignEntity->getInstantiationClassShortName();
        $methodName = self::METHOD_ADD_TO_PREFIX . $collection->getMethodName();
        $method = $this->codeGenerator->newPublicMethod($methodName);
        $method->addParameter($foreignClass, 'entity');
        // add reference to entity
        $reference = $collection->getForeignReference();
        $method->addLine('$entity->set' . $reference->getMethodName() . '($this);');
        // check if collection is alread array
        $member = '$this->' . $collection->getName();
        $method->addIfStart($member . ' === null');
        $method->addLine($member . ' = [];');
        $method->addIfEnd();
        // add entity to collection
        $method->addLine($member . '[] = $entity;');
        $method->end();
    }
    /**
     * @return string
     */
    protected function getServiceClassInvocationSignature()
    {
        $pkList = [];
        foreach ($this->entity->getPrimaryKeyAttributeList() as $pkAttribute) {
            $pkList[] = '$this->get' . $pkAttribute->getMethodName() . '(true, $connectionName)';
        }
        return implode(", ", $pkList);
    }
}