public function addToString() { $classGenerator = $this->getClassGenerator(); $methodGenerator = new CodeGenerator\MethodGenerator(); $methodGenerator->setName('__toString'); $idAttribute = 'id'; $body = <<<METHODBODY \$attributes = array(\$this->{$idAttribute}); METHODBODY; foreach (static::$secondaryProperties as $attributeName => $propertyName) { $body .= <<<METHODBODY if (!empty(\$this->{$propertyName})) { \$attributes[] = "{$attributeName}={\$this->{$propertyName}}"; } METHODBODY; } $body .= <<<METHODBODY foreach (\$this->attributes as \$attributeName => \$value) { if (!empty(\$value)) { \$attributes[] = "{\$attributeName}={\$value}"; } } return implode('!', \$attributes); METHODBODY; $methodGenerator->setBody($body); $classGenerator->addMethodFromGenerator($methodGenerator); }
/** * Build method factory * * @param ClassGenerator $generator */ public function buildFactory(ClassGenerator $generator) { $docBlock = new DocBlockGenerator('@return ' . $this->config->getName()); $factory = new MethodGenerator(); $factory->setDocBlock($docBlock); $factory->setName('factory'); $factory->setBody('return new ' . $this->config->getName() . '();'); $generator->addMethodFromGenerator($factory); }
/** * @param string $name * @param Code $code * @return PhpMethod */ public function addMethod($name, Code $code) { $this->code[$name] = $code; $method = new MethodGenerator(); $method->setName($name); $method->setVisibility(MethodGenerator::VISIBILITY_PUBLIC); // $code remains object until turning to string $method->setBody($code); $this->addMethodFromGenerator($method); return $method; }
/** * Build method factory * * @param ClassGenerator $generator * @param \Scaffold\State $state */ public function buildFactory(ClassGenerator $generator, State $state) { $repository = ucfirst($state->getRepositoryModel()->getClassName()); $docBlock = new DocBlockGenerator(); $docBlock->setTag(new Tag\GenericTag('return', $this->config->getName())); $factory = new MethodGenerator(); $factory->setDocBlock($docBlock); $factory->setName('factory'); $factory->setBody('return $this->get' . $repository . '()->factory();'); $generator->addMethodFromGenerator($factory); }
public function postRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $table = $part->getTable(); $tags = array(array('name' => 'return', 'description' => 'array')); $docblock = new DocBlockGenerator('Initialize indexes'); $docblock->setTags($tags); $resultIndexList = array(); $indexList = $table->getIndex(); foreach ($indexList as $index) { $resIndex = $index->toArray(); $resIndex['column_list'] = array(); switch ($index->getType()) { case AbstractIndex::TYPE_PRIMARY: $resIndex['type'] = new ValueGenerator('AbstractModel::INDEX_PRIMARY', ValueGenerator::TYPE_CONSTANT); break; case AbstractIndex::TYPE_KEY: $resIndex['type'] = new ValueGenerator('AbstractModel::INDEX_KEY', ValueGenerator::TYPE_CONSTANT); break; case AbstractIndex::TYPE_UNIQUE: $resIndex['type'] = new ValueGenerator('AbstractModel::INDEX_UNIQUE', ValueGenerator::TYPE_CONSTANT); break; } foreach ($resIndex['columns'] as $col) { $resIndex['column_list'][] = $col['column_name']; } unset($resIndex['columns']); $resultIndexList[$index->getName()] = $resIndex; } $property = new PropertyGenerator('indexList', $resultIndexList, PropertyGenerator::FLAG_PROTECTED); $body = preg_replace("#^(\\s*)protected #", "\\1", $property->generate()) . "\n"; $method = new MethodGenerator(); $method->setName('initIndexList'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setFinal(true); $method->setDocBlock($docblock); $method->setBody(<<<EOS {$body} \$this->indexList = \$indexList; \$this->setupIndexList(); EOS ); $file->getClass()->addMethodFromGenerator($method); }
/** * Generate an __invoke method */ protected function addInvokeMethod() { // set action body $body = ['// add view helper code here', '$output = \'\';', '', 'return $output;']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('__invoke'); $method->setBody($body); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Called when view helper is executed', null, [new ReturnTag(['string'])])); } // add method $this->addMethodFromGenerator($method); }
/** * Generate an init method */ protected function addInitMethod() { // set action body $body = array('// add input objects here'); $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('init'); $method->setBody($body); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Generate input filter by adding inputs')); } // add method $this->addMethodFromGenerator($method); }
/** * Generate an __invoke method */ protected function addInvokeMethod() { // set action body $body = ['// add controller plugin code here']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('__invoke'); $method->setBody($body); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Called when controller plugin is executed', null, [new ReturnTag(['mixed'])])); } // add method $this->addMethodFromGenerator($method); }
/** * Generate an init method */ protected function addInitMethod() { // set action body $body = ['// add form elements and form configuration here']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('init'); $method->setBody($body); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Generate form by adding elements')); } // add method $this->addMethodFromGenerator($method); }
public function postRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $table = $part->getTable(); $tags = array(array('name' => 'var', 'description' => 'array значения по-умолчанию для полей')); $docblock = new \Zend\Code\Generator\DocBlockGenerator('Значения по-умолчанию для полей'); $docblock->setTags($tags); $columnCollection = $table->getColumn(); if (!$columnCollection) { return; } $defaults = ''; /** @var $column Column */ foreach ($columnCollection as $column) { $columnName = $column->getName(); $defaultValue = $column->getColumnDefault(); if (substr($columnName, -5) == '_date') { $defaults .= "\$this->setDefaultRule('" . $columnName . "', date('Y-m-d H:i:s'));\n "; } elseif ($defaultValue == 'CURRENT_TIMESTAMP') { $defaults .= "\$this->setDefaultRule('" . $columnName . "', date('Y-m-d H:i:s'));\n "; } elseif (!empty($defaultValue)) { $defaults .= '$this->setDefaultRule(\'' . $columnName . '\', \'' . (string) $defaultValue . '\');' . "\n "; } elseif (is_null($defaultValue) && $column->isNullable()) { $defaults .= '$this->setDefaultRule(\'' . $columnName . '\', null);' . "\n "; } } $tags = array(array('name' => 'return', 'description' => 'void')); $docblock = new DocBlockGenerator('Инициализация значений по-умолчанию'); $docblock->setTags($tags); $method = new MethodGenerator(); $method->setName('initDefaultsRules'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setFinal(true); $method->setDocBlock($docblock); $method->setBody(<<<EOS {$defaults} \$this->setupDefaultsRules(); EOS ); $file->getClass()->addMethodFromGenerator($method); }
/** * Generate a filter method */ protected function addFilterMethod() { // set action body $body = ['// add filter code here', 'return $value;']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('filter'); $method->setBody($body); $method->setParameters([new ParameterGenerator('value', 'mixed')]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Called when filter is executed', null, [new ParamTag('value', ['mixed']), new ReturnTag(['mixed'])])); } // add method $this->addMethodFromGenerator($method); }
/** * Generate the create service method * * @param string $className * @param string $managerName */ protected function addCreateServiceMethod($className, $managerName) { // set action body $body = array('/** @var ServiceLocatorAwareInterface $' . $managerName . ' */', '$serviceLocator = $' . $managerName . '->getServiceLocator();', '', '$instance = new ' . $className . '();', '', 'return $instance;'); $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('createService'); $method->setBody($body); $method->setParameters(array(new ParameterGenerator($managerName, 'ServiceLocatorInterface'))); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Create service', null, array(new ParamTag($managerName, array('ServiceLocatorInterface')), new ReturnTag(array($className))))); } // add method $this->addMethodFromGenerator($method); }
/** * Generate a isValid method */ protected function addIsValidMethod() { // set action body $body = ['$this->setValue((string) $value);', '', '// add validation code here', '$isValid = true;', '', 'if (!$isValid) {', ' $this->error(self::INVALID);', ' return false;', '}', '', 'return true;']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('isValid'); $method->setBody($body); $method->setParameters([new ParameterGenerator('value', 'mixed')]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Called when validator is executed', null, [new ParamTag('value', ['mixed']), new ReturnTag(['mixed'])])); } // add method $this->addMethodFromGenerator($method); }
/** * Generate the create service method * * @param string $className * @param string $moduleName * @param string $tableName */ protected function addCreateServiceMethod($className, $moduleName, $tableName) { $managerName = 'serviceLocator'; $tableGatewayName = ucfirst($tableName) . 'TableGateway'; $tableGatewayService = $moduleName . '\\' . $this->config['namespaceTableGateway'] . '\\' . ucfirst($tableName); // set action body $body = ['/** @var ' . $tableGatewayName . ' $tableGateway */', '$tableGateway = $serviceLocator->get(\'' . $tableGatewayService . '\');', '', '$instance = new ' . $className . '($tableGateway);', '', 'return $instance;']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('createService'); $method->setBody($body); $method->setParameters([new ParameterGenerator($managerName, 'ServiceLocatorInterface')]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Create service', null, [new ParamTag($managerName, ['ServiceLocatorInterface']), new ReturnTag([$className])])); } // add method $this->addMethodFromGenerator($method); }
public function transform() { $classGenerator = $this->getService()->getClassGenerator(); if ($createContactBatch = $classGenerator->getMethod('CreateContactBatch')) { $newMethod = new CodeGenerator\MethodGenerator(); $newMethod->setName($createContactBatch->getName()); $docblock = $createContactBatch->getDocBlock(); $docblock->setTag((new CodeGenerator\DocBlock\Tag\ParamTag())->setParamName('id')->setDataType('int')); $newMethod->setDocBlock($docblock); $parameters = $createContactBatch->getParameters(); $idParameter = new CodeGenerator\ParameterGenerator(); $idParameter->setName('id'); $createContactBatch->setParameter($idParameter); $newMethod->setParameters(array("id" => $idParameter, "CreateContactBatch" => $parameters['CreateContactBatch'])); $body = $createContactBatch->getBody(); $body = str_replace('array()', 'array($id)', $body); $newMethod->setBody($body); $classGenerator->removeMethod('CreateContactBatch'); $classGenerator->addMethodFromGenerator($newMethod); } }
public function fixPathParameters($classGenerator, $method, $newParameters) { $oldMethod = $classGenerator->getMethod($method); $newMethod = new CodeGenerator\MethodGenerator(); $newMethod->setName($oldMethod->getName()); $docblock = $oldMethod->getDocBlock(); $parameters = $oldMethod->getParameters(); $body = $oldMethod->getBody(); $newParameterSpec = array(); foreach ($newParameters as $name => $type) { $docblock->setTag((new CodeGenerator\DocBlock\Tag\ParamTag())->setParamName($name)->setDataType($type)); $newParameter = new CodeGenerator\ParameterGenerator(); $newParameter->setName($name); $newMethod->setParameter($newParameter); $newParameterSpec[$name] = $newParameter; } $newMethod->setParameters(array_merge($newParameterSpec, array($method => $parameters[$method]))); $body = str_replace('array()', 'array($' . implode(', $', array_keys($newParameters)) . ')', $body); $newMethod->setDocBlock($docblock); $newMethod->setBody($body); $classGenerator->removeMethod($method); $classGenerator->addMethodFromGenerator($newMethod); }
public function postRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $tableNameAsCamelCase = $part->getTable()->getNameAsCamelCase(); $tags = array(array('name' => 'return', 'description' => '\\Model\\' . $tableNameAsCamelCase . 'Model экземпляр модели')); $docblock = new DocBlockGenerator('Получить экземпляр модели ' . $tableNameAsCamelCase); $docblock->setTags($tags); $method = new MethodGenerator(); $method->setName('getInstance'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setStatic(true); $method->setDocBlock($docblock); $method->setBody(<<<EOS return parent::getInstance(); EOS ); $file->getClass()->addMethodFromGenerator($method); }
/** * @param PartInterface $part */ public function preRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); /** * @var $table \Model\Cluster\Schema\Table */ $table = $part->getTable(); /** @var $columnList Column[] */ $columnList = $table->getColumn(); foreach ($columnList as $column) { $columnName = $column->getName(); $columnComment = $column->getComment(); if ($columnComment) { $shortDescription = "Get " . mb_strtolower($columnComment, 'UTF-8') . ' (' . $column->getTable()->getName() . '.' . $columnName . ')'; } else { $shortDescription = 'Get ' . $column->getTable()->getName() . '.' . $columnName; } $docBlock = new DocBlockGenerator($shortDescription); $docBlock->setTags(array(array('name' => 'return', 'description' => $column->getTypeAsPhp()))); $method = new MethodGenerator(); $method->setName('get' . $column->getNameAsCamelCase()); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setDocBlock($docBlock); $method->setBody(<<<EOS return \$this->get('{$columnName}'); EOS ); $part->getFile()->getClass()->addMethodFromGenerator($method); } }
public function postRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $tableName = $part->getTable()->getName(); $schema = $part->getTable()->getSchema()->getName(); $docblock = new DocBlockGenerator('Конструктор'); $method = new MethodGenerator(); $method->setName('__construct'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setDocBlock($docblock); $method->setBody(<<<EOS \$this->setName('{$tableName}'); \$this->setDbAdapterName('{$schema}_db'); parent::__construct(); EOS ); $file->getClass()->addMethodFromGenerator($method); }
/** * @group ZF-6444 */ public function testMethodWithFinalModifierIsNotEmittedWhenMethodIsAbstract() { $methodGenerator = new MethodGenerator(); $methodGenerator->setName('foo'); $methodGenerator->setParameters(array('one')); $methodGenerator->setFinal(true); $methodGenerator->setAbstract(true); $expected = <<<EOS abstract public function foo(\$one) { } EOS; $this->assertEquals($expected, $methodGenerator->generate()); }
protected function viewStub($part) { /** * @var $part \Model\Generator\Part\Model */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $table = $part->getTable(); $tableNameAsCamelCase = $table->getNameAsCamelCase(); $entity = $tableNameAsCamelCase . 'Entity'; $collection = $tableNameAsCamelCase . 'Collection'; $p = new \Zend\Code\Generator\ParameterGenerator('cond'); $p->setType('Cond'); $p->setDefaultValue(null); $params[] = $p; $docblock = new DocBlockGenerator('Получить объект условий в виде представления \'Extended\' $param Cond $cond $return Cond '); $method = new MethodGenerator(); $method->setName('getCondAsExtendedView'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setFinal(false); $method->setDocBlock($docblock); $method->setParameters($params); $method->setBody(<<<EOS \$cond = \$this->prepareCond(\$cond); \$cond->where(array('status' => 'active')); return \$cond; EOS ); $file->getClass()->addMethodFromGenerator($method); $p = new \Zend\Code\Generator\ParameterGenerator('cond'); $p->setType('Cond'); $p->setDefaultValue(null); $params[] = $p; $docblock = new DocBlockGenerator('Получить элемент в виде представления \'Extended\' @param Cond $cond @return ' . $entity); $method = new MethodGenerator(); $method->setName('getAsExtendedView'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setFinal(false); $method->setDocBlock($docblock); $method->setParameters($params); $method->setBody(<<<EOS \$cond = \$this->getCondAsExtendedView(\$cond); return \$this->get(\$cond); EOS ); $file->getClass()->addMethodFromGenerator($method); $p = new \Zend\Code\Generator\ParameterGenerator('cond'); $p->setType('Cond'); $p->setDefaultValue(null); $params[] = $p; $docblock = new DocBlockGenerator('Получить коллекцию в виде представления \'Extended\' @param Cond $cond @return ' . $collection); $method = new MethodGenerator(); $method->setName('getCollectionAsExtendedView'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setFinal(false); $method->setDocBlock($docblock); $method->setParameters($params); $method->setBody(<<<EOS \$cond = \$this->getCondAsExtendedView(\$cond); return \$this->getCollection(\$cond); EOS ); $file->getClass()->addMethodFromGenerator($method); }
/** * Generate an hydrate method */ protected function addHydrateMethod() { // set action body $body = ['// add extended hydrator logic here', 'return parent::hydrate($data, $object);']; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('hydrate'); $method->setBody($body); $method->setParameters([new ParameterGenerator('data', 'array'), new ParameterGenerator('object', 'object')]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Hydrate an object by populating data', null, [new ParamTag('data', ['array']), new ParamTag('object', ['object']), new ReturnTag(['object'])])); } // add method $this->addMethodFromGenerator($method); }
private function buildClass($replacement) { $type = $this->splitNsandClass($replacement['originalFullyQualifiedType']); $class = new ClassGenerator(); $class->setName($type['class']); $class->setNamespaceName($type['ns']); $class->setExtendedClass('\\Brads\\Ppm\\Proxy'); $properties = []; $methods = []; $implementedInterfaces = []; foreach ($versions as $version => $info) { foreach ($info['files'] as $file) { echo "Parsing: " . $this->vendorDir . '/' . $package . '/' . $version . '/' . $file . "\n"; $parsedFile = new ReflectionFile($this->vendorDir . '/' . $package . '/' . $version . '/' . $file); $parsedClass = $parsedFile->getFileNamespace($info['toNs'])->getClass($info['toNs'] . '\\' . $type['class']); if ($parsedClass->getInterfaceNames() != null) { $implementedInterfaces = array_merge($implementedInterfaces, $parsedClass->getInterfaceNames()); } foreach ($parsedClass->getMethods() as $method) { if ($method->isPublic()) { $generatedMethod = new MethodGenerator(); $generatedMethod->setName($method->name); $generatedMethod->setBody('echo "Hello world!";'); $generatedMethod->setAbstract($method->isAbstract()); $generatedMethod->setFinal($method->isFinal()); $generatedMethod->setStatic($method->isStatic()); $generatedMethod->setVisibility(MethodGenerator::VISIBILITY_PUBLIC); foreach ($method->getParameters() as $param) { $generatedParam = new ParameterGenerator(); $generatedParam->setName($param->name); if ($param->hasType()) { $generatedParam->setType($param->getType()); } //$generatedParam->setDefaultValue($param->getDefaultValue()); $generatedParam->setPosition($param->getPosition()); $generatedParam->setVariadic($param->isVariadic()); $generatedParam->setPassedByReference($param->isPassedByReference()); $generatedMethod->setParameter($generatedParam); } $existingMethod = Linq::from($methods)->firstOrDefault(null, function (MethodGenerator $v) use($method) { return $v->getName() == $method->name; }); if ($existingMethod != null) { $existingParams = $existingMethod->getParameters(); foreach ($generatedMethod->getParameters() as $newParam) { $existingParam = Linq::from($existingParams)->firstOrDefault(function (ParameterGenerator $v) { return $v->getName() == $newParam->getName(); }); if ($existingParam == null) { $existingMethod->setParameter($newParam); } } } else { $methods[] = $generatedMethod; } } } foreach ($parsedClass->getProperties() as $prop) { //$properties[] = PropertyGenerator::fromReflection($prop); } } } $class->setImplementedInterfaces($implementedInterfaces); $class->addMethods($methods); $class->addProperties($properties); return (new FileGenerator(['classes' => [$class]]))->generate(); }
/** * Generate the create service method * * @param string $className * @param $moduleName * @param string $entityModule * @param $entityClass */ protected function addCreateServiceMethod($className, $moduleName, $entityModule, $entityClass) { // prepare repository params $repositoryClass = str_replace('Entity', '', $entityClass) . 'Repository'; $repositoryParam = lcfirst($repositoryClass); $repositoryService = $entityModule . '\\' . $this->config['namespaceRepository'] . '\\' . str_replace('Entity', '', $entityClass); // prepare form params if (in_array($className, ['CreateController', 'UpdateController'])) { $formClass = str_replace('Entity', '', $entityClass) . 'DataForm'; $formParam = lcfirst($formClass); $formService = $moduleName . '\\Data\\Form'; } elseif (in_array($className, ['DeleteController'])) { $formClass = str_replace('Entity', '', $entityClass) . 'DeleteForm'; $formParam = lcfirst($formClass); $formService = $moduleName . '\\Delete\\Form'; } else { $formClass = null; $formParam = null; $formService = null; } // set action body $body = []; $body[] = '$serviceLocator = $controllerManager->getServiceLocator();'; $body[] = ''; if ($formClass) { $body[] = '/** @var ServiceLocatorInterface $formElementManager */'; $body[] = '$formElementManager = $serviceLocator->get(\'FormElementManager\');'; $body[] = ''; } $body[] = '/** @var ' . $repositoryClass . ' $' . $repositoryParam . ' */'; $body[] = '$' . $repositoryParam . ' = $serviceLocator->get(\'' . $repositoryService . '\');'; $body[] = ''; if ($formClass) { $body[] = '/** @var ' . $formClass . ' $' . $formParam . ' */'; $body[] = '$' . $formParam . ' = $formElementManager->get(\'' . $formService . '\');'; $body[] = ''; } $body[] = '$instance = new ' . $className . '();'; $body[] = '$instance->set' . $repositoryClass . '($' . $repositoryParam . ');'; if ($formClass) { $body[] = '$instance->set' . $formClass . '($' . $formParam . ');'; } $body[] = ''; $body[] = 'return $instance;'; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('createService'); $method->setBody($body); $method->setParameters([new ParameterGenerator('controllerManager', 'ServiceLocatorInterface')]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Create service', null, [new ParamTag('controllerManager', ['ServiceLocatorInterface', 'ServiceLocatorAwareInterface']), new ReturnTag([$className])])); } // add method $this->addMethodFromGenerator($method); }
/** * Create a PhpMethod code generation object named after a given alias * * @param string $alias * @param string $class Class to which alias refers * @return MethodGenerator */ protected function getCodeGenMethodFromAlias($alias, $class) { $alias = $this->normalizeAlias($alias); $method = new MethodGenerator(); $method->setName($alias); $method->setBody(sprintf('return $this->get(\'%s\');', $class)); return $method; }
public function testSetMethodNameAlreadyExistsThrowsException() { $methodA = new MethodGenerator(); $methodA->setName("foo"); $methodB = new MethodGenerator(); $methodB->setName("foo"); $classGenerator = new ClassGenerator(); $classGenerator->addMethodFromGenerator($methodA); $this->setExpectedException( 'Zend\Code\Generator\Exception\InvalidArgumentException', 'A method by name foo already exists in this class.' ); $classGenerator->addMethodFromGenerator($methodB); }
/** * Generate the getConfig() method * * @return MethodGenerator */ protected function generateGetConfigMethod() { // create method body $body = new ValueGenerator(); $body->initEnvironmentConstants(); $body->setValue('include __DIR__ . \'/config/module.config.php\''); // create method $method = new MethodGenerator(); $method->setName('getConfig'); $method->setBody('return ' . $body->generate() . ';' . AbstractGenerator::LINE_FEED); // add optional doc block if ($this->flagCreateApiDocs) { $method->setDocBlock(new DocBlockGenerator('Get module configuration', null, array($this->generateReturnTag('array')))); } return $method; }
/** * @param ClassGenerator $class * @param Attribute $tagAttribute */ protected function addAttribute(ClassGenerator $class, Attribute $tagAttribute) { $methodName = str_replace(':', '-', $tagAttribute->getName()); $methodName = ucfirst($this->stringToCamelCaseConverter->convert($methodName, '-')); $method = new MethodGenerator(); if ($tagAttribute->isFlag()) { $method->setName('is' . $methodName); $method->setParameter(new ParameterGenerator('v', 'bool', true)); } else { $method->setName('set' . $methodName); $method->setParameter(new ParameterGenerator('v', 'string')); } $body = []; $body[] = '$this->attributes[\'' . $tagAttribute->getName() . '\'] = $v;'; $body[] = 'return $this;'; $method->setBody(implode(PHP_EOL, $body)); $docBlock = new DocBlockGenerator(); $docBlock->setTag(new GenericTag('var', 'string')); $docBlock->setTag(new GenericTag('return', '$this')); if (!is_null($tagAttribute->getDescription())) { $docBlock->setLongDescription($tagAttribute->getDescription()); } $method->setDocBlock($docBlock); $class->addMethodFromGenerator($method); }
/** * Generate an hydrate method */ protected function addHydrateMethod() { $refTableData = $this->loadedTables[$this->refTableName]; /** @var ColumnObject $firstColumn */ $firstColumn = reset($refTableData['columns']); // set action body $body = []; $body[] = 'if (isset($data[\'' . $this->refTableName . '.' . $firstColumn->getName() . '\'])) {'; /** @var ColumnObject $column */ foreach ($refTableData['columns'] as $column) { $body[] = ' $' . $column->getName() . ' = $data[\'' . $this->refTableName . '.' . $column->getName() . '\'];'; } $body[] = '} else {'; /** @var ColumnObject $column */ foreach ($refTableData['columns'] as $column) { $body[] = ' $' . $column->getName() . ' = $value;'; } $body[] = '}'; $body[] = ''; $body[] = '$' . $this->refTableName . ' = new ' . $this->entityClass . '();'; $body[] = '$' . $this->refTableName . '->exchangeArray('; $body[] = ' ['; /** @var ColumnObject $column */ foreach ($refTableData['columns'] as $column) { $body[] = ' \'' . $column->getName() . '\' => $' . $column->getName() . ','; } $body[] = ' ]'; $body[] = ');'; $body[] = ''; $body[] = 'return $' . $this->refTableName . ';'; $body = implode(AbstractGenerator::LINE_FEED, $body); // create method $method = new MethodGenerator(); $method->setName('hydrate'); $method->setBody($body); $method->setParameters([new ParameterGenerator('value'), new ParameterGenerator('data', 'array', [])]); // check for api docs if ($this->config['flagAddDocBlocks']) { $method->setDocBlock(new DocBlockGenerator('Hydrate an entity by populating data', null, [new ParamTag('value'), new ParamTag('data', ['array']), new ReturnTag([$this->entityClass])])); } // add method $this->addMethodFromGenerator($method); }
public function postRun(PartInterface $part) { /** * @var $part \Model\Generator\Part\Entity */ /** * @var $file \Model\Code\Generator\FileGenerator */ $file = $part->getFile(); $columnCollection = $part->getTable()->getColumn(); $template = ''; /** @var $column Column */ foreach ($columnCollection as $column) { $name = $column->getName(); $requiredFlag = !($column->isNullable() || $column->getName() == 'id'); if ($columnConfig = $part->getColumntConfig($column)) { if ($columnConfig && isset($columnConfig['validators'])) { foreach ($columnConfig['validators'] as $validator) { if (isset($validator['params'])) { $validatorParams = $this->prepareValidatorParams($validator['params'], $column); $validatorParams = $this->varExportMin($validatorParams, true); } else { $validatorParams = null; } if ($validatorParams && $validatorParams != 'NULL') { $template .= "\$this->addValidatorRule('{$name}', Model::getValidatorAdapter()->getValidatorInstance('{$validator['name']}', {$validatorParams}), " . ($requiredFlag ? 'true' : 'false') . ");\n"; } else { $template .= "\$this->addValidatorRule('{$name}', Model::getValidatorAdapter()->getValidatorInstance('{$validator['name']}'), " . ($requiredFlag ? 'true' : 'false') . ");\n"; } } } } } $template = rtrim($template, "\r\n, "); //$tableNameAsCamelCase = $part->getTable()->getNameAsCamelCase(); $tags = array(array('name' => 'return', 'description' => 'array Model массив с фильтрами по полям')); $docblock = new DocBlockGenerator('Получить правила для фильтрации '); $docblock->setTags($tags); $method = new MethodGenerator(); $method->setName('initValidatorRules'); $method->setVisibility(AbstractMemberGenerator::VISIBILITY_PUBLIC); $method->setStatic(false); $method->setFinal(true); $method->setDocBlock($docblock); $method->setBody(<<<EOS {$template} \$this->setupValidatorRules(); EOS ); $file->getClass()->addMethodFromGenerator($method); }