Test Failed
Pull Request — 1.1 (#49)
by Guido
17:25 queued 04:21
created

FluentDriver::setFluentFactory()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 4
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 1
CRAP Score 1

Importance

Changes 0
Metric Value
cc 1
eloc 2
nc 1
nop 1
dl 0
loc 4
rs 10
c 0
b 0
f 0
ccs 1
cts 1
cp 1
crap 1
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
        $this->fluentFactory = function (ClassMetadata $metadata) {
33 4
            return new Builder(new ClassMetadataBuilder($metadata));
34 4
        };
35
36
        $this->mappers = new MapperSet();
37 92
        $this->addMappings($mappings);
38 92
    }
39 92
40
    /**
41
     * Loads the metadata for the specified class into the provided container.
42
     *
43
     * @param string        $className
44
     * @param ClassMetadata $metadata
45
     */
46
    public function loadMetadataForClass($className, ClassMetadata $metadata)
47 6
    {
48
        $this->mappers->getMapperFor($className)->map(
49 6
            $this->getFluent($metadata)
50 5
        );
51 5
    }
52 5
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
    public function getAllClassNames()
61 12
    {
62
        return $this->mappers->getClassNames();
63 12
    }
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
    public function isTransient($className)
74 4
    {
75
        return
76
            !$this->mappers->hasMapperFor($className) ||
77 4
            $this->mappers->getMapperFor($className)->isTransient();
78 4
    }
79
80
    /**
81
     * Adds an array of mapping classes / objects to the driver.
82
     *
83
     * @param string[]|Mapping[] $mappings
84 92
     *
85
     * @throws MappingException
86 92
     * @throws InvalidArgumentException
87 82
     */
88 1
    public function addMappings(array $mappings = [])
89
    {
90
        foreach ($mappings as $mapping) {
91 81
            $this->addMapping($mapping);
92
        }
93 81
    }
94 1
95
    /**
96
     * @param string|Mapping $mapping
97 80
     *
98 92
     * @throws MappingException
99 92
     * @throws InvalidArgumentException
100
     *
101
     * @return void
102
     */
103
    public function addMapping($mapping)
104
    {
105
        $this->mappers->add($mapping instanceof Mapping ?
106
            $mapping : $this->createMapping($mapping)
107
        );
108 87
    }
109
110 87
    /**
111 87
     * @return MapperSet
112
     */
113
    public function getMappers()
114
    {
115
        return $this->mappers;
116 5
    }
117
118 5
    /**
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
    public function setFluentFactory(callable $factory)
126
    {
127
        $this->fluentFactory = $factory;
128 1
    }
129
130 1
    /**
131 1
     * @param ClassMetadata $metadata
132
     *
133
     * @return Fluent
134
     */
135
    protected function getFluent(ClassMetadata $metadata)
136
    {
137
        return call_user_func($this->fluentFactory, $metadata);
138 5
    }
139
140 5
    /**
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
    protected function createMapping($class)
150
    {
151
        if (!class_exists($class)) {
152
            throw new InvalidArgumentException("Mapping class [{$class}] does not exist");
153
        }
154
155
        $mapping = new $class();
156
157
        if (!$mapping instanceof Mapping) {
158
            throw new InvalidArgumentException("Mapping class [{$class}] should implement ".Mapping::class);
159
        }
160
161
        return $mapping;
162
    }
163
}
164