public function postRun(PartInterface $part)
{
/**
* @var $part \Model\Generator\Part\Entity
*/
/**
* @var $file \Model\Code\Generator\FileGenerator
*/
$file = $part->getFile();
$tableNameAsCamelCase = $part->getTable()->getNameAsCamelCase();
$tableNameAsVar = $part->getTable()->getNameAsVar();
//$tableComment = $part->getTable()->getComment();
$tags = array(array('name' => 'param', 'description' => 'array|' . $tableNameAsCamelCase . 'Model $' . $tableNameAsVar), array('name' => 'return', 'description' => '\\Model\\Entity\\' . $tableNameAsCamelCase));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('Добавить ' . $tableNameAsCamelCase);
$docblock->setTags($tags);
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('add' . $tableNameAsCamelCase);
$method->setParameter(new \Zend\Code\Generator\ParameterGenerator($tableNameAsVar, 'mixed'));
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
\${$tableNameAsVar}Id = null;
\$result = ModelResult();
\${$tableNameAsVar} = new \${$tableNameAsVar}Entity(\${$tableNameAsVar});
\${$tableNameAsVar}Data = \${$tableNameAsVar}->toArray();
// Фильтруем входные данные
\${$tableNameAsVar}Data = \$this->addFilter(\${$tableNameAsVar}Data);
// Валидируем данные
\$validator = \$this->addValidate(\${$tableNameAsVar}Data);
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();
$tableNameAsCamelCase = $table->getNameAsCamelCase();
$tags = array(array('name' => 'see', 'description' => 'parent::setupDefaultEntityType()'));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('Устанавливаем тип entity по-умолчанию');
$docblock->setTags($tags);
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('setupDefaultEntityType');
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PROTECTED);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
\$this->defaultEntityType = '\\Model\\Entity\\{$tableNameAsCamelCase}Entity';
EOS
);
$file->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();
$tableNameAsCamelCase = $part->getTable()->getNameAsCamelCase();
$tags = array(array('name' => 'return', 'description' => 'void'));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('Настраиваем текущую сущность');
$docblock->setTags($tags);
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('setupEntity');
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
\$this->setName('{$tableName}');
\$this->setEntityName('{$tableName}');
\$this->setEntityClassName('\\Model\\Entity\\{$tableNameAsCamelCase}Entity');
\$this->setCollectionClassName('\\Model\\Collection\\{$tableNameAsCamelCase}Collection');
EOS
);
$file->getClass()->addMethodFromGenerator($method);
}
/**
* @param \Model\Generator\Part\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();
$maxColNameLen = $table->getMaxColumnNameLength();
$columnList = $table->getColumn();
$docblock = new DocBlockGenerator('Настраиваем типы данных');
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('setupDataTypes');
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PROTECTED);
$method->setDocBlock($docblock);
$dataTypesText = '';
foreach ($columnList as $column) {
$columnName = str_pad('\'' . $column->getName() . '\'', $maxColNameLen + 2, ' ', STR_PAD_RIGHT);
$columnType = $column->getTypeAsDataTypeConstant();
$dataTypesText .= " {$columnName} => {$columnType},\n";
}
$linkList = $table->getLink();
foreach ($linkList as $link) {
$foreignName = $link->getForeignTable()->getName();
$columnComment = $link->getForeignColumn()->getComment();
switch ($link->getLinkType()) {
case AbstractLink::LINK_TYPE_ONE_TO_MANY:
case AbstractLink::LINK_TYPE_MANY_TO_MANY:
$columnName = str_pad('\'_' . $foreignName . '_collection' . '\'', $maxColNameLen + 2, ' ', STR_PAD_RIGHT);
$dataTypesText .= " {$columnName} => 'Model\\\\Collection\\\\" . $link->getForeignColumn()->getTable()->getNameAsCamelCase() . "Collection',\n";
case AbstractLink::LINK_TYPE_ONE_TO_ONE:
case AbstractLink::LINK_TYPE_MANY_TO_ONE:
$columnName = str_pad('\'_' . $foreignName . '\'', $maxColNameLen + 2, ' ', STR_PAD_RIGHT);
$dataTypesText .= " {$columnName} => 'Model\\\\Entity\\\\" . $link->getForeignColumn()->getTable()->getNameAsCamelCase() . "Entity',\n";
}
}
$dataTypesText = rtrim($dataTypesText, "\r\n ,");
$method->setBody(<<<EOS
\$this->dataTypes = array(
{$dataTypesText});
EOS
);
$file->getClass()->addMethodFromGenerator($method);
}
/**
* Add method from MethodGenerator
*
* @param Zend\Code\Generator\MethodGenerator $method
* @return Magento_Di_Generator_CodeGenerator_Zend
* @throws InvalidArgumentException
*/
public function addMethodFromGenerator(Zend\Code\Generator\MethodGenerator $method)
{
if (!is_string($method->getName())) {
throw new InvalidArgumentException('addMethodFromGenerator() expects string for name');
}
return parent::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();
// $tableNameAsCamelCase = $table->getNameAsCamelCase();
/** @var $columnList Column[] */
$columnList = $table->getColumn();
foreach ($columnList as $column) {
if ($column->getColumnType() != 'enum' || substr($column->getName(), 0, 3) == 'is_' || count($column->getEnumValuesAsArray()) < 1) {
continue;
}
$columnName = $column->getName();
foreach ($column->getEnumValuesAsArray() as $enumValue) {
$shortDescr = 'Проверить на соответствие ' . $column->getTable()->getName() . '.' . $columnName . ' значению ' . $enumValue;
$docblock = new \Zend\Code\Generator\DocBlockGenerator($shortDescr);
$docblock->setTags(array(array('name' => 'return', 'description' => 'bool')));
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('is' . $column->getNameAsCamelCase() . implode('', array_map('ucfirst', explode('_', $enumValue))));
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
if ($part->hasPlugin('ConstantList', AbstractPart::PART_MODEL)) {
$modelName = $table->getNameAsCamelCase() . 'Model';
$file->addUse("\\Model\\{$modelName}");
$enumValue = $modelName . '::' . strtoupper($column->getName() . '_' . $enumValue);
} else {
$enumValue = "'{$enumValue}'";
}
$method->setBody(<<<EOS
return \$this->get('{$columnName}') == {$enumValue};
EOS
);
$part->getFile()->getClass()->addMethodFromGenerator($method);
}
}
foreach ($columnList as $column) {
if ($column->getColumnType() != 'enum' || substr($column->getName(), 0, 3) != 'is_' || count($column->getEnumValuesAsArray()) != 2) {
continue;
}
$inc = 0;
foreach ($column->getEnumValuesAsArray() as $enumValue) {
if (in_array($enumValue, array('y', 'n'))) {
$inc++;
}
}
if ($inc != 2) {
continue;
}
$columnName = $column->getName();
$shortDescr = 'Проверить флаг ' . $column->getTable()->getName() . '.' . $columnName;
$docblock = new \Zend\Code\Generator\DocBlockGenerator($shortDescr);
$docblock->setTags(array(array('name' => 'return', 'description' => 'bool')));
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('is' . substr($column->getNameAsCamelCase(), 2));
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
return \$this->get('{$columnName}') == 'y';
EOS
);
$part->getFile()->getClass()->addMethodFromGenerator($method);
}
}
protected function execute(InputInterface $input, OutputInterface $output)
{
$path = new PathBuilder();
$path = $path->addPart('config')->addPart('autoload')->addPart('database.eloquentzf2.config.local.php')->getPath();
$dbConfig = (include $path);
$dbConfig = $dbConfig['database_eloquent'];
$moduleName = $input->getArgument('ModuleName');
$modelName = $input->getArgument('ModelName');
$tblName = $input->getArgument('TableName');
$entityPath = $input->getArgument('EntityRelativePath');
if ($dbConfig['host'] == 'localhost') {
$dbConfig['host'] = '127.0.0.1';
}
$dbh = new \PDO('mysql:host=' . $dbConfig['host'] . ';dbname=' . $dbConfig['database'], $dbConfig['username'], $dbConfig['password'] ? $dbConfig['password'] : null);
$columns = [];
$priKey = null;
$autoInc = true;
$enums = [];
foreach ($dbh->query('select * from information_schema.columns where table_schema = \'' . $dbConfig['database'] . '\' and table_name = \'' . $tblName . '\'') as $row) {
$columns[] = $row['COLUMN_NAME'];
if (strtolower($row['COLUMN_KEY']) == 'pri') {
if (!$priKey) {
$priKey = $row['COLUMN_NAME'];
if ($row['EXTRA'] != 'auto_increment') {
$autoInc = false;
}
} else {
$priKey = [$priKey];
$priKey[] = $row['COLUMN_NAME'];
$autoInc = false;
}
}
if (strtolower($row['DATA_TYPE']) == 'enum') {
$enums[$row['COLUMN_NAME']] = explode(',', rtrim(str_replace('enum(', '', $row['COLUMN_TYPE']), ')'));
$enums[$row['COLUMN_NAME']] = array_map(function ($el) {
return trim($el, "'");
}, $enums[$row['COLUMN_NAME']]);
}
if (strtolower($row['DATA_TYPE']) == 'tinyint') {
$enums[$row['COLUMN_NAME']] = [true, false];
}
}
$fillable = [];
foreach ($columns as $column) {
if (is_array($priKey)) {
if (!in_array($column, $priKey)) {
$fillable[] = $column;
}
} else {
if ($column != $priKey) {
$fillable[] = $column;
}
}
}
foreach ($fillable as $k => $v) {
if (in_array($v, ['created_at', 'updated_at', 'deleted_at'])) {
unset($fillable[$k]);
}
}
$path = new PathBuilder();
$path = $path->addPart('module')->addPart($moduleName)->addPart('src')->addPart($moduleName)->addPart('Entity');
if ($entityPath) {
$entityPath = trim($entityPath, '/');
$path->addPart($entityPath);
}
$path = $path->addPart($modelName . '.php')->getPath();
if (file_exists($path)) {
throw new \RuntimeException('A file with the same model name already exists: ' . $path);
}
$prop1 = new \Zend\Code\Generator\PropertyGenerator('incrementing', $autoInc, \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$prop1->setDocBlock('@var bool');
$prop2 = new \Zend\Code\Generator\PropertyGenerator('table', $tblName, \Zend\Code\Generator\PropertyGenerator::FLAG_PROTECTED);
$prop2->setDocBlock('@var string');
$prop3 = new \Zend\Code\Generator\PropertyGenerator('primaryKey', $priKey, \Zend\Code\Generator\PropertyGenerator::FLAG_PROTECTED);
$prop3->setDocBlock(is_array($priKey) ? '@var array' : '@var string');
$prop4 = new \Zend\Code\Generator\PropertyGenerator('fillable', $fillable, \Zend\Code\Generator\PropertyGenerator::FLAG_PROTECTED);
$prop4->setDocBlock('@var array The attributes that are mass assignable.');
$class = new ClassGenerator($modelName, $moduleName . '\\Entity', null, 'AbstractEntity', [], [$prop1, $prop2, $prop3, $prop4], []);
if ($entityPath) {
$class->setNamespaceName($class->getNamespaceName() . '\\' . trim(str_replace('/', '\\', $entityPath), '/'));
}
$dbGenerator = new \Zend\Code\Generator\DocBlockGenerator();
foreach ($columns as $column) {
$dbGenerator->setTag(['name' => 'property', 'description' => 'string $' . $column]);
}
$class->addUse('MdoBundleEloquent\\Entity\\AbstractEntity');
$class->setDocBlock($dbGenerator);
// Set timestamps = false if no created_at / updated_at
if (!in_array('created_at', $columns) && !in_array('updated_at', $columns)) {
$prop = new \Zend\Code\Generator\PropertyGenerator('timestamps', false, \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$prop->setDocBlock('@var bool');
$class->addPropertyFromGenerator($prop);
}
// Set getUpdatedAtColumn() to null if no created_at
if (!in_array('created_at', $columns) && in_array('updated_at', $columns)) {
$method = new \Zend\Code\Generator\MethodGenerator('getCreatedAtColumn', [], \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$method->setDocBlock('@return null');
$class->addMethodFromGenerator($method);
$method = new \Zend\Code\Generator\MethodGenerator('setCreatedAt', ['value'], \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$method->setDocBlock('@return null');
$class->addMethodFromGenerator($method);
}
// Set getCreatedAtColumn() to null if no created_at
if (!in_array('updated_at', $columns) && in_array('created_at', $columns)) {
$method = new \Zend\Code\Generator\MethodGenerator('getUpdatedAtColumn', [], \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$method->setDocBlock('@return null');
$class->addMethodFromGenerator($method);
$method = new \Zend\Code\Generator\MethodGenerator('setUpdatedAt', ['value'], \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$method->setDocBlock('@return null');
$class->addMethodFromGenerator($method);
}
// Set use soft delete trait if deleted_at
if (in_array('deleted_at', $columns)) {
$class->addTrait('SoftDeletingTrait');
$prop = new \Zend\Code\Generator\PropertyGenerator('dates', ['deleted_at'], \Zend\Code\Generator\PropertyGenerator::FLAG_PROTECTED);
$prop->setDocBlock('@var array');
$class->addPropertyFromGenerator($prop);
$class->addUse('Illuminate\\Database\\Eloquent\\SoftDeletingTrait');
}
$method = new \Zend\Code\Generator\MethodGenerator('getDates', [], \Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC, 'return [];');
$method->setDocBlock('@return null');
$class->addMethodFromGenerator($method);
// Set enums
foreach ($enums as $enumColName => $values) {
foreach ($values as $value) {
if (is_bool($value)) {
$name = $value === TRUE ? 'yes' : 'no';
$prop = new \Zend\Code\Generator\PropertyGenerator(strtoupper($enumColName . '_' . $name), (int) $value);
} else {
$prop = new \Zend\Code\Generator\PropertyGenerator(strtoupper($enumColName . '_' . $value), $value);
}
$prop->setConst(true);
$class->addPropertyFromGenerator($prop);
}
$method = new \Zend\Code\Generator\MethodGenerator('get' . ucfirst($enumColName) . 'Values', [], \Zend\Code\Generator\PropertyGenerator::FLAG_STATIC);
$method->setDocBlock('@return array');
$body = 'return [';
foreach ($values as $value) {
if (is_bool($value)) {
$name = $value === TRUE ? 'yes' : 'no';
} else {
$name = $value;
}
$body .= "\r\n" . ' ' . "'" . $name . "'" . ' => self::' . strtoupper($enumColName . '_' . $name) . ',';
}
$body .= "\r\n" . '];';
$method->setBody($body);
$class->addMethodFromGenerator($method);
}
$builder = new PathBuilder();
$builder->make_path($path);
file_put_contents($path, '<?php' . "\r\n" . $class->generate());
$output->writeln('<info>write</info>: ' . $path);
// Now that we wrote the eloquent class
// Let's build the model validator class
$class = new ClassGenerator($modelName . 'Validator', $moduleName . '\\Validator', null, 'AbstractFieldset', ['InputFilterProviderInterface'], [], []);
$class->addUse('MdoBundle\\Form\\AbstractFieldset');
$class->addUse('MdoBundle\\Validator\\ValidatorTrait\\ValidationRulesTrait');
$class->addUse('Zend\\InputFilter\\InputFilterProviderInterface');
$class->addUse('Zend\\Mvc\\I18n\\Translator');
$class->addTrait('ValidationRulesTrait');
$codeName = strtolower($modelName);
$body = <<<phpCode
parent::__construct('{$codeName}');
\$this->validationRules = [
phpCode;
foreach ($fillable as $item) {
$body .= <<<phpCode
'{$item}' => [
'required',
'validators' => [
[
'name' => 'NotEmpty',
'break_chain_on_failure' => true,
'options' => [
'message' => \$translator->translate('You must provide the field {$item}'),
],
],
]
],
phpCode;
}
$body .= <<<phpCode
];
return \$this->addElements([
phpCode;
foreach ($fillable as $item) {
$body .= <<<phpCode
'{$item}',
phpCode;
}
$body .= <<<phpCode
]);
phpCode;
$class->addMethod('__construct', [new ParameterGenerator('translator', 'Translator')], [], $body);
$path = new PathBuilder();
$path = $path->addPart('module')->addPart($moduleName)->addPart('src')->addPart($moduleName)->addPart('Validator')->addPart($modelName . 'Validator.php')->getPath();
if (file_exists($path)) {
throw new \RuntimeException('A file with the same model validator name already exists: ' . $path);
}
file_put_contents($path, '<?php' . "\r\n" . $class->generate());
$output->writeln('<info>write</info>: ' . $path);
}
/**
* @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();
$linkList = $table->getLink();
foreach ($linkList as $link) {
$foreignEntity = $link->getForeignEntity();
$columnName = $link->getForeignColumn()->getName();
$columnComment = $link->getForeignColumn()->getComment();
switch ($link->getLinkType()) {
case AbstractLink::LINK_TYPE_ONE_TO_MANY:
case AbstractLink::LINK_TYPE_MANY_TO_MANY:
if ($columnComment) {
$shortDescr = "Получить список " . mb_strtolower($columnComment, 'UTF-8') . ' (' . $table->getName() . '.' . $columnName . ')';
} else {
$shortDescr = 'Получить список ' . $link->getForeignEntity();
}
$docblock = new \Zend\Code\Generator\DocBlockGenerator($shortDescr);
$docblock->setTags(array(array('name' => 'return', 'description' => '\\Model\\Collection\\' . $link->getForeignColumn()->getTable()->getNameAsCamelCase() . 'Collection')));
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('get' . $link->getForeignEntityAsCamelCase() . 'Collection');
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
return \$this->get('_{$foreignEntity}_collection');
EOS
);
try {
$part->getFile()->getClass()->addMethodFromGenerator($method);
} catch (\Exception $e) {
}
case AbstractLink::LINK_TYPE_ONE_TO_ONE:
case AbstractLink::LINK_TYPE_MANY_TO_ONE:
if ($columnComment) {
$shortDescr = "Получить связанную сущность" . mb_strtolower($columnComment, 'UTF-8') . ' (' . $table->getName() . '.' . $columnName . ')';
} else {
$shortDescr = 'Получить связанную сущность ' . $link->getForeignEntity();
}
$docblock = new \Zend\Code\Generator\DocBlockGenerator($shortDescr);
$docblock->setTags(array(array('name' => 'return', 'description' => '\\Model\\Entity\\' . $link->getForeignColumn()->getTable()->getNameAsCamelCase() . 'Entity')));
$method = new \Zend\Code\Generator\MethodGenerator();
$method->setName('get' . $link->getForeignEntityAsCamelCase());
$method->setVisibility(\Zend\Code\Generator\AbstractMemberGenerator::VISIBILITY_PUBLIC);
$method->setDocBlock($docblock);
$method->setBody(<<<EOS
return \$this->get('_{$foreignEntity}');
EOS
);
try {
$part->getFile()->getClass()->addMethodFromGenerator($method);
} catch (\Exception $e) {
}
}
}
}
/**
* @param string $proxyName
* @return \Zend\Code\Generator\ClassGenerator
*/
private function createProxyClass($proxyName)
{
$proxyClass = new Generator\ClassGenerator($proxyName);
$proxyClass->setExtendedClass($this->fullClassName);
foreach ($this->injectableArguments as $injectableArgument) {
$injectorMethod = new \Zend\Code\Generator\MethodGenerator('propertyInjection' . $injectableArgument->getName());
$injectorMethod->setBody($injectableArgument->getProcessingBody());
$proxyClass->addMethodFromGenerator($injectorMethod);
}
return $proxyClass;
}
