Passed
Push — feature/unit-tests ( 84c996...7c5ff9 )
by Daniel
05:24
created

TablePrefixExtension::__construct()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 3
Code Lines 1

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1

Importance

Changes 2
Bugs 0 Features 0
Metric Value
eloc 1
c 2
b 0
f 0
dl 0
loc 3
ccs 2
cts 2
cp 1
rs 10
cc 1
nc 1
nop 1
crap 1
1
<?php
2
3
/*
4
 * This file is part of the Silverback API Component Bundle Project
5
 *
6
 * (c) Daniel West <[email protected]>
7
 *
8
 * For the full copyright and license information, please view the LICENSE
9
 * file that was distributed with this source code.
10
 */
11
12
declare(strict_types=1);
13
14
namespace Silverback\ApiComponentBundle\Doctrine\Extension;
15
16
use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
17
use Doctrine\ORM\Mapping\ClassMetadata;
18
use Doctrine\ORM\Mapping\ClassMetadataInfo;
19
use Symfony\Component\Serializer\NameConverter\CamelCaseToSnakeCaseNameConverter;
20
21
/**
22
 * @author Daniel West <[email protected]>
23
 */
24
class TablePrefixExtension
25
{
26
    private $prefix;
27
28 3
    public function __construct(?string $prefix = '_acb_')
29
    {
30 3
        $this->prefix = $prefix;
31 3
    }
32
33 3
    private function supportsClass(ClassMetadata $classMetadata): bool
34
    {
35 3
        if (null === $this->prefix) {
36
            return false;
37
        }
38
39 3
        if (0 !== strpos($classMetadata->getReflectionClass()->getNamespaceName(), 'Silverback\ApiComponentBundle\\')) {
40 3
            return false;
41
        }
42
43 3
        return true;
44
    }
45
46 3
    private function setPrimaryTable(ClassMetadata $classMetadata): void
47
    {
48 3
        $converter = new CamelCaseToSnakeCaseNameConverter();
49 3
        if (!$classMetadata->isInheritanceTypeSingleTable() || $classMetadata->getName() === $classMetadata->rootEntityName) {
50 3
            $classMetadata->setPrimaryTable([
51 3
                'name' => $this->prefix . $converter->normalize($classMetadata->getTableName()),
52
            ]);
53
        }
54 3
    }
55
56 3
    private function setJoinTableName(ClassMetadata $classMetadata): void
57
    {
58 3
        foreach ($classMetadata->getAssociationMappings() as $fieldName => $mapping) {
59 3
            if (ClassMetadataInfo::MANY_TO_MANY === $mapping['type'] && $mapping['isOwningSide'] && !\array_key_exists('inherited', $mapping)) {
60 3
                $mappedTableName = $mapping['joinTable']['name'];
61 3
                $classMetadata->associationMappings[$fieldName]['joinTable']['name'] = $this->prefix . $mappedTableName;
62
            }
63
        }
64 3
    }
65
66 3
    public function loadClassMetadata(LoadClassMetadataEventArgs $eventArgs): void
67
    {
68 3
        $classMetadata = $eventArgs->getClassMetadata();
69 3
        if (!$this->supportsClass($classMetadata)) {
70 3
            return;
71
        }
72 3
        $this->setPrimaryTable($classMetadata);
73 3
        $this->setJoinTableName($classMetadata);
74 3
    }
75
}
76