Sirius\Orm\CodeGenerator\EntityBaseGenerator

Complexity

Total Complexity

4

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Importance

Changes

0

<?php
class EntityBaseGenerator
{
    /**
     * @var ClassType
     */
    protected $class;

    /**
     * @var PhpNamespace
     */
    protected $namespace;

    /**
     * @var Mapper
     */
    protected $mapper;
    /**
     * @var Dumper
     */
    protected $dumper;

    public function __construct(Mapper $mapper)
    {
        $this->dumper    = new Dumper();
        $this->mapper    = $mapper;
        $this->namespace = new PhpNamespace($mapper->getEntityNamespace());
        $this->class     = new ClassType($mapper->getEntityClass() . 'Base', $this->namespace);
        $this->class->setAbstract(true);
    }

    public function getClass()
    {
        $this->build();

        return $this->class;
    }

    protected function build()
    {
        if ($this->mapper->getEntityStyle() === Mapper::ENTITY_STYLE_PROPERTIES) {
            $this->namespace->addUse(\Sirius\Orm\Entity\GenericEntity::class);
            $this->class->setExtends('GenericEntity');
        } else {
            $this->namespace->addUse(\Sirius\Orm\Entity\ClassMethodsEntity::class);
            $this->class->setExtends('ClassMethodsEntity');
        }

        $constructor = $this->class->addMethod('__construct')
                                   ->setBody('parent::__construct($attributes, $state);');
        $constructor->addParameter('attributes')
                    ->setType('array')
                    ->setDefaultValue([]);
        $constructor->addParameter('state')
                    ->setType('string')
                    ->setDefaultValue(null);

        $this->class = $this->mapper->getOrm()->applyObservers($this->mapper->getName() . '_base_entity', $this->class);
    }