/**
* @param \Model\Generator\Part\Model|\Model\Generator\Part\PartInterface $part
*/
public function preRun(PartInterface $part)
{
/**
* @var $file \Model\Code\Generator\FileGenerator
*/
$file = $part->getFile();
/** @var Table $table */
$table = $part->getTable();
/** @var string $tableName */
$tableName = $table->getName();
/** @var $linkList|Link[] \Model\Cluster\Schema\Table\Column */
$linkList = $table->getLink();
foreach ($linkList as $link) {
$_name = $link->getForeignEntity();
switch ($link->getLinkType()) {
case AbstractLink::LINK_TYPE_ONE_TO_ONE:
case AbstractLink::LINK_TYPE_ONE_TO_MANY:
case AbstractLink::LINK_TYPE_MANY_TO_ONE:
case AbstractLink::LINK_TYPE_MANY_TO_MANY:
$name = 'WITH_' . strtoupper($_name);
$property = new \Zend\Code\Generator\PropertyGenerator($name, strtolower($_name), \Zend\Code\Generator\PropertyGenerator::FLAG_CONSTANT);
$tags = array(array('name' => 'const'), array('name' => 'var', 'description' => 'string'));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('WITH сущность ' . $link->getForeignColumn()->getTable()->getNameAsCamelCase());
$docblock->setTags($tags);
$property->setDocBlock($docblock);
self::$_data[$tableName][$name] = $property;
$name = 'WITH_' . strtoupper($_name) . '_COUNT';
$property = new \Zend\Code\Generator\PropertyGenerator($name, strtolower($_name) . '_count', \Zend\Code\Generator\PropertyGenerator::FLAG_CONSTANT);
$tags = array(array('name' => 'const'), array('name' => 'var', 'description' => 'string'));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('WITH сущность ' . $link->getForeignColumn()->getTable()->getNameAsCamelCase());
$docblock->setTags($tags);
$property->setDocBlock($docblock);
self::$_data[$tableName][$name] = $property;
break;
}
switch ($link->getLinkType()) {
case AbstractLink::LINK_TYPE_ONE_TO_MANY:
case AbstractLink::LINK_TYPE_MANY_TO_MANY:
$name = 'WITH_' . strtoupper($_name) . "_COLLECTION";
$property = new \Zend\Code\Generator\PropertyGenerator($name, $_name . '_collection', \Zend\Code\Generator\PropertyGenerator::FLAG_CONSTANT);
$tags = array(array('name' => 'const'), array('name' => 'var', 'description' => 'string'));
$docblock = new \Zend\Code\Generator\DocBlockGenerator('WITH сущность коллекции ' . $link->getForeignColumn()->getTable()->getNameAsCamelCase());
$docblock->setTags($tags);
$property->setDocBlock($docblock);
self::$_data[$tableName][$name] = $property;
break;
}
}
}
/**
* Add property from PropertyGenerator
*
* @param Zend\Code\Generator\PropertyGenerator $property
* @throws InvalidArgumentException
* @return Magento_Di_Generator_CodeGenerator_Zend
*/
public function addPropertyFromGenerator(Zend\Code\Generator\PropertyGenerator $property)
{
if (!is_string($property->getName())) {
throw new InvalidArgumentException('addPropertyFromGenerator() expects string for name');
}
return parent::addPropertyFromGenerator($property);
}
/**
* generates headers
*
* @param \Zend\Code\Generator\ClassGenerator $class
* @param DataSetDescriptorInterface $dataSet
*/
protected function generateHeaderProperty(ClassGenerator $class, DataSetDescriptorInterface $dataSet)
{
$property = new \Zend\Code\Generator\PropertyGenerator("headers");
$property->setFlags(\Zend\Code\Generator\PropertyGenerator::FLAG_PROTECTED);
$value = array();
foreach ($dataSet->listGenerator() as $field) {
$value[$field->getName()] = array('title' => ucfirst(preg_replace("/[^a-z0-9]/i", " ", $field->getName())), 'width' => '100', 'filters' => $this->getFieldType($field) == 'bool' ? array(null => 'All', 1 => 'True', 0 => 'False') : "text");
}
$value['edit'] = array('title' => 'Edit', 'width' => '100');
$value['delete'] = array('title' => 'Delete', 'width' => '100');
$property->setDefaultValue($value);
$class->addPropertyFromGenerator($property);
}
/**
* creates property for given field
*
* @param FieldDescriptorInterface $column
* @param string $name
* @return \Zend\Code\Generator\PropertyGenerator
*/
protected function createProperty(FieldDescriptorInterface $column, $name)
{
$type = $this->getFieldType($column);
$docblock = new \Zend\Code\Generator\DocblockGenerator('Column: ' . $column->getName());
$docblock->setTag(array("name" => "var", "description" => $type));
if ($column instanceof ReferenceFieldInterface) {
if ($column->referencedFieldName()) {
$docblock->setLongDescription("Reference to " . $column->referencedDataSetName() . "." . $column->referencedFieldName());
}
}
$property = new \Zend\Code\Generator\PropertyGenerator();
$property->addFlag(\Zend\Code\Generator\PropertyGenerator::FLAG_PUBLIC);
$property->setName(lcfirst($name));
$property->setDocBlock($docblock);
return $property;
}
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);
}
