From ab6effd64caefa732ef0994d6c3048f5cd474bf6 Mon Sep 17 00:00:00 2001 From: Daniel Date: Fri, 22 Nov 2013 15:25:30 -0500 Subject: WSAT beta release. --- framework/Wsat/TWsatARGenerator.php | 368 +++++++++++++++++++++++------------- 1 file changed, 232 insertions(+), 136 deletions(-) (limited to 'framework/Wsat/TWsatARGenerator.php') diff --git a/framework/Wsat/TWsatARGenerator.php b/framework/Wsat/TWsatARGenerator.php index 7b39fce1..433e1640 100644 --- a/framework/Wsat/TWsatARGenerator.php +++ b/framework/Wsat/TWsatARGenerator.php @@ -1,136 +1,232 @@ -_ar_manager = TActiveRecordManager::getInstance(); - $this->_opnamespace = $_opnamespace; - } - - // - public function generate($tableName, $clasName = '') { - $conn = $this->_ar_manager->getDbConnection(); - $gateway = $this->_ar_manager->getRecordGateway(); - $tableInfo = $gateway->getTableInfo($conn, $tableName); - - if (count($tableInfo->getColumns()) === 0) { - throw new TIOException("Unable to find table or view $tableName in " . $conn->getConnectionString() . "."); - } else { - $properties = array(); - foreach ($tableInfo->getColumns() as $field => $column) - $properties[] = $this->generateProperty($field, $column); - } - - $clasName = empty($clasName) ? $this->_getClassName($tableName) : $clasName; - $class = $this->generateClass($properties, $tableName, $clasName); - $op_file = Prado::getPathOfNamespace($this->_opnamespace); - if (!is_dir($op_file)) { - mkdir($op_file, 0777, true); - } - - $output = $op_file . DIRECTORY_SEPARATOR . $clasName . ".php"; - file_put_contents($output, $class); - } - - public function generateAll() { - foreach ($this->_getAllTableNames() as $tableName) { - $this->generate($tableName); - } - } - -// -//----------------------------------------------------------------------------- - // - private function getDbConnection() { - $con = $this->_ar_manager->getDbConnection(); - $con->Active = true; - return $con; - } - - private function _getAllTableNames() { - $con = $this->getDbConnection(); - $command = $con->createCommand("Show Tables"); - $dataReader = $command->query(); - $dataReader->bindColumn(1, $table); - $tables = array(); - while ($dataReader->read()) { - $tables[] = $table; - } - $con->setActive(false); - return $tables; - } - - private function _getClassName($tableName) { - return ucfirst($tableName); - } - - public function renderAllTablesInformation() { - $conn = $this->_ar_manager->getDbConnection(); - $gateway = $this->_ar_manager->getRecordGateway(); - - foreach ($this->_getAllTableNames() as $table_name) { - echo $table_name . "
"; - - $tableInfo = $gateway->getTableInfo($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; - } - - protected function generateClass($properties, $tablename, $class) { - $props = implode("\n", $properties); - $date = date('Y-m-d h:i:s'); - return << -EOD; - } - -// -} - -?> +_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; + } + +// +} + +?> -- cgit v1.2.3