HttpCache.php 4.05 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
<?php
/**
 * @link http://www.yiiframework.com/
 * @copyright Copyright (c) 2008 Yii Software LLC
 * @license http://www.yiiframework.com/license/
 */

namespace yii\web;

use Yii;
use yii\base\ActionFilter;
use yii\base\Action;

/**
15 16
 * The HttpCache provides functionality for caching via HTTP Last-Modified and Etag headers
 *
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
 * @author Da:Sourcerer <webmaster@dasourcerer.net>
 * @author Qiang Xue <qiang.xue@gmail.com>
 * @since 2.0
 */
class HttpCache extends ActionFilter
{
	/**
	 * @var callback a PHP callback that returns the UNIX timestamp of the last modification time.
	 * The callback's signature should be:
	 *
	 * ~~~
	 * function ($action, $params)
	 * ~~~
	 *
	 * where `$action` is the [[Action]] object that this filter is currently handling;
Qiang Xue committed
32
	 * `$params` takes the value of [[params]]. The callback should return a UNIX timestamp.
33 34 35 36 37 38 39 40 41 42 43
	 */
	public $lastModified;
	/**
	 * @var callback a PHP callback that generates the Etag seed string.
	 * The callback's signature should be:
	 *
	 * ~~~
	 * function ($action, $params)
	 * ~~~
	 *
	 * where `$action` is the [[Action]] object that this filter is currently handling;
Qiang Xue committed
44 45
	 * `$params` takes the value of [[params]]. The callback should return a string serving
	 * as the seed for generating an Etag.
46 47 48 49 50 51 52
	 */
	public $etagSeed;
	/**
	 * @var mixed additional parameters that should be passed to the [[lastModified]] and [[etagSeed]] callbacks.
	 */
	public $params;
	/**
Qiang Xue committed
53
	 * @var string HTTP cache control header. If null, the header will not be sent.
54
	 */
Qiang Xue committed
55
	public $cacheControlHeader = 'max-age=3600, public';
56 57 58 59 60 61 62 63 64

	/**
	 * This method is invoked right before an action is to be executed (after all possible filters.)
	 * You may override this method to do last-minute preparation for the action.
	 * @param Action $action the action to be executed.
	 * @return boolean whether the action should continue to be executed.
	 */
	public function beforeAction($action)
	{
Qiang Xue committed
65
		$verb = Yii::$app->getRequest()->getMethod();
Qiang Xue committed
66
		if ($verb !== 'GET' && $verb !== 'HEAD' || $this->lastModified === null && $this->etagSeed === null) {
67 68 69 70 71 72 73 74 75 76 77 78
			return true;
		}

		$lastModified = $etag = null;
		if ($this->lastModified !== null) {
			$lastModified = call_user_func($this->lastModified, $action, $this->params);
		}
		if ($this->etagSeed !== null) {
			$seed = call_user_func($this->etagSeed, $action, $this->params);
			$etag = $this->generateEtag($seed);
		}

79 80
		$this->sendCacheControlHeader();
		$response = Yii::$app->getResponse();
81
		if ($etag !== null) {
Qiang Xue committed
82
			$response->getHeaders()->set('Etag', $etag);
83 84
		}

Qiang Xue committed
85
		if ($this->validateCache($lastModified, $etag)) {
Qiang Xue committed
86
			$response->setStatusCode(304);
Qiang Xue committed
87
			return false;
Qiang Xue committed
88 89 90
		}

		if ($lastModified !== null) {
Qiang Xue committed
91
			$response->getHeaders()->set('Last-Modified', gmdate('D, d M Y H:i:s', $lastModified) . ' GMT');
92
		}
93
		return true;
94 95
	}

Qiang Xue committed
96 97 98 99 100 101 102 103
	/**
	 * Validates if the HTTP cache contains valid content.
	 * @param integer $lastModified the calculated Last-Modified value in terms of a UNIX timestamp.
	 * If null, the Last-Modified header will not be validated.
	 * @param string $etag the calculated ETag value. If null, the ETag header will not be validated.
	 * @return boolean whether the HTTP cache is still valid.
	 */
	protected function validateCache($lastModified, $etag)
104
	{
Qiang Xue committed
105 106 107 108
		if ($lastModified !== null && (!isset($_SERVER['HTTP_IF_MODIFIED_SINCE']) || @strtotime($_SERVER['HTTP_IF_MODIFIED_SINCE']) < $lastModified)) {
			return false;
		} else {
			return $etag === null || isset($_SERVER['HTTP_IF_NONE_MATCH']) && $_SERVER['HTTP_IF_NONE_MATCH'] === $etag;
109 110 111 112 113 114 115
		}
	}

	/**
	 * Sends the cache control header to the client
	 * @see cacheControl
	 */
116
	protected function sendCacheControlHeader()
117
	{
Qiang Xue committed
118
		session_cache_limiter('public');
119
		$headers = Yii::$app->getResponse()->getHeaders();
Qiang Xue committed
120
		$headers->set('Pragma');
Qiang Xue committed
121
		if ($this->cacheControlHeader !== null) {
Qiang Xue committed
122
			$headers->set('Cache-Control', $this->cacheControlHeader);
Qiang Xue committed
123
		}
124 125 126
	}

	/**
Qiang Xue committed
127
	 * Generates an Etag from the given seed string.
128 129 130 131 132 133 134
	 * @param string $seed Seed for the ETag
	 * @return string the generated Etag
	 */
	protected function generateEtag($seed)
	{
		return '"' . base64_encode(sha1($seed, true)) . '"';
	}
Zander Baldwin committed
135
}