filter out incorrect domains

Author: edalzell

src/Http/Controllers/SitemapController.php

@@ -3,17 +3,19 @@
 namespace Statamic\SeoPro\Http\Controllers;
 
 use Carbon\Carbon;
+use Illuminate\Http\Request;
 use Illuminate\Routing\Controller;
 use Illuminate\Support\Facades\Cache;
 use Statamic\SeoPro\Sitemap\Sitemap;
 
 class SitemapController extends Controller
 {
-    public function index()
+    public function index(Request $request)
     {
         abort_unless(config('statamic.seo-pro.sitemap.enabled'), 404);
 
         $cacheUntil = Carbon::now()->addMinutes(config('statamic.seo-pro.sitemap.expire'));
+        $domain = $request->schemeAndHttpHost();
 
         if (config('statamic.seo-pro.sitemap.pagination.enabled', false)) {
             $content = Cache::remember(Sitemap::CACHE_KEY.'_index', $cacheUntil, function () {
@@ -23,10 +25,10 @@ public function index()
                 ])->render();
             });
         } else {
-            $content = Cache::remember(Sitemap::CACHE_KEY, $cacheUntil, function () {
+            $content = Cache::remember(Sitemap::CACHE_KEY.'_'.$domain, $cacheUntil, function () use ($domain) {
                 return view('seo-pro::sitemap', [
                     'xml_header' => '<?xml version="1.0" encoding="UTF-8"?>',
-                    'pages' => app(Sitemap::class)->pages(),
+                    'pages' => app(Sitemap::class)->forDomain($domain)->pages(),
                 ])->render();
             });
         }

src/Sitemap/Sitemap.php

@@ -4,9 +4,10 @@
 
 use Illuminate\Support\Collection as IlluminateCollection;
 use Illuminate\Support\LazyCollection;
+use Statamic\Entries\Entry;
 use Statamic\Facades\Blink;
 use Statamic\Facades\Collection;
-use Statamic\Facades\Entry;
+use Statamic\Facades\Entry as EntryFacade;
 use Statamic\Facades\Taxonomy;
 use Statamic\SeoPro\Cascade;
 use Statamic\SeoPro\GetsSectionDefaults;
@@ -18,6 +19,8 @@ class Sitemap
 
     const CACHE_KEY = 'seo-pro.sitemap';
 
+    private ?string $domain = null;
+
     public function pages(): array
     {
         return collect()
@@ -88,6 +91,13 @@ public function paginatedSitemaps(): array
             ->all();
     }
 
+    public function forDomain(string $domain): self
+    {
+        $this->domain = $domain;
+
+        return $this;
+    }
+
     protected function getPages($items)
     {
         return $items
@@ -123,7 +133,7 @@ protected function publishedEntriesQuery()
             ->values()
             ->all();
 
-        return Entry::query()
+        return EntryFacade::query()
             ->whereIn('collection', $collections)
             ->whereNotNull('uri')
             ->whereStatus('published')
@@ -132,7 +142,15 @@ protected function publishedEntriesQuery()
 
     protected function publishedEntries(): LazyCollection
     {
-        return $this->publishedEntriesQuery()->lazy();
+        return $this
+            ->publishedEntriesQuery()
+            ->lazy()
+            ->unless(
+                is_null($this->domain),
+                fn (LazyCollection $c) => $c->filter(
+                    fn (Entry $entry) => str($entry->permalink)->startsWith($this->domain)
+                )
+            );
     }
 
     protected function publishedEntriesForPage(int $page, int $perPage): IlluminateCollection