<?php
/**
 * モデルクラスの自動生成
 * @author Daijiro Abe
 * @date   2007.12.06
 */

require_once("adodb/adodb.inc.php");

class GenerateModels
{
    var $dsn;
    var $dir = "classes";
    var $sep = "/";
    
    function GenerateModels($dsn)
    {
        $this->dsn = $dsn;
    }
    
    /**
     * 生成ディレクトリを返します
     * @return 生成ディレクトリ名
     */
    function getDirectory()
    {
        $dir = $this->dir;
        if(substr($dir, strlen($dir)-1, 1) != $this->sep)
            $dir .= $this->sep;
        return $dir;
    }
    
    /**
     * 生成ディレクトリを設定します
     * @param $dir 生成ディレクトリ
     */
    function setDirectory($dir)
    {
        $this->dir = $dir;
    }
    
    /**
     * ディレクトリセパレータを返します
     * @return ディレクトリセパレータ
     */
    function getSeparator()
    {
        return $this->sep;
    }
    
    /**
     * ディレクトリセパレータを設定します
     * @param $sep ディレクトリセパレータ
     */
    function setSeparator($sep)
    {
        $this->sep = $sep;
    }
    
    /**
     * クラスファイルの生成
     * @return 失敗時にfalse、それ以外でtrue
     */
    function generate()
    {
        $conn = &ADONewConnection($this->dsn);
        
        $date = date("Y/m/d");
        $dsn = $this->dsn;
        
        $fp = fopen($this->dir . $this->sep . "CommonDAO.php", "w");
        if(!$fp)
            return false;
        $body = <<< PHP_END
<?php
/**
 * @class  CommonDAO
 * @brief  共通DAOクラス
 * @author Daijiro Abe
 * @date   {$date}
 */

require_once(dirname(__FILE__) . '/../lib/Capsule/AbstractDAO.php');
require_once(dirname(__FILE__) . '/../lib/Capsule/AbstractDTO.php');
require_once(dirname(__FILE__) . '/../lib/Capsule/Collection.php');

class CommonDAO extends AbstractDAO
{
    function CommonDAO()
    {
        \$this->connect("{$dsn}");
    }
}
PHP_END;
        if(fwrite($fp, $body) === FALSE)
            return false;
        fclose($fp);
        
        $tables = $conn->MetaTables("TABLES");
        foreach($tables as $table)
        {
            $columns = $conn->MetaColumnNames($table, true);
            $pkeys = $conn->MetaPrimaryKeys($table);

            $class_name = ucfirst($table);
            $dirpath = $this->dir . $this->sep . $class_name;
            if(!file_exists($dirpath))
                mkdir($dirpath);
            $dirpath .= $this->sep;
            
            // DTOクラスの生成
            $dto_class = $class_name;
            $dto_path = $dirpath . $dto_class . ".php";
            $fp = fopen($dto_path, "w");
            if(!$fp)
                return false;
            
            $body = <<< PHP_END
<?php
/**
 * {$dto_class}クラス
 * @author Daijiro Abe
 * @date   {$date}
 */

class {$dto_class} extends AbstractDTO
{

PHP_END;
            foreach($columns as $column)
            {
                $column = strtolower($column);
                $method = ucfirst($column);
                $body .= "\tfunction get$method()\n";
                $body .= "\t{\n";
                $body .= "\t\treturn \$this->get(\"$column\");\n";
                $body .= "\t}\n";
                $body .= "\tfunction set$method(\$$column)\n";
                $body .= "\t{\n";
                $body .= "\t\t\$this->set(\"$column\", \$$column);\n";
                $body .= "\t}\n";
            }
            
            $body .= "}\n";
            
            if(fwrite($fp, $body) === FALSE)
                return false;
            fclose($fp);
            
            // Collectionクラスの生成
            $collection_class = $class_name . "Collection";
            $collection_path = $dirpath . $collection_class . ".php";
            $fp = fopen($collection_path, "w");
            if(!$fp)
                return false;
            
            $body = <<< PHP_END
<?php
/**
 * {$collection_class}クラス
 * @author Daijiro Abe
 * @date   {$date}
 */

require_once(dirname(__FILE__) . "/{$class_name}.php");

class {$collection_class} extends Collection
{
    function &makeObject(\$array)
    {
        \$obj = new {$class_name}(\$array);
        return \$obj;
    }
}

PHP_END;
            if(fwrite($fp, $body) === FALSE)
                return false;
            fclose($fp);
            
            // DAOクラスの生成
            $dao_class = $class_name . "DAO";
            $dao_path = $dirpath . $dao_class . ".php";
            $fp = fopen($dao_path, "w");
            if(!$fp)
                return false;
            
            $body = <<< PHP_END
<?php
/**
 * {$dao_class}クラス
 * @author Daijiro Abe
 * @date   {$date}
 */

require_once(dirname(__FILE__) . "/../CommonDAO.php");
require_once(dirname(__FILE__) . "/{$class_name}Collection.php");

class {$dao_class} extends CommonDAO
{

PHP_END;
            if($pkeys)
            {
                $body .= "\tfunction &findByPrimaryKey(";
                foreach($pkeys as $index => $pkey)
                {
                    $pkey = strtolower($pkey);
                    if($index > 0)
                        $body .= ", ";
                    $body .= "$" . $pkey;
                }
                $body .= ")\n";
                $body .= "\t{\n";
                $body .= "\t\treturn \$this->queryOne(\"select * from $table where ";
                foreach($pkeys as $index => $pkey)
                {
                    $pkey = strtolower($pkey);
                    if($index > 0)
                        $body .= " AND ";
                    $body .= "$pkey = ?";
                }
                $body .= "\", array(";
                foreach($pkeys as $index => $pkey)
                {
                    $pkey = strtolower($pkey);
                    if($index > 0)
                        $body .= ", ";
                    $body .= "$" . $pkey;
                }
                $body .= "));\n";
                $body .= "\t}\n";
            }
            
            $body .= "\tfunction &findAll()\n";
            $body .= "\t{\n";
            $body .= "\t\treturn \$this->query(\"select * from $table\");\n";
            $body .= "\t}\n";
            
            $body .= "\tfunction insert(\$".$table.")\n";
            $body .= "\t{\n";
            $body .= "\t\treturn \$this->doInsert(\"$table\", \$".$table."->get());\n";
            $body .= "\t}\n";
            
            if($pkeys)
            {
                $body .= "\tfunction update(\$$table)\n";
                $body .= "\t{\n";
                foreach($pkeys as $index => $pkey)
                {
                    $pkey = strtolower($pkey);
                    $body .= "\t\t\$" . $pkey . " = \$this->qstr(\$" . $table . "->get(\"$pkey\"));\n";
                }
                $body .= "\t\treturn \$this->doUpdate(\"$table\", \$".$table."->get(), \"";
                foreach($pkeys as $index => $pkey)
                {
                    $pkey = strtolower($pkey);
                    if($index > 0)
                        $body .= " AND ";
                    $body .= "$pkey = \$".$pkey;
                }
                $body .= "\");\n";
                $body .= "\t}\n";
            }
            
            $body .= "\tfunction &createCollection(\$result)\n";
            $body .= "\t{\n";
            $body .= "\t\t\$collection = new " . $collection_class . "(\$result);\n";
            $body .= "\t\treturn \$collection;\n";
            $body .= "\t}\n";
            $body .= "}\n";
            if(fwrite($fp, $body) === FALSE)
                return false;
            fclose($fp);
        }
        return true;
    }
}