<?php

/*

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 *

 * This software consists of voluntary contributions made by many individuals

 * and is licensed under the LGPL. For more information, see

 * <http://www.doctrine-project.org>.

*/

namespace Doctrine\DBAL\Migrations\Configuration;

use Doctrine\DBAL\Connection,

    Doctrine\DBAL\Migrations\MigrationException,

    Doctrine\DBAL\Migrations\Version,

    Doctrine\DBAL\Migrations\OutputWriter,

    Doctrine\DBAL\Schema\Table,

    Doctrine\DBAL\Schema\Column,

    Doctrine\DBAL\Types\Type;

/**

 * Default Migration Configurtion object used for configuring an instance of

 * the Migration class. Set the connection, version table name, register migration

 * classes/versions, etc.

 *

 * @license     http://www.opensource.org/licenses/lgpl-license.php LGPL

 * @link        www.doctrine-project.org

 * @since       2.0

 * @author      Jonathan H. Wage <jonwage@gmail.com>

 */

class Configuration

{

    /**

     * Name of this set of migrations

     *

     * @var string

     */

    private $name;

    /**

     * Flag for whether or not the migration table has been created

     *

     * @var bool

     */

    private $migrationTableCreated = false;

    /**

     * Connection instance to use for migrations

     *

     * @var Connection

     */

    private $connection;

    /**

     * OutputWriter instance for writing output during migrations

     * 

     * @var OutputWriter

     */

    private $outputWriter;

    /**

     * The migration table name to track versions in

     *

     * @var string

     */

    private $migrationsTableName = 'doctrine_migration_versions';

    /**

     * The path to a directory where new migration classes will be written

     * 

     * @var string

     */

    private $migrationsDirectory;

    /**

     * Namespace the migration classes live in

     *

     * @var string

     */

    private $migrationsNamespace;

    /**

     * Array of the registered migrations

     *

     * @var array

     */

    private $migrations = array();

    /**

     * Construct a migration configuration object.

     *

     * @param Connection $connection      A Connection instance

     * @param OutputWriter $outputWriter  A OutputWriter instance

     */

    public function __construct(Connection $connection, OutputWriter $outputWriter = null)

    {

        $this->connection = $connection;

        if ($outputWriter === null) {

            $outputWriter = new OutputWriter();

        }

        $this->outputWriter = $outputWriter;

    }

    /**

     * Validation that this instance has all the required properties configured

     *

     * @return void

     * @throws MigrationException

     */

    public function validate()

    {

        if ( ! $this->migrationsNamespace) {

            throw MigrationException::migrationsNamespaceRequired();

        }

        if ( ! $this->migrationsDirectory) {

            throw MigrationException::migrationsDirectoryRequired();

        }

    }

    /**

     * Set the name of this set of migrations

     *

     * @param string $name The name of this set of migrations

     */

    public function setName($name)

    {

        $this->name = $name;

    }

    /**

     * Returns the name of this set of migrations

     *

     * @return string $name The name of this set of migrations

     */

    public function getName()

    {

        return $this->name;

    }

    /**

     * Returns the OutputWriter instance

     *

     * @return OutputWriter $outputWriter  The OutputWriter instance

     */

    public function getOutputWriter()

    {

        return $this->outputWriter;

    }

    /**

     * Returns a timestamp version as a formatted date

     *

     * @param string $version 

     * @return string $formattedVersion The formatted version

     */

    public function formatVersion($version)

    {

        return sprintf('%s-%s-%s %s:%s:%s',

            substr($version, 0, 4),

            substr($version, 4, 2),

            substr($version, 6, 2),

            substr($version, 8, 2),

            substr($version, 10, 2),

            substr($version, 12, 2)

        );

    }

    /**

     * Returns the Connection instance

     *

     * @return Connection $connection  The Connection instance

     */

    public function getConnection()

    {

        return $this->connection;

    }

    /**

     * Set the migration table name

     *

     * @param string $tableName The migration table name

     */

    public function setMigrationsTableName($tableName)

    {

        $this->migrationsTableName = $tableName;

    }

    /**

     * Returns the migration table name

     *

     * @return string $migrationsTableName The migration table name

     */

    public function getMigrationsTableName()

    {

        return $this->migrationsTableName;

    }

    /**

     * Set the new migrations directory where new migration classes are generated

     *

     * @param string $migrationsDirectory The new migrations directory 

     */

    public function setMigrationsDirectory($migrationsDirectory)

    {

        $this->migrationsDirectory = $migrationsDirectory;

    }

    /**

     * Returns the new migrations directory where new migration classes are generated

     *

     * @return string $migrationsDirectory The new migrations directory

     */

    public function getMigrationsDirectory()

    {

        return $this->migrationsDirectory;

    }

    /**

     * Set the migrations namespace

     *

     * @param string $migrationsNamespace The migrations namespace

     */

    public function setMigrationsNamespace($migrationsNamespace)

    {

        $this->migrationsNamespace = $migrationsNamespace;

    }

    /**

     * Returns the migrations namespace

     *

     * @return string $migrationsNamespace The migrations namespace

     */

    public function getMigrationsNamespace()

    {

        return $this->migrationsNamespace;

    }

    /**

     * Register migrations from a given directory. Recursively finds all files

     * with the pattern VersionYYYYMMDDHHMMSS.php as the filename and registers

     * them as migrations.

     *

     * @param string $path  The root directory to where some migration classes live.

     * @return $migrations  The array of migrations registered.

     */

    public function registerMigrationsFromDirectory($path)

    {

        $path = realpath($path);

        $path = rtrim($path, '/');

        $files = glob($path . '/Version*.php');

        $versions = array();

        if ($files) {

            foreach ($files as $file) {

                require_once($file);

                $info = pathinfo($file);

                $version = substr($info['filename'], 7);

                $class = $this->migrationsNamespace . '\\' . $info['filename'];

                $versions[] = $this->registerMigration($version, $class);

            }

        }

        return $versions;

    }

    /**

     * Register a single migration version to be executed by a AbstractMigration

     * class.

     *

     * @param string $version  The version of the migration in the format YYYYMMDDHHMMSS.

     * @param string $class    The migration class to execute for the version.

     */

    public function registerMigration($version, $class)

    {

        $version = (string) $version;

        $class = (string) $class;

        if (isset($this->migrations[$version])) {

            throw MigrationException::duplicateMigrationVersion($version, get_class($this->migrations[$version]));

        }

        $version = new Version($this, $version, $class);

        $this->migrations[$version->getVersion()] = $version;

        ksort($this->migrations);

        return $version;

    }

    /**

     * Register an array of migrations. Each key of the array is the version and

     * the value is the migration class name.

     *

     *

     * @param array $migrations

     * @return void

     */

    public function registerMigrations(array $migrations)

    {

        $versions = array();

        foreach ($migrations as $version => $class) {

            $versions[] = $this->registerMigration($version, $class);

        }

        return $versions;

    }

    /**

     * Get the array of registered migration versions.

     *

     * @return array $migrations

     */

    public function getMigrations()

    {

        return $this->migrations;

    }

    /**

     * Returns the Version instance for a given version in the format YYYYMMDDHHMMSS.

     *

     * @param string $version   The version string in the format YYYYMMDDHHMMSS.

     * @return Version $version

     * @throws MigrationException $exception Throws exception if migration version does not exist.

     */

    public function getVersion($version)

    {

        if ( ! isset($this->migrations[$version])) {

            throw MigrationException::unknownMigrationVersion($version);

        }

        return $this->migrations[$version];

    }

    /**

     * Check if a version exists.

     *

     * @param string $version

     * @return bool $exists

     */

    public function hasVersion($version)

    {

        return isset($this->migrations[$version]) ? true : false;

    }

    /**

     * Check if a version has been migrated or not yet

     *

     * @param Version $version

     * @return bool $migrated

     */

    public function hasVersionMigrated(Version $version)

    {

        $this->createMigrationTable();

        $version = $this->connection->fetchColumn("SELECT version FROM " . $this->migrationsTableName . " WHERE version = ?", array($version->getVersion()));

        return $version !== false ? true : false;

    }

    /**

     * Returns all migrated versions from the versions table, in an array.

     *

     * @return array $migrated

     */

    public function getMigratedVersions()

    {

        $this->createMigrationTable();

        $ret = $this->connection->fetchAll("SELECT version FROM " . $this->migrationsTableName);

        $versions = array();

        foreach ($ret as $version) {

            $versions[] = current($version);

        }

        return $versions;

    }

    /**

     * Returns the current migrated version from the versions table.

     *

     * @return bool $currentVersion

     */

    public function getCurrentVersion()

    {

        $this->createMigrationTable();

        $sql = "SELECT version FROM " . $this->migrationsTableName . " ORDER BY version DESC";

        $sql = $this->connection->getDatabasePlatform()->modifyLimitQuery($sql, 1);

        $result = $this->connection->fetchColumn($sql);

        return $result !== false ? (string) $result : '0';

    }

    /**

     * Returns the total number of executed migration versions

     *

     * @return integer $count

     */

    public function getNumberOfExecutedMigrations()

    {

        $this->createMigrationTable();

        $result = $this->connection->fetchColumn("SELECT COUNT(version) FROM " . $this->migrationsTableName);

        return $result !== false ? $result : 0;

    }

    /**

     * Returns the total number of available migration versions

     *

     * @return integer $count

     */

    public function getNumberOfAvailableMigrations()

    {

        return count($this->migrations);

    }

    /**

     * Returns the latest available migration version.

     *

     * @return string $version  The version string in the format YYYYMMDDHHMMSS.

     */

    public function getLatestVersion()

    {

        $versions = array_keys($this->migrations);

        $latest = end($versions);

        return $latest !== false ? (string) $latest : '0';

    }

    /**

     * Create the migration table to track migrations with.

     *

     * @return bool $created  Whether or not the table was created.

     */

    public function createMigrationTable()

    {

        $this->validate();

        if ($this->migrationTableCreated) {

            return false;

        }

        $schema = $this->connection->getSchemaManager()->createSchema();

        if ( ! $schema->hasTable($this->migrationsTableName)) {

            $columns = array(

                'version' => new Column('version', Type::getType('string'), array('length' => 255)),

            );

            $table = new Table($this->migrationsTableName, $columns);

            $table->setPrimaryKey(array('version'));

            $this->connection->getSchemaManager()->createTable($table);

            $this->migrationTableCreated = true;

            return true;

        }