Tarantool\Mapper\Plugin\Annotation

Complexity

Total Complexity

77

Size/Duplication

Total Lines	346
Duplicated Lines	2.89 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

13 Methods

Rating	Name	Duplication	Size	Complexity
B	register()	0	31	8
A	validateSpace()	0	16	5
A	getSpace()	0	6	2
A	isExtension()	0	4	1
A	getExtensions()	0	14	4
F	migrate()	0	171	36
A	getEntityClass()	0	11	4
A	setEntityPostfix()	0	5	1
A	setRepositoryPostfix()	0	5	1
A	getRepositorySpaceName()	0	4	1
B	getSpaceName()	10	23	6
A	isReference()	0	4	1
B	getTarantoolType()	0	25	7

<?php

namespace Tarantool\Mapper\Plugin;

use Closure;
use Exception;
use LogicException;
use phpDocumentor\Reflection\DocBlockFactory;
use phpDocumentor\Reflection\Types\ContextFactory;
use ReflectionClass;
use ReflectionMethod;
use ReflectionProperty;
use Tarantool\Mapper\Entity;
use Tarantool\Mapper\Plugin\NestedSet;
use Tarantool\Mapper\Repository;
use Tarantool\Mapper\Space;

class Annotation extends UserClasses
{
    protected $entityClasses = [];
    protected $entityPostfix;

    protected $repositoryClasses = [];
    protected $repositoryPostifx;

    protected $extensions;

    public function register($class)
    {
        $isEntity = is_subclass_of($class, Entity::class);

Due to PHP Bug #53727, is_subclass_of might return inconsistent results on some PHP versions if \Tarantool\Mapper\Entity::class can be an interface. If so, you could instead use ReflectionClass::implementsInterface.

        $isRepository = is_subclass_of($class, Repository::class);

Due to PHP Bug #53727, is_subclass_of might return inconsistent results on some PHP versions if \Tarantool\Mapper\Repository::class can be an interface. If so, you could instead use ReflectionClass::implementsInterface.

        if (!$isEntity && !$isRepository) {
            throw new Exception("Invalid registration for $class");
        }

        if ($isEntity) {
            if ($class == Entity::class) {
                throw new Exception("Invalid entity registration for $class");
            }
            $this->entityClasses[] = $class;
        }

        if ($isRepository) {
            if ($class == Repository::class) {
                throw new Exception("Invalid repository registration for $class");
            }
            $this->repositoryClasses[] = $class;
        }

        $space = $this->getSpaceName($class);
        if ($isEntity) {
            $this->mapEntity($space, $class);
        } else {
            $this->mapRepository($space, $class);
        }
        return $this;
    }

    public function validateSpace($space)
    {
        foreach ($this->entityClasses as $class) {
            if ($this->getSpaceName($class) == $space) {
                return true;
            }
        }

        foreach ($this->repositoryClasses as $class) {
            if ($this->getSpaceName($class) == $space) {
                return true;
            }
        }

        return parent::validateSpace($space);
    }

    public function getSpace($instance)
    {
        $class = get_class($instance);
        $target = $this->isExtension($class) ? $this->getExtensions()[$class] : $class;
        return $this->getSpaceName($target);
    }

    public function isExtension($class)
    {
        return array_key_exists($class, $this->getExtensions());
    }

    public function getExtensions()
    {
        if (is_null($this->extensions)) {
            $this->extensions = [];
            foreach ($this->entityClasses as $entity) {
                $reflection = new ReflectionClass($entity);
                $parentEntity = $reflection->getParentClass()->getName();
                if (in_array($parentEntity, $this->entityClasses)) {
                    $this->extensions[$entity] = $parentEntity;
                }
            }
        }
        return $this->extensions;
    }

    public function migrate($extensionInstances = true)
    {
        $factory = DocBlockFactory::createInstance();
        $contextFactory = new ContextFactory();

        $schema = $this->mapper->getSchema();

        $computes = [];
        foreach ($this->entityClasses as $entity) {
            if ($this->isExtension($entity)) {
                continue;
            }

            $spaceName = $this->getSpaceName($entity);

            $engine = 'memtx';
            if (array_key_exists($spaceName, $this->repositoryMapping)) {
                $repositoryClass = $this->repositoryMapping[$spaceName];
                $repositoryReflection = new ReflectionClass($repositoryClass);
                $repositoryProperties = $repositoryReflection->getDefaultProperties();
                if (array_key_exists('engine', $repositoryProperties)) {
                    $engine = $repositoryProperties['engine'];
                }
            }

            if ($schema->hasSpace($spaceName)) {
                $space = $schema->getSpace($spaceName);
                if ($space->getEngine() != $engine) {
                    throw new Exception("Space engine can't be updated");
                }
            } else {
                $space = $schema->createSpace($spaceName, [
                    'engine' => $engine,
                    'properties' => [],
                ]);
            }

            $class = new ReflectionClass($entity);

            foreach ($class->getProperties(ReflectionProperty::IS_PUBLIC) as $property) {
                $context = $contextFactory->createFromReflector($property);
                $description = $factory->create($property->getDocComment(), $context);
                $tags = $description->getTags('var');

                if (!count($tags)) {
                    throw new Exception("No var tag for ".$entity.'::'.$property->getName());
                }

                $byTypes = [];
                foreach ($tags as $candidate) {
                    $byTypes[$candidate->getName()] = $candidate;
                }

                if (!array_key_exists('var', $byTypes)) {
                    throw new Exception("No var tag for ".$entity.'::'.$property->getName());
                }

                $propertyName = $property->getName();
                $phpType = $byTypes['var']->getType();

                if (array_key_exists('type', $byTypes)) {
                    $type = (string) $byTypes['type']->getDescription();
                } else {
                    $type = $this->getTarantoolType($phpType);
                }

                $isNullable = true;

                if (array_key_exists('required', $byTypes)) {
                    $isNullable = false;
                }

                if (!$space->hasProperty($propertyName)) {
                    $opts = [
                        'is_nullable' => $isNullable,
                    ];
                    if ($this->isReference($phpType)) {
                        $opts['reference'] = $this->getSpaceName((string) $phpType);
                    }
                    if (array_key_exists('default', $byTypes)) {
                        $opts['default'] = $schema->formatValue($type, (string) $byTypes['default']);
                    }
                    $space->addProperty($propertyName, $type, $opts);
                }
            }
            if ($this->mapper->hasPlugin(NestedSet::class)) {
                $nested = $this->mapper->getPlugin(NestedSet::class);
                if ($nested->isNested($space)) {
                    $nested->addIndexes($space);
                }
            }
            if (in_array($entity, $this->extensions)) {
                if (!$space->hasProperty('class')) {
                    throw new Exception("$entity has extensions, but not class property is defined");
                }
                $space->addIndex('class');
            }

            if ($class->hasMethod('compute')) {
                $computes[] = $spaceName;
            }
        }


        foreach ($this->repositoryClasses as $repository) {
            $spaceName = $this->getSpaceName($repository);

            if (!$schema->hasSpace($spaceName)) {
                throw new Exception("Repository $spaceName has no entity definition");
            }

            $this->mapRepository($spaceName, $repository);

            $space = $schema->getSpace($spaceName);

            $class = new ReflectionClass($repository);
            $properties = $class->getDefaultProperties();

            if (array_key_exists('indexes', $properties)) {
                foreach ($properties['indexes'] as $i => $index) {

The expression $properties['indexes'] of type null|integer|double|string|boolean|array is not guaranteed to be traversable. How about adding an additional type check?

                    if (!is_array($index)) {
                        $index = (array) $index;
                    }
                    if (!array_key_exists('fields', $index)) {
                        $index = ['fields' => $index];
                    }

                    $index['if_not_exists'] = true;
                    try {
                        $space->addIndex($index);
                    } catch (Exception $e) {
                        $presentation = json_encode($properties['indexes'][$i]);
                        throw new Exception("Failed to add index $presentation. ". $e->getMessage(), 0, $e);
                    }
                }
            }
        }
        foreach ($schema->getSpaces() as $space) {
            if ($space->isSystem()) {
                continue;
            }
            if (!count($space->getIndexes())) {
                if (!$space->hasProperty('id')) {
                    throw new Exception("No primary index on ". $space->getName());
                }
                $space->addIndex(['id']);
            }
        }

        foreach ($computes as $spaceName) {
            $method = new ReflectionMethod($this->entityMapping[$spaceName], 'compute');
            $type = (string) $method->getParameters()[0]->getType();
            $sourceSpace = array_search($type, $this->entityMapping);
            if (!$sourceSpace) {
                throw new Exception("Invalid compute source $type");
            }
            $compute = Closure::fromCallable([$this->entityMapping[$spaceName], 'compute']);
            $this->mapper->getPlugin(Compute::class)->register($sourceSpace, $spaceName, $compute);
        }

        if ($extensionInstances) {
            foreach ($this->extensions as $class => $target) {
                $space = $this->getSpaceName($target);
                $this->mapper->findOrCreate($space, [
                    'class' => $class,
                ]);
            }
        }

        return $this;
    }

    public function getEntityClass(Space $space, array $data)
    {
        $class = parent::getEntityClass($space, $data);
        if (in_array($class, $this->getExtensions())) {
            if (!array_key_exists('class', $data) || !$data['class']) {
                throw new LogicException("Extension without class defined");
            }
            return $data['class'];
        }
        return $class;
    }

    public function setEntityPostfix($postfix)
    {
        $this->entityPostfix = $postfix;
        return $this;
    }

    public function setRepositoryPostfix($postfix)
    {
        $this->repositoryPostifx = $postfix;
        return $this;
    }

    private $spaceNames = [];

    public function getRepositorySpaceName($class)
    {
        return array_search($class, $this->repositoryMapping);
    }

    public function getSpaceName($class)
    {
        if (!array_key_exists($class, $this->spaceNames)) {
            $reflection = new ReflectionClass($class);
            $className = $reflection->getShortName();

            if ($reflection->isSubclassOf(Repository::class)) {

This code seems to be duplicated across your project.

                if ($this->repositoryPostifx) {
                    $className = substr($className, 0, strlen($className) - strlen($this->repositoryPostifx));
                }
            }

            if ($reflection->isSubclassOf(Entity::class)) {

This code seems to be duplicated across your project.

                if ($this->entityPostfix) {
                    $className = substr($className, 0, strlen($className) - strlen($this->entityPostfix));
                }
            }

            $this->spaceNames[$class] = $this->mapper->getSchema()->toUnderscore($className);
        }

        return $this->spaceNames[$class];
    }

    private $tarantoolTypes = [];

    private function isReference(string $type)
    {
        return $type[0] == '\\';
    }

    private function getTarantoolType(string $type)
    {
        if (array_key_exists($type, $this->tarantoolTypes)) {
            return $this->tarantoolTypes[$type];
        }

        if ($this->isReference($type)) {
            return $this->tarantoolTypes[$type] = 'unsigned';
        }

        switch ($type) {
            case 'mixed':
            case 'array':
                return $this->tarantoolTypes[$type] = '*';

            case 'float':
                return $this->tarantoolTypes[$type] = 'number';

            case 'int':
                return $this->tarantoolTypes[$type] = 'unsigned';

            default:
                return $this->tarantoolTypes[$type] = 'string';
        }
    }
}