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