Allow to run a invokable class when tenant is selected

henryavila · May 4, 2023 · 6434712 · 6434712
1 parent 634859a
commit 6434712
Show file tree

Hide file tree

Showing 4 changed files with 99 additions and 92 deletions.
diff --git a/README.md b/README.md
@@ -21,6 +21,8 @@ If the `Tenant` has `domains` relationship, allow to define the current tenant b
 The `domains` relation model, must contain an `fqdn` attribute with the fqdn domain 
 
 
+Allow to run a invokable class when tenant is selected (See config file)
+
 ## Installation
 
 You can install the package via composer:

diff --git a/composer.json b/composer.json
@@ -1,6 +1,6 @@
 {
     "name": "henryavila/laravel-nova-multitenancy",
-    "version": "2.1.0",
+    "version": "2.2.0",
     "description": "Integrate the multitenancy single database in Laravel Nova.",
     "keywords": [
         "henryavila",

diff --git a/config/nova-multitenancy.php b/config/nova-multitenancy.php
@@ -38,8 +38,8 @@
 
     'tenant_not_found_message' => 'Você precisa estar vinculado a alguma conta.',
 
-    // This callback will always be executed after the Tenant has been set as current tenant
-    'execute_after_tenant_selection' => fn () => null,
+    // This invokable class will be executed after the Tenant has been set as current tenant
+    'invoke_after_tenant_selection' => null,
 
     /*
      * This class is responsible for determining which tenant should be current

diff --git a/src/Http/Middleware/SetTenantMiddleware.php b/src/Http/Middleware/SetTenantMiddleware.php
@@ -9,93 +9,98 @@
 
 class SetTenantMiddleware
 {
-    /**
-     * Handle an incoming request.
-     *
-     * @param  \Illuminate\Http\Request  $request
-     * @return mixed
-     */
-    public function handle($request, Closure $next)
-    {
-        $route = $request->route();
-        $routeName = $route ? $route->getName() : null;
-
-        // Don't run middleware if in any route marked to be ignored
-        if (in_array($routeName, config('nova-multitenancy.routes_to_ignore'))) {
-            return $next($request);
-        }
-
-        $skipRoute = LaravelNovaMultitenancy::SKIP_ROUTE;
-        if (isset($route->defaults[$skipRoute]) && $route->defaults[$skipRoute] === true) {
-            return $next($request);
-        }
-
-        // If tenant is already defined don't run this middleware
-        if (Tenant::current() !== null) {
-            return $next($request);
-        }
-
-        // Set tenant based on Domain
-        $domainClass = config('nova-multitenancy.domain_class');
-        if (! empty($domainClass)) {
-            $host = $request->getHost();
-            $fqdn = config('nova-multitenancy.fqdn_column') ?? 'fqdn';
-
-            $tenantByDomain = $domainClass::where($fqdn, $host)->first()?->tenant;
-
-            if (! empty($tenantByDomain)) {
-                $this->selectTenant($tenantByDomain);
-
-                return $next($request);
-            }
-        }
-
-        // If the user came from Tenant Selection, let's get data from session and save the tenant
-        $tenantId = session()->get(Tenant::TENANT_SELECTOR_SESSION_ID);
-
-        if (! empty($tenantId)) {
-            /** @var Tenant $tenant */
-            $tenant = LaravelNovaMultitenancy::getTenantModel($tenantId);
-            if (empty($tenant)) {
-                abort(404);
-            }
-
-            $this->selectTenant($tenant);
-
-            return $next($request);
-        }
-
-        // From this point, we have sure that: we have no tenant selected and we are not in login or select tenant routes
-        // (and not in any other route marked to be ignored)
-
-        // Allow The super Admin to enter without a Tenant Selected
-        $user = Auth::user();
-
-        // The Tenant must be set just to users that can't manage Tenants
-        $totalTenants = $user?->tenants->count() ?? 0;
-
-        if ($totalTenants === 1) {
-            $this->selectTenant($user->tenants->first());
-
-            return $next($request);
-        }
-
-        if ($user?->isSuperAdmin() || $totalTenants > 1) {
-            return redirect()->route('select-tenant');
-        }
-
-        Auth::logout();
-        abort(403, config('nova-multitenancy.tenant_not_found_message'));
-    }
-
-    private function selectTenant(Tenant $tenant): void
-    {
-        $finishCallback = config('nova-multitenancy.execute_after_tenant_selection');
-
-        $tenant->makeCurrent();
-
-        if ($finishCallback instanceof Closure) {
-            $finishCallback();
-        }
-    }
+	/**
+	 * Handle an incoming request.
+	 *
+	 * @param \Illuminate\Http\Request $request
+	 * @return mixed
+	 */
+	public function handle($request, Closure $next)
+	{
+		$route = $request->route();
+		$routeName = $route ? $route->getName() : null;
+
+		// Don't run middleware if in any route marked to be ignored
+		if (in_array($routeName, config('nova-multitenancy.routes_to_ignore'))) {
+			return $next($request);
+		}
+
+		$skipRoute = LaravelNovaMultitenancy::SKIP_ROUTE;
+		if (isset($route->defaults[$skipRoute]) && $route->defaults[$skipRoute] === true) {
+			return $next($request);
+		}
+
+		// If tenant is already defined don't run this middleware
+		if (Tenant::current() !== null) {
+			return $next($request);
+		}
+
+		// Set tenant based on Domain
+		$domainClass = config('nova-multitenancy.domain_class');
+		if (!empty($domainClass)) {
+			$host = $request->getHost();
+			$fqdn = config('nova-multitenancy.fqdn_column') ?? 'fqdn';
+
+			$tenantByDomain = $domainClass::where($fqdn, $host)->first()?->tenant;
+
+			if (!empty($tenantByDomain)) {
+				$this->selectTenant($tenantByDomain);
+
+				return $next($request);
+			}
+		}
+
+		// If the user came from Tenant Selection, let's get data from session and save the tenant
+		$tenantId = session()->get(Tenant::TENANT_SELECTOR_SESSION_ID);
+
+		if (!empty($tenantId)) {
+			/** @var Tenant $tenant */
+			$tenant = LaravelNovaMultitenancy::getTenantModel($tenantId);
+			if (empty($tenant)) {
+				abort(404);
+			}
+
+			$this->selectTenant($tenant);
+
+			return $next($request);
+		}
+
+		// From this point, we have sure that: we have no tenant selected and we are not in login or select tenant routes
+		// (and not in any other route marked to be ignored)
+
+		// Allow The super Admin to enter without a Tenant Selected
+		$user = Auth::user();
+
+		// The Tenant must be set just to users that can't manage Tenants
+		$totalTenants = $user?->tenants->count() ?? 0;
+
+		if ($totalTenants === 1) {
+			$this->selectTenant($user->tenants->first());
+
+			return $next($request);
+		}
+
+		if ($user?->isSuperAdmin() || $totalTenants > 1) {
+			return redirect()->route('select-tenant');
+		}
+
+		Auth::logout();
+		abort(403, config('nova-multitenancy.tenant_not_found_message'));
+	}
+
+	private function selectTenant(Tenant $tenant): void
+	{
+		$tenant->makeCurrent();
+
+		try {
+			$invokable = config('nova-multitenancy.invoke_after_tenant_selection');
+
+			if (class_exists($invokable)) {
+				(new $invokable)();
+			}
+
+		} catch (\Exception) {
+		}
+
+	}
 }