TenantAtlas/app/Filament/Pages/ChooseTenant.php

<?php

declare(strict_types=1);

namespace App\Filament\Pages;

use App\Models\Tenant;
use App\Models\User;
use App\Models\UserTenantPreference;
use App\Services\Tenants\TenantOperabilityService;
use App\Support\Tenants\TenantLifecyclePresentation;
use App\Support\Workspaces\WorkspaceContext;
use Filament\Facades\Filament;
use Filament\Pages\Page;
use Illuminate\Database\Eloquent\Collection;
use Illuminate\Support\Facades\Schema;

class ChooseTenant extends Page
{
    protected static string $layout = 'filament-panels::components.layout.simple';

    protected static bool $shouldRegisterNavigation = false;

    protected static bool $isDiscovered = false;

    protected static ?string $slug = 'choose-tenant';

    protected static ?string $title = 'Choose tenant';

    protected string $view = 'filament.pages.choose-tenant';

    /**
     * Disable the simple-layout topbar to prevent lazy-loaded
     * DatabaseNotifications from triggering Livewire update 404s.
     */
    protected function getLayoutData(): array
    {
        return [
            'hasTopbar' => false,
        ];
    }

    /**
     * @return Collection<int, Tenant>
     */
    public function getTenants(): Collection
    {
        $user = auth()->user();

        if (! $user instanceof User) {
            return Tenant::query()->whereRaw('1 = 0')->get();
        }

        $tenants = $user->getTenants(Filament::getCurrentOrDefaultPanel());

        if ($tenants instanceof Collection) {
            return app(TenantOperabilityService::class)->filterSelectable($tenants);
        }

        return app(TenantOperabilityService::class)->filterSelectable(collect($tenants));
    }

    public function selectTenant(int $tenantId): void
    {
        $user = auth()->user();

        if (! $user instanceof User) {
            abort(403);
        }

        $workspaceId = app(WorkspaceContext::class)->currentWorkspaceId(request());

        if ($workspaceId === null) {
            $this->redirect(route('filament.admin.pages.choose-workspace'));

            return;
        }

        $tenant = Tenant::query()
            ->where('workspace_id', $workspaceId)
            ->whereKey($tenantId)
            ->first();

        if (! $tenant instanceof Tenant) {
            abort(404);
        }

        if (! $user->canAccessTenant($tenant)) {
            abort(404);
        }

        if (! app(TenantOperabilityService::class)->canSelectAsContext($tenant)) {
            abort(404);
        }

        $this->persistLastTenant($user, $tenant);

        if (! app(WorkspaceContext::class)->rememberTenantContext($tenant, request())) {
            abort(404);
        }

        $this->redirect(TenantDashboard::getUrl(panel: 'tenant', tenant: $tenant));
    }

    public function tenantLifecyclePresentation(Tenant $tenant): TenantLifecyclePresentation
    {
        return TenantLifecyclePresentation::fromTenant($tenant);
    }

    private function persistLastTenant(User $user, Tenant $tenant): void
    {
        if (Schema::hasColumn('users', 'last_tenant_id')) {
            $user->forceFill(['last_tenant_id' => $tenant->getKey()])->save();

            return;
        }

        if (! Schema::hasTable('user_tenant_preferences')) {
            return;
        }

        UserTenantPreference::query()->updateOrCreate(
            ['user_id' => $user->getKey(), 'tenant_id' => $tenant->getKey()],
            ['last_used_at' => now()]
        );
    }
}