shlink/data/migrations/Version20201023090929.php

<?php

declare(strict_types=1);

namespace ShlinkMigrations;

use Doctrine\DBAL\Schema\Schema;
use Doctrine\DBAL\Types\Types;
use Doctrine\Migrations\AbstractMigration;

final class Version20201023090929 extends AbstractMigration
{
    private const IMPORT_SOURCE_COLUMN = 'import_source';

    public function up(Schema $schema): void
    {
        $shortUrls = $schema->getTable('short_urls');
        $this->skipIf($shortUrls->hasColumn(self::IMPORT_SOURCE_COLUMN));

        $shortUrls->addColumn(self::IMPORT_SOURCE_COLUMN, Types::STRING, [
            'length' => 255,
            'notnull' => false,
        ]);
        $shortUrls->addColumn('import_original_short_code', Types::STRING, [
            'length' => 255,
            'notnull' => false,
        ]);

        $shortUrls->addUniqueIndex(
            [self::IMPORT_SOURCE_COLUMN, 'import_original_short_code', 'domain_id'],
            'unique_imports',
        );
    }

    public function down(Schema $schema): void
    {
        $shortUrls = $schema->getTable('short_urls');
        $this->skipIf(! $shortUrls->hasColumn(self::IMPORT_SOURCE_COLUMN));

        $shortUrls->dropColumn(self::IMPORT_SOURCE_COLUMN);
        $shortUrls->dropColumn('import_original_short_code');
        $shortUrls->dropIndex('unique_imports');
    }

    public function isTransactional(): bool
    {
        return $this->connection->getDatabasePlatform()->getName() !== 'mysql';
    }
}