Passed
Pull Request — main (#140)
by Daniel
07:55 queued 02: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 0
CRAP Score 2

Importance

Changes 0
Metric Value
cc 1
eloc 1
nc 1
nop 1
dl 0
loc 3
ccs 0
cts 2
cp 0
crap 2
rs 10
c 0
b 0
f 0
1
<?php
2
3
/*
4
 * This file is part of the Silverback API Components 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\ApiComponentsBundle\Doctrine\Extension\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 ?string $prefix;
27
28
    public function __construct(?string $prefix = '_acb_')
29
    {
30
        $this->prefix = $prefix;
31
    }
32
33
    private function supportsClass(ClassMetadata $classMetadata): bool
34
    {
35
        if (null === $this->prefix) {
36
            return false;
37
        }
38
39
        // Fix: doctrine console command - may not have a class
40
        if (!$reflectionClass = $classMetadata->getReflectionClass()) {
41
            return false;
42
        }
43
44
        if (0 !== strpos($reflectionClass->getNamespaceName(), 'Silverback\ApiComponentsBundle\\')) {
45
            return false;
46
        }
47
48
        return true;
49
    }
50
51
    private function setPrimaryTable(ClassMetadata $classMetadata): void
52
    {
53
        $converter = new CamelCaseToSnakeCaseNameConverter();
54
        if (!$classMetadata->isInheritanceTypeSingleTable() || $classMetadata->getName() === $classMetadata->rootEntityName) {
55
            $classMetadata->setPrimaryTable(
56
                [
57
                    'name' => $this->prefix . $converter->normalize($classMetadata->getTableName()),
58
                ]
59
            );
60
        }
61
    }
62
63
    private function setJoinTableName(ClassMetadata $classMetadata): void
64
    {
65
        foreach ($classMetadata->getAssociationMappings() as $fieldName => $mapping) {
66
            if (ClassMetadataInfo::MANY_TO_MANY === $mapping['type'] && $mapping['isOwningSide'] && !\array_key_exists('inherited', $mapping)) {
67
                $mappedTableName = $mapping['joinTable']['name'];
68
                $classMetadata->associationMappings[$fieldName]['joinTable']['name'] = $this->prefix . $mappedTableName;
69
            }
70
        }
71
    }
72
73
    public function loadClassMetadata(LoadClassMetadataEventArgs $eventArgs): void
74
    {
75
        $classMetadata = $eventArgs->getClassMetadata();
76
        if (!$this->supportsClass($classMetadata)) {
77
            return;
78
        }
79
        $this->setPrimaryTable($classMetadata);
80
        $this->setJoinTableName($classMetadata);
81
    }
82
}
83