1:  2:  3:  4:  5:  6:  7:  8:  9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 44: 45: 46: 47: 48: 49: 

<?php

    namespace Inlm\SchemaGenerator\Extractors;

    use CzProject\SqlSchema;
    use Inlm\SchemaGenerator\Bridges;
    use Inlm\SchemaGenerator\IExtractor;
    use Inlm\SchemaGenerator\Utils\Generator;
    use Inlm\SchemaGenerator\Utils\DataTypeParser;
    use Nette;


    class DibiExtractor implements IExtractor
    {
        /** @var \DibiConnection|\Dibi\Connection */
        private $connection;

        /** @var string[] */
        private $ignoredTables;


        /**
         * @param  \Dibi\Connection|\DibiConnection
         * @param  string[]
         */
        public function __construct($connection, array $ignoredTables = array())
        {
            Bridges\Dibi::validateConnection($connection);
            $this->connection = $connection;
            $this->ignoredTables = $ignoredTables;
        }


        /**
         * @return SqlSchema\Schema
         */
        public function generateSchema(array $options = array(), array $customTypes = array(), $databaseType = NULL)
        {
            $dibiDriver = $this->connection->getDriver();

            if (Bridges\Dibi::isMysqlDriver($dibiDriver)) {
                $generator = new Bridges\DibiMysql($this->connection);
                return $generator->generateSchema($this->ignoredTables);
            }

            throw new \Inlm\SchemaGenerator\UnsupportedException('Driver ' . get_class($dibiDriver) . ' is not supported.');
        }
    }