Added urlResolver() method to Middleware class

Author: pascalbaljet

src/Inertia.php

@@ -13,6 +13,7 @@
  * @method static void flushShared()
  * @method static void version(\Closure|string|null $version)
  * @method static string getVersion()
+ * @method static void resolveUrlUsing(\Closure|null $urlResolver = null)
  * @method static \Inertia\OptionalProp optional(callable $callback)
  * @method static \Inertia\LazyProp lazy(callable $callback)
  * @method static \Inertia\DeferProp defer(callable $callback, string $group = 'default')

src/Middleware.php

@@ -69,6 +69,16 @@ public function rootView(Request $request)
         return $this->rootView;
     }
 
+    /**
+     * Defines a callback that returns the relative URL.
+     *
+     * @return Closure|null
+     */
+    public function urlResolver()
+    {
+        return null;
+    }
+
     /**
      * Handle the incoming request.
      *
@@ -83,6 +93,10 @@ public function handle(Request $request, Closure $next)
         Inertia::share($this->share($request));
         Inertia::setRootView($this->rootView($request));
 
+        if ($urlResolver = $this->urlResolver()) {
+            Inertia::resolveUrlUsing($urlResolver);
+        }
+
         $response = $next($request);
         $response->headers->set('Vary', Header::INERTIA);
 

src/ResponseFactory.php

@@ -96,7 +96,7 @@ public function getVersion(): string
         return (string) $version;
     }
 
-    public function resolveUrlUsing(?Closure $urlResolver): void
+    public function resolveUrlUsing(?Closure $urlResolver = null): void
     {
         $this->urlResolver = $urlResolver;
     }

tests/MiddlewareTest.php

@@ -11,6 +11,7 @@
 use Inertia\AlwaysProp;
 use Inertia\Inertia;
 use Inertia\Middleware;
+use Inertia\Tests\Stubs\CustomUrlResolverMiddleware;
 use Inertia\Tests\Stubs\ExampleMiddleware;
 use LogicException;
 
@@ -122,6 +123,21 @@ public function test_it_will_instruct_inertia_to_reload_on_a_version_mismatch():
         self::assertEmpty($response->getContent());
     }
 
+    public function test_the_url_can_be_resolved_with_a_custom_resolver()
+    {
+        $this->prepareMockEndpoint(middleware: new CustomUrlResolverMiddleware);
+
+        $response = $this->withoutExceptionHandling()->get('/', [
+            'X-Inertia' => 'true',
+        ]);
+
+        $response->assertSuccessful();
+        $response->assertJson([
+            'component' => 'User/Edit',
+            'url' => '/my-custom-url',
+        ]);
+    }
+
     public function test_validation_errors_are_registered_as_of_default(): void
     {
         Route::middleware([StartSession::class, ExampleMiddleware::class])->get('/', function () {

tests/Stubs/CustomUrlResolverMiddleware.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace Inertia\Tests\Stubs;
+
+use Illuminate\Http\Request;
+use Illuminate\Routing\ResponseFactory;
+use Inertia\Middleware;
+use PHPUnit\Framework\Assert;
+
+class CustomUrlResolverMiddleware extends Middleware
+{
+    public function urlResolver()
+    {
+        return function ($request, ResponseFactory $otherDependency) {
+            Assert::assertInstanceOf(Request::class, $request);
+            Assert::assertInstanceOf(ResponseFactory::class, $otherDependency);
+
+            return '/my-custom-url';
+        };
+    }
+}