/**
* Build generators
*
* @param State|State $state
* @return \Scaffold\State|void
*/
public function build(State $state)
{
$model = $state->getControllerModel();
$generator = new ClassGenerator($model->getName());
$generator->setExtendedClass('AbstractRestfulController');
$generator->addUse('Zend\\Mvc\\Controller\\AbstractRestfulController');
$generator->addUse('Doctrine\\ORM\\EntityManager');
$generator->addUse('Zend\\Form\\Form');
$generator->addUse($state->getServiceTraitModel()->getName());
$generator->addTrait($state->getServiceTraitModel()->getClassName());
$this->addProperty($generator, 'entityManager', 'EntityManager');
$this->buildGetEntityManager($generator, $state);
$model->setGenerator($generator);
}
protected function write(State $state, InputInterface $input, OutputInterface $output)
{
$writeState = new State($this->configWriter);
$writeState->addModel($state->getControllerModel());
parent::write($writeState, $input, $output);
}
