PhpCfdi\SatCatalogosPopulate\Database\Repository

Complexity

Total Complexity

15

Size/Duplication

Total Lines	85
Duplicated Lines	0 %

Test Coverage

Coverage

80.95%

Importance

Changes	1
Bugs	0	Features	0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	execute()	0	6	2
A	queryArray()	0	12	3
A	hasTable()	0	4	1
A	queryRow()	0	12	3
A	getRecordCount()	0	4	1
A	__construct()	0	8	2
A	escapeName()	0	3	1
A	queryOne()	0	5	1
A	pdo()	0	3	1

<?php

declare(strict_types=1);

namespace PhpCfdi\SatCatalogosPopulate\Database;

use PDO;
use PDOException;
use RuntimeException;

class Repository
{
    /** @var PDO */
    private $pdo;

    public function __construct(string $dbfile)
    {
        if (':memory:' !== $dbfile) {
            // TODO: validate other sources ?
            $dbfile = '//' . $dbfile;
        }
        $this->pdo = new PDO('sqlite:' . $dbfile, '', '', [
            PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
        ]);
    }

    public function pdo(): PDO
    {
        return $this->pdo;
    }

    public function hasTable(string $table): bool
    {
        $sql = 'SELECT count(*) FROM sqlite_master WHERE type = :type AND name = :table;';
        return (1 === (int) $this->queryOne($sql, ['type' => 'table', 'table' => $table]));
    }

    public function getRecordCount(string $table): int
    {
        $sql = 'SELECT count(*) FROM ' . $this->escapeName($table) . ';';
        return (int) $this->queryOne($sql);
    }

    public function execute(string $sql, array $values = []): void
    {
        $stmt = $this->pdo->prepare($sql);
        if (false === $stmt->execute($values)) {
            $errorInfo = $stmt->errorInfo();
            throw new PDOException(sprintf('[%s] %s', $errorInfo[1] ?? 'UNDEF', $errorInfo[2] ?? 'Unknown error'));
        }
    }

    public function queryArray(string $sql, array $values = []): array
    {
        $stmt = $this->pdo->prepare($sql);
        if (false === $stmt->execute($values)) {
            throw new RuntimeException("Unable to execute $sql");
        }
        $table = [];
        while (false !== $row = $stmt->fetch(PDO::FETCH_ASSOC)) {
            $table[] = $row;
        }

        return $table;
    }

    public function queryRow(string $sql, array $values = []): array
    {
        $stmt = $this->pdo->prepare($sql);
        if (false === $stmt->execute($values)) {
            throw new RuntimeException("Unable to execute $sql");
        }
        $row = $stmt->fetch(PDO::FETCH_ASSOC);
        if (is_array($row)) {
            return $row;
        }

        return [];
    }

    /**
     * @param string $sql
     * @param mixed[] $values
     * @return mixed
     */
    public function queryOne(string $sql, array $values = [])
    {
        $stmt = $this->pdo->prepare($sql);
        $stmt->execute($values);
        return $stmt->fetchColumn(0);
    }

    public function escapeName(string $name): string
    {
        return '"' . str_replace('"', '""', $name) . '"';
    }
}