* @link http://www.pradosoft.com/ * @copyright Copyright © 2005-2013 PradoSoft * @license http://www.pradosoft.com/license/ * @version $Id$ * @since 3.3 * @package Wsat */ class TWsatARGenerator { /** * Gets the current Db connection, the connection object is obtained from * the TActiveRecordManager if connection is currently null. * @return TDbConnection current db connection for this object. */ private $_conn; /** * @return TActiveRecordGateway record table gateway. */ private $_gateway; /** * Output folder where AR classes will be generated. */ private $_op_file; /** * Class name prefix */ private $_clas_prefix; /** * Class name sufix */ private $_class_sufix; /** * all table relations array */ private $_relations; /** * unquote chars * @var array */ private $uq_chars = array('[', ']', '"', '`', "'"); function __construct() { $ar_manager = TActiveRecordManager::getInstance(); $this->_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 setClassSufix($_clas_sufix) { $this->_class_sufix = $_clas_sufix; } //----------------------------------------------------------------------------- // 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 . $this->_class_sufix; } 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($this->uq_chars, "", $column->ColumnName); } elseif ($column->PHPType == "string" && $column->DBType != "date") { $property = str_replace($this->uq_chars, "", $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; } //
} ?>