<?php 
 
/* 
 * This file is part of the Nelmio SecurityBundle. 
 * 
 * (c) Nelmio <hello@nelm.io> 
 * 
 * For the full copyright and license information, please view the LICENSE 
 * file that was distributed with this source code. 
 */ 
 
namespace Nelmio\SecurityBundle\EventListener; 
 
use Symfony\Component\HttpKernel\Event\FilterResponseEvent; 
use Symfony\Component\HttpKernel\Event\ResponseEvent; 
use Symfony\Component\HttpKernel\HttpKernelInterface; 
 
/** 
 * Referrer Policy Listener. 
 * 
 * @author Andrej Hudec <pulzarraider@gmail.com> 
 * @final 
 */ 
class ReferrerPolicyListener 
{ 
    private $policies; 
 
    public function __construct(array $policies) 
    { 
        $this->policies = $policies; 
    } 
 
    /** 
     * @param FilterResponseEvent|ResponseEvent $e 
     */ 
    public function onKernelResponse($e) 
    { 
        // Compatibility with Symfony < 5 and Symfony >=5 
        if (!$e instanceof FilterResponseEvent && !$e instanceof ResponseEvent) { 
            throw new \InvalidArgumentException(\sprintf('Expected instance of type %s, %s given', \class_exists(ResponseEvent::class) ? ResponseEvent::class : FilterResponseEvent::class, \is_object($e) ? \get_class($e) : \gettype($e))); 
        } 
 
        if (HttpKernelInterface::MASTER_REQUEST !== $e->getRequestType()) { 
            return; 
        } 
 
        $response = $e->getResponse(); 
 
        $response->headers->set('Referrer-Policy', implode(', ', $this->policies)); 
    } 
}