/**
* @param State|\Scaffold\State $state
* @return State|void
*/
public function build(State $state)
{
$model = $state->getModel('options-factory');
$generator = new ClassGenerator($model->getName());
$generator->setImplementedInterfaces(['FactoryInterface']);
$model->setGenerator($generator);
$generator->addUse('Zend\\ServiceManager\\FactoryInterface');
$generator->addUse('Zend\\ServiceManager\\ServiceLocatorInterface');
$options = $state->getModel('options');
$key = $options->getServiceName();
$key = substr($key, 0, -7);
$body = <<<EOF
\$config = \$serviceLocator->get('Config');
return new {$options->getClassName()}(
isset(\$config['{$key}'])
? \$config['{$key}']
: []
);
EOF;
$method = new MethodGenerator('createService');
$method->setParameter(new ParameterGenerator('serviceLocator', 'ServiceLocatorInterface'));
$method->setBody($body);
$doc = new DocBlockGenerator('');
$doc->setTag(new Tag(['name' => 'inhertidoc']));
$method->setDocBlock($doc);
$generator->addMethodFromGenerator($method);
}
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);
}
protected function getConstructor()
{
$defaultValue = new ValueGenerator([], ValueGenerator::TYPE_ARRAY);
$setterParam = new ParameterGenerator('properties', 'array', $defaultValue);
$methodGenerator = new MethodGenerator('__construct', [$setterParam]);
$methodGenerator->setBody('$this->properties = $properties;');
return $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;
}
public function addTestIdMethod(ClassGenerator $generator, State $state)
{
$method = new MethodGenerator('testGetSetId');
$class = $state->getEntityModel()->getClassName();
$code = <<<EOF
\$object = new {$class}();
\$object->setId(123);
\$this->assertEquals(123, \$object->getId());
EOF;
$method->setBody($code);
$generator->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' => '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 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 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 = ['// 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);
}
/**
* 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);
}
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 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 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);
}
/**
* Build generators
*
* @param State|\Scaffold\State $state
* @return \Scaffold\State|void
*/
public function build(State $state)
{
$model = $this->model;
$generator = new ClassGenerator($model->getName());
$generator->addUse('Zend\\ServiceManager\\FactoryInterface');
$generator->addUse('Zend\\ServiceManager\\ServiceLocatorInterface');
$generator->addUse('Zend\\ServiceManager\\ServiceManager');
$generator->addUse($state->getServiceModel()->getName());
$generator->setImplementedInterfaces(['FactoryInterface']);
$method = new MethodGenerator('createService');
$method->setParameter(new ParameterGenerator('serviceLocator', 'ServiceLocatorInterface'));
$method->setBody('return new ' . $state->getServiceModel()->getClassName() . '($serviceLocator);');
$doc = new DocBlockGenerator('Create service');
$doc->setTag(new Tag\GenericTag('param', 'ServiceLocatorInterface|ServiceManager $serviceLocator'));
$doc->setTag(new Tag\GenericTag('return', $state->getServiceModel()->getClassName()));
$method->setDocBlock($doc);
$generator->addMethodFromGenerator($method);
$model->setGenerator($generator);
}
/**
* 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();
$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);
}
/**
* @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();
$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);
}
/**
* 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);
}
/**
* 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);
}
/**
* Add a selectWith() method if table has an external dependency
*
* @return MethodGenerator
*/
protected function addSelectWithMethod()
{
$foreignKeys = $this->loadedTables[$this->tableName]['foreignKeys'];
if (empty($foreignKeys)) {
return true;
}
$body = [];
/** @var ConstraintObject $foreignKey */
foreach ($foreignKeys as $foreignKey) {
$refTableName = $foreignKey->getReferencedTableName();
$refTableColumns = $this->loadedTables[$refTableName]['columns'];
$body[] = '$select->join(';
$body[] = ' \'' . $refTableName . '\',';
$body[] = ' \'' . $this->tableName . '.' . $foreignKey->getColumns()[0] . ' = ' . $refTableName . '.' . $foreignKey->getReferencedColumns()[0] . '\',';
$body[] = ' [';
/** @var ColumnObject $column */
foreach ($refTableColumns as $column) {
$body[] = ' \'' . $refTableName . '.' . $column->getName() . '\' => \'' . $column->getName() . '\',';
}
$body[] = ' ]';
$body[] = ');';
$body[] = '';
}
$body[] = 'return parent::selectWith($select);';
$body = implode(AbstractGenerator::LINE_FEED, $body);
$this->addUse('Zend\\Db\\ResultSet\\ResultSetInterface');
$this->addUse('Zend\\Db\\Sql\\Select');
$selectMethod = new MethodGenerator('selectWith');
$selectMethod->addFlag(MethodGenerator::FLAG_PUBLIC);
$selectMethod->setParameter(new ParameterGenerator('select', 'Select'));
$selectMethod->setDocBlock(new DocBlockGenerator('Add join tables', null, [['name' => 'param', 'description' => 'Select $select'], ['name' => 'return', 'description' => 'ResultSetInterface']]));
$selectMethod->setBody($body);
$this->addMethodFromGenerator($selectMethod);
return true;
}
/**
* @param string $columnName
* @param ConstraintObject $foreignKey
*/
protected function addOptionsSetter($columnName, ConstraintObject $foreignKey)
{
$body = '$this->' . $columnName . 'Options = $' . $columnName . 'Options;';
$parameter = new ParameterGenerator($columnName . 'Options', 'array');
$setMethodName = 'set' . ucfirst($columnName) . 'Options';
$setMethod = new MethodGenerator($setMethodName);
$setMethod->addFlag(MethodGenerator::FLAG_PUBLIC);
$setMethod->setParameter($parameter);
$setMethod->setDocBlock(new DocBlockGenerator('Set ' . $columnName . ' options', null, [['name' => 'param', 'description' => 'array $' . $columnName . 'Options']]));
$setMethod->setBody($body);
$this->addMethodFromGenerator($setMethod);
}
public function testMethodBodyGetterAndSetter()
{
$method = new MethodGenerator();
$method->setBody('Foo');
$this->assertEquals('Foo', $method->getBody());
}
/**
* 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);
}
protected function buildDelete(ClassGenerator $generator, State $state)
{
$body = '$this->getEntityManager()->remove($model);';
$method = new MethodGenerator('delete');
$method->setParameter(new ParameterGenerator('model', $state->getEntityModel()->getClassName()));
$method->setDocBlock(new DocBlockGenerator());
$method->getDocBlock()->setTag(new Tag\GenericTag('param', $state->getEntityModel()->getClassName() . ' $model'));
$method->setBody($body);
$generator->addMethodFromGenerator($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 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;
}
