Fixed issue with no entity directory in symfony 3.4

src/Maker/MakeEntity.php

@@ -99,20 +99,6 @@ public function interact(InputInterface $input, ConsoleStyle $io, Command $comma
             return;
         }
 
-        $entityFinder = $this->fileManager->createFinder('src/Entity/')
-            // remove if/when we allow entities in subdirectories
-            ->depth('<1')
-            ->name('*.php');
-        $classes = [];
-        /** @var SplFileInfo $item */
-        foreach ($entityFinder as $item) {
-            if (!$item->getRelativePathname()) {
-                continue;
-            }
-
-            $classes[] = str_replace(['.php', '/'], ['', '\\'], $item->getRelativePathname());
-        }
-
         $argument = $command->getDefinition()->getArgument('name');
         $question = $this->createEntityClassQuestion($argument->getDescription());
         $value = $io->askQuestion($question);
@@ -464,25 +450,39 @@ private function printAvailableTypes(ConsoleStyle $io)
         $printSection($allTypes);
     }
 
-    private function createEntityClassQuestion(string $questionText): Question
+    private function getEntityClasses(): array
     {
-        $entityFinder = $this->fileManager->createFinder('src/Entity/')
-            // remove if/when we allow entities in subdirectories
-            ->depth('<1')
-            ->name('*.php');
+        $entityPath = $this->generator->getRootNamespace() ?
+            'src/'.$this->generator->getRootNamespace().'/Entity/' :
+            'src/Entity/';
+
         $classes = [];
-        /** @var SplFileInfo $item */
-        foreach ($entityFinder as $item) {
-            if (!$item->getRelativePathname()) {
-                continue;
-            }
+        try {
+            $entityFinder = $this->fileManager->createFinder($entityPath)
+                // remove if/when we allow entities in subdirectories
+                ->depth('<1')
+                ->name('*.php');
+
+            /** @var SplFileInfo $item */
+            foreach ($entityFinder as $item) {
+                if (!$item->getRelativePathname()) {
+                    continue;
+                }
 
-            $classes[] = str_replace('/', '\\', str_replace('.php', '', $item->getRelativePathname()));
+                $classes[] = str_replace('/', '\\', str_replace('.php', '', $item->getRelativePathname()));
+            }
+        } catch (\InvalidArgumentException $e) {
+            // no entity directory exists nothing to auto-complete
         }
 
+        return $classes;
+    }
+
+    private function createEntityClassQuestion(string $questionText): Question
+    {
         $question = new Question($questionText);
         $question->setValidator([Validator::class, 'notBlank']);
-        $question->setAutocompleterValues($classes);
+        $question->setAutocompleterValues($this->getEntityClasses());
 
         return $question;
     }