ResponseCacheStrategy.php

<?php

/*
 * This file is part of the Symfony package.
 *
 * (c) Fabien Potencier <fabien@symfony.com>
 *
 * This code is partially based on the Rack-Cache library by Ryan Tomayko,
 * which is released under the MIT license.
 * (based on commit 02d2b48d75bcb63cf1c0c7149c077ad256542801)
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Component\HttpKernel\HttpCache;

use Symfony\Component\HttpFoundation\Response;

/**
 * ResponseCacheStrategy knows how to compute the Response cache HTTP header
 * based on the different response cache headers.
 *
 * This implementation changes the master response TTL to the smallest TTL received
 * or force validation if one of the surrogates has validation cache strategy.
 *
 * @author Fabien Potencier <fabien@symfony.com>
 */
class ResponseCacheStrategy implements ResponseCacheStrategyInterface
{
    /**
     * Cache-Control headers that are sent to the final response if they appear in ANY of the responses.
     */
    private static $overrideDirectives = array('no-cache', 'no-store', 'no-transform', 'must-revalidate', 'proxy-revalidate');

    /**
     * Cache-Control headers that are sent to the final response if the appear in ALL of the responses.
     */
    private static $inheritDirectives = array('immutable');

    private $embeddedResponses = 0;
    private $isNotCacheableResponseEmbedded = false;
    private $age = 0;
    private $flagDirectives = array(
        'no-cache' => null,
        'no-store' => null,
        'no-transform' => null,
        'must-revalidate' => null,
        'proxy-revalidate' => null,
        'public' => null,
        'private' => null,
        'immutable' => null,
    );
    private $ageDirectives = array(
        'max-age' => null,
        's-maxage' => null,
        'expires' => null,
    );

    /**
     * {@inheritdoc}
     */
    public function add(Response $response)
    {
        ++$this->embeddedResponses;

        foreach (self::$overrideDirectives as $directive) {
            if ($response->headers->hasCacheControlDirective($directive)) {
                $this->flagDirectives[$directive] = true;
            }
        }

        foreach (self::$inheritDirectives as $directive) {
            if (false !== $this->flagDirectives[$directive]) {
                $this->flagDirectives[$directive] = $response->headers->hasCacheControlDirective($directive);
            }
        }

        if (false !== $this->flagDirectives['public']) {
            $this->flagDirectives['public'] = $response->headers->hasCacheControlDirective('public');
        }

        if (false !== $this->flagDirectives['private']) {
            $this->flagDirectives['private'] = $response->headers->hasCacheControlDirective('private') || $response->headers->hasCacheControlDirective('public');
        }

        $age = $response->getAge();
        $this->age = max($this->age, $age);

        if ($this->isUncacheable($response)) {
            $this->isNotCacheableResponseEmbedded = true;
            return;
        }

        $this->storeRelativeAgeDirective('max-age', $response->headers->getCacheControlDirective('max-age'), $age);
        $this->storeRelativeAgeDirective('s-maxage', $response->headers->getCacheControlDirective('s-maxage'), $age);

        $expires = $response->getExpires();
        $expires = null !== $expires ? $expires->format('U') - $response->getDate()->format('U') : null;
        $this->storeRelativeAgeDirective('expires', $expires >= 0 ? $expires : null, 0);
    }

    /**
     * {@inheritdoc}
     */
    public function update(Response $response)
    {
        // if we have no embedded Response, do nothing
        if (0 === $this->embeddedResponses) {
            return;
        }

        // Remove validation related headers of the master response,
        // because some of the response content comes from at least
        // one embedded response (which likely has a different caching strategy).
        $response->setEtag(null);
        $response->setLastModified(null);

        $this->add($response);

        $response->headers->set('Age', $this->age);

        if ($this->isNotCacheableResponseEmbedded) {
            $response->setExpires($response->getDate());

            if ($this->flagDirectives['no-store']) {
                $response->headers->set('Cache-Control', 'no-cache, no-store');
            } else {
                $response->headers->set('Cache-Control', 'no-cache');
            }

            return;
        }

        $flags = array_filter($this->flagDirectives);

        if (isset($flags['public'], $flags['private'])) {
            unset($flags['private']);
        }

        $response->headers->set('Cache-Control', implode(', ', array_keys($flags)));

        if ($this->ageDirectives['max-age']) {
            $response->headers->addCacheControlDirective('max-age', $this->ageDirectives['max-age'] + $this->age);
        }

        if ($this->ageDirectives['s-maxage']) {
            $response->headers->addCacheControlDirective('s-maxage', $this->ageDirectives['s-maxage'] + $this->age);
        }

        if ($this->ageDirectives['expires']) {
            $response->setExpires($response->getDate()->modify('+'.$this->ageDirectives['expires'].' seconds'));
        }
    }

    /**
     * RFC2616, Section 13.4
     *
     * @see https://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4
     *
     * @return bool
     */
    private function isUncacheable(Response $response)
    {
        // A response received with a status code of 200, 203, 300, 301 or 410
        // MAY be stored by a cache […] unless a cache-control directive prohibits caching.
        if ($response->headers->hasCacheControlDirective('no-cache') || $response->headers->getCacheControlDirective('no-store')) {
            return true;
        }

        if (in_array($response->getStatusCode(), array(200, 203, 300, 301, 410))) {
            return false;
        }

        // A response received with any other status code (e.g. status codes 302 and 307)
        // MUST NOT be returned in a reply to a subsequent request unless there are
        // cache-control directives or another header(s) that explicitly allow it.
        $cacheControl = array('max-age', 's-maxage', 'must-revalidate', 'proxy-revalidate', 'public', 'private', 'must-revalidate');
        foreach ($cacheControl as $key) {
            if ($response->headers->hasCacheControlDirective($key)) {
                return false;
            }
        }

        if ($response->headers->has('Expires')) {
            return false;
        }

        return true;
    }

    /**
     * Stores the cache directive relative to the age of the response.
     *
     * @param string   $cacheKey
     * @param int|null $value
     * @param int      $age
     */
    private function storeRelativeAgeDirective($cacheKey, $value, $age)
    {
        if (null === $value) {
            $this->ageDirectives[$cacheKey] = false;
        }

        if (false !== $this->ageDirectives[$cacheKey]) {
            $value = $value - $age;
            $this->ageDirectives[$cacheKey] = null !== $this->ageDirectives[$cacheKey] ? min($this->ageDirectives[$cacheKey], $value) : $value;
        }
    }
}