SqliteRowCounter.php

<?php

declare(strict_types=1);

namespace PhpDb\Sqlite\Pdo\Feature;

use Closure;
use PhpDb\Adapter\Driver\Feature\AbstractFeature;
use PhpDb\Adapter\Driver\Pdo;
use PhpDb\Adapter\Driver\Pdo\Statement;

use function str_contains;
use function strtolower;

/**
 * SqliteRowCounter
 */
class SqliteRowCounter extends AbstractFeature
{
    public function getCountForStatement(Pdo\Statement $statement): int
    {
        $countStmt = clone $statement;
        $sql       = $statement->getSql();
        if (empty($sql) || ! str_contains(strtolower($sql), 'select')) {
            return 0;
        }
        $countSql = 'SELECT COUNT(*) as "count" FROM (' . $sql . ')';
        $countStmt->prepare($countSql);
        $result   = $countStmt->execute();
        $countRow = $result->getResource()->fetch(\PDO::FETCH_ASSOC);
        unset($statement, $result);

        return (int) $countRow['count'];
    }

    public function getCountForSql(string $sql): int
    {
        if (empty($sql) || ! str_contains(strtolower($sql), 'select')) {
            return 0;
        }
        $countSql = 'SELECT COUNT(*) as count FROM (' . $sql . ')';
        /** @var \PDO $pdo */
        $pdo      = $this->driver->getConnection()->getResource();
        $result   = $pdo->query($countSql);
        $countRow = $result->fetch(\PDO::FETCH_ASSOC);

        return (int) $countRow['count'];
    }

    public function getRowCountClosure(Statement|string|null $context): Closure
    {
        return function () use ($context): int {
            return $context instanceof Pdo\Statement
                ? $this->getCountForStatement($context)
                : $this->getCountForSql($context);
        };
    }
}