add cleanup for ratelimit table

[uguu.git] / src / Classes / expireChecker.php

<?php

namespace Pomf\Uguu\Classes;
use PDO;
class expireChecker
{
    public PDO $DB;
    public string $dbType;
    public array $CONFIG;
    public string $timeUnit;

    public function checkDB(): bool | array {
        if(is_int($this->CONFIG['expireTime'])) {
            $this->timeUnit = strtoupper($this->CONFIG['expireTimeUnit']);
            if(!in_array($this->timeUnit, ['SECONDS', 'MINUTES', 'HOURS', 'DAYS', 'WEEKS', 'MONTHS', 'YEARS'])){
                $this->timeUnit = "HOURS";
            }
            $query = match ($this->dbType) {
                'pgsql' => 'SELECT id, filename FROM files WHERE date < EXTRACT(epoch from NOW() - INTERVAL \'' . $this->CONFIG['expireTime'] . ' ' . $this->timeUnit . '\')',
                default => 'SELECT id, filename FROM files WHERE date <= strftime(\'%s\', datetime(\'now\', \'-' . $this->CONFIG['expireTime'] . ' ' . $this->timeUnit . '\'));'
            };
            $q = $this->DB->prepare($query);
            $q->execute();
            $result = $q->fetchAll(PDO::FETCH_ASSOC);
            $q->closeCursor();
            $returnArray = [
                'ids' => [],
                'filenames' => []
            ];
            foreach ($result as $array){
                $returnArray['ids'][] = $array['id'];
                $returnArray['filenames'][] = $array['filename'];
            }
            return $returnArray;
        } else {
            return false;
        }
    }

    public function cleanRateLimitDB(): void {
        $query = match ($this->dbType) {
            'pgsql' => 'DELETE FROM ratelimit WHERE time < EXTRACT(epoch from NOW() - INTERVAL \'24 HOURS\')',
            default => 'DELETE FROM ratelimit WHERE time <= strftime(\'%s\', datetime(\'now\', \'-24 HOURS\'));'
        };
        $q = $this->DB->prepare($query);
        $q->execute();
        $q->closeCursor();
    }
    public function deleteFiles(array $filenames): void {
        foreach ($filenames as $filename) {
            unlink($this->CONFIG['FILES_ROOT'] . $filename);
        }
    }

    public function deleteFromDB(array $ids): void {
        foreach ($ids as $id) {
            $query = match ($this->dbType) {
                'pgsql' => 'DELETE FROM files WHERE id = (:id)',
                default => 'DELETE FROM files WHERE id = (:id)'
            };
            $q = $this->DB->prepare($query);
            $q->bindValue(':id', $id);
            $q->execute();
            $q->closeCursor();
        }
    }

    /**
     * Reads the config.json file and populates the CONFIG property with the settings
     * Also assembles the PDO DB connection and registers error handlers.
     *
     */
    public function __construct()
    {
        $this->response = new Response('json');
        if (!file_exists(__DIR__ . '/../config.json')) {
            $this->response->error(500, 'Cant read settings file.');
        }
        $this->CONFIG = json_decode(
            file_get_contents(__DIR__ . '/../config.json'),
            true,
        );
        ini_set('display_errors', 0);
        $this->dbType = $this->CONFIG['DB_MODE'];
        $this->DB = new PDO(
            $this->CONFIG['DB_MODE'] . ':' . $this->CONFIG['DB_PATH'],
            $this->CONFIG['DB_USER'],
            $this->CONFIG['DB_PASS']
        );
    }
}