Add code to ask class of the new Entity

Author: vacho

src/Command/GeneratorEntityCommand.php

@@ -24,7 +24,8 @@ protected function configure()
     $this
         ->setDefinition(array(
             new InputOption('module','',InputOption::VALUE_REQUIRED, 'The name of the module'),
-            new InputOption('entity','',InputOption::VALUE_REQUIRED, 'The name of the entity')
+            new InputOption('entity','',InputOption::VALUE_REQUIRED, 'The name of the entity'),
+            new InputOption('class','',InputOption::VALUE_REQUIRED, 'The class of the entity')
         ))
         ->setName('generate:entity')
         ->setDescription('Generate entity')
@@ -37,10 +38,11 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
     $module = $input->getOption('module');
     $entity = $input->getOption('entity');
-
+    $class = $input->getOption('class');
+    
     $this
       ->getGenerator()
-      ->generate($module, $entity);
+      ->generate($module, $entity, $class);
 
     $errors = [];
     $dialog->writeGeneratorSummary($output, $errors);
@@ -73,6 +75,26 @@ protected function interact(InputInterface $input, OutputInterface $output)
     }
     $input->setOption('entity', $entity);
 
+    // --entity option
+    $entity = $input->getOption('entity');
+    if (!$entity) {
+    	$entity = $dialog->ask(
+    			$output,
+    			$dialog->getQuestion('Enter the entity name', '')
+    	);
+    }
+    $input->setOption('entity', $entity);
+    
+    // --class option
+    $class = $input->getOption('class');
+    if (!$class) {
+    	$class = $dialog->askConfirmation(
+    			$output,
+    			$dialog->getQuestion('Enter the class of the entity', 'ConfigEntityType', '?'),
+    			true
+    	);
+    }
+    $input->setOption('class', $class);    
   }
 
 

src/Generator/EntityGenerator.php

@@ -15,14 +15,17 @@ class EntityGenerator extends Generator
    * @param  string $class_name   Class name
    * @param  array  $services     List of services
    */
-  public function generate($module, $entity)
+  public function generate($module, $entity, $class)
   {
 
     $parameters = [
       'module' => $module,
-      'entity' => $entity
+      'entity' => $entity,
+      'class' => $class,
     ];
 
+    //switch ConfigEntityType or ContentEntityType
+    
     $this->renderFile(
       'module/config/schema/entity.schema.yml.twig',
       $this->getModulePath($module). '/config/schema/' . $entity . '.schema.yml',