/**
* Generate class
* @return string generated source code
*/
public function generate()
{
$table = new ModelGenerator_Table_Table($this->_options['tableName']);
$className = $this->_getNamer()->formatClassName($this->_config->base->classname);
$templates = array('tags' => array());
foreach ($this->_options['docblock'] as $tag => $value) {
$templates['tags'][] = array('name' => $tag, 'description' => $value);
}
////////////////////////////////////////////
// create model base
////////////////////////////////////////////
$methods = array();
$properties = array();
$tmp = array();
foreach ($table->getProperties() as $property) {
$tmp[] = array('name' => 'property', 'description' => $property['type'] . ' $' . $property['name'] . ' ' . $property['desc']);
switch ($property['type']) {
case 'string':
$property['defaultValue'] = '';
break;
case 'int':
case 'float':
case 'double':
$property['defaultValue'] = null;
break;
default:
$property['defaultValue'] = '';
break;
}
$properties[] = new Zend_CodeGenerator_Php_Property(array('name' => '_' . $property['name'], 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('tags' => array(array('name' => 'var', 'description' => $property['type'])))), 'defaultValue' => $property['defaultValue'], 'visibility' => 'private'));
// getters and setters
if (false !== strpos($property['desc'], 'enum')) {
$types = rtrim(str_replace(array('enum('), '', $property['desc']), ')');
$types = explode(',', $types);
$typesImploded = implode(', ', $types);
$enumBody = 'in_array($' . $property['name'] . ', array(' . $typesImploded . ')) ? $' . $property['name'] . ' : ' . $types[0];
$body = '$' . $property['name'] . ' = (' . $property['type'] . ') $' . $property['name'] . ';' . PHP_EOL . '$this->_' . $property['name'] . ' = ' . $enumBody . ';' . PHP_EOL . 'return $this;';
} else {
$body = '$this->_' . $property['name'] . ' = (' . $property['type'] . ') $' . $property['name'] . ';' . PHP_EOL . 'return $this;';
}
$methods[] = new Zend_CodeGenerator_Php_Method(array('name' => 'set' . $this->_getNamer()->formatMethodName($property['name']), 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('shortDescription' => 'Set ' . $property['name'] . ' (' . $property['desc'] . ')', 'tags' => array(new Zend_CodeGenerator_Php_Docblock_Tag_Param(array('paramName' => $property['name'], 'dataType' => $property['type'])), array('name' => 'return', 'description' => $className)))), 'parameters' => array(array('name' => $property['name'])), 'body' => $body));
$methods[] = new Zend_CodeGenerator_Php_Method(array('name' => 'get' . $this->_getNamer()->formatMethodName($property['name']), 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('shortDescription' => 'Get ' . $property['name'] . ' (' . $property['desc'] . ')', 'tags' => array(array('name' => 'return', 'description' => $property['type'])))), 'parameters' => array(), 'body' => 'return $this->_' . $property['name'] . ';'));
}
$toArrayBody[] = '$data = array(';
foreach ($table->getProperties() as $property) {
$toArrayBody[] = "\t" . '\'' . $property['name'] . '\' => $this->_' . $property['name'] . ',';
}
$toArrayBody[] = ');';
$toArrayBody[] = '';
$toArrayBody[] = 'return $data;';
$methods[] = new Zend_CodeGenerator_Php_Method(array('name' => 'toArray', 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('shortDescription' => 'Convert entity properties to array and return it', 'tags' => array(array('name' => 'return', 'description' => 'array')))), 'parameters' => array(), 'body' => implode(PHP_EOL, $toArrayBody)));
$modelBase = new Zend_CodeGenerator_Php_Class(array('name' => $className, 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('shortDescription' => $className . PHP_EOL . '*DO NOT* edit this file.', 'tags' => array_merge($tmp, $templates['tags']))), 'methods' => $methods, 'properties' => $properties));
$modelBaseFile = new Zend_CodeGenerator_Php_File(array('classes' => array($modelBase)));
return $modelBaseFile->generate();
}
/**
* Generate class
* @return string generated source code
*/
public function generate()
{
$table = new ModelGenerator_Table_Table($this->_options['tableName']);
$className = $this->_getNamer()->formatClassName($this->_config->baseMapper->classname);
$templates = array('tags' => array());
foreach ($this->_options['docblock'] as $tag => $value) {
$templates['tags'][] = array('name' => $tag, 'description' => $value);
}
$methods = array();
$tableReferences = array();
foreach ($table->getUniqueKeys() as $key) {
$methods[] = new Zend_CodeGenerator_Php_Method(array('name' => 'findBy' . $this->_getNamer()->formatMethodName($key), 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('tags' => array(new Zend_CodeGenerator_Php_Docblock_Tag_Param(array('paramName' => $key, 'dataType' => 'mixed')), array('name' => 'return', 'description' => $this->_getNamer()->formatClassName($this->_config->classname))))), 'parameters' => array(array('name' => $key)), 'body' => 'return $this->findOne(array(\'' . $key . ' = ?\' => $' . $key . '));'));
}
$modelTableBase = new Zend_CodeGenerator_Php_Class(array('name' => $className, 'docblock' => new Zend_CodeGenerator_Php_Docblock(array('shortDescription' => $className . PHP_EOL . '*DO NOT* edit this file.', 'tags' => array_merge($templates['tags']))), 'extendedClass' => 'Zend_Db_Table_Abstract', 'properties' => array(array('name' => '_name', 'visiblity' => 'protected', 'defaultValue' => $table->getName()), array('name' => '_primary', 'visiblity' => 'protected', 'defaultValue' => $table->getPrimary()), array('name' => '_dependantTables', 'visiblity' => 'protected', 'defaultValue' => $table->getDependantTables())), 'methods' => $methods));
$modelTableBaseFile = new Zend_CodeGenerator_Php_File(array('classes' => array($modelTableBase)));
return $modelTableBaseFile->generate();
}
