Completed
Push — master ( 5a8e55...7a79d7 )
by Benjamin
07:38 queued 05:28
created

CMSResolverSubscriber::getSubscribedEvents()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 6
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 0 Features 0
Metric Value
c 1
b 0
f 0
dl 0
loc 6
rs 9.4285
cc 1
eloc 3
nc 1
nop 0
1
<?php
2
3
namespace Alpixel\Bundle\CMSBundle\Listener;
4
5
use Alpixel\Bundle\CMSBundle\Entity\Block;
6
use Alpixel\Bundle\CMSBundle\Entity\Node;
7
use Doctrine\Common\EventSubscriber;
8
use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
9
use Doctrine\ORM\Events;
10
11
class CMSResolverSubscriber implements EventSubscriber
12
{
13
    private $contentTypes;
14
    private $blocks;
15
16
    public function __construct($contentTypes = [], $blocks = [])
17
    {
18
        $this->blocks = $blocks;
19
        $this->contentTypes = $contentTypes;
20
    }
21
22
    public function getSubscribedEvents()
23
    {
24
        return [
25
            Events::loadClassMetadata,
26
        ];
27
    }
28
29
    /**
30
     * @param LoadClassMetadataEventArgs $eventArgs
31
     */
32
    public function loadClassMetadata(LoadClassMetadataEventArgs $eventArgs)
33
    {
34
        $metadata = $eventArgs->getClassMetadata();
35
36
        $discriminatorMap = [];
37
        if (Node::class === $metadata->getName()) {
38
            foreach ($this->contentTypes as $key => $contentType) {
39
                $discriminatorMap[$key] = $contentType['class'];
40
            }
41
        } elseif (Block::class === $metadata->getName()) {
42
            foreach ($this->blocks as $key => $block) {
43
                if (!empty($block['class'])) {
44
                    $discriminatorMap[$key] = $block['class'];
45
                }
46
            }
47
        } else {
48
            return;
49
        }
50
51
        if (!empty($discriminatorMap)) {
52
            $metadata->setDiscriminatorMap($discriminatorMap);
53
        }
54
    }
55
}
56