Skip to content

Commit e919b78

Browse files
Adding EntityManagerInterface use statement if not already added
1 parent bbb7967 commit e919b78

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/Maker/MakeCrud.php

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,12 @@ public function generate(InputInterface $input, ConsoleStyle $io, Generator $gen
163163
Request::class,
164164
Response::class,
165165
Route::class,
166-
EntityManagerInterface::class,
167166
]);
168167

168+
if (EntityManagerInterface::class !== $repositoryClassName) {
169+
$useStatements->addUseStatement(EntityManagerInterface::class);
170+
}
171+
169172
$generator->generateController(
170173
$controllerClassDetails->getFullName(),
171174
'crud/controller/Controller.tpl.php',
@@ -259,6 +262,10 @@ public function generate(InputInterface $input, ConsoleStyle $io, Generator $gen
259262
$useStatements->addUseStatement(EntityRepository::class);
260263
}
261264

265+
if (EntityManagerInterface::class !== $repositoryClassName) {
266+
$useStatements->addUseStatement(EntityManagerInterface::class);
267+
}
268+
262269
$generator->generateFile(
263270
'tests/Controller/'.$testClassDetails->getShortName().'.php',
264271
$usesEntityManager ? 'crud/test/Test.EntityManager.tpl.php' : 'crud/test/Test.tpl.php',

0 commit comments

Comments
 (0)