From 22276c273cc158cf12fd7b63e581b08cdae7b6f1 Mon Sep 17 00:00:00 2001 From: Daniel Date: Sat, 30 Nov 2013 02:58:02 -0500 Subject: Adding Many To Many relationships support! --- framework/Wsat/TWsatARGenerator.php | 457 +++++++++++++++++++----------------- 1 file changed, 245 insertions(+), 212 deletions(-) (limited to 'framework/Wsat/TWsatARGenerator.php') diff --git a/framework/Wsat/TWsatARGenerator.php b/framework/Wsat/TWsatARGenerator.php index a4846306..5c812ef5 100644 --- a/framework/Wsat/TWsatARGenerator.php +++ b/framework/Wsat/TWsatARGenerator.php @@ -12,244 +12,277 @@ 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; - } + /** + * 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(); + } -//----------------------------------------------------------------------------- - // - public function generate($tableName) - { - $tableInfo = $this->_gateway->getTableInfo($this->_conn, $tableName); - if (count($tableInfo->getColumns()) === 0) + /** + * Destructor. + * Disconnect the db connection. + */ + public function __destruct() + { + if ($this->_conn !== null) + $this->_conn->Active = false; + } + + public function setOpFile($op_file_namespace) { - throw new Exception("Unable to find table or view $tableName in " . $this->_conn->getConnectionString() . "."); - } else + $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) { - $properties = array(); - foreach ($tableInfo->getColumns() as $field => $column) - $properties[] = $this->generateProperty($field, $column); - $toString = $this->_buildSmartToString($tableInfo); + $this->_class_sufix = $_clas_sufix; } - $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 generate($tableName) + { + $tableInfo = $this->_gateway->getTableInfo($this->_conn, $tableName); + $this->_commonGenerate($tableName, $tableInfo); + } - public function generateAll() - { - foreach ($this->_getAllTableNames() as $tableName) + public function generateAll() { - if ($tableName == "pradocache") - { - continue; - } - $this->generate($tableName); + foreach ($this->_getAllTableNames() as $tableName) + { + if ($tableName == "pradocache") + continue; + $tableInfo = $this->_gateway->getTableInfo($this->_conn, $tableName); + if (!empty($this->_relations)) + { + // Cancel generation of M-M relationships middle table + if (count($tableInfo->getPrimaryKeys()) === 2 && count($tableInfo->getColumns()) === 2)//M-M relationships + continue; + } + $this->_commonGenerate($tableName, $tableInfo); + } } - } - public function buildRelations() - { - $this->_relations = array(); - foreach ($this->_getAllTableNames() as $table_name) + public function buildRelations() { - $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 - ); - } + $this->_relations = array(); + foreach ($this->_getAllTableNames() as $table_name) + { + $tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name); + $pks = $tableInfo->getPrimaryKeys(); + $fks = $tableInfo->getForeignKeys(); + + if (count($pks) === 2 && count($tableInfo->getColumns()) === 2)//M-M relationships + { + $table_name_mm = $fks[0]["table"]; + $table_name_mm2 = $fks[1]["table"]; + + $this->_relations[$table_name_mm][] = array( + "prop_name" => strtolower($table_name_mm2), + "rel_type" => "self::MANY_TO_MANY", + "ref_class_name" => $this->_getProperClassName($table_name_mm2), + "prop_ref" => $table_name + ); + + $this->_relations[$table_name_mm2][] = array( + "prop_name" => strtolower($table_name_mm), + "rel_type" => "self::MANY_TO_MANY", + "ref_class_name" => $this->_getProperClassName($table_name_mm), + "prop_ref" => $table_name + ); + continue; + } + foreach ($fks as $fk_data)//1-M relationships + { + $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) + // + + private function _commonGenerate($tableName, $tableInfo) { - echo $table_name . "
"; + 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); + } - $tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name); - echo "Table info:" . "
"; - echo "
";
-            var_dump($tableInfo);
-            echo "
"; + $clasName = $this->_getProperClassName($tableName); + $class = $this->generateClass($properties, $tableName, $clasName, $toString); + $output = $this->_op_file . DIRECTORY_SEPARATOR . $clasName . ".php"; + file_put_contents($output, $class); } - } -//----------------------------------------------------------------------------- + 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; + } - 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 + private function _getProperClassName($tableName) { - 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") + $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) { - $property = str_replace($this->uq_chars, "", $column->ColumnName); - break; + echo $table_name . "
"; + + $tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name); + echo "Table info:" . "
"; + echo "
";
+                        var_dump($tableInfo);
+                        echo "
"; } - } - } catch (Exception $ex) + } + +//----------------------------------------------------------------------------- + + 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) { - Prado::trace($ex->getMessage()); + $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) + { + Prado::trace($ex->getMessage()); + } + $code .= "\n\t\treturn \$this->$property;"; + $code .= "\n\t}"; + return $code; } - $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 <<_renderRelations($tablename); + $date = date('Y-m-d h:i:s'); + $env_user = getenv("username"); + return << } \ No newline at end of file -- cgit v1.2.3