Identification middleware & tests

src/Contracts/Domain.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace Stancl\Tenancy\Contracts;
+
+/**
+ * @property-read Tenant $tenant
+ * 
+ * @see \Stancl\Tenancy\Database\Models\Domain
+ */
+interface Domain
+{
+    public function tenant();
+}

src/Database/Models/Domain.php

@@ -8,7 +8,8 @@ use Stancl\Tenancy\Events\DomainCreated;
 use Stancl\Tenancy\Events\DomainDeleted;
 use Stancl\Tenancy\Events\DomainSaved;
 use Stancl\Tenancy\Events\DomainUpdated;
-use Stancl\Tenancy\Exceptions\DomainsOccupiedByOtherTenantException;
+use Stancl\Tenancy\Exceptions\DomainOccupiedByOtherTenantException;
+use Stancl\Tenancy\Contracts;
 
 /**
  * @property string $domain
@@ -16,7 +17,7 @@ use Stancl\Tenancy\Exceptions\DomainsOccupiedByOtherTenantException;
  *
  * @property-read Tenant $tenant
  */
-class Domain extends Model
+class Domain extends Model implements Contracts\Domain
 {
     public $guarded = [];
     public $casts = [
@@ -28,7 +29,7 @@ class Domain extends Model
         $ensureDomainIsNotOccupied = function (Domain $self) {
             if ($domain = Domain::where('domain', $self->domain)->first()) {
                 if ($domain->getKey() !== $self->getKey()) {
-                    throw new DomainsOccupiedByOtherTenantException;
+                    throw new DomainOccupiedByOtherTenantException($self->domain);
                 }
             }
         };

src/Exceptions/DomainOccupiedByOtherTenantException.php

@@ -0,0 +1,15 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Exceptions;
+
+use Exception;
+
+class DomainOccupiedByOtherTenantException extends Exception
+{
+    public function __construct($domain)
+    {
+        parent::__construct("The $domain domain is occupied by another tenant.");
+    }
+}

src/Exceptions/DomainsOccupiedByOtherTenantException.php

@@ -1,15 +0,0 @@
-<?php
-
-declare(strict_types=1);
-
-namespace Stancl\Tenancy\Exceptions;
-
-use Stancl\Tenancy\Contracts\TenantCannotBeCreatedException;
-
-class DomainsOccupiedByOtherTenantException extends TenantCannotBeCreatedException
-{
-    public function reason(): string
-    {
-        return "One or more of the tenant's domains are already occupied by another tenant.";
-    }
-}

src/Exceptions/NotASubdomainException.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace Stancl\Tenancy\Exceptions;
+
+use Exception;
+
+class NotASubdomainException extends Exception
+{
+    public function __construct(string $hostname)
+    {
+        parent::__construct("Hostname $hostname does not include a subdomain.");
+    }
+}

src/Middleware/IdentificationMiddleware.php

@@ -0,0 +1,36 @@
+<?php
+
+namespace Stancl\Tenancy\Middleware;
+
+use Stancl\Tenancy\Contracts\TenantCouldNotBeIdentifiedException;
+use Stancl\Tenancy\Contracts\TenantResolver;
+use Stancl\Tenancy\Tenancy;
+
+abstract class IdentificationMiddleware
+{
+    /** @var callable */
+    public static $onFail;
+
+    /** @var Tenancy */
+    protected $tenancy;
+
+    /** @var TenantResolver */
+    protected $resolver;
+
+    public function initializeTenancy($request, $next, ...$resolverArguments)
+    {
+        try {
+            $this->tenancy->initialize(
+                $this->resolver->resolve(...$resolverArguments)
+            );
+        } catch (TenantCouldNotBeIdentifiedException $e) {
+            $onFail = static::$onFail ?? function ($e) {
+                throw $e;
+            };
+
+            return $onFail($e, $request, $next);
+        }
+
+        return $next($request);
+    }
+}

src/Middleware/InitializeTenancy.php

@@ -1,45 +0,0 @@
-<?php
-
-declare(strict_types=1);
-
-namespace Stancl\Tenancy\Middleware;
-
-use Closure;
-use Stancl\Tenancy\Exceptions\TenantCouldNotBeIdentifiedException;
-
-class InitializeTenancy
-{
-    /** @var callable */
-    protected $onFail;
-
-    public function __construct(callable $onFail = null)
-    {
-        $this->onFail = $onFail ?? function ($e) {
-            throw $e;
-        };
-    }
-
-    /**
-     * Handle an incoming request.
-     *
-     * @param  \Illuminate\Http\Request  $request
-     * @param  \Closure  $next
-     * @return mixed
-     */
-    public function handle($request, Closure $next)
-    {
-        if (tenancy()->initialized) {
-            return $next($request);
-        }
-
-        if (! in_array($request->getHost(), config('tenancy.exempt_domains', []), true)) {
-            try {
-                tenancy()->init($request->getHost());
-            } catch (TenantCouldNotBeIdentifiedException $e) {
-                return ($this->onFail)($e, $request, $next);
-            }
-        }
-
-        return $next($request);
-    }
-}

src/Middleware/InitializeTenancyByDomain.php

@@ -0,0 +1,41 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Middleware;
+
+use Closure;
+use Stancl\Tenancy\Resolvers\DomainTenantResolver;
+use Stancl\Tenancy\Tenancy;
+
+class InitializeTenancyByDomain extends IdentificationMiddleware
+{
+    /** @var callable|null */
+    public static $onFail;
+
+    /** @var Tenancy */
+    protected $tenancy;
+
+    /** @var DomainTenantResolver */
+    protected $resolver;
+
+    public function __construct(Tenancy $tenancy, DomainTenantResolver $resolver)
+    {
+        $this->tenancy = $tenancy;
+        $this->resolver = $resolver;
+    }
+
+    /**
+     * Handle an incoming request.
+     *
+     * @param  \Illuminate\Http\Request  $request
+     * @param  \Closure  $next
+     * @return mixed
+     */
+    public function handle($request, Closure $next)
+    {
+        return $this->initializeTenancy(
+            $request, $next, $request->getHost()
+        );
+    }
+}

src/Middleware/InitializeTenancyByPath.php

@@ -8,7 +8,7 @@ use Illuminate\Routing\Route;
 use Stancl\Tenancy\Resolvers\PathTenantResolver;
 use Stancl\Tenancy\Tenancy;
 
-class InitializeTenancyByPath
+class InitializeTenancyByPath extends IdentificationMiddleware
 {
     /** @var Tenancy */
     protected $tenancy;
@@ -27,14 +27,15 @@ class InitializeTenancyByPath
         /** @var Route $route */
         $route = $request->route();
 
+        // todo test the behavior described by the comment
         // Only initialize tenancy if tenant is the first parameter
         // We don't want to initialize tenancy if the tenant is
         // simply injected into some route controller action.
         if ($route->parameterNames()[0] === 'tenant') {
-            $this->tenancy->initialize(
-                $this->resolver->resolve($route)
+            return $this->initializeTenancy(
+                $request, $next, $route
             );
-        }
+        } // todo else case should probably throw exception about malformed route? or do we just leave that as the developer's responsibility?
 
         return $next($request);
     }

src/Middleware/InitializeTenancyBySubdomain.php

@@ -0,0 +1,67 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Middleware;
+
+use Closure;
+use Illuminate\Http\Response;
+use Stancl\Tenancy\Exceptions\NotASubdomainException;
+
+class InitializeTenancyBySubdomain extends InitializeTenancyByDomain
+{
+    /** @var callable|null */
+    public static $onInvalidSubdomain;
+
+    /**
+     * The index of the subdomain fragment in the hostname
+     * split by `.`. 0 for first fragment, 1 if you prefix
+     * your subdomain fragments with `www`.
+     *
+     * @var int
+     */
+    public static $subdomainIndex = 0;
+
+    /**
+     * Handle an incoming request.
+     *
+     * @param  \Illuminate\Http\Request  $request
+     * @param  \Closure  $next
+     * @return mixed
+     */
+    public function handle($request, Closure $next)
+    {
+        $subdomain = $this->makeSubdomain($request->getHost());
+
+        // If a non-string, like a Response instance was returned
+        // from makeSubdomain() - due to NotASubDomainException
+        // being thrown, we abort by returning the value now.
+        if (! is_string($subdomain)) {
+            return $subdomain;
+        }
+
+        return $this->initializeTenancy(
+            $request, $next, $subdomain
+        );
+    }
+
+    /** @return string|Response|mixed */
+    protected function makeSubdomain(string $hostname)
+    {
+        $parts = explode('.', $hostname);
+
+        // If we're on localhost or an IP address, then we're not visiting a subdomain.
+        if (in_array(count($parts), [1, 4])) {
+            $handle = static::$onInvalidSubdomain ?? function ($e) {
+                throw $e;
+            };
+
+            return $handle(new NotASubdomainException($hostname));
+        }
+
+        // todo should we verify that the subdomain belongs to one of our central domains?
+        // if yes, then write a test for it.
+
+        return $parts[static::$subdomainIndex];
+    }
+}

src/Middleware/PreventAccessFromTenantDomains.php

@@ -1,75 +0,0 @@
-<?php
-
-declare(strict_types=1);
-
-namespace Stancl\Tenancy\Middleware;
-
-use Closure;
-use Illuminate\Routing\Route;
-use Illuminate\Support\Facades\Route as Router;
-
-/**
- * Prevent access from tenant domains to central routes and vice versa.
- */
-class PreventAccessFromTenantDomains
-{
-    /** @var callable */
-    protected $central404;
-
-    public function __construct(callable $central404 = null)
-    {
-        $this->central404 = $central404 ?? function () {
-            return 404;
-        };
-    }
-
-    /**
-     * Handle an incoming request.
-     *
-     * @param  \Illuminate\Http\Request  $request
-     * @param  \Closure  $next
-     * @return mixed
-     */
-    public function handle($request, Closure $next)
-    {
-        // If the route is universal, always let the request pass.
-        if ($this->routeHasMiddleware($request->route(), 'universal')) {
-            return $next($request);
-        }
-
-        // If the domain is not in exempt domains, it's a tenant domain.
-        // Tenant domains can't have routes without tenancy middleware.
-        $isExemptDomain = in_array($request->getHost(), config('tenancy.exempt_domains'));
-        $isTenantDomain = ! $isExemptDomain;
-
-        $isTenantRoute = $this->routeHasMiddleware($request->route(), 'tenancy');
-
-        if ($isTenantDomain && ! $isTenantRoute) { // accessing web routes from tenant domains
-            return redirect(config('tenancy.home_url'));
-        }
-
-        if ($isExemptDomain && $isTenantRoute) { // accessing tenant routes on web domains
-            return ($this->central404)($request, $next);
-        }
-
-        return $next($request);
-    }
-
-    public static function routeHasMiddleware(Route $route, $middleware): bool
-    {
-        if (in_array($middleware, $route->middleware(), true)) {
-            return true;
-        }
-
-        // Loop one level deep and check if the route's middleware
-        // groups have a `tenancy` middleware group inside them
-        $middlewareGroups = Router::getMiddlewareGroups();
-        foreach ($route->gatherMiddleware() as $inner) {
-            if (! $inner instanceof Closure && isset($middlewareGroups[$inner]) && in_array($middleware, $middlewareGroups[$inner], true)) {
-                return true;
-            }
-        }
-
-        return false;
-    }
-}

src/Resolvers/DomainTenantResolver.php

@@ -2,6 +2,7 @@
 
 namespace Stancl\Tenancy\Resolvers;
 
+use Stancl\Tenancy\Contracts\Domain;
 use Stancl\Tenancy\Contracts\Tenant;
 use Stancl\Tenancy\Contracts\TenantResolver;
 use Stancl\Tenancy\Exceptions\TenantCouldNotBeIdentifiedOnDomainException;
@@ -10,6 +11,7 @@ class DomainTenantResolver implements TenantResolver
 {
     public function resolve(...$args): Tenant
     {
+        /** @var Domain $domain */
         $domain = config('tenancy.domain_model')::where('domain', $args[0])->first();
 
         if ($domain) {
@@ -18,4 +20,4 @@ class DomainTenantResolver implements TenantResolver
 
         throw new TenantCouldNotBeIdentifiedOnDomainException($domain);
     }
-}
+}

src/Tenancy.php

@@ -3,7 +3,7 @@
 namespace Stancl\Tenancy;
 
 use Stancl\Tenancy\Contracts\TenancyBootstrapper;
-use Stancl\Tenancy\Database\Models\Tenant;
+use Stancl\Tenancy\Database\Models\Tenant; // todo contract
 
 class Tenancy
 {
@@ -18,6 +18,11 @@ class Tenancy
 
     public function initialize(Tenant $tenant): void
     {
+        // todo the id is something that should be on the contract, with a method
+        if ($this->initialized && $this->tenant->id === $tenant->id) {
+            return;
+        }
+
         $this->tenant = $tenant;
 
         $this->initialized = true;