Drupal\TqExtension\Utils\DatabaseManager

Complexity

Total Complexity

16

Size/Duplication

Total Lines	151
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Test Coverage

Coverage

0%

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
C	__construct()	0	34	7
A	__destruct()	0	7	1
A	exist()	0	4	1
A	drop()	0	6	2
A	create()	0	6	2
A	copy()	0	6	1
A	exec()	0	15	2

<?php
/**
 * @author Sergii Bondarenko, <sb@firstvector.org>
 */
namespace Drupal\TqExtension\Utils;

class DatabaseManager
{
    /**
     * MySQL and MySQLDump login arguments.
     *
     * @var string
     */
    private $credentials = '-u%s -p%s -h%s -P%s';
    /**
     * Name of original database.
     *
     * @var string
     */
    private $source = '';
    /**
     * Name of temporary database that will store data from original.
     *
     * @var string
     */
    private $temporary = '';
    /**
     * Name of an object where this class is called.
     *
     * @var string
     */
    private $callee;

    /**
     * @param string $connection
     *   Database connection name (key in $databases array from settings.php).
     * @param string $callee
     *   Must be the value of "self::class" of callee object.
     */
    public function __construct($connection, $callee)
    {
        if (!defined('DRUPAL_ROOT') || !function_exists('conf_path')) {
            throw new \RuntimeException('Drupal is not bootstrapped.');
        }

        if (!class_exists($callee)) {
            throw new \InvalidArgumentException(sprintf('An object of "%s" type does not exist.', $callee));
        }

        $databases = [];

        require sprintf('%s/%s/settings.php', DRUPAL_ROOT, conf_path());

        if (empty($databases[$connection])) {
            throw new \InvalidArgumentException(sprintf('The "%s" database connection does not exist.', $connection));
        }

        $db = $databases[$connection]['default'];

        foreach (['port' => 3306, 'host' => '127.0.0.1'] as $option => $default) {
            if (empty($db[$option])) {
                $db[$option] = $default;
            }
        }

        $this->callee = $callee;
        $this->source = $db['database'];
        $this->temporary = "tqextension_$this->source";
        $this->credentials = sprintf($this->credentials, $db['username'], $db['password'], $db['host'], $db['port']);

        // Drop and create temporary DB and copy source into it.
        $this->copy($this->source, $this->temporary);
    }

    /**
     * Restore original database.
     */
    public function __destruct()
    {
        // Drop and create source DB and copy temporary into it.
        $this->copy($this->temporary, $this->source);
        // Kill temporary DB.
        $this->drop($this->temporary);
    }

    /**
     * @param string $name
     *   Name of the database to check.
     *
     * @return bool
     *   Checking state.
     */
    public function exist($name)
    {
        return !empty($this->exec("mysql -e 'show databases' | grep '^$name$'"));
    }

    /**
     * @param string $name
     *   Name of the database to drop.
     */
    public function drop($name)
    {
        if ($this->exist($name)) {
            $this->exec("mysql -e '%s database $name;'", __FUNCTION__);
        }
    }

    /**
     * @param string $name
     *   Name of the database to create.
     */
    public function create($name)
    {
        if (!$this->exist($name)) {
            $this->exec("mysql -e '%s database $name;'", __FUNCTION__);
        }
    }

    /**
     * @param string $source
     *   Source DB name.
     * @param string $destination
     *   Name of the new DB.
     */
    public function copy($source, $destination)
    {
        $this->drop($destination);
        $this->create($destination);
        $this->exec("mysqldump $source | mysql $destination");
    }

    /**
     * Executes a shell command.
     *
     * @param string $command
     *   Command to execute.
     *
     * @return string
     *   Result of a shell command.
     */
    private function exec($command)
    {
        // Adding credentials after "mysql" and "mysqldump" commands.
        $command = preg_replace(
            '/(mysql(?:dump)?)/',
            "\\1 $this->credentials",
            vsprintf($command, array_slice(func_get_args(), 1))
        );

        if (method_exists($this->callee, 'debug')) {
            call_user_func([$this->callee, 'debug'], [$command]);
        }

        return trim(shell_exec($command));
    }
}