Completed
Pull Request — master (#10)
by Mariusz
03:23
created

EntityClassHelper::supplementEntitySource()   A

Complexity

Conditions 3
Paths 3

Size

Total Lines 14
Code Lines 8

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 3
eloc 8
nc 3
nop 1
dl 0
loc 14
rs 9.4285
c 0
b 0
f 0
1
<?php
2
3
namespace Xsolve\Associate\DoctrineOrm\Source;
4
5
use Xsolve\Associate\DoctrineOrm\Metadata\ClassMetadataAdapter;
6
use Xsolve\Associate\DoctrineOrm\Metadata\MetadataAdapterProvider;
7
8
class EntityClassHelper
9
{
10
    /**
11
     * @var MetadataAdapterProvider
12
     */
13
    protected $metadataAdapterProvider;
14
15
    /**
16
     * @param MetadataAdapterProvider $metadataAdapterProvider
17
     */
18
    public function __construct(MetadataAdapterProvider $metadataAdapterProvider)
19
    {
20
        $this->metadataAdapterProvider = $metadataAdapterProvider;
21
    }
22
23
    /**
24
     * @param EntitySource $entitySource
25
     *
26
     * @throws \Exception if one common entity class cannot be determined
27
     */
28
    public function supplementEntitySource(EntitySource $entitySource): void
29
    {
30
        if ($entitySource->hasEntityClass()) {
31
            return;
32
        }
33
34
        $entityClassName = $this->metadataAdapterProvider->getClassNameForEntities($entitySource->getEntities());
35
        $entitySource->setEntityClass($entityClassName);
36
37
        $classMetadataAdapter = $this->metadataAdapterProvider->getClassMetadataAdapterByClassName($entityClassName);
38
        if (!$classMetadataAdapter instanceof ClassMetadataAdapter) {
39
            throw new \Exception();
40
        }
41
        $entitySource->setClassMetadataAdapter($classMetadataAdapter);
42
    }
43
}
44