<?php

/**
 * @author Daniel Sampedro Bello <darthdaniel85@gmail.com>
 * @link http://www.pradosoft.com/
 * @copyright Copyright &copy; 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;
    }

//-----------------------------------------------------------------------------    
    // <editor-fold defaultstate="collapsed" desc="Main APIs">
    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
                );
            }
        }
    }

// </editor-fold>
//-----------------------------------------------------------------------------
    // <editor-fold defaultstate="collapsed" desc="Common Methods">

    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 . "<br>";

            $tableInfo = $this->_gateway->getTableInfo($this->_conn, $table_name);
            echo "Table info:" . "<br>";
            echo "<pre>";
            var_dump($tableInfo);
            echo "</pre>";
        }
    }

//-----------------------------------------------------------------------------

    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)
        {
            Prado::trace($ex->getMessage());
        }
        $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
<?php
/**
 * Auto generated by PRADO - WSAT on $date.
 * @author prado_user_name               
 */
class $classname extends TActiveRecord 
{       
	const TABLE='$tablename';

$props

	public static function finder(\$className=__CLASS__) {
                return parent::finder(\$className);
	}
                
$relations
                
$toString
}
EOD;
    }

// </editor-fold>
}