formatting

Author: taylorotwell

src/Illuminate/Database/Console/Migrations/FreshCommand.php

@@ -35,9 +35,15 @@ public function handle()
             return;
         }
 
-        $this->dropAllTables(
-            $database = $this->input->getOption('database')
-        );
+        $database = $this->input->getOption('database');
+
+        if ($this->option('drop-views')) {
+            $this->dropAllViews($database);
+
+            $this->info('Dropped all views successfully.');
+        }
+
+        $this->dropAllTables($database);
 
         $this->info('Dropped all tables successfully.');
 
@@ -65,6 +71,19 @@ protected function dropAllTables($database)
                     ->dropAllTables();
     }
 
+    /**
+     * Drop all of the database views.
+     *
+     * @param  string  $database
+     * @return void
+     */
+    protected function dropAllViews($database)
+    {
+        $this->laravel['db']->connection($database)
+                    ->getSchemaBuilder()
+                    ->dropAllViews();
+    }
+
     /**
      * Determine if the developer has requested database seeding.
      *
@@ -100,6 +119,8 @@ protected function getOptions()
         return [
             ['database', null, InputOption::VALUE_OPTIONAL, 'The database connection to use.'],
 
+            ['drop-views', null, InputOption::VALUE_NONE, 'Drop all tables and views.'],
+
             ['force', null, InputOption::VALUE_NONE, 'Force the operation to run when in production.'],
 
             ['path', null, InputOption::VALUE_OPTIONAL, 'The path to the migrations files to be executed.'],

src/Illuminate/Database/Schema/Builder.php

@@ -203,6 +203,18 @@ public function dropAllTables()
         throw new LogicException('This database driver does not support dropping all tables.');
     }
 
+    /**
+     * Drop all views from the database.
+     *
+     * @return void
+     *
+     * @throws \LogicException
+     */
+    public function dropAllViews()
+    {
+        throw new LogicException('This database driver does not support dropping all views.');
+    }
+
     /**
      * Rename a table on the schema.
      *

src/Illuminate/Database/Schema/Grammars/MySqlGrammar.php

@@ -369,6 +369,17 @@ public function compileDropAllTables($tables)
         return 'drop table '.implode(',', $this->wrapArray($tables));
     }
 
+    /**
+     * Compile the SQL needed to drop all views.
+     *
+     * @param  array  $views
+     * @return string
+     */
+    public function compileDropAllViews($views)
+    {
+        return 'drop view '.implode(',', $this->wrapArray($views));
+    }
+
     /**
      * Compile the SQL needed to retrieve all table names.
      *
@@ -379,6 +390,16 @@ public function compileGetAllTables()
         return 'SHOW FULL TABLES WHERE table_type = \'BASE TABLE\'';
     }
 
+    /**
+     * Compile the SQL needed to retrieve all view names.
+     *
+     * @return string
+     */
+    public function compileGetAllViews()
+    {
+        return 'SHOW FULL TABLES WHERE table_type = \'VIEW\'';
+    }
+
     /**
      * Compile the command to enable foreign key constraints.
      *

src/Illuminate/Database/Schema/Grammars/PostgresGrammar.php

@@ -205,6 +205,17 @@ public function compileDropAllTables($tables)
         return 'drop table "'.implode('","', $tables).'" cascade';
     }
 
+    /**
+     * Compile the SQL needed to drop all views.
+     *
+     * @param  string  $views
+     * @return string
+     */
+    public function compileDropAllViews($views)
+    {
+        return 'drop view "'.implode('","', $views).'" cascade';
+    }
+
     /**
      * Compile the SQL needed to retrieve all table names.
      *
@@ -216,6 +227,17 @@ public function compileGetAllTables($schema)
         return "select tablename from pg_catalog.pg_tables where schemaname = '{$schema}'";
     }
 
+    /**
+     * Compile the SQL needed to retrieve all view names.
+     *
+     * @param  string  $schema
+     * @return string
+     */
+    public function compileGetAllViews($schema)
+    {
+        return "select viewname from pg_catalog.pg_views where schemaname = '{$schema}'";
+    }
+
     /**
      * Compile a drop column command.
      *

src/Illuminate/Database/Schema/Grammars/SQLiteGrammar.php

@@ -231,6 +231,16 @@ public function compileDropAllTables()
         return "delete from sqlite_master where type in ('table', 'index', 'trigger')";
     }
 
+    /**
+     * Compile the SQL needed to drop all views.
+     *
+     * @return string
+     */
+    public function compileDropAllViews()
+    {
+        return "delete from sqlite_master where type in ('view')";
+    }
+
     /**
      * Compile a drop column command.
      *

src/Illuminate/Database/Schema/MySqlBuilder.php

@@ -64,6 +64,30 @@ public function dropAllTables()
         $this->enableForeignKeyConstraints();
     }
 
+    /**
+     * Drop all views from the database.
+     *
+     * @return void
+     */
+    public function dropAllViews()
+    {
+        $views = [];
+
+        foreach ($this->getAllViews() as $row) {
+            $row = (array) $row;
+
+            $views[] = reset($row);
+        }
+
+        if (empty($views)) {
+            return;
+        }
+
+        $this->connection->statement(
+            $this->grammar->compileDropAllViews($views)
+        );
+    }
+
     /**
      * Get all of the table names for the database.
      *
@@ -75,4 +99,16 @@ protected function getAllTables()
             $this->grammar->compileGetAllTables()
         );
     }
+
+    /**
+     * Get all of the view names for the database.
+     *
+     * @return array
+     */
+    protected function getAllViews()
+    {
+        return $this->connection->select(
+            $this->grammar->compileGetAllViews()
+        );
+    }
 }

src/Illuminate/Database/Schema/PostgresBuilder.php

@@ -51,6 +51,30 @@ public function dropAllTables()
         );
     }
 
+    /**
+     * Drop all views from the database.
+     *
+     * @return void
+     */
+    public function dropAllViews()
+    {
+        $views = [];
+
+        foreach ($this->getAllViews() as $row) {
+            $row = (array) $row;
+
+            $views[] = reset($row);
+        }
+
+        if (empty($views)) {
+            return;
+        }
+
+        $this->connection->statement(
+            $this->grammar->compileDropAllViews($views)
+        );
+    }
+
     /**
      * Get all of the table names for the database.
      *
@@ -63,6 +87,18 @@ protected function getAllTables()
         );
     }
 
+    /**
+     * Get all of the view names for the database.
+     *
+     * @return array
+     */
+    protected function getAllViews()
+    {
+        return $this->connection->select(
+            $this->grammar->compileGetAllViews($this->connection->getConfig('schema'))
+        );
+    }
+
     /**
      * Get the column listing for a given table.
      *

src/Illuminate/Database/Schema/SQLiteBuilder.php

@@ -22,6 +22,20 @@ public function dropAllTables()
         $this->connection->select($this->grammar->compileDisableWriteableSchema());
     }
 
+    /**
+     * Drop all views from the database.
+     *
+     * @return void
+     */
+    public function dropAllViews()
+    {
+        $this->connection->select($this->grammar->compileEnableWriteableSchema());
+
+        $this->connection->select($this->grammar->compileDropAllViews());
+
+        $this->connection->select($this->grammar->compileDisableWriteableSchema());
+    }
+
     /**
      * Empty the database file.
      *

src/Illuminate/Foundation/Testing/RefreshDatabase.php

@@ -50,7 +50,9 @@ protected function refreshInMemoryDatabase()
     protected function refreshTestDatabase()
     {
         if (! RefreshDatabaseState::$migrated) {
-            $this->artisan('migrate:fresh');
+            $this->artisan('migrate:fresh', $this->shouldDropViews() ? [
+                '--drop-views' => true,
+            ] : []);
 
             $this->app[Kernel::class]->setArtisan(null);
 
@@ -101,4 +103,15 @@ protected function connectionsToTransact()
         return property_exists($this, 'connectionsToTransact')
                             ? $this->connectionsToTransact : [null];
     }
+
+    /**
+     * Determine if views should be dropped when refreshing the database.
+     *
+     * @return bool
+     */
+    protected function shouldDropViews()
+    {
+        return property_exists($this, 'dropViews')
+                            ? $this->dropViews : false;
+    }
 }

tests/Database/DatabaseMySqlSchemaGrammarTest.php

@@ -941,6 +941,13 @@ public function testDropAllTables()
         $this->assertEquals('drop table `alpha`,`beta`,`gamma`', $statement);
     }
 
+    public function testDropAllViews()
+    {
+        $statement = $this->getGrammar()->compileDropAllViews(['alpha', 'beta', 'gamma']);
+
+        $this->assertEquals('drop view `alpha`,`beta`,`gamma`', $statement);
+    }
+
     public function testGrammarsAreMacroable()
     {
         // compileReplace macro.

tests/Database/DatabasePostgresSchemaGrammarTest.php

@@ -790,6 +790,13 @@ public function testDropAllTablesEscapesTableNames()
         $this->assertEquals('drop table "alpha","beta","gamma" cascade', $statement);
     }
 
+    public function testDropAllViewsEscapesTableNames()
+    {
+        $statement = $this->getGrammar()->compileDropAllViews(['alpha', 'beta', 'gamma']);
+
+        $this->assertEquals('drop view "alpha","beta","gamma" cascade', $statement);
+    }
+
     protected function getConnection()
     {
         return m::mock('Illuminate\Database\Connection');