/**
* 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);
}
