tenancy/src/DatabaseManager.php

<?php

namespace Stancl\Tenancy;

use Illuminate\Bus\Queueable;
use Illuminate\Support\Facades\DB;
use Illuminate\Queue\SerializesModels;
use Illuminate\Queue\InteractsWithQueue;
use Illuminate\Foundation\Bus\Dispatchable;
use Illuminate\Contracts\Queue\ShouldQueue;
use Illuminate\Database\DatabaseManager as BaseDatabaseManager;
use Stancl\Tenancy\Interfaces\DatabaseCreator;

class DatabaseManager
{
    public function __construct(BaseDatabaseManager $database)
    {
        $this->database = $database;
    }

    public function connect(string $database)
    {
        $this->createTenantConnection($database);
        $this->database->setDefaultConnection('tenant');
        $this->database->reconnect('tenant');
    }

    public function connectToTenant($tenant)
    {
        $this->connect(tenant()->getDatabaseName($tenant));
    }

    public function disconnect()
    {
        $this->database->reconnect('default');
        $this->database->setDefaultConnection('default');
    }

    public function create(string $name, string $driver = null)
    {
        $this->createTenantConnection($name);
        $driver = $driver ?: $this->getDriver();

        $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.");
        }

        if (config('tenancy.queue_database_creation', false)) {
            QueuedDatabaseCreator::dispatch(app($databaseCreators[$driver], $name));
        } else {
            app($databaseCreators[$driver])->createDatabase($name);
        }
    }

    public function delete()
    {
        // todo: delete database. similar to create()
    }

    public function getDriver(): ?string
    {
        return config("database.connections.tenant.driver");
    }

    public function createTenantConnection(string $database_name)
    {
        // Create the `tenancy` database connection.
        $based_on = config('tenancy.database.based_on') ?: config('database.default');
        config()->set([
            'database.connections.tenant' => config('database.connections.' . $based_on)
        ]);

        // Change DB name
        $database_name = $this->getDriver() === "sqlite" ? database_path($database_name) : $database_name;
        config()->set(['database.connections.tenant.database' => $database_name]);
    }
}

class QueuedDatabaseCreator implements ShouldQueue
{
    use Dispatchable, InteractsWithQueue, Queueable, SerializesModels;

    /**
     * Create a new job instance.
     *
     * @param DatabaseCreator $databaseCreator
     * @param string $databaseName
     * @return void
     */
    public function __construct(DatabaseCreator $databaseCreator, string $databaseName)
    {
        $this->databaseCreator = $databaseCreator;
        $this->databaseName = $databaseName;
    }

    /**
     * Execute the job.
     *
     * @return void
     */
    public function handle()
    {
        $this->databaseCreator->createDatabase($databaseName);
    }
}