Completed
Pull Request — 1.1 (#49)
by Guido
27:15 queued 24:38
created

FluentDriver::addMappings()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 6
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 5
CRAP Score 2

Importance

Changes 0
Metric Value
cc 2
eloc 3
nc 2
nop 1
dl 0
loc 6
ccs 5
cts 5
cp 1
crap 2
rs 9.4285
c 0
b 0
f 0
1
<?php
2
3
namespace LaravelDoctrine\Fluent;
4
5
use Doctrine\Common\Persistence\Mapping\ClassMetadata;
6
use Doctrine\Common\Persistence\Mapping\Driver\MappingDriver;
7
use Doctrine\ORM\Mapping\Builder\ClassMetadataBuilder;
8
use Doctrine\ORM\Mapping\MappingException;
9
use InvalidArgumentException;
10
use LaravelDoctrine\Fluent\Builders\Builder;
11
use LaravelDoctrine\Fluent\Mappers\MapperSet;
12
13
class FluentDriver implements MappingDriver
14
{
15
    /**
16
     * @var MapperSet
17
     */
18
    protected $mappers;
19
20
    /**
21
     * @var callable
22
     */
23
    protected $fluentFactory;
24
25
    /**
26
     * Initializes a new FluentDriver that will load given Mapping classes / objects.
27
     *
28
     * @param string[]|Mapping[] $mappings
29
     */
30
    public function __construct(array $mappings = [])
31
    {
32 20
        $this->fluentFactory = function (ClassMetadata $metadata) {
33 20
            return new Builder(new ClassMetadataBuilder($metadata));
34
        };
35
36 372
        $this->mappers = new MapperSet();
37 372
        $this->addMappings($mappings);
38 372
    }
39
40
    /**
41
     * Loads the metadata for the specified class into the provided container.
42
     *
43
     * @param string        $className
44
     * @param ClassMetadata $metadata
45
     */
46 28
    public function loadMetadataForClass($className, ClassMetadata $metadata)
47
    {
48 28
        $this->mappers->getMapperFor($className)->map(
49 24
            $this->getFluent($metadata)
50 12
        );
51 24
    }
52
53
    /**
54
     * Gets the names of all mapped classes known to this driver.
55
     *
56
     * @throws MappingException
57
     *
58
     * @return string[] The names of all mapped classes known to this driver.
59
     */
60 52
    public function getAllClassNames()
61
    {
62 52
        return $this->mappers->getClassNames();
63
    }
64
65
    /**
66
     * Returns whether the class with the specified name should have its metadata loaded.
67
     * This is only the case if it is either mapped as an Entity or a MappedSuperclass.
68
     *
69
     * @param string $className
70
     *
71
     * @return bool
72
     */
73 16
    public function isTransient($className)
74
    {
75
        return
76 16
            !$this->mappers->hasMapperFor($className) ||
77 16
            $this->mappers->getMapperFor($className)->isTransient();
78
    }
79
80
    /**
81
     * Adds an array of mapping classes / objects to the driver.
82
     *
83
     * @param string[]|Mapping[] $mappings
84
     *
85
     * @throws MappingException
86
     * @throws InvalidArgumentException
87
     */
88 372
    public function addMappings(array $mappings = [])
89
    {
90 372
        foreach ($mappings as $mapping) {
91 332
            $this->addMapping($mapping);
92 186
        }
93 372
    }
94
95
    /**
96
     * @param string|Mapping $mapping
97
     *
98
     * @throws MappingException
99
     * @throws InvalidArgumentException
100
     *
101
     * @return void
102
     */
103 360
    public function addMapping($mapping)
104
    {
105 360
        $this->mappers->add($mapping instanceof Mapping ?
106 360
            $mapping : $this->createMapping($mapping)
107 176
        );
108 352
    }
109
110
    /**
111
     * @return MapperSet
112
     */
113 24
    public function getMappers()
114
    {
115 24
        return $this->mappers;
116
    }
117
118
    /**
119
     * Override the default Fluent factory method with a custom one.
120
     * Use this to implement your own Fluent builder.
121
     * The method will receive a ClassMetadata object as its only argument.
122
     *
123
     * @param callable $factory
124
     */
125 4
    public function setFluentFactory(callable $factory)
126
    {
127 4
        $this->fluentFactory = $factory;
128 4
    }
129
130
    /**
131
     * @param ClassMetadata $metadata
132
     *
133
     * @return Fluent
134
     */
135 24
    protected function getFluent(ClassMetadata $metadata)
136
    {
137 24
        return call_user_func($this->fluentFactory, $metadata);
138
    }
139
140
    /**
141
     * Create a mapping object from a mapping class, assuming an empty constructor.
142
     *
143
     * @param string $class
144
     *
145
     * @throws InvalidArgumentException
146
     *
147
     * @return Mapping
148
     */
149 332
    protected function createMapping($class)
150
    {
151 332
        if (!class_exists($class)) {
152 4
            throw new InvalidArgumentException("Mapping class [{$class}] does not exist");
153
        }
154
155 328
        $mapping = new $class();
156
157 328
        if (!$mapping instanceof Mapping) {
158 4
            throw new InvalidArgumentException("Mapping class [{$class}] should implement ".Mapping::class);
159
        }
160
161 324
        return $mapping;
162
    }
163
}
164