php-yaoi/php-yaoi

<?php

namespace Yaoi\Database\Driver;

use SQLite3;
use SQLite3Result;
use Yaoi\Database;
use Yaoi\Database\Driver;
use Yaoi\Sql\Symbol;
use Yaoi\Undefined;

class Sqlite extends Driver
{

    /**
     * @var SQLite3
     */
    private $dbHandle;

    private function connect()
    {
        if (null === $this->dbHandle) {
            $this->dbHandle = new SQLite3($this->dsn->path);
        }
    }


    /**
     * @param $statement
     * @return SQLite3Result
     */
    public function query($statement)
    {
        if (null === $this->dbHandle) {
            $this->connect();
        }
        try {
            $result = @$this->dbHandle->query($statement);
            return $result;
        } catch (\Exception $exception) {
            $dbException = new Database\Exception($exception->getMessage(), $exception->getCode());
            $dbException->query = $statement;
            throw $dbException;
        }
    }

    public function lastInsertId()
    {
        return $this->dbHandle->lastInsertRowID();
    }

    /**
     * @param SQLite3Result $result
     * @return int
     */
    public function rowsAffected($result)
    {
        return $this->dbHandle->changes();
    }

    public function escape($value)
    {
        if (null === $this->dbHandle) {
            $this->connect();
        }
        if ($value instanceof Undefined) {
            $value = null;
        }

        return $this->dbHandle->escapeString($value);
    }

    /**
     * @param SQLite3Result $result
     */
    public function rewind($result)
    {
        $result->reset();
    }

    /**
     * @param SQLite3Result $result
     * @return array|false
     */
    public function fetchAssoc($result)
    {
        $row = $result->fetchArray(SQLITE3_ASSOC);
        return $row ? $row : null;
    }

    public function queryErrorMessage($result)
    {
        return $this->dbHandle->lastErrorCode() . ' ' . $this->dbHandle->lastErrorMsg();
    }

    public function disconnect()
    {
        if ($this->dbHandle) {
            $this->dbHandle->close();
        }
    }

    public function quoteSymbol(Symbol $symbol)
    {
        $result = '';
        foreach ($symbol->names as $name) {
            $result .= '`' . str_replace('`', '``', $name) . '`.';
        }
        if ($result) {
            $result = substr($result, 0, -1);
        }
        return $result;
    }

    public function getDialect()
    {
        return Database::DIALECT_SQLITE;
    }

    /**
     * @return Contract
     */
    public function getUtility()
    {
        return new Database\Sqlite\Utility();
    }


}