Skip to content

Commit d3487ef

Browse files
patrickbrouwerspatrickbrouwers
committed
Add path loader to fluent driver
1 parent 99f1182 commit d3487ef

File tree

2 files changed

+170
-4
lines changed

2 files changed

+170
-4
lines changed

‎src/FluentDriver.php

+92-2
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,15 @@
66
use Doctrine\Common\Persistence\Mapping\Driver\MappingDriver;
77
use Doctrine\ORM\Mapping\Builder\ClassMetadataBuilder;
88
use Doctrine\ORM\Mapping\MappingException;
9+
use FilesystemIterator;
910
use InvalidArgumentException;
1011
use LaravelDoctrine\Fluent\Builders\Builder;
1112
use LaravelDoctrine\Fluent\Mappers\MapperSet;
13+
use RecursiveDirectoryIterator;
14+
use RecursiveIteratorIterator;
15+
use RecursiveRegexIterator;
16+
use ReflectionClass;
17+
use RegexIterator;
1218

1319
class FluentDriver implements MappingDriver
1420
{
@@ -22,20 +28,37 @@ class FluentDriver implements MappingDriver
2228
*/
2329
protected $fluentFactory;
2430

31+
/**
32+
* The file extension of mapping documents.
33+
*
34+
* @var string
35+
*/
36+
protected $fileExtension = '.php';
37+
2538
/**
2639
* Initializes a new FileDriver that looks in the given path(s) for mapping
2740
* documents and operates in the specified operating mode.
2841
*
2942
* @param string[] $mappings
43+
* @param array $paths
44+
*
45+
* @throws MappingException
3046
*/
31-
public function __construct(array $mappings = [])
47+
public function __construct(array $mappings = [], array $paths = [])
3248
{
3349
$this->fluentFactory = function (ClassMetadata $metadata) {
3450
return new Builder(new ClassMetadataBuilder($metadata));
3551
};
3652

3753
$this->mappers = new MapperSet();
38-
$this->addMappings($mappings);
54+
55+
if (!empty($paths)) {
56+
$this->loadPaths($paths);
57+
}
58+
59+
if (!empty($mappings)) {
60+
$this->addMappings($mappings);
61+
}
3962
}
4063

4164
/**
@@ -78,8 +101,75 @@ public function isTransient($className)
78101
$this->mappers->getMapperFor($className)->isTransient();
79102
}
80103

104+
/**
105+
* @param array $paths
106+
*
107+
* @throws MappingException
108+
*/
109+
public function loadPaths(array $paths)
110+
{
111+
$classes = [];
112+
$includedFiles = [];
113+
114+
foreach ($paths as $path) {
115+
if (!is_dir($path)) {
116+
throw MappingException::fileMappingDriversRequireConfiguredDirectoryPath($path);
117+
}
118+
119+
$iterator = new RegexIterator(
120+
new RecursiveIteratorIterator(
121+
new RecursiveDirectoryIterator($path, FilesystemIterator::SKIP_DOTS),
122+
RecursiveIteratorIterator::LEAVES_ONLY
123+
),
124+
'/^.+'.preg_quote($this->fileExtension).'$/i',
125+
RecursiveRegexIterator::GET_MATCH
126+
);
127+
128+
foreach ($iterator as $file) {
129+
$sourceFile = $file[0];
130+
131+
if (!preg_match('(^phar:)i', $sourceFile)) {
132+
$sourceFile = realpath($sourceFile);
133+
}
134+
135+
require_once $sourceFile;
136+
137+
$includedFiles[] = $sourceFile;
138+
}
139+
140+
$declared = get_declared_classes();
141+
142+
foreach ($declared as $className) {
143+
$rc = new ReflectionClass($className);
144+
$sourceFile = $rc->getFileName();
145+
146+
if (!in_array($sourceFile, $includedFiles)) {
147+
continue;
148+
}
149+
150+
if ($rc->isAbstract() || $rc->isInterface()) {
151+
continue;
152+
}
153+
154+
if (!$rc->implementsInterface(Mapping::class)) {
155+
continue;
156+
}
157+
158+
if ($this->isTransient($className)) {
159+
160+
/** @var Mapping $mapping */
161+
$mapping = $rc->newInstanceWithoutConstructor();
162+
163+
$this->addMapping($mapping);
164+
}
165+
}
166+
}
167+
}
168+
81169
/**
82170
* @param string[] $mappings
171+
*
172+
* @throws InvalidArgumentException
83173
*/
84174
public function addMappings(array $mappings = [])
85175
{

‎tests/FluentDriverTest.php

+78-2
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ public function test_it_should_load_metadata_for_mappings_passed_as_constructor_
8282
StubEntity::class,
8383
new ClassMetadataInfo(StubEntity::class)
8484
);
85+
8586
$this->assertInstanceOf(
8687
EntityMapper::class,
8788
$driver->getMappers()->getMapperFor(StubEntity::class)
@@ -91,6 +92,7 @@ public function test_it_should_load_metadata_for_mappings_passed_as_constructor_
9192
StubEmbeddable::class,
9293
new ClassMetadataInfo(StubEmbeddable::class)
9394
);
95+
9496
$this->assertInstanceOf(
9597
EmbeddableMapper::class,
9698
$driver->getMappers()->getMapperFor(StubEmbeddable::class)
@@ -100,6 +102,7 @@ public function test_it_should_load_metadata_for_mappings_passed_as_constructor_
100102
StubMappedSuperClass::class,
101103
new ClassMetadataInfo(StubMappedSuperClass::class)
102104
);
105+
103106
$this->assertInstanceOf(
104107
MappedSuperClassMapper::class,
105108
$driver->getMappers()->getMapperFor(StubMappedSuperClass::class)
@@ -126,9 +129,81 @@ public function test_can_add_array_of_new_mappings()
126129
);
127130
}
128131

132+
public function test_can_load_mappings_through_file_path()
133+
{
134+
$driver = new FluentDriver([], [__DIR__ . '/' . 'Stubs/Mappings']);
135+
136+
$this->assertContains(
137+
StubEntity::class,
138+
$driver->getAllClassNames()
139+
);
140+
141+
$this->assertContains(
142+
StubEmbeddable::class,
143+
$driver->getAllClassNames()
144+
);
145+
146+
$this->assertContains(
147+
StubMappedSuperClass::class,
148+
$driver->getAllClassNames()
149+
);
150+
151+
$driver->loadMetadataForClass(
152+
StubEntity::class,
153+
new ClassMetadataInfo(StubEntity::class)
154+
);
155+
156+
$this->assertInstanceOf(
157+
EntityMapper::class,
158+
$driver->getMappers()->getMapperFor(StubEntity::class)
159+
);
160+
161+
$driver->loadMetadataForClass(
162+
StubEmbeddable::class,
163+
new ClassMetadataInfo(StubEmbeddable::class)
164+
);
165+
166+
$this->assertInstanceOf(
167+
EmbeddableMapper::class,
168+
$driver->getMappers()->getMapperFor(StubEmbeddable::class)
169+
);
170+
171+
$driver->loadMetadataForClass(
172+
StubMappedSuperClass::class,
173+
new ClassMetadataInfo(StubMappedSuperClass::class)
174+
);
175+
176+
$this->assertInstanceOf(
177+
MappedSuperClassMapper::class,
178+
$driver->getMappers()->getMapperFor(StubMappedSuperClass::class)
179+
);
180+
}
181+
182+
public function test_can_load_paths()
183+
{
184+
$driver = new FluentDriver();
185+
$driver->loadPaths([__DIR__ . '/' . 'Stubs/Mappings']);
186+
187+
$this->assertContains(
188+
StubEntity::class,
189+
$driver->getAllClassNames()
190+
);
191+
192+
$this->assertContains(
193+
StubEmbeddable::class,
194+
$driver->getAllClassNames()
195+
);
196+
197+
$this->assertContains(
198+
StubMappedSuperClass::class,
199+
$driver->getAllClassNames()
200+
);
201+
}
202+
129203
public function test_the_given_mapping_class_should_exist()
130204
{
131-
$this->setExpectedException(\InvalidArgumentException::class, 'Mapping class [Tests\DoesnExist] does not exist');
205+
$this->setExpectedException(\InvalidArgumentException::class,
206+
'Mapping class [Tests\DoesnExist] does not exist');
132207

133208
$driver = new FluentDriver;
134209

@@ -139,7 +214,8 @@ public function test_the_given_mapping_class_should_exist()
139214

140215
public function test_the_given_mapping_class_should_implement_mapping()
141216
{
142-
$this->setExpectedException(\InvalidArgumentException::class, 'Mapping class [Tests\Stubs\Entities\StubEntity] should implement LaravelDoctrine\Fluent\Mapping');
217+
$this->setExpectedException(\InvalidArgumentException::class,
218+
'Mapping class [Tests\Stubs\Entities\StubEntity] should implement LaravelDoctrine\Fluent\Mapping');
143219

144220
$driver = new FluentDriver;
145221

0 commit comments

Comments
 (0)