Rougin\Describe\Drivers\MySQLDriver

Complexity

Total Complexity

14

Size/Duplication

Total Lines	118
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

100%

Importance

Changes	4
Bugs	1	Features	2

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	showTables()	0	13	2
C	getTable()	0	77	11

<?php

namespace Rougin\Describe\Drivers;

use PDO;
use Rougin\Describe\Column;

/**
 * MySQL Driver
 *
 * A database driver extension for MySQL.
 *
 * @package  Describe
 * @category Drivers
 * @author   Rougin Royce Gutib <rougingutib@gmail.com>
 */
class MySQLDriver implements DriverInterface
{
    protected $database;
    protected $columns = [];
    protected $pdo;

    /**
     * @param PDO    $pdo
     * @param string $database
     */
    public function __construct(PDO $pdo, $database)
    {
        $this->database = $database;
        $this->pdo = $pdo;
    }

    /**
     * Returns the result.
     *
     * @return array
     */
    public function getTable($table)
    {
        $this->columns = [];

        $information = $this->pdo->prepare('DESCRIBE ' . $table);
        $information->execute();
        $information->setFetchMode(PDO::FETCH_OBJ);

        if ($stripped = strpos($table, '.')) {
            $table = substr($table, $stripped + 1);
        }

        while ($row = $information->fetch()) {
            preg_match('/(.*?)\((.*?)\)/', $row->Type, $match);

            $column = new Column;

            if ($row->Extra == 'auto_increment') {
                $column->setAutoIncrement(true);
            }

            if ($row->Null == 'YES') {
                $column->setNull(true);
            }

            switch ($row->Key) {
                case 'PRI':
                    $column->setPrimary(true);

                    break;
                
                case 'MUL':
                    $column->setForeign(true);

                    break;

                case 'UNI':
                    $column->setUnique(true);

                    break;
            }

            $column->setDataType($row->Type);
            $column->setDefaultValue($row->Default);
            $column->setField($row->Field);

            if (isset($match[1])) {
                $column->setDataType($match[1]);
                $column->setLength($match[2]);
            }

            $query = 'SELECT COLUMN_NAME as "column",' .
                'REFERENCED_COLUMN_NAME as "referenced_column",' .
                'CONCAT(' .
                    'REFERENCED_TABLE_SCHEMA, ".",' .
                    'REFERENCED_TABLE_NAME' .
                ') as "referenced_table"' .
                'FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE ' .
                'WHERE CONSTRAINT_SCHEMA = "' . $this->database . '" ' .
                'AND TABLE_NAME = "' . $table . '";';

            $foreignTable = $this->pdo->prepare($query);
            $foreignTable->execute();
            $foreignTable->setFetchMode(PDO::FETCH_OBJ);

            while ($foreignRow = $foreignTable->fetch()) {
                if ($foreignRow->column == $row->Field) {
                    $column->setReferencedField($foreignRow->referenced_column);
                    $column->setReferencedTable($foreignRow->referenced_table);
                }
            }

            array_push($this->columns, $column);
        }

        return $this->columns;
    }

    /**
     * Shows the list of tables.
     *
     * @return array
     */
    public function showTables()
    {
        $tables = [];

        $information = $this->pdo->prepare('SHOW TABLES');
        $information->execute();

        while ($row = $information->fetch()) {
            array_push($tables, $row[0]);
        }

        return $tables;
    }
}