1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 

<?php

/**
 * Simple Machines Forum (SMF)
 *
 * @package SMF
 * @author Simple Machines http://www.simplemachines.org
 * @copyright 2019 Simple Machines and individual contributors
 * @license http://www.simplemachines.org/about/smf/license.php BSD
 *
 * @version 2.1 RC1
 */

if (!defined('SMF'))
    die('Hacking attempt...');

/**
 * SQLite Cache API class
 *
 * @package cacheAPI
 */
class sqlite_cache extends cache_api
{
    /**
     * @var string The path to the current $cachedir directory.
     */
    private $cachedir = null;

    /**
     * @var SQLite3
     */
    private $cacheDB = null;

    /**
     * @var int
     */
    private $cacheTime = 0;

    public function __construct()
    {
        parent::__construct();

        // Set our default cachedir.
        $this->setCachedir();
    }

    /**
     * {@inheritDoc}
     */
    public function connect()
    {
        $database = $this->cachedir . '/' . 'SQLite3Cache.db3';
        $this->cacheDB = new SQLite3($database);
        $this->cacheDB->busyTimeout(1000);
        if (filesize($database) == 0)
        {
            $this->cacheDB->exec('CREATE TABLE cache (key text unique, value blob, ttl int);');
            $this->cacheDB->exec('CREATE INDEX ttls ON cache(ttl);');
        }
        $this->cacheTime = time();
    }

    /**
     * {@inheritDoc}
     */
    public function isSupported($test = false)
    {
        $supported = class_exists("SQLite3") && is_writable($this->cachedir);

        if ($test)
            return $supported;

        return parent::isSupported() && $supported;
    }

    /**
     * {@inheritDoc}
     */
    public function getData($key, $ttl = null)
    {
        $ttl = time() - $ttl;
        $query = 'SELECT value FROM cache WHERE key = \'' . $this->cacheDB->escapeString($key) . '\' AND ttl >= ' . $ttl . ' LIMIT 1';
        $result = $this->cacheDB->query($query);

        $value = null;
        while ($res = $result->fetchArray(SQLITE3_ASSOC))
            $value = $res['value'];

        return !empty($value) ? $value : null;
    }

    /**
     * {@inheritDoc}
     */
    public function putData($key, $value, $ttl = null)
    {
        $ttl = $this->cacheTime + $ttl;
        $query = 'REPLACE INTO cache VALUES (\'' . $this->cacheDB->escapeString($key) . '\', \'' . $this->cacheDB->escapeString($value) . '\', ' . $this->cacheDB->escapeString($ttl) . ');';
        $result = $this->cacheDB->exec($query);

        return $result;
    }

    /**
     * {@inheritDoc}
     */
    public function cleanCache($type = '')
    {
        if ($type == 'expired')
            $query = 'DELETE FROM cache WHERE ttl >= ' . time() . ';';
        else
            $query = 'DELETE FROM cache;';

        $result = $this->cacheDB->exec($query);

        $query = 'VACUUM;';
        $this->cacheDB->exec($query);

        return $result;
    }

    /**
     * {@inheritDoc}
     */
    public function cacheSettings(array &$config_vars)
    {
        global $context, $txt;

        $config_vars[] = $txt['cache_sqlite_settings'];
        $config_vars[] = array('cachedir_sqlite', $txt['cachedir_sqlite'], 'file', 'text', 36, 'cache_sqlite_cachedir');

        if (!isset($context['settings_post_javascript']))
            $context['settings_post_javascript'] = '';

        $context['settings_post_javascript'] .= '
            $("#cache_accelerator").change(function (e) {
                var cache_type = e.currentTarget.value;
                $("#cachedir_sqlite").prop("disabled", cache_type != "sqlite");
            });';
    }

    /**
     * Sets the $cachedir or uses the SMF default $cachedir..
     *
     * @access public
     *
     * @param string $dir A valid path
     *
     * @return boolean If this was successful or not.
     */
    public function setCachedir($dir = null)
    {
        global $cachedir, $cachedir_sqlite;

        // If its invalid, use SMF's.
        if (is_null($dir) || !is_writable($dir))
            if (is_null($cachedir_sqlite) || !is_writable($cachedir_sqlite))
                $this->cachedir = $cachedir;
            else
                $this->cachedir = $cachedir_sqlite;
        else
            $this->cachedir = $dir;
    }

    /**
     * {@inheritDoc}
     */
    public function getVersion()
    {
        $temp = $this->cacheDB->version();
        return $temp['versionString'];
    }

    /**
     * {@inheritDoc}
     */
    public function housekeeping()
    {
        $this->cleanCache('expired');
    }
}

?>