app/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     /**
  28      * @param Schema $schema
  29      */
  30     public function up(Schema $schema)
  31     {
  32         $clientsTable = $schema->getTable($this->getTable('oauth2_clients'));
  33
  34         $this->skipIf($clientsTable->hasColumn('user_id'), 'It seems that you already played this migration.');
  35
  36         $clientsTable->addColumn('user_id', 'integer', ['notnull' => false]);
  37
  38         $clientsTable->addForeignKeyConstraint(
  39             $this->getTable('user'),
  40             ['user_id'],
  41             ['id'],
  42             ['onDelete' => 'CASCADE'],
  43             $this->constraintName
  44         );
  45     }
  46
  47     /**
  48      * @param Schema $schema
  49      */
  50     public function down(Schema $schema)
  51     {
  52         $clientsTable = $schema->getTable($this->getTable('oauth2_clients'));
  53
  54         $this->skipIf(!$clientsTable->hasColumn('user_id'), 'It seems that you already played this migration.');
  55
  56         $clientsTable->dropColumn('user_id', 'integer');
  57
  58         if ('sqlite' !== $this->connection->getDatabasePlatform()->getName()) {
  59             $clientsTable->removeForeignKey($this->constraintName);
  60         }
  61     }
  62
  63     private function getTable($tableName)
  64     {
  65         return $this->container->getParameter('database_table_prefix') . $tableName;
  66     }
  67 }