dimtrovich / php-db-dumper

src/Importer.php

<?php

/**
 * This file is part of dimtrovich/db-dumper".
 *
 * (c) 2024 Dimitri Sitchet Tomkeu <devcode.dst@gmail.com>
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

namespace Dimtrovich\DbDumper;

use Dimtrovich\DbDumper\Compressor\Factory as CompressorFactory;
use Dimtrovich\DbDumper\Exceptions\Exception;
use PDOException;

/**
 * @method void onTableCreate(callable(string $tableName) $callback)
 * @method void onTableInsert(callable(string $tableName, int $rowCount) $callback)
 */
class Importer
{
    use Dumper;

The trait Dimtrovich\DbDumper\Dumper requires some properties which are not provided by Dimtrovich\DbDumper\Importer: $init_commands, $compress

    /**
     * Primary function, triggers restoration.
     *
     * @param string $filename Name of file to read sql dump to
     */
    public function process(string $filename): void
    {
        $extension = pathinfo($filename, PATHINFO_EXTENSION);

        switch ($extension) {
            case 'gz':
            case 'gzip':
                $this->compressor = CompressorFactory::create(Option::COMPRESSION_GZIP);
                break;

            case 'bz2':
            case 'bzip2':
                $this->compressor = CompressorFactory::create(Option::COMPRESSION_BZIP2);
                break;

            case 'sql':
                $this->compressor = CompressorFactory::create(Option::COMPRESSION_NONE);
                break;

            default:
                throw Exception::unavailableDriverForcompression($extension);

It seems like $extension can also be of type array; however, parameter $extension of Dimtrovich\DbDumper\Exceptions\Exception::unavailableDriverForcompression() does only seem to accept string, maybe add an additional type check?

        }

        if (null === $filename = $this->getFile($filename)) {
            throw Exception::failledToRead(func_get_arg(0));
        }

        if ($this->option->disable_foreign_keys_check && '' !== $disableForeignKeysCheck = $this->adapter->startDisableForeignKeysCheck()) {
            $this->pdo->exec($disableForeignKeysCheck);
        }

        /**
         * Read backup file line by line
         */
        $handle = fopen($filename, 'rb');

        if (! $handle) {

$handle is of type resource, thus it always evaluated to false.

            throw Exception::failledToRead($filename);
        }

        $buffer = '';

        try {
            while (! feof($handle)) {
                $line = fgets($handle);

                if (substr($line, 0, 2) === '--' || ! $line) {
                    continue; // skip comments
                }

                $buffer .= $line;

                // if it has a semicolon at the end, it's the end of the query
                if (';' === substr(rtrim($line), -1, 1)) {
                    if (false !== $affectedRows = $this->pdo->exec($buffer)) {
                        if (preg_match('/^CREATE TABLE `([^`]+)`/i', $buffer, $tableName)) {
                            $this->event->emit('table.create', $tableName[1]);
                        }
                        if (preg_match('/^INSERT INTO `([^`]+)`/i', $buffer, $tableName)) {
                            $this->event->emit('table.insert', $tableName[1], $affectedRows);
                        }
                    }

                    $buffer = '';
                }
            }
        } catch (PDOException $e) {
            throw Exception::pdoException($e->getMessage(), $buffer);
        } finally {
            fclose($handle);
            unlink($filename);
        }

        if ($this->option->disable_foreign_keys_check && '' !== $enableForeignKeysCheck = $this->adapter->endDisableForeignKeysCheck()) {
            $this->pdo->exec($enableForeignKeysCheck);
        }
    }

    /**
     * Return unzipped file
     */
    private function getFile(string $source): ?string
    {
        $pathinfo = pathinfo($source);

        $dest = $pathinfo['dirname'] . '/' . date('Ymd_His', time()) . '_' . $pathinfo['filename'];

        // Remove $dest file if exists
        if (file_exists($dest) && ! unlink($dest)) {
            return null;
        }

        // Open gzipped and destination files in binary mode
        $this->compressor->open($source, 'rb');
        if (! $dstFile = fopen($dest, 'wb')) {
            return null;
        }

        if (! fwrite($dstFile, $this->compressor->read())) {
            return null;
        }

        fclose($dstFile);
        $this->compressor->close();

        return $dest;
    }
}