Inject all assets into cached response

Author: aerni

src/Replacers/AssetsReplacer.php

@@ -2,13 +2,14 @@
 
 namespace Jonassiewertsen\Livewire\Replacers;
 
-use Illuminate\Http\Response;
 use Illuminate\Support\Str;
-use Livewire\Features\SupportAutoInjectedAssets\SupportAutoInjectedAssets;
-use Livewire\Features\SupportScriptsAndAssets\SupportScriptsAndAssets;
-use Livewire\Livewire;
-use Livewire\Mechanisms\FrontendAssets\FrontendAssets;
+use Illuminate\Http\Response;
+use Statamic\StaticCaching\Cacher;
 use Statamic\StaticCaching\Replacer;
+use Statamic\StaticCaching\Cachers\NullCacher;
+use Livewire\Mechanisms\FrontendAssets\FrontendAssets;
+use Livewire\Features\SupportScriptsAndAssets\SupportScriptsAndAssets;
+use Livewire\Features\SupportAutoInjectedAssets\SupportAutoInjectedAssets;
 
 class AssetsReplacer implements Replacer
 {
@@ -18,31 +19,54 @@ public function prepareResponseToCache(Response $responseToBeCached, Response $i
             return;
         }
 
-        if (! $assets = SupportScriptsAndAssets::getAssets()) {
+        // Don't disturb Livewire's assets injection when caching is off.
+        if (app(Cacher::class) instanceof NullCacher) {
             return;
         }
 
+        $assetsHead = '';
+        $assetsBody = '';
+
+        $assets = array_values(SupportScriptsAndAssets::getAssets());
+
+        if (count($assets) > 0) {
+            foreach ($assets as $asset) {
+                $assetsHead .= $asset."\n";
+            }
+        }
+
+        if ($this->shouldInjectLivewireAssets($initialResponse)) {
+            $assetsHead .= FrontendAssets::styles()."\n";
+            $assetsBody .= FrontendAssets::scripts()."\n";
+
+            /**
+             * Ensure Livewire injects its assets on the initial request.
+             * @see \Livewire\Features\SupportAutoInjectedAssets\SupportAutoInjectedAssets
+             */
+            app(FrontendAssets::class)->hasRenderedStyles = false;
+            app(FrontendAssets::class)->hasRenderedScripts = false;
+        }
+
         $responseToBeCached->setContent(
-            SupportAutoInjectedAssets::injectAssets(
-                html: $content,
-                assetsHead: implode('', $assets),
-                assetsBody: ''
-            )
+            SupportAutoInjectedAssets::injectAssets($content, $assetsHead, $assetsBody)
         );
     }
 
-    public function replaceInCachedResponse(Response $response)
+    protected function shouldInjectLivewireAssets(Response $response): bool
     {
         if (Str::contains($response, FrontendAssets::scripts())) {
-            return;
+            return false;
         }
 
         if (Str::contains($response, FrontendAssets::scriptConfig())) {
-            return;
+            return false;
         }
 
-        app(FrontendAssets::class)->hasRenderedScripts = false;
+        return true;
+    }
 
-        Livewire::forceAssetInjection();
+    public function replaceInCachedResponse(Response $response)
+    {
+        //
     }
 }

src/ServiceProvider.php

@@ -30,8 +30,8 @@ public function bootAddon(): void
     protected function bootReplacers(): void
     {
         config()->set('statamic.static_caching.replacers', array_merge(
-            config('statamic.static_caching.replacers'),
-            config('statamic-livewire.replacers')
+            config('statamic-livewire.replacers'),
+            config('statamic.static_caching.replacers')
         ));
     }