Added isDatabaseVersioned issue #27

byjg · May 6, 2020 · 3d74841 · 3d74841
1 parent 2f614ed
commit 3d74841
Show file tree

Hide file tree

Showing 5 changed files with 46 additions and 0 deletions.
diff --git a/src/Database/AbstractDatabase.php b/src/Database/AbstractDatabase.php
@@ -121,4 +121,19 @@ public function updateVersionTable()
         $this->createVersion();
         $this->setVersion($currentVersion, Migration::VERSION_STATUS_UNKNOWN);
     }
+
+    public function isDatabaseVersioned()
+    {
+        $count = $this->getDbDriver()->getScalar(
+            'SELECT count(*) FROM information_schema.tables ' .
+            ' WHERE table_schema = [[schema]] ' .
+            '  AND table_name = [[table]] ',
+            [
+                "schema" => ltrim($this->getDbDriver()->getUri()->getPath(), "/"),
+                "table" => $this->getMigrationTable()
+            ]
+        );
+
+        return ($count !== 0);
+    }
 }
diff --git a/src/Database/DatabaseInterface.php b/src/Database/DatabaseInterface.php
@@ -27,6 +27,8 @@ public function setVersion($version, $status);
 
     public function createVersion();
 
+    public function isDatabaseVersioned();
+
     public function getDbDriver();
 
     public function getMigrationTable();

diff --git a/src/Database/SqliteDatabase.php b/src/Database/SqliteDatabase.php
@@ -53,4 +53,16 @@ protected function executeSqlInternal($sql)
     {
         $this->getDbDriver()->execute($sql);
     }
+
+    public function isDatabaseVersioned()
+    {
+        $count = $this->getDbDriver()->getScalar(
+            "SELECT count(*) FROM sqlite_master WHERE type='table' AND name=[[table]]",
+            [
+                "table" => $this->getMigrationTable()
+            ]
+        );
+
+        return (intval($count) !== 0);
+    }
 }
diff --git a/src/Migration.php b/src/Migration.php
@@ -101,6 +101,11 @@ public function getDbCommand()
         return $this->dbCommand;
     }
 
+    public function getMigrationTable()
+    {
+        return $this->migrationTable;
+    }
+
     /**
      * @return mixed
      * @throws \ByJG\DbMigration\Exception\DatabaseDoesNotRegistered
@@ -346,4 +351,9 @@ public function addCallbackProgress(callable $callable)
     {
         $this->callableProgress = $callable;
     }
+
+    public function isDatabaseVersioned()
+    {
+        return $this->getDbCommand()->isDatabaseVersioned();
+    }
 }
diff --git a/tests/BaseDatabase.php b/tests/BaseDatabase.php
@@ -45,6 +45,13 @@ public function testVersion0()
         $this->assertVersion0();
     }
 
+    public function testIsDatabaseVersioned()
+    {
+        $this->assertFalse($this->migrate->isDatabaseVersioned());
+        $this->migrate->reset();
+        $this->assertTrue($this->migrate->isDatabaseVersioned());
+    }
+
     /**
      * @throws \ByJG\DbMigration\Exception\DatabaseDoesNotRegistered
      * @throws \ByJG\DbMigration\Exception\DatabaseIsIncompleteException