Select.php

<?php

namespace Nulpunkt\Yesql\Statement;

class Select implements Statement
{
    private $sql;
    private $modline;
    private $rowFunc;
    private $rowClass;
    private $stmt;

    public function __construct($sql, $modline)
    {
        $this->sql = $sql;
        $this->modline = $modline;
        $this->rowFunc = $this->getRowFunc();
        $this->rowClass = $this->getRowClass();
    }

    public function execute($db, $args)
    {
        if (!$this->stmt) {
            $this->stmt = $db->prepare($this->sql);
        }

        $this->stmt->execute($args);

        if ($this->rowClass) {
            $this->stmt->setFetchMode(\PDO::FETCH_CLASS, $this->rowClass);
        } else {
            $this->stmt->setFetchMode(\PDO::FETCH_ASSOC);
        }

        $res = array_map([$this, 'prepareElement'], $this->stmt->fetchAll());

        return $this->oneOrMany() == 'one' ? @$res[0] : $res;
    }

    private function oneOrMany()
    {
        preg_match("/\boneOrMany:\s*(one|many)/", $this->modline, $m);
        return isset($m[1]) ? $m[1] : "many";
    }

    private function getRowClass()
    {
        preg_match('/rowClass:\s*(\S+)/', $this->modline, $m);
        $c = @$m[1];

        if ($c && !class_exists($c)) {
            throw new \Nulpunkt\Yesql\Exception\ClassNotFound("{$c} is not a class");
        }

        return $c;
    }

    private function getRowFunc()
    {
        preg_match('/rowFunc:\s*(\S+)/', $this->modline, $m);
        $f = isset($m[1]) ? $m[1] : [$this, 'identity'];

        if ($f && !is_callable($f)) {
            throw new \Nulpunkt\Yesql\Exception\MethodMissing("{$f} is not callable");
        }

        return $f;
    }

    private function prepareElement($res)
    {
        return call_user_func($this->rowFunc, $res);
    }

    private function identity($e)
    {
        return $e;
    }
}