<?php

/**

 * Zend Framework

 *

 * LICENSE

 *

 * This source file is subject to the new BSD license that is bundled

 * with this package in the file LICENSE.txt.

 * It is also available through the world-wide-web at this URL:

 * http://framework.zend.com/license/new-bsd

 * If you did not receive a copy of the license and are unable to

 * obtain it through the world-wide-web, please send an email

 * to license@zend.com so we can send you a copy immediately.

 *

 * @category   Zend

 * @package    Zend_Db

 * @subpackage Statement

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 * @version    $Id: Mysqli.php 20096 2010-01-06 02:05:09Z bkarwin $

 */

/**

 * @see Zend_Db_Statement

 */

require_once 'Zend/Db/Statement.php';

/**

 * Extends for Mysqli

 *

 * @category   Zend

 * @package    Zend_Db

 * @subpackage Statement

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 */

class Zend_Db_Statement_Mysqli extends Zend_Db_Statement

{

    /**

     * Column names.

     *

     * @var array

     */

    protected $_keys;

    /**

     * Fetched result values.

     *

     * @var array

     */

    protected $_values;

    /**

     * @var array

     */

    protected $_meta = null;

    /**

     * @param  string $sql

     * @return void

     * @throws Zend_Db_Statement_Mysqli_Exception

     */

    public function _prepare($sql)

    {

        $mysqli = $this->_adapter->getConnection();

        $this->_stmt = $mysqli->prepare($sql);

        if ($this->_stmt === false || $mysqli->errno) {

            /**

             * @see Zend_Db_Statement_Mysqli_Exception

             */

            require_once 'Zend/Db/Statement/Mysqli/Exception.php';

            throw new Zend_Db_Statement_Mysqli_Exception("Mysqli prepare error: " . $mysqli->error, $mysqli->errno);

        }

    }

    /**

     * Binds a parameter to the specified variable name.

     *

     * @param mixed $parameter Name the parameter, either integer or string.

     * @param mixed $variable  Reference to PHP variable containing the value.

     * @param mixed $type      OPTIONAL Datatype of SQL parameter.

     * @param mixed $length    OPTIONAL Length of SQL parameter.

     * @param mixed $options   OPTIONAL Other options.

     * @return bool

     * @throws Zend_Db_Statement_Mysqli_Exception

     */

    protected function _bindParam($parameter, &$variable, $type = null, $length = null, $options = null)

    {

        return true;

    }

    /**

     * Closes the cursor and the statement.

     *

     * @return bool

     */

    public function close()

    {

        if ($this->_stmt) {

            $r = $this->_stmt->close();

            $this->_stmt = null;

            return $r;

        }

        return false;

    }

    /**

     * Closes the cursor, allowing the statement to be executed again.

     *

     * @return bool

     */

    public function closeCursor()

    {

        if ($stmt = $this->_stmt) {

            $mysqli = $this->_adapter->getConnection();

            while ($mysqli->more_results()) {

                $mysqli->next_result();

            }

            $this->_stmt->free_result();

            return $this->_stmt->reset();

        }

        return false;

    }

    /**

     * Returns the number of columns in the result set.

     * Returns null if the statement has no result set metadata.

     *

     * @return int The number of columns.

     */

    public function columnCount()

    {

        if (isset($this->_meta) && $this->_meta) {

            return $this->_meta->field_count;

        }

        return 0;

    }

    /**

     * Retrieves the error code, if any, associated with the last operation on

     * the statement handle.

     *

     * @return string error code.

     */

    public function errorCode()

    {

        if (!$this->_stmt) {

            return false;

        }

        return substr($this->_stmt->sqlstate, 0, 5);

    }

    /**

     * Retrieves an array of error information, if any, associated with the

     * last operation on the statement handle.

     *

     * @return array

     */

    public function errorInfo()

    {

        if (!$this->_stmt) {

            return false;

        }

        return array(

            substr($this->_stmt->sqlstate, 0, 5),

            $this->_stmt->errno,

            $this->_stmt->error,

        );

    }

    /**

     * Executes a prepared statement.

     *

     * @param array $params OPTIONAL Values to bind to parameter placeholders.

     * @return bool

     * @throws Zend_Db_Statement_Mysqli_Exception

     */

    public function _execute(array $params = null)

    {

        if (!$this->_stmt) {

            return false;

        }

        // if no params were given as an argument to execute(),

        // then default to the _bindParam array

        if ($params === null) {

            $params = $this->_bindParam;

        }

        // send $params as input parameters to the statement

        if ($params) {

            array_unshift($params, str_repeat('s', count($params)));

            $stmtParams = array();

            foreach ($params as $k => &$value) {

                $stmtParams[$k] = &$value;

            }

            call_user_func_array(

                array($this->_stmt, 'bind_param'),

                $stmtParams

                );

        }

        // execute the statement

        $retval = $this->_stmt->execute();

        if ($retval === false) {

            /**

             * @see Zend_Db_Statement_Mysqli_Exception

             */

            require_once 'Zend/Db/Statement/Mysqli/Exception.php';

            throw new Zend_Db_Statement_Mysqli_Exception("Mysqli statement execute error : " . $this->_stmt->error, $this->_stmt->errno);

        }

        // retain metadata

        if ($this->_meta === null) {

            $this->_meta = $this->_stmt->result_metadata();

            if ($this->_stmt->errno) {

                /**

                 * @see Zend_Db_Statement_Mysqli_Exception

                 */

                require_once 'Zend/Db/Statement/Mysqli/Exception.php';

                throw new Zend_Db_Statement_Mysqli_Exception("Mysqli statement metadata error: " . $this->_stmt->error, $this->_stmt->errno);

            }

        }

        // statements that have no result set do not return metadata

        if ($this->_meta !== false) {

            // get the column names that will result

            $this->_keys = array();

            foreach ($this->_meta->fetch_fields() as $col) {

                $this->_keys[] = $this->_adapter->foldCase($col->name);

            }

            // set up a binding space for result variables

            $this->_values = array_fill(0, count($this->_keys), null);

            // set up references to the result binding space.

            // just passing $this->_values in the call_user_func_array()

            // below won't work, you need references.

            $refs = array();

            foreach ($this->_values as $i => &$f) {

                $refs[$i] = &$f;

            }

            $this->_stmt->store_result();

            // bind to the result variables

            call_user_func_array(

                array($this->_stmt, 'bind_result'),

                $this->_values

            );

        }

        return $retval;

    }

    /**

     * Fetches a row from the result set.

     *

     * @param int $style  OPTIONAL Fetch mode for this fetch operation.

     * @param int $cursor OPTIONAL Absolute, relative, or other.

     * @param int $offset OPTIONAL Number for absolute or relative cursors.

     * @return mixed Array, object, or scalar depending on fetch mode.

     * @throws Zend_Db_Statement_Mysqli_Exception

     */

    public function fetch($style = null, $cursor = null, $offset = null)

    {

        if (!$this->_stmt) {

            return false;

        }

        // fetch the next result

        $retval = $this->_stmt->fetch();

        switch ($retval) {

            case null: // end of data

            case false: // error occurred

                $this->_stmt->reset();

                return false;

            default:

                // fallthrough

        }

        // make sure we have a fetch mode

        if ($style === null) {

            $style = $this->_fetchMode;

        }

        // dereference the result values, otherwise things like fetchAll()

        // return the same values for every entry (because of the reference).

        $values = array();

        foreach ($this->_values as $key => $val) {

            $values[] = $val;

        }

        $row = false;

        switch ($style) {

            case Zend_Db::FETCH_NUM:

                $row = $values;

                break;

            case Zend_Db::FETCH_ASSOC:

                $row = array_combine($this->_keys, $values);

                break;

            case Zend_Db::FETCH_BOTH:

                $assoc = array_combine($this->_keys, $values);

                $row = array_merge($values, $assoc);

                break;

            case Zend_Db::FETCH_OBJ:

                $row = (object) array_combine($this->_keys, $values);

                break;

            case Zend_Db::FETCH_BOUND:

                $assoc = array_combine($this->_keys, $values);

                $row = array_merge($values, $assoc);

                return $this->_fetchBound($row);

                break;

            default:

                /**

                 * @see Zend_Db_Statement_Mysqli_Exception

                 */

                require_once 'Zend/Db/Statement/Mysqli/Exception.php';

                throw new Zend_Db_Statement_Mysqli_Exception("Invalid fetch mode '$style' specified");

                break;

        }

        return $row;

    }

    /**

     * Retrieves the next rowset (result set) for a SQL statement that has

     * multiple result sets.  An example is a stored procedure that returns

     * the results of multiple queries.

     *

     * @return bool

     * @throws Zend_Db_Statement_Mysqli_Exception

     */

    public function nextRowset()

    {

        /**

         * @see Zend_Db_Statement_Mysqli_Exception

         */

        require_once 'Zend/Db/Statement/Mysqli/Exception.php';

        throw new Zend_Db_Statement_Mysqli_Exception(__FUNCTION__.'() is not implemented');

    }

    /**

     * Returns the number of rows affected by the execution of the

     * last INSERT, DELETE, or UPDATE statement executed by this

     * statement object.

     *

     * @return int     The number of rows affected.

     */

    public function rowCount()

    {

        if (!$this->_adapter) {

            return false;

        }

        $mysqli = $this->_adapter->getConnection();

        return $mysqli->affected_rows;

    }

}