<?php

namespace App\Http\Middleware;

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

class PrivateTwoFactorResponse
{
    public function handle(Request $request, Closure $next): Response
    {
        if (! str_starts_with($request->route()?->getName() ?? '', 'two-factor.')) {
            return $next($request);
        }
        try {
            $response = $next($request);
        } catch (ValidationException $error) {
            // Exclude authentication material from old input without affecting other forms' code fields.
            $request->request->remove('code');
            $request->request->remove('recovery_code');
            $request->json()->remove('code');
            $request->json()->remove('recovery_code');
            throw $error;
        }
        if ($request->hasSession()) {
            $request->session()->forget(['_old_input.code', '_old_input.recovery_code']);
        }
        $response->headers->set('Cache-Control', 'private, no-store, max-age=0');

        return $response;
    }
}