Use Database Prefix (#71)

* Use database prefix (if applicable)

* cs fix

* Add test for not having a prefix specified

Author: bennett-treptow

src/GeneratorManagers/MySQLGeneratorManager.php

@@ -2,7 +2,9 @@
 
 namespace LaravelMigrationGenerator\GeneratorManagers;
 
+use Illuminate\Support\Str;
 use Illuminate\Support\Facades\DB;
+use LaravelMigrationGenerator\Definitions\TableDefinition;
 use LaravelMigrationGenerator\Generators\MySQL\ViewGenerator;
 use LaravelMigrationGenerator\Generators\MySQL\TableGenerator;
 use LaravelMigrationGenerator\GeneratorManagers\Interfaces\GeneratorManagerInterface;
@@ -29,4 +31,14 @@ public function init()
             }
         }
     }
+
+    public function addTableDefinition(TableDefinition $tableDefinition): BaseGeneratorManager
+    {
+        $prefix = config('database.connections.' . DB::getDefaultConnection() . '.prefix', '');
+        if (! empty($prefix) && Str::startsWith($tableDefinition->getTableName(), $prefix)) {
+            $tableDefinition->setTableName(Str::replaceFirst($prefix, '', $tableDefinition->getTableName()));
+        }
+
+        return parent::addTableDefinition($tableDefinition);
+    }
 }

tests/Unit/GeneratorManagers/MySQLGeneratorManagerTest.php

@@ -4,6 +4,7 @@
 
 use Tests\TestCase;
 use Mockery\MockInterface;
+use Illuminate\Support\Facades\DB;
 use LaravelMigrationGenerator\Definitions\IndexDefinition;
 use LaravelMigrationGenerator\Definitions\TableDefinition;
 use LaravelMigrationGenerator\Definitions\ColumnDefinition;
@@ -52,4 +53,33 @@ public function test_can_sort_tables()
         $this->assertCount(4, $sorted);
         $this->assertStringContainsString('$table->dropForeign', $sorted[3]->formatter()->stubTableDown());
     }
+
+    public function test_can_remove_database_prefix()
+    {
+        $connection = DB::getDefaultConnection();
+        config()->set('database.connections.' . $connection . '.prefix', 'wp_');
+
+        $mocked = $this->partialMock(MySQLGeneratorManager::class, function (MockInterface $mock) {
+            $mock->shouldReceive('init');
+        });
+
+        $definition = (new TableDefinition())->setTableName('wp_posts');
+        $mocked->addTableDefinition($definition);
+        $this->assertEquals('posts', $definition->getTableName());
+
+        $definition = (new TableDefinition())->setTableName('posts');
+        $mocked->addTableDefinition($definition);
+        $this->assertEquals('posts', $definition->getTableName());
+
+
+        config()->set('database.connections.' . $connection . '.prefix', '');
+
+        $definition = (new TableDefinition())->setTableName('wp_posts');
+        $mocked->addTableDefinition($definition);
+        $this->assertEquals('wp_posts', $definition->getTableName());
+
+        $definition = (new TableDefinition())->setTableName('posts');
+        $mocked->addTableDefinition($definition);
+        $this->assertEquals('posts', $definition->getTableName());
+    }
 }