Update session flash on partial reloads

[DamianUS]

Great work guys! Inertia is amazing. I'm working with Laravel and React, and I have one doubt/proposal to know your opinion.

Right now, if I try to reload partially a resource:

<Button icon={<RedoOutlined/>} type="link" onClick={() => {router.reload({only:['sensorLocation','flash']})}}/>

Even though, such as suggested by the documentation, in HandleInertiaRequests I add:

public function share(Request $request): array
    {
        return array_merge(parent::share($request), [
            'auth' => [
                'user' => $request->user(),
            ],
            'ziggy' => function () use ($request) {
                return array_merge((new Ziggy)->toArray(), [
                    'location' => $request->url(),
                ]);
            },
            'flash' => [
                'notification' => fn () => $request->session()->get('notification'),
            ],
        ]);
    }

It needs two requests since the retrieval of flash is being done before actually executing the controller method.

Due to this, I can't show alerts/notifications as response to ajax-style actions.

I think this should be a great addition to Inertia and a major requirement in any modern UX. I can imagine there can be many workarounds, but I achieved what I tried to do by creating a post-controller global middleware.

<?php

namespace App\Http\Middleware;

use Closure;
use Illuminate\Http\Request;
use Symfony\Component\HttpFoundation\Response;

class HandleInertiaPartialReloadFlash
{
    /**
     * Handle an incoming request.
     *
     * @param  \Closure(\Illuminate\Http\Request): (\Symfony\Component\HttpFoundation\Response)  $next
     */
    public function handle(Request $request, Closure $next): Response
    {
        $response = $next($request);
        try{
            if($request->header('x-inertia-partial-data')){
                $data = $response->getData();
                $data->props->flash->notification = $request->session()->get('notification');
                $response->setData($data);
            }
        }
        catch (\Exception $e){
            return $response;
        }
        return $response;
    }
}

I wanted to ask you if you see there are some hidden risks in such approach or that I'm missing something basic that would meet the requirement I mentioned. I'm new to Inertia!

Thanks in advance.