PHPPackage\PDOWrapper\Database

Complexity

Total Complexity

14

Size/Duplication

Total Lines	125
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	databases()	0	10	2
A	create()	0	11	2
A	attribute()	0	8	2
A	name()	0	8	2
A	info()	0	12	3
A	tables()	0	10	2

<?php
declare(strict_types=1);

/*
 +-----------------------------------------------------------------------------+
 | PHPPackage - PDO Wrapper
 +-----------------------------------------------------------------------------+
 | Copyright (c)2018 (http://github.com/phppackage/pdo-wrapper)
 +-----------------------------------------------------------------------------+
 | This source file is subject to MIT License
 | that is bundled with this package in the file LICENSE.
 |
 | If you did not receive a copy of the license and are unable to
 | obtain it through the world-wide-web, please send an email
 | to lawrence@cherone.co.uk so we can send you a copy immediately.
 +-----------------------------------------------------------------------------+
 | Authors:
 |   Lawrence Cherone <lawrence@cherone.co.uk>
 +-----------------------------------------------------------------------------+
 */

namespace PHPPackage\PDOWrapper;

class Database
{
    /**
     * @var \PHPPackage\PDOWrapper\PDO
     */
    private $pdo;
    
    /**
     * @var array PDO attribute keys.
     */
    private $attributes = array(
        'AUTOCOMMIT', 'ERRMODE', 'CASE', 'CLIENT_VERSION', 'CONNECTION_STATUS',
        'ORACLE_NULLS', 'PERSISTENT', 'PREFETCH', 'SERVER_INFO', 'SERVER_VERSION',
        'TIMEOUT', 'DRIVER_NAME'
    );

    /**
     * @param \PHPPackage\PDOWrapper\PDO $pdo
     */
    public function __construct(PDO $pdo)
    {
        $this->pdo = $pdo;
    }
    
    /**
     * Enumarate PDO attributes
     *
     * @param string $key Pick out a attribute by key
     * @return mixed
     */
    public function attribute(string $key = null) 
    {
        $key = 'PDO::ATTR_'.strtoupper($key);

        try {
            return $this->pdo->getAttribute(constant($key));
        } catch (\PDOException $e) {
            return;
        }
    }
    
    /**
     * Enumarate PDO attributes
     *
     * @param string $key Pick out a attribute by key
     * @return mixed
     */
    public function info(string $key = null)
    {
        if (!is_null($key)) {
            return $this->attribute($key);
        }
        
        $return = [];
        foreach ($this->attributes as $value) {
            $return['PDO::ATTR_'.$value] = $this->attribute($value);
        }
        
        return $return;
    }

    /**
     * Get database name from dsn
     *
     * @throws RuntimeException
     * @return string
     */
    public function name(string $dsn): string
    {
        // match database from dsn & set working vars
        if (!preg_match('/dbname=(\w+);/', $dsn, $results)) {
            throw new \RuntimeException('Could not match database name from dsn');
        }

        return $results[1];
    }

    /**
     * Create database
     *
     * @return bool
     */
    public function create(string $name, $username, $password): bool
    {
        if (!in_array($name, $this->databases())) {
            return (bool) $this->pdo->exec("
                CREATE DATABASE `$name`;
                CREATE USER '{$username}'@'%' IDENTIFIED BY '{$password}';
                GRANT ALL ON `$name`.* TO '{$username}'@'%';
                FLUSH PRIVILEGES;
            ");
        }
        return false;
    }

    /**
     * Returns an array of databases
     * @return mixed
     */
    public function databases(): array
    {
        $stmt = $this->pdo->query("SHOW DATABASES");

        $result = [];
        while ($row = $stmt->fetchColumn(0)) {
            $result[] = $row;
        }

        return $result;
    }

    /**
     * Returns an array of tables
     * @return array
     */
    public function tables(): array
    {
        $stmt = $this->pdo->query("SHOW TABLES");

        $result = [];
        while ($row = $stmt->fetchColumn(0)) {
            $result[] = $row;
        }

        return $result;
    }
}