DoctrineMigrations/Version20161024212538.php

   1 <?php
   2
   3 namespace Application\Migrations;
   4
   5 use Doctrine\DBAL\Migrations\AbstractMigration;
   6 use Doctrine\DBAL\Schema\Schema;
   7 use Symfony\Component\DependencyInjection\ContainerAwareInterface;
   8 use Symfony\Component\DependencyInjection\ContainerInterface;
   9
  10 /**
  11  * Added user_id column on oauth2_clients to prevent users to delete API clients from other users
  12  */
  13 class Version20161024212538 extends AbstractMigration implements ContainerAwareInterface
  14 {
  15     /**
  16      * @var ContainerInterface
  17      */
  18     private $container;
  19
  20     private $constraintName = 'IDX_user_oauth_client';
  21
  22     public function setContainer(ContainerInterface $container = null)
  23     {
  24         $this->container = $container;
  25     }
  26
  27     private function getTable($tableName)
  28     {
  29         return $this->container->getParameter('database_table_prefix').$tableName;
  30     }
  31
  32     /**
  33      * @param Schema $schema
  34      */
  35     public function up(Schema $schema)
  36     {
  37         $clientsTable = $schema->getTable($this->getTable('oauth2_clients'));
  38
  39         $this->skipIf($clientsTable->hasColumn('user_id'), 'It seems that you already played this migration.');
  40
  41         $clientsTable->addColumn('user_id', 'integer', ['notnull' => false]);
  42
  43         $clientsTable->addForeignKeyConstraint(
  44             $this->getTable('user'),
  45             ['user_id'],
  46             ['id'],
  47             ['onDelete' => 'CASCADE'],
  48             $this->constraintName
  49         );
  50     }
  51
  52     /**
  53      * @param Schema $schema
  54      */
  55     public function down(Schema $schema)
  56     {
  57         $clientsTable = $schema->getTable($this->getTable('oauth2_clients'));
  58
  59         $this->skipIf(!$clientsTable->hasColumn('user_id'), 'It seems that you already played this migration.');
  60
  61         $clientsTable->dropColumn('user_id', 'integer');
  62
  63         if ($this->connection->getDatabasePlatform()->getName() != 'sqlite') {
  64             $clientsTable->removeForeignKey($this->constraintName);
  65         }
  66     }
  67 }