Merge 597eebd into 27e9fb4

archtechx · Apr 14, 2021 · fe152b7 · fe152b7
2 parents 27e9fb4 + 597eebd
commit fe152b7
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 6 deletions.
diff --git a/src/Resolvers/Contracts/CachedTenantResolver.php b/src/Resolvers/Contracts/CachedTenantResolver.php
@@ -37,7 +37,11 @@ public function resolve(...$args): Tenant
         $key = $this->getCacheKey(...$args);
 
         if ($this->cache->has($key)) {
-            return $this->cache->get($key);
+            $tenant = $this->cache->get($key);
+
+            $this->resolved($tenant, ...$args);
+
+            return $tenant;
         }
 
         $tenant = $this->resolveWithoutCache(...$args);
@@ -64,6 +68,10 @@ public function getCacheKey(...$args): string
 
     abstract public function resolveWithoutCache(...$args): Tenant;
 
+    public function resolved(Tenant $tenant, ...$args): void
+    {
+    }
+
     /**
      * Get all the arg combinations for resolve() that can be used to find this tenant.
      *

diff --git a/src/Resolvers/DomainTenantResolver.php b/src/Resolvers/DomainTenantResolver.php
@@ -4,6 +4,7 @@
 
 namespace Stancl\Tenancy\Resolvers;
 
+use Illuminate\Database\Eloquent\Builder;
 use Stancl\Tenancy\Contracts\Domain;
 use Stancl\Tenancy\Contracts\Tenant;
 use Stancl\Tenancy\Exceptions\TenantCouldNotBeIdentifiedOnDomainException;
@@ -28,18 +29,35 @@ class DomainTenantResolver extends Contracts\CachedTenantResolver
 
     public function resolveWithoutCache(...$args): Tenant
     {
-        /** @var Domain $domain */
-        $domain = config('tenancy.domain_model')::where('domain', $args[0])->first();
+        $domain = $args[0];
 
-        if ($domain) {
-            static::$currentDomain = $domain;
+        /** @var Tenant|null $tenant */
+        $tenant = config('tenancy.tenant_model')::query()
+            ->whereHas('domains', function (Builder $query) use ($domain) {
+                $query->where('domain', $domain);
+            })
+            ->with('domains')
+            ->first();
 
-            return $domain->tenant;
+        if ($tenant) {
+            $this->setCurrentDomain($tenant, $domain);
+
+            return $tenant;
         }
 
         throw new TenantCouldNotBeIdentifiedOnDomainException($args[0]);
     }
 
+    public function resolved(Tenant $tenant, ...$args): void
+    {
+        $this->setCurrentDomain($tenant, $args[0]);
+    }
+
+    protected function setCurrentDomain(Tenant $tenant, string $domain): void
+    {
+        static::$currentDomain = $tenant->domains->where('domain', $domain)->first();
+    }
+
     public function getArgsForTenant(Tenant $tenant): array
     {
         $tenant->unsetRelation('domains');