Passed
Pull Request — feature/publishable (#43)
by Vincent
06:29
created

TablePrefixExtension::loadClassMetadata()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 8
Code Lines 5

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 6
CRAP Score 2

Importance

Changes 0
Metric Value
eloc 5
c 0
b 0
f 0
dl 0
loc 8
ccs 6
cts 6
cp 1
rs 10
cc 2
nc 2
nop 1
crap 2
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\Extension\Doctrine\ORM;
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 9
    public function __construct(?string $prefix = '_acb_')
29
    {
30 9
        $this->prefix = $prefix;
31 9
    }
32
33 9
    private function supportsClass(ClassMetadata $classMetadata): bool
34
    {
35 9
        if (null === $this->prefix) {
36
            return false;
37
        }
38
39 9
        if (0 !== strpos($classMetadata->getReflectionClass()->getNamespaceName(), 'Silverback\ApiComponentBundle\\')) {
40 9
            return false;
41
        }
42
43 9
        return true;
44
    }
45
46 9
    private function setPrimaryTable(ClassMetadata $classMetadata): void
47
    {
48 9
        $converter = new CamelCaseToSnakeCaseNameConverter();
49 9
        if (!$classMetadata->isInheritanceTypeSingleTable() || $classMetadata->getName() === $classMetadata->rootEntityName) {
50 9
            $classMetadata->setPrimaryTable([
51 9
                'name' => $this->prefix . $converter->normalize($classMetadata->getTableName()),
52
            ]);
53
        }
54 9
    }
55
56 9
    private function setJoinTableName(ClassMetadata $classMetadata): void
57
    {
58 9
        foreach ($classMetadata->getAssociationMappings() as $fieldName => $mapping) {
59 9
            if (ClassMetadataInfo::MANY_TO_MANY === $mapping['type'] && $mapping['isOwningSide'] && !\array_key_exists('inherited', $mapping)) {
60 9
                $mappedTableName = $mapping['joinTable']['name'];
61 9
                $classMetadata->associationMappings[$fieldName]['joinTable']['name'] = $this->prefix . $mappedTableName;
62
            }
63
        }
64 9
    }
65
66 9
    public function loadClassMetadata(LoadClassMetadataEventArgs $eventArgs): void
67
    {
68 9
        $classMetadata = $eventArgs->getClassMetadata();
69 9
        if (!$this->supportsClass($classMetadata)) {
70 9
            return;
71
        }
72 9
        $this->setPrimaryTable($classMetadata);
73 9
        $this->setJoinTableName($classMetadata);
74 9
    }
75
}
76