_conn = $ar_manager->getDbConnection();
$this->_conn->Active = true;
$this->_gateway = $ar_manager->getRecordGateway();
}
/**
* Destructor.
* Disconnect the db connection.
*/
public function __destruct() {
if ($this->_conn !== null)
$this->_conn->Active = false;
}
public function setOpFile($op_file_namespace) {
$op_file = Prado::getPathOfNamespace($op_file_namespace);
if (empty($op_file)) {
throw new Exception("You need to fix your output folder namespace.");
}
if (!is_dir($op_file)) {
mkdir($op_file, 0777, true);
}
$this->_op_file = $op_file;
}
public function setClasPrefix($_clas_prefix) {
$this->_clas_prefix = $_clas_prefix;
}
//-----------------------------------------------------------------------------
//
public function generate($tableName) {
$tableInfo = $this->_gateway->getTableInfo($this->_conn, $tableName);
if (count($tableInfo->getColumns()) === 0) {
throw new Exception("Unable to find table or view $tableName in " . $this->_conn->getConnectionString() . ".");
} else {
$properties = array();
foreach ($tableInfo->getColumns() as $field => $column)
$properties[] = $this->generateProperty($field, $column);
$toString = $this->_buildSmartToString($tableInfo);
}
$clasName = $this->_getProperClassName($tableName);
$class = $this->generateClass($properties, $tableName, $clasName, $toString);
$output = $this->_op_file . DIRECTORY_SEPARATOR . $clasName . ".php";
file_put_contents($output, $class);
}
public function generateAll() {
foreach ($this->_getAllTableNames() as $tableName) {
if ($tableName == "pradocache") {
continue;
}
$this->generate($tableName);
}
}
public function buildRelations() {
$this->_relations = array();
foreach ($this->_getAllTableNames() as $table_name) {
$tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name);
foreach ($tableInfo->getForeignKeys() as $fk_data) {
$owner_table = $fk_data["table"];
$slave_table = $table_name;
$fk_prop = key($fk_data["keys"]);
$this->_relations[$owner_table][] = array(
"prop_name" => strtolower($slave_table),
"rel_type" => "self::HAS_MANY",
"ref_class_name" => $this->_getProperClassName($slave_table),
"prop_ref" => $fk_prop
);
$this->_relations[$slave_table][] = array(
"prop_name" => strtolower($owner_table),
"rel_type" => "self::BELONGS_TO",
"ref_class_name" => $this->_getProperClassName($owner_table),
"prop_ref" => $fk_prop
);
}
}
}
//
//-----------------------------------------------------------------------------
//
private function _getAllTableNames() {
$command = $this->_conn->createCommand("Show Tables");
$dataReader = $command->query();
$dataReader->bindColumn(1, $table);
$tables = array();
while ($dataReader->read()) {
$tables[] = $table;
}
return $tables;
}
private function _getProperClassName($tableName) {
$table_name_words = str_replace("_", " ", strtolower($tableName));
$final_conversion = str_replace(" ", "", ucwords($table_name_words));
return $this->_clas_prefix . $final_conversion;
}
public function renderAllTablesInformation() {
foreach ($this->_getAllTableNames() as $table_name) {
echo $table_name . "
";
$tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name);
echo "Table info:" . "
";
echo "";
var_dump($tableInfo);
echo "";
}
}
//-----------------------------------------------------------------------------
protected function generateProperty($field, $column) {
$prop = '';
$name = '$' . $field;
/* TODO use in version 2.0 */
// $type = $column->getPHPType();
$prop .= "\tpublic $name;";
return $prop;
}
private function _renderRelations($tablename) {
if (!isset($this->_relations[$tablename])) {
return "";
}
$code = "\tpublic static \$RELATIONS = array (";
foreach ($this->_relations[$tablename] as $rel_data) {
$code .= "\n\t\t'" . $rel_data["prop_name"] . "' => array(" . $rel_data["rel_type"] . ", '" . $rel_data["ref_class_name"] . "', '" . $rel_data["prop_ref"] . "'),";
}
$code = substr($code, 0, -1);
$code .= "\n\t);";
return $code;
}
private function _buildSmartToString($tableInfo) {
$code = "\tpublic function __toString() {";
$property = "throw new THttpException(500, 'Not implemented yet.');";
try {
foreach ($tableInfo->getColumns() as $column) {
if (isset($column->IsPrimaryKey) && $column->IsPrimaryKey) {
$property = str_replace(array("`", "'", '"'), "", $column->ColumnName);
} elseif ($column->DbType == "varchar") {
$property = str_replace(array("`", "'", '"'), "", $column->ColumnName);
break;
}
}
} catch (Exception $ex) {
}
$code .= "\n\t\treturn \$this->$property;";
$code .= "\n\t}";
return $code;
}
protected function generateClass($properties, $tablename, $classname, $toString) {
$props = implode("\n", $properties);
$relations = $this->_renderRelations($tablename);
$date = date('Y-m-d h:i:s');
return <<
EOD;
}
//
}
?>