<?php

/*

 * This file is part of the Symfony package.

 *

 * (c) Fabien Potencier <fabien@symfony.com>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

namespace Symfony\Component\HttpKernel\Profiler;

use Symfony\Component\HttpKernel\DataCollector\DataCollectorInterface;

/**

 * SqliteProfilerStorage stores profiling information in a SQLite database.

 *

 * @author Fabien Potencier <fabien@symfony.com>

 */

class SqliteProfilerStorage extends PdoProfilerStorage

{

    /**

     * @throws \RuntimeException When neither of SQLite3 or PDO_SQLite extension is enabled

     */

    protected function initDb()

    {

        if (null === $this->db || $this->db instanceof \SQLite3) {

            if ('sqlite' !== substr($this->dsn, 0, 6 )) {

                throw new \RuntimeException('You are trying to use Sqlite with a wrong dsn. "'.$this->dsn.'"');

            }

            if (class_exists('SQLite3')) {

                $db = new \SQLite3(substr($this->dsn, 7, strlen($this->dsn)), \SQLITE3_OPEN_READWRITE | \SQLITE3_OPEN_CREATE);

                if (method_exists($db, 'busyTimeout')) {

                    // busyTimeout only exists for PHP >= 5.3.3

                    $db->busyTimeout(1000);

                }

            } elseif (class_exists('PDO') && in_array('sqlite', \PDO::getAvailableDrivers(), true)) {

                $db = new \PDO($this->dsn);

            } else {

                throw new \RuntimeException('You need to enable either the SQLite3 or PDO_SQLite extension for the profiler to run properly.');

            }

            $db->exec('CREATE TABLE IF NOT EXISTS sf_profiler_data (token STRING, data STRING, ip STRING, url STRING, time INTEGER, parent STRING, created_at INTEGER)');

            $db->exec('CREATE INDEX IF NOT EXISTS data_created_at ON sf_profiler_data (created_at)');

            $db->exec('CREATE INDEX IF NOT EXISTS data_ip ON sf_profiler_data (ip)');

            $db->exec('CREATE INDEX IF NOT EXISTS data_url ON sf_profiler_data (url)');

            $db->exec('CREATE INDEX IF NOT EXISTS data_parent ON sf_profiler_data (parent)');

            $db->exec('CREATE UNIQUE INDEX IF NOT EXISTS data_token ON sf_profiler_data (token)');

            $this->db = $db;

        }

        return $this->db;

    }

    protected function exec($db, $query, array $args = array())

    {

        if ($db instanceof \SQLite3) {

            $stmt = $this->prepareStatement($db, $query);

            foreach ($args as $arg => $val) {

                $stmt->bindValue($arg, $val, is_int($val) ? \SQLITE3_INTEGER : \SQLITE3_TEXT);

            }

            $res = $stmt->execute();

            if (false === $res) {

                throw new \RuntimeException(sprintf('Error executing SQLite query "%s"', $query));

            }

            $res->finalize();

        } else {

            parent::exec($db, $query, $args);

        }

    }

    protected function fetch($db, $query, array $args = array())

    {

        $return = array();

        if ($db instanceof \SQLite3) {

            $stmt = $this->prepareStatement($db, $query, true);

            foreach ($args as $arg => $val) {

                $stmt->bindValue($arg, $val, is_int($val) ? \SQLITE3_INTEGER : \SQLITE3_TEXT);

            }

            $res = $stmt->execute();

            while ($row = $res->fetchArray(\SQLITE3_ASSOC)) {

                $return[] = $row;

            }

            $res->finalize();

            $stmt->close();

        } else {

            $return = parent::fetch($db, $query, $args);

        }

        return $return;

    }

    /**

     * {@inheritdoc}

     */

    protected function buildCriteria($ip, $url, $limit)

    {

        $criteria = array();

        $args = array();

        if ($ip = preg_replace('/[^\d\.]/', '', $ip)) {

            $criteria[] = 'ip LIKE :ip';

            $args[':ip'] = '%'.$ip.'%';

        }

        if ($url) {

            $criteria[] = 'url LIKE :url ESCAPE "\"';

            $args[':url'] = '%'.addcslashes($url, '%_\\').'%';

        }

        return array($criteria, $args);

    }

    protected function close($db)

    {

        if ($db instanceof \SQLite3) {

            $db->close();

        }

    }

}