From 623c8494badd3d78d6796072514c08c17651835c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Samuel=20=C5=A0tancl?= <samuel.stancl@gmail.com>
Date: Thu, 7 Feb 2019 14:18:53 +0100
Subject: [PATCH] SOLIDify database creation

---
 src/DatabaseCreators/MySQLDatabaseCreator.php  | 13 +++++++++++++
 src/DatabaseCreators/SQLiteDatabaseCreator.php | 15 +++++++++++++++
 src/DatabaseManager.php                        | 12 ++++++------
 src/Interfaces/DatabaseCreator.php             | 14 ++++++++++++++
 src/config/tenancy.php                         |  5 ++++-
 5 files changed, 52 insertions(+), 7 deletions(-)
 create mode 100644 src/DatabaseCreators/MySQLDatabaseCreator.php
 create mode 100644 src/DatabaseCreators/SQLiteDatabaseCreator.php
 create mode 100644 src/Interfaces/DatabaseCreator.php

diff --git a/src/DatabaseCreators/MySQLDatabaseCreator.php b/src/DatabaseCreators/MySQLDatabaseCreator.php
new file mode 100644
index 00000000..c3b72283
--- /dev/null
+++ b/src/DatabaseCreators/MySQLDatabaseCreator.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace Stancl\Tenancy\DatabaseCreators;
+
+use Stancl\Tenancy\Interfaces\DatabaseCreator;
+
+class MySQLDatabaseCreator implements DatabaseCreator
+{
+    public function createDatabase(string $name): bool
+    {
+        return DB::statement("CREATE DATABASE `$name` CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci");
+    }
+}
diff --git a/src/DatabaseCreators/SQLiteDatabaseCreator.php b/src/DatabaseCreators/SQLiteDatabaseCreator.php
new file mode 100644
index 00000000..78ad2521
--- /dev/null
+++ b/src/DatabaseCreators/SQLiteDatabaseCreator.php
@@ -0,0 +1,15 @@
+<?php
+
+namespace Stancl\Tenancy\DatabaseCreators;
+
+use Stancl\Tenancy\Interfaces\DatabaseCreator;
+
+class SQLiteDatabaseCreator implements DatabaseCreator
+{
+    public function createDatabase(string $name): bool
+    {
+        fclose(fopen(database_path($name), 'w'));
+
+        return true;
+    }
+}
diff --git a/src/DatabaseManager.php b/src/DatabaseManager.php
index 745efddf..05a8039b 100644
--- a/src/DatabaseManager.php
+++ b/src/DatabaseManager.php
@@ -34,14 +34,14 @@ class DatabaseManager
     {
         $this->createTenantConnection($name);
         $driver = $driver ?: $this->getDriver();
-        if ($driver === "sqlite") {
-            $f = fopen(database_path($name), 'w');
-            fclose($f);
-            
-            return;
+
+        $databaseCreators = config('tenancy.database_creators');
+
+        if (! array_key_exists($driver, $databaseCreators)) {
+            throw new \Exception("Database could not be created: no database creator for driver $driver is registered.");
         }
 
-        return DB::statement("CREATE DATABASE `$name` CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci");
+        app($databaseCreators[$driver])->createDatabase($name);
     }
 
     public function getDriver(): ?string
diff --git a/src/Interfaces/DatabaseCreator.php b/src/Interfaces/DatabaseCreator.php
new file mode 100644
index 00000000..9b6fd52a
--- /dev/null
+++ b/src/Interfaces/DatabaseCreator.php
@@ -0,0 +1,14 @@
+<?php
+
+namespace Stancl\Tenancy\Interfaces;
+
+interface DatabaseCreator
+{
+    /**
+     * Create a database.
+     *
+     * @param string $name Name of the database.
+     * @return void
+     */
+    public function createDatabase(string $name): bool;
+}
diff --git a/src/config/tenancy.php b/src/config/tenancy.php
index 50b6ad75..be5e00da 100644
--- a/src/config/tenancy.php
+++ b/src/config/tenancy.php
@@ -21,7 +21,6 @@ return [
     'cache' => [
         'prefix_base' => 'tenant',
     ],
-
     'filesystem' => [
         'suffix_base' => 'tenant',
         // Disks which should be suffixed with the suffix_base + tenant UUID.
@@ -30,4 +29,8 @@ return [
             // 's3',
         ],
     ],
+    'database_creators' => [
+        'sqlite' => 'Stancl\Tenancy\DatabaseCreators\SQLiteDatabaseCreator',
+        'mysql' => 'Stancl\Tenancy\DatabaseCreators\MySQLDatabaseCreator',
+    ],
 ];