tidy and pass tests

statamic · Feb 12, 2025 · 87921e1 · 87921e1
1 parent 92a95b7
commit 87921e1
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/src/Tokens/Handlers/LivePreview.php b/src/Tokens/Handlers/LivePreview.php
@@ -4,8 +4,10 @@
 
 use Closure;
 use Facades\Statamic\CP\LivePreview as Facade;
+use Illuminate\Support\Collection;
 use Statamic\Contracts\Tokens\Token;
-use Statamic\Facades\Site;
+use Statamic\Facades\Site as Sites;
+use Statamic\Sites\Site;
 
 class LivePreview
 {
@@ -17,20 +19,26 @@ public function handle(Token $token, $request, Closure $next)
 
         $response = $next($request);
 
-        if (Site::multiEnabled()) {
-            $validURLs = Site::all()
-                ->map(function ($site) {
-                    $parts = parse_url($site->absoluteUrl());
-
-                    return $parts['scheme'].'://'.$parts['host'];
-                })->values()
+        if (Sites::multiEnabled()) {
+            /** @var Collection */
+            $siteURLs = Sites::all()
+                ->map(fn (Site $site) => $this->getSchemeAndHost($site))
+                ->values()
                 ->unique()
                 ->join(' ');
-            $response->headers->set('Content-Security-Policy "frame-ancestors '.$validURLs.'"');
+
+            $response->headers->set('Content-Security-Policy', "frame-ancestors '$siteURLs'");
         }
 
         $response->headers->set('X-Statamic-Live-Preview', true);
 
         return $response;
     }
+
+    private function getSchemeAndHost(Site $site): string
+    {
+        $parts = parse_url($site->absoluteUrl());
+
+        return $parts['scheme'].'://'.$parts['host'];
+    }
 }