From 9de4924933dc1e382576eff3b598287f102e232c Mon Sep 17 00:00:00 2001
From: Joseph Roberts <joseph.roberts@usa.com>
Date: Tue, 30 Jan 2024 01:52:04 -0500
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=9A=20PSR-7=20&=20PSR-15=20support=20f?=
 =?UTF-8?q?or=20WordPress=20routes=20(#334)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---------

Co-authored-by: QWp6t <QWp6t@users.noreply.github.com>
Co-authored-by: Brandon <brandon@tendency.me>
---
 src/Roots/Acorn/Bootloader.php | 86 +++++++++++++++++++++++-----------
 1 file changed, 59 insertions(+), 27 deletions(-)

diff --git a/src/Roots/Acorn/Bootloader.php b/src/Roots/Acorn/Bootloader.php
index bc1cc3ba..9122c8ad 100644
--- a/src/Roots/Acorn/Bootloader.php
+++ b/src/Roots/Acorn/Bootloader.php
@@ -3,6 +3,7 @@
 namespace Roots\Acorn;
 
 use Illuminate\Contracts\Foundation\Application as ApplicationContract;
+use Illuminate\Http\Response;
 use Illuminate\Support\Env;
 use Illuminate\Support\Facades\Facade;
 use Illuminate\Support\Str;
@@ -201,26 +202,74 @@ protected function bootHttp(ApplicationContract $app)
     {
         $kernel = $app->make(\Illuminate\Contracts\Http\Kernel::class);
         $request = \Illuminate\Http\Request::capture();
-        $time = microtime();
 
         $app->instance('request', $request);
+
         Facade::clearResolvedInstance('request');
 
         $kernel->bootstrap($request);
 
-        add_filter('do_parse_request', function ($doParse, \WP $wp, $extraQueryVars) use ($app, $request) {
-            if (! $app->make('router')->getRoutes()->match($request)) {
+        $this->registerWordPressRoute($app);
+
+        /** @var \Illuminate\Routing\Route $route */
+        $route = $app->make('router')->getRoutes()->match($request);
+
+        /** @var array $config */
+        $config = $app->config->get('router.wordpress', ['web' => 'web', 'api' => 'api']);
+
+        $this->registerRequestHandler($kernel, $request, $route, $config);
+    }
+
+    /**
+     * Register the WordPress route.
+     *
+     * @return void
+     */
+    protected function registerWordPressRoute(ApplicationContract $app)
+    {
+        $app->make('router')
+            ->any('{any?}', fn () => tap(response(), function (Response $response) {
+                foreach (headers_list() as $header) {
+                    header_remove($header);
+                    $response->header(...explode(': ', $header));
+                }
+
+                $response->setContent(ob_get_clean());
+            }))
+            ->where('any', '.*')
+            ->name('wordpress_request');
+    }
+
+    /**
+     * Register the request handler.
+     *
+     * @return void
+     */
+    protected function registerRequestHandler(
+        \Illuminate\Contracts\Http\Kernel $kernel,
+        \Illuminate\Http\Request $request,
+        ?\Illuminate\Routing\Route $route,
+        array $config
+    ) {
+        add_filter('do_parse_request', function ($doParse, \WP $wp, $extraQueryVars) use ($route) {
+            if (! $route) {
                 return $doParse;
             }
 
             return apply_filters('acorn/router/do_parse_request', $doParse, $wp, $extraQueryVars);
         }, 100, 3);
 
-        $app->make('router')
-            ->any('{any?}', fn () => response()->json(['message' => "wordpress_request_{$time}"]))
-            ->where('any', '.*');
+        if ($route->getName() !== 'wordpress_request') {
+            add_action('parse_request', fn () => $this->handleRequest($kernel, $request));
+
+            return;
+        }
+
+        $route->middleware(preg_match('/^wp-json(\/.*)?/', $request->path()) ? $config['api'] : $config['web']);
+
+        ob_start();
 
-        add_action('parse_request', fn () => $this->handleRequest($time, $kernel, $request));
+        add_action('shutdown', fn () => $this->handleRequest($kernel, $request), PHP_INT_MAX);
     }
 
     /**
@@ -229,33 +278,16 @@ protected function bootHttp(ApplicationContract $app)
      * @return void
      */
     protected function handleRequest(
-        string $time,
         \Illuminate\Contracts\Http\Kernel $kernel,
         \Illuminate\Http\Request $request
     ) {
         $response = $kernel->handle($request);
 
-        if (
-            $response instanceof \Symfony\Component\HttpFoundation\Response
-            && ! $response->isServerError()
-        ) {
-            return;
-        }
+        $body = $response->send();
 
-        if (
-            in_array(false, [
-                $response instanceof \Illuminate\Http\JsonResponse,
-                is_string($response->getContent()),
-                $data = json_decode($response->getContent()),
-                isset($data->message) && $data->message == "wordpress_request_{$time}",
-            ])
-        ) {
-            $body = $response->send();
+        $kernel->terminate($request, $body);
 
-            $kernel->terminate($request, $body);
-
-            exit;
-        }
+        exit((int) $response->isServerError());
     }
 
     /**