<?php

namespace queasy\db\query;

use PDO;

use queasy\db\Db;

class CustomQuery extends Query
{
    private $config;

    public function __construct(PDO $pdo, $config)
    {
        $this->config = $config;

        parent::__construct($pdo, $config['sql']);
    }

    /**
     * Executes SQL query and returns all selected rows.
     *
     * @param array $params Query parameters
     *
     * @return array Returned data depends on query, usually it is an array (or affected rows count for queries like INSERT, DELETE or UPDATE)
     *
     * @throws DbException On error
     */
    public function run(array $params = array(), array $options = array())
    {
        $config = $this->config;
        if (is_object($config) && method_exists($config, 'toArray')) {
            $config = $config->toArray();
        }

        $options = $options + (isset($config['options'])? $config['options']: array());

        $statement = parent::run($params, $options);

        $returns = isset($this->config['returns'])? $this->config['returns']: null;

        if ($returns) {
            $fetchMode = isset($this->config['fetchMode'])? $this->config['fetchMode']: null;
            $fetchArg = isset($this->config['fetchArg'])? $this->config['fetchArg']: null;
            switch ($returns) {
                case Db::RETURN_ONE:
                    return (Db::FETCH_CLASS === $fetchMode)
                        ? $statement->fetchObject($fetchArg? $fetchArg: 'stdClass')
                        : $statement->fetch($fetchMode);

                case Db::RETURN_ALL:
                    return (Db::FETCH_CLASS === $fetchMode)
                        ? $statement->fetchAll($fetchMode, $fetchArg)
                        : $statement->fetchAll($fetchMode);

                case Db::RETURN_VALUE:
                    $row = $statement->fetch();
                    $value = array_shift($row);
                    return $value;

                default:
                    return $statement;
            }
        }

        return $statement;
    }
}