amtgard/ORK3

<?php

include_once('class.YapoDb.php');

class YapoMysql extends YapoDb {

    private $DBH;
    
    private $Data;
    
    function __construct($host, $dbname, $user, $password) {
        $this->DBH = new PDO("mysql:host=$host;dbname=$dbname", $user, $password);
        $this->DBH->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING);
    }
    
    function TableDescription($table) {
        $Keys = $this->DataSet("SHOW KEYS IN $table");
        $Fields = $this->DataSet("describe $table");
        $this->Clear();
        
        $keys = array();
        
        while ($Keys->Next()) {
            if (!isset($keys[$Keys->Key_name]) || !is_array($keys[$Keys->Key_name]))
                $keys[$Keys->Key_name] = array('Unique'=>!$Keys->Non_unique,'Columns'=>array());
            $keys[$Keys->Key_name]['Columns'][] = $Keys->Column_name;
        }
        
        
        $fields = array();
        $primary_key = false;
        while ($Fields->Next()) {
            preg_match("/(.+)\((.+)\)/", $Fields->Type, $matches);
            $fields[$Fields->Field] = array(
                    'MajorType' => count($matches) < 3 ? $Fields->Type : $matches[1],
                    'MinorType' => count($matches) < 3 ? $Fields->Type : $matches[2],
                    'Type' => $Fields->Type,
                    'Null' => $Fields->Null=="NO"?false:true,
                    'Key' => $Fields->Key,
                    'Extra' => $Fields->Extra
                );
            if (strtoupper($Fields->Key) == 'PRI') $primary_key = $Fields->Field;
        }
        
        return array("Keys" => $keys, "Fields" => $fields, "PrimaryKey" => $primary_key);
    }    
    
    function GetLastInsertId() {
        return $this->DBH->lastInsertId();
    }
    
    function GetCore($table) {
        return new YapoCoreMysql($this, $table);
    }
    
    function Execute($sql) {
        if ($this->Debug) {
            echo $sql;
            print_r($this->Data);
        }
        $cnt = 3;
        do {
            $Query = $this->DBH->prepare($sql);
            if (count($this->Data) > 0)
                $Query->execute($this->Data);
            else
                $Query->execute();
            $failed = $this->handle_errors($cnt--, $Query);
        } while (!$failed);
    }
    
    function DataSet($sql) {
        if ($this->Debug) {
            echo $sql;
            print_r($this->Data);
        }
        $cnt = 3;
        do {
            $Query = $this->DBH->prepare($sql);
            if (count($this->Data) > 0)
                $Query->execute($this->Data);
            else
                $Query->execute();
            $failed = $this->handle_errors($cnt--, $Query);
        } while (!$failed);
        return new YapoResultSet($Query, $sql);
    }
    
    function Clear() {
        $this->Data = array();
    }
    
    function __set($field, $value) {
        if (is_object($value)) die("you cannot insert an object.");
        $this->Data[":$field"] = $value;
    }
    
    function SetData($Data) {
        $this->Data = $Data;
    }
    
    function ValidateField($field_def, $value) {
        if (stristr($field_def['MajorType'], 'int') || 
            stristr($field_def['MajorType'], 'float') || 
            stristr($field_def['MajorType'], 'double') || 
            stristr($field_def['MajorType'], 'real') ||
            stristr($field_def['MajorType'], 'decimal') ||
            stristr($field_def['MajorType'], 'numeric')) {
            return $value;
        } else if (strtoupper($field_def['MajorType']) == 'TIME') {
            return "'" . date("H:i:s", strtotime($value)) . "'";
        } else if (stristr($field_def['MajorType'], 'time')) {
            return "'" . date("Y-m-d H:i:s", strtotime($value)) . "'";
        } else if (stristr($field_def['MajorType'], 'date')) {
            return "'" . date("Y-m-d", strtotime($value)) . "'";
        } else if (strtoupper($field_def['MajorType']) == 'YEAR') {
            return "'" . date("Y", strtotime($value)) . "'";
        } else if (stristr($field_def['MajorType'], 'text')) {
            // incomplete
        }
    }

}


?>