tenancy/src/StorageDrivers/Database/DatabaseStorageDriver.php

<?php

declare(strict_types=1);

namespace Stancl\Tenancy\StorageDrivers\Database;

use Illuminate\Foundation\Application;
use Illuminate\Support\Carbon;
use Stancl\Tenancy\Contracts\StorageDriver;
use Stancl\Tenancy\DatabaseManager;
use Stancl\Tenancy\Exceptions\DomainsOccupiedByOtherTenantException;
use Stancl\Tenancy\Exceptions\TenantCouldNotBeIdentifiedException;
use Stancl\Tenancy\Exceptions\TenantWithThisIdAlreadyExistsException;
use Stancl\Tenancy\StorageDrivers\Database\DomainModel as Domains;
use Stancl\Tenancy\StorageDrivers\Database\TenantModel as Tenants;
use Stancl\Tenancy\Tenant;

class DatabaseStorageDriver implements StorageDriver
{
    /** @var Application */
    protected $app;

    /** @var \Illuminate\Database\Connection */
    protected $centralDatabase;

    /** @var Tenant The default tenant. */
    protected $tenant;

    public function __construct(Application $app)
    {
        $this->app = $app;
        $this->centralDatabase = $app->make(DatabaseManager::class)->getCentralConnection();
    }

    public function findByDomain(string $domain): Tenant
    {
        $id = $this->getTenantIdByDomain($domain);
        if (! $id) {
            throw new TenantCouldNotBeIdentifiedException($domain);
        }

        return $this->findById($id);
    }

    public function findById(string $id): Tenant
    {
        return Tenant::fromStorage(Tenants::find($id)->decoded())
            ->withDomains($this->getTenantDomains($id));
    }

    protected function getTenantDomains($id)
    {
        return Domains::where('tenant_id', $id)->get()->map(function ($model) {
            return $model->domain;
        })->toArray();
    }

    public function ensureTenantCanBeCreated(Tenant $tenant): void
    {
        if (Tenants::find($tenant->id)) {
            throw new TenantWithThisIdAlreadyExistsException($tenant->id);
        }

        if (Domains::whereIn('domain', $tenant->domains)->exists()) {
            throw new DomainsOccupiedByOtherTenantException;
        }
    }

    public function withDefaultTenant(Tenant $tenant): self
    {
        $this->tenant = $tenant;

        return $this;
    }

    public function getTenantIdByDomain(string $domain): ?string
    {
        return Domains::where('domain', $domain)->first()->tenant_id ?? null;
    }

    public function createTenant(Tenant $tenant): void
    {
        $this->centralDatabase->transaction(function () use ($tenant) {
            Tenants::create(['id' => $tenant->id, 'data' => '{}'])->toArray();

            $domainData = [];
            foreach ($tenant->domains as $domain) {
                $domainData[] = ['domain' => $domain, 'tenant_id' => $tenant->id, 'created_at' => Carbon::now()];
            }

            Domains::insert($domainData);
        });
    }

    public function updateTenant(Tenant $tenant): void
    {
        $this->centralDatabase->transaction(function () use ($tenant) {
            Tenants::find($tenant->id)->putMany($tenant->data);

            $original_domains = Domains::where('tenant_id', $tenant->id)->get()->map(function ($model) {
                return $model->domain;
            })->toArray();
            $deleted_domains = array_diff($original_domains, $tenant->domains);

            Domains::whereIn('domain', $deleted_domains)->delete();

            foreach ($tenant->domains as $domain) {
                Domains::firstOrCreate([
                    'tenant_id' => $tenant->id,
                    'domain' => $domain,
                ]);
            }
        });
    }

    public function deleteTenant(Tenant $tenant): void
    {
        $this->centralDatabase->transaction(function () use ($tenant) {
            Tenants::find($tenant->id)->delete();
            Domains::where('tenant_id', $tenant->id)->delete();
        });
    }

    /**
     * Get all tenants.
     *
     * @param string[] $ids
     * @return Tenant[]
     */
    public function all(array $ids = []): array
    {
        return Tenants::getAllTenants($ids)->map(function ($data) {
            return Tenant::fromStorage($data)->withDomains($this->getTenantDomains($data['id']));
        })->toArray();
    }

    /**
     * Get the current tenant.
     *
     * @return Tenant
     */
    protected function currentTenant()
    {
        return $this->tenant ?? $this->app[Tenant::class];
    }

    public function get(string $key, Tenant $tenant = null)
    {
        $tenant = $tenant ?? $this->currentTenant();

        return Tenants::find($tenant->id)->get($key);
    }

    public function getMany(array $keys, Tenant $tenant = null): array
    {
        $tenant = $tenant ?? $this->currentTenant();

        return Tenants::find($tenant->id)->getMany($keys);
    }

    public function put(string $key, $value, Tenant $tenant = null): void
    {
        $tenant = $tenant ?? $this->currentTenant();
        Tenants::find($tenant->id)->put($key, $value);
    }

    public function putMany(array $kvPairs, Tenant $tenant = null): void
    {
        $tenant = $tenant ?? $this->currentTenant();
        Tenants::find($tenant->id)->putMany($kvPairs);
    }
}