Session.php 21 KB
Newer Older
Qiang Xue committed
1 2 3
<?php
/**
 * @link http://www.yiiframework.com/
Qiang Xue committed
4
 * @copyright Copyright (c) 2008 Yii Software LLC
Qiang Xue committed
5 6 7
 * @license http://www.yiiframework.com/license/
 */

Qiang Xue committed
8 9
namespace yii\web;

Qiang Xue committed
10
use Yii;
Qiang Xue committed
11 12 13
use yii\base\Component;
use yii\base\InvalidParamException;

Qiang Xue committed
14
/**
Qiang Xue committed
15
 * Session provides session data management and the related configurations.
Qiang Xue committed
16
 *
Qiang Xue committed
17
 * Session is a Web application component that can be accessed via `Yii::$app->session`.
18
 *
Qiang Xue committed
19 20
 * To start the session, call [[open()]]; To complete and send out session data, call [[close()]];
 * To destroy the session, call [[destroy()]].
Qiang Xue committed
21
 *
Qiang Xue committed
22 23
 * By default, [[autoStart]] is true which means the session will be started automatically
 * when the session component is accessed the first time.
Qiang Xue committed
24
 *
Qiang Xue committed
25
 * Session can be used like an array to set and get session data. For example,
Qiang Xue committed
26
 *
Qiang Xue committed
27 28 29 30 31 32 33 34
 * ~~~
 * $session = new Session;
 * $session->open();
 * $value1 = $session['name1'];  // get session variable 'name1'
 * $value2 = $session['name2'];  // get session variable 'name2'
 * foreach ($session as $name => $value) // traverse all session variables
 * $session['name3'] = $value3;  // set session variable 'name3'
 * ~~~
Qiang Xue committed
35
 *
Qiang Xue committed
36
 * Session can be extended to support customized session storage.
Qiang Xue committed
37 38 39 40
 * To do so, override [[useCustomStorage()]] so that it returns true, and
 * override these methods with the actual logic about using custom storage:
 * [[openSession()]], [[closeSession()]], [[readSession()]], [[writeSession()]],
 * [[destroySession()]] and [[gcSession()]].
Qiang Xue committed
41
 *
Qiang Xue committed
42 43 44 45 46
 * Session also supports a special type of session data, called *flash messages*.
 * A flash message is available only in the current request and the next request.
 * After that, it will be deleted automatically. Flash messages are particularly
 * useful for displaying confirmation messages. To use flash messages, simply
 * call methods such as [[setFlash()]], [[getFlash()]].
Qiang Xue committed
47
 *
48
 * @property array $allFlashes Flash messages (key => message). This property is read-only.
Qiang Xue committed
49
 * @property array $cookieParams The session cookie parameters. This property is read-only.
50 51 52 53 54 55
 * @property integer $count The number of session variables. This property is read-only.
 * @property string $flash The key identifying the flash message. Note that flash messages and normal session
 * variables share the same name space. If you have a normal session variable using the same name, its value will
 * be overwritten by this method. This property is write-only.
 * @property float $gCProbability The probability (percentage) that the GC (garbage collection) process is
 * started on every session initialization, defaults to 1 meaning 1% chance.
56
 * @property string $id The current session ID.
57 58 59
 * @property boolean $isActive Whether the session has started. This property is read-only.
 * @property SessionIterator $iterator An iterator for traversing the session variables. This property is
 * read-only.
60
 * @property string $name The current session name.
61 62 63
 * @property string $savePath The current session save path, defaults to '/tmp'.
 * @property integer $timeout The number of seconds after which data will be seen as 'garbage' and cleaned up.
 * The default value is 1440 seconds (or the value of "session.gc_maxlifetime" set in php.ini).
64 65 66 67 68
 * @property boolean|null $useCookies The value indicating whether cookies should be used to store session
 * IDs.
 * @property boolean $useCustomStorage Whether to use custom storage. This property is read-only.
 * @property boolean $useTransparentSessionID Whether transparent sid support is enabled or not, defaults to
 * false.
69
 *
Qiang Xue committed
70
 * @author Qiang Xue <qiang.xue@gmail.com>
Qiang Xue committed
71
 * @since 2.0
Qiang Xue committed
72
 */
Qiang Xue committed
73
class Session extends Component implements \IteratorAggregate, \ArrayAccess, \Countable
Qiang Xue committed
74 75
{
	/**
Qiang Xue committed
76
	 * @var boolean whether the session should be automatically started when the session component is initialized.
Qiang Xue committed
77
	 */
Qiang Xue committed
78
	public $autoStart = true;
Qiang Xue committed
79 80 81 82
	/**
	 * @var string the name of the session variable that stores the flash message data.
	 */
	public $flashVar = '__flash';
83
	/**
84
	 * @var array parameter-value pairs to override default session cookie parameters that are used for session_set_cookie_params() function
85
	 * Array may have the following possible keys: 'lifetime', 'path', 'domain', 'secure', 'httpOnly'
86
	 * @see http://www.php.net/manual/en/function.session-set-cookie-params.php
87
	 */
88
	private $_cookieParams = ['httpOnly' => true];
89

Qiang Xue committed
90 91 92 93 94 95 96
	/**
	 * Initializes the application component.
	 * This method is required by IApplicationComponent and is invoked by application.
	 */
	public function init()
	{
		parent::init();
Qiang Xue committed
97
		if ($this->autoStart) {
Qiang Xue committed
98
			$this->open();
Qiang Xue committed
99
		}
Alexander Makarov committed
100
		register_shutdown_function([$this, 'close']);
Qiang Xue committed
101 102 103 104
	}

	/**
	 * Returns a value indicating whether to use custom session storage.
Qiang Xue committed
105 106 107
	 * This method should be overridden to return true by child classes that implement custom session storage.
	 * To implement custom session storage, override these methods: [[openSession()]], [[closeSession()]],
	 * [[readSession()]], [[writeSession()]], [[destroySession()]] and [[gcSession()]].
Qiang Xue committed
108 109 110 111 112 113 114 115
	 * @return boolean whether to use custom storage.
	 */
	public function getUseCustomStorage()
	{
		return false;
	}

	/**
Qiang Xue committed
116
	 * Starts the session.
Qiang Xue committed
117 118 119
	 */
	public function open()
	{
120 121
		if (session_status() == PHP_SESSION_ACTIVE) {
			return;
Qiang Xue committed
122 123
		}

124 125 126 127 128 129 130 131 132 133
		if ($this->getUseCustomStorage()) {
			@session_set_save_handler(
				[$this, 'openSession'],
				[$this, 'closeSession'],
				[$this, 'readSession'],
				[$this, 'writeSession'],
				[$this, 'destroySession'],
				[$this, 'gcSession']
			);
		}
Qiang Xue committed
134

135
		$this->setCookieParamsInternal();
136

137
		@session_start();
Qiang Xue committed
138

139 140 141 142 143 144
		if (session_id() == '') {
			$error = error_get_last();
			$message = isset($error['message']) ? $error['message'] : 'Failed to start session.';
			Yii::error($message, __METHOD__);
		} else {
			$this->updateFlashCounters();
Qiang Xue committed
145 146 147 148 149 150 151 152
		}
	}

	/**
	 * Ends the current session and store session data.
	 */
	public function close()
	{
Qiang Xue committed
153
		if (session_id() !== '') {
Qiang Xue committed
154
			@session_write_close();
Qiang Xue committed
155
		}
Qiang Xue committed
156 157 158 159 160 161 162
	}

	/**
	 * Frees all session variables and destroys all data registered to a session.
	 */
	public function destroy()
	{
Qiang Xue committed
163
		if (session_id() !== '') {
Qiang Xue committed
164 165 166 167 168 169 170 171
			@session_unset();
			@session_destroy();
		}
	}

	/**
	 * @return boolean whether the session has started
	 */
Qiang Xue committed
172
	public function getIsActive()
Qiang Xue committed
173
	{
174
		return session_status() == PHP_SESSION_ACTIVE;
Qiang Xue committed
175 176 177 178 179
	}

	/**
	 * @return string the current session ID
	 */
Qiang Xue committed
180
	public function getId()
Qiang Xue committed
181 182 183 184 185 186 187
	{
		return session_id();
	}

	/**
	 * @param string $value the session ID for the current session
	 */
Qiang Xue committed
188
	public function setId($value)
Qiang Xue committed
189 190 191 192 193
	{
		session_id($value);
	}

	/**
Qiang Xue committed
194 195
	 * Updates the current session ID with a newly generated one .
	 * Please refer to [[http://php.net/session_regenerate_id]] for more details.
Qiang Xue committed
196 197
	 * @param boolean $deleteOldSession Whether to delete the old associated session file or not.
	 */
Qiang Xue committed
198
	public function regenerateID($deleteOldSession = false)
Qiang Xue committed
199 200 201 202 203 204 205
	{
		session_regenerate_id($deleteOldSession);
	}

	/**
	 * @return string the current session name
	 */
Qiang Xue committed
206
	public function getName()
Qiang Xue committed
207 208 209 210 211
	{
		return session_name();
	}

	/**
Qiang Xue committed
212 213
	 * @param string $value the session name for the current session, must be an alphanumeric string.
	 * It defaults to "PHPSESSID".
Qiang Xue committed
214
	 */
Qiang Xue committed
215
	public function setName($value)
Qiang Xue committed
216 217 218 219 220 221 222 223 224 225 226 227 228
	{
		session_name($value);
	}

	/**
	 * @return string the current session save path, defaults to '/tmp'.
	 */
	public function getSavePath()
	{
		return session_save_path();
	}

	/**
Qiang Xue committed
229 230
	 * @param string $value the current session save path. This can be either a directory name or a path alias.
	 * @throws InvalidParamException if the path is not a valid directory
Qiang Xue committed
231 232 233
	 */
	public function setSavePath($value)
	{
Qiang Xue committed
234
		$path = Yii::getAlias($value);
Qiang Xue committed
235
		if (is_dir($path)) {
Qiang Xue committed
236
			session_save_path($path);
Qiang Xue committed
237
		} else {
Qiang Xue committed
238
			throw new InvalidParamException("Session save path is not a valid directory: $value");
Qiang Xue committed
239
		}
Qiang Xue committed
240 241 242 243 244 245 246 247
	}

	/**
	 * @return array the session cookie parameters.
	 * @see http://us2.php.net/manual/en/function.session-get-cookie-params.php
	 */
	public function getCookieParams()
	{
Qiang Xue committed
248 249 250 251 252
		$params = session_get_cookie_params();
		if (isset($params['httponly'])) {
			$params['httpOnly'] = $params['httponly'];
			unset($params['httponly']);
		}
253
		return array_merge($params, $this->_cookieParams);
Qiang Xue committed
254 255 256 257
	}

	/**
	 * Sets the session cookie parameters.
258 259
	 * The cookie parameters passed to this method will be merged with the result
	 * of `session_get_cookie_params()`.
Qiang Xue committed
260
	 * @param array $value cookie parameters, valid keys include: `lifetime`, `path`, `domain`, `secure` and `httpOnly`.
Qiang Xue committed
261
	 * @throws InvalidParamException if the parameters are incomplete.
Qiang Xue committed
262 263
	 * @see http://us2.php.net/manual/en/function.session-set-cookie-params.php
	 */
264 265 266 267 268 269 270 271 272 273 274 275
	public function setCookieParams(array $value)
	{
		$this->_cookieParams = $value;
	}

	/**
	 * Sets the session cookie parameters.
	 * This method is called by [[open()]] when it is about to open the session.
	 * @throws InvalidParamException if the parameters are incomplete.
	 * @see http://us2.php.net/manual/en/function.session-set-cookie-params.php
	 */
	private function setCookieParamsInternal()
Qiang Xue committed
276
	{
Qiang Xue committed
277
		$data = $this->getCookieParams();
Qiang Xue committed
278
		extract($data);
Qiang Xue committed
279 280
		if (isset($lifetime, $path, $domain, $secure, $httpOnly)) {
			session_set_cookie_params($lifetime, $path, $domain, $secure, $httpOnly);
Qiang Xue committed
281
		} else {
282
			throw new InvalidParamException('Please make sure cookieParams contains these elements: lifetime, path, domain, secure and httpOnly.');
Qiang Xue committed
283
		}
Qiang Xue committed
284 285 286
	}

	/**
Qiang Xue committed
287 288 289
	 * Returns the value indicating whether cookies should be used to store session IDs.
	 * @return boolean|null the value indicating whether cookies should be used to store session IDs.
	 * @see setUseCookies()
Qiang Xue committed
290
	 */
Qiang Xue committed
291
	public function getUseCookies()
Qiang Xue committed
292
	{
Qiang Xue committed
293
		if (ini_get('session.use_cookies') === '0') {
Qiang Xue committed
294 295 296
			return false;
		} elseif (ini_get('session.use_only_cookies') === '1') {
			return true;
Qiang Xue committed
297
		} else {
Qiang Xue committed
298
			return null;
Qiang Xue committed
299
		}
Qiang Xue committed
300 301 302
	}

	/**
Qiang Xue committed
303 304 305 306 307 308 309 310
	 * Sets the value indicating whether cookies should be used to store session IDs.
	 * Three states are possible:
	 *
	 * - true: cookies and only cookies will be used to store session IDs.
	 * - false: cookies will not be used to store session IDs.
	 * - null: if possible, cookies will be used to store session IDs; if not, other mechanisms will be used (e.g. GET parameter)
	 *
	 * @param boolean|null $value the value indicating whether cookies should be used to store session IDs.
Qiang Xue committed
311
	 */
Qiang Xue committed
312
	public function setUseCookies($value)
Qiang Xue committed
313
	{
Qiang Xue committed
314
		if ($value === false) {
Qiang Xue committed
315 316
			ini_set('session.use_cookies', '0');
			ini_set('session.use_only_cookies', '0');
Qiang Xue committed
317 318 319
		} elseif ($value === true) {
			ini_set('session.use_cookies', '1');
			ini_set('session.use_only_cookies', '1');
Qiang Xue committed
320
		} else {
Qiang Xue committed
321 322
			ini_set('session.use_cookies', '1');
			ini_set('session.use_only_cookies', '0');
Qiang Xue committed
323 324 325 326
		}
	}

	/**
Qiang Xue committed
327
	 * @return float the probability (percentage) that the GC (garbage collection) process is started on every session initialization, defaults to 1 meaning 1% chance.
Qiang Xue committed
328 329 330
	 */
	public function getGCProbability()
	{
Qiang Xue committed
331
		return (float)(ini_get('session.gc_probability') / ini_get('session.gc_divisor') * 100);
Qiang Xue committed
332 333 334
	}

	/**
Qiang Xue committed
335 336
	 * @param float $value the probability (percentage) that the GC (garbage collection) process is started on every session initialization.
	 * @throws InvalidParamException if the value is not between 0 and 100.
Qiang Xue committed
337 338 339
	 */
	public function setGCProbability($value)
	{
Qiang Xue committed
340
		if ($value >= 0 && $value <= 100) {
Qiang Xue committed
341 342 343
			// percent * 21474837 / 2147483647 ≈ percent * 0.01
			ini_set('session.gc_probability', floor($value * 21474836.47));
			ini_set('session.gc_divisor', 2147483647);
Qiang Xue committed
344
		} else {
Qiang Xue committed
345
			throw new InvalidParamException('GCProbability must be a value between 0 and 100.');
Qiang Xue committed
346
		}
Qiang Xue committed
347 348 349 350 351 352 353
	}

	/**
	 * @return boolean whether transparent sid support is enabled or not, defaults to false.
	 */
	public function getUseTransparentSessionID()
	{
Qiang Xue committed
354
		return ini_get('session.use_trans_sid') == 1;
Qiang Xue committed
355 356 357 358 359 360 361
	}

	/**
	 * @param boolean $value whether transparent sid support is enabled or not.
	 */
	public function setUseTransparentSessionID($value)
	{
Qiang Xue committed
362
		ini_set('session.use_trans_sid', $value ? '1' : '0');
Qiang Xue committed
363 364 365
	}

	/**
Qiang Xue committed
366 367
	 * @return integer the number of seconds after which data will be seen as 'garbage' and cleaned up.
	 * The default value is 1440 seconds (or the value of "session.gc_maxlifetime" set in php.ini).
Qiang Xue committed
368 369 370 371 372 373 374 375 376 377 378
	 */
	public function getTimeout()
	{
		return (int)ini_get('session.gc_maxlifetime');
	}

	/**
	 * @param integer $value the number of seconds after which data will be seen as 'garbage' and cleaned up
	 */
	public function setTimeout($value)
	{
Qiang Xue committed
379
		ini_set('session.gc_maxlifetime', $value);
Qiang Xue committed
380 381 382 383
	}

	/**
	 * Session open handler.
Qiang Xue committed
384
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
385 386 387 388 389
	 * Do not call this method directly.
	 * @param string $savePath session save path
	 * @param string $sessionName session name
	 * @return boolean whether session is opened successfully
	 */
Qiang Xue committed
390
	public function openSession($savePath, $sessionName)
Qiang Xue committed
391 392 393 394 395 396
	{
		return true;
	}

	/**
	 * Session close handler.
Qiang Xue committed
397
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
398 399 400 401 402 403 404 405 406 407
	 * Do not call this method directly.
	 * @return boolean whether session is closed successfully
	 */
	public function closeSession()
	{
		return true;
	}

	/**
	 * Session read handler.
Qiang Xue committed
408
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
409 410 411 412 413 414 415 416 417 418 419
	 * Do not call this method directly.
	 * @param string $id session ID
	 * @return string the session data
	 */
	public function readSession($id)
	{
		return '';
	}

	/**
	 * Session write handler.
Qiang Xue committed
420
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
421 422 423 424 425
	 * Do not call this method directly.
	 * @param string $id session ID
	 * @param string $data session data
	 * @return boolean whether session write is successful
	 */
Qiang Xue committed
426
	public function writeSession($id, $data)
Qiang Xue committed
427 428 429 430 431 432
	{
		return true;
	}

	/**
	 * Session destroy handler.
Qiang Xue committed
433
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
434 435 436 437 438 439 440 441 442 443 444
	 * Do not call this method directly.
	 * @param string $id session ID
	 * @return boolean whether session is destroyed successfully
	 */
	public function destroySession($id)
	{
		return true;
	}

	/**
	 * Session GC (garbage collection) handler.
Qiang Xue committed
445
	 * This method should be overridden if [[useCustomStorage()]] returns true.
Qiang Xue committed
446 447 448 449 450 451 452 453 454 455 456 457
	 * Do not call this method directly.
	 * @param integer $maxLifetime the number of seconds after which data will be seen as 'garbage' and cleaned up.
	 * @return boolean whether session is GCed successfully
	 */
	public function gcSession($maxLifetime)
	{
		return true;
	}

	/**
	 * Returns an iterator for traversing the session variables.
	 * This method is required by the interface IteratorAggregate.
Qiang Xue committed
458
	 * @return SessionIterator an iterator for traversing the session variables.
Qiang Xue committed
459 460 461
	 */
	public function getIterator()
	{
Qiang Xue committed
462
		return new SessionIterator;
Qiang Xue committed
463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485
	}

	/**
	 * Returns the number of items in the session.
	 * @return integer the number of session variables
	 */
	public function getCount()
	{
		return count($_SESSION);
	}

	/**
	 * Returns the number of items in the session.
	 * This method is required by Countable interface.
	 * @return integer number of items in the session.
	 */
	public function count()
	{
		return $this->getCount();
	}

	/**
	 * Returns the session variable value with the session variable name.
Qiang Xue committed
486 487
	 * If the session variable does not exist, the `$defaultValue` will be returned.
	 * @param string $key the session variable name
Qiang Xue committed
488 489 490
	 * @param mixed $defaultValue the default value to be returned when the session variable does not exist.
	 * @return mixed the session variable value, or $defaultValue if the session variable does not exist.
	 */
Qiang Xue committed
491
	public function get($key, $defaultValue = null)
Qiang Xue committed
492 493 494 495 496 497
	{
		return isset($_SESSION[$key]) ? $_SESSION[$key] : $defaultValue;
	}

	/**
	 * Adds a session variable.
Qiang Xue committed
498 499
	 * If the specified name already exists, the old value will be overwritten.
	 * @param string $key session variable name
Qiang Xue committed
500 501
	 * @param mixed $value session variable value
	 */
Qiang Xue committed
502
	public function set($key, $value)
Qiang Xue committed
503
	{
Qiang Xue committed
504
		$_SESSION[$key] = $value;
Qiang Xue committed
505 506 507 508
	}

	/**
	 * Removes a session variable.
Qiang Xue committed
509
	 * @param string $key the name of the session variable to be removed
Qiang Xue committed
510 511 512 513
	 * @return mixed the removed value, null if no such session variable.
	 */
	public function remove($key)
	{
Qiang Xue committed
514 515
		if (isset($_SESSION[$key])) {
			$value = $_SESSION[$key];
Qiang Xue committed
516 517
			unset($_SESSION[$key]);
			return $value;
Qiang Xue committed
518
		} else {
Qiang Xue committed
519
			return null;
Qiang Xue committed
520
		}
Qiang Xue committed
521 522 523 524 525
	}

	/**
	 * Removes all session variables
	 */
Qiang Xue committed
526
	public function removeAll()
Qiang Xue committed
527
	{
Qiang Xue committed
528
		foreach (array_keys($_SESSION) as $key) {
Qiang Xue committed
529
			unset($_SESSION[$key]);
Qiang Xue committed
530
		}
Qiang Xue committed
531 532 533 534 535 536
	}

	/**
	 * @param mixed $key session variable name
	 * @return boolean whether there is the named session variable
	 */
Qiang Xue committed
537
	public function has($key)
Qiang Xue committed
538 539 540 541 542 543 544 545 546 547 548 549
	{
		return isset($_SESSION[$key]);
	}

	/**
	 * @return array the list of all session variables in array
	 */
	public function toArray()
	{
		return $_SESSION;
	}

Qiang Xue committed
550 551 552 553 554 555
	/**
	 * Updates the counters for flash messages and removes outdated flash messages.
	 * This method should only be called once in [[init()]].
	 */
	protected function updateFlashCounters()
	{
Alexander Makarov committed
556
		$counters = $this->get($this->flashVar, []);
Qiang Xue committed
557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576
		if (is_array($counters)) {
			foreach ($counters as $key => $count) {
				if ($count) {
					unset($counters[$key], $_SESSION[$key]);
				} else {
					$counters[$key]++;
				}
			}
			$_SESSION[$this->flashVar] = $counters;
		} else {
			// fix the unexpected problem that flashVar doesn't return an array
			unset($_SESSION[$this->flashVar]);
		}
	}

	/**
	 * Returns a flash message.
	 * A flash message is available only in the current request and the next request.
	 * @param string $key the key identifying the flash message
	 * @param mixed $defaultValue value to be returned if the flash message does not exist.
577 578
	 * @param boolean $delete whether to delete this flash message right after this method is called.
	 * If false, the flash message will be automatically deleted after the next request.
Qiang Xue committed
579 580
	 * @return mixed the flash message
	 */
581
	public function getFlash($key, $defaultValue = null, $delete = false)
Qiang Xue committed
582
	{
Alexander Makarov committed
583
		$counters = $this->get($this->flashVar, []);
584 585 586 587 588 589 590 591 592
		if (isset($counters[$key])) {
			$value = $this->get($key, $defaultValue);
			if ($delete) {
				$this->removeFlash($key);
			}
			return $value;
		} else {
			return $defaultValue;
		}
Qiang Xue committed
593 594 595 596 597 598 599 600
	}

	/**
	 * Returns all flash messages.
	 * @return array flash messages (key => message).
	 */
	public function getAllFlashes()
	{
Alexander Makarov committed
601 602
		$counters = $this->get($this->flashVar, []);
		$flashes = [];
Qiang Xue committed
603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618
		foreach (array_keys($counters) as $key) {
			if (isset($_SESSION[$key])) {
				$flashes[$key] = $_SESSION[$key];
			}
		}
		return $flashes;
	}

	/**
	 * Stores a flash message.
	 * A flash message is available only in the current request and the next request.
	 * @param string $key the key identifying the flash message. Note that flash messages
	 * and normal session variables share the same name space. If you have a normal
	 * session variable using the same name, its value will be overwritten by this method.
	 * @param mixed $value flash message
	 */
Qiang Xue committed
619
	public function setFlash($key, $value = true)
Qiang Xue committed
620
	{
Alexander Makarov committed
621
		$counters = $this->get($this->flashVar, []);
Qiang Xue committed
622 623 624 625 626 627 628 629 630 631 632 633 634 635 636
		$counters[$key] = 0;
		$_SESSION[$key] = $value;
		$_SESSION[$this->flashVar] = $counters;
	}

	/**
	 * Removes a flash message.
	 * Note that flash messages will be automatically removed after the next request.
	 * @param string $key the key identifying the flash message. Note that flash messages
	 * and normal session variables share the same name space.  If you have a normal
	 * session variable using the same name, it will be removed by this method.
	 * @return mixed the removed flash message. Null if the flash message does not exist.
	 */
	public function removeFlash($key)
	{
Alexander Makarov committed
637
		$counters = $this->get($this->flashVar, []);
Qiang Xue committed
638 639 640 641 642 643 644 645 646 647 648 649 650 651
		$value = isset($_SESSION[$key], $counters[$key]) ? $_SESSION[$key] : null;
		unset($counters[$key], $_SESSION[$key]);
		$_SESSION[$this->flashVar] = $counters;
		return $value;
	}

	/**
	 * Removes all flash messages.
	 * Note that flash messages and normal session variables share the same name space.
	 * If you have a normal session variable using the same name, it will be removed
	 * by this method.
	 */
	public function removeAllFlashes()
	{
Alexander Makarov committed
652
		$counters = $this->get($this->flashVar, []);
Qiang Xue committed
653 654 655 656 657 658 659
		foreach (array_keys($counters) as $key) {
			unset($_SESSION[$key]);
		}
		unset($_SESSION[$this->flashVar]);
	}

	/**
Qiang Xue committed
660
	 * Returns a value indicating whether there is a flash message associated with the specified key.
Qiang Xue committed
661 662 663 664 665 666 667 668
	 * @param string $key key identifying the flash message
	 * @return boolean whether the specified flash message exists
	 */
	public function hasFlash($key)
	{
		return $this->getFlash($key) !== null;
	}

Qiang Xue committed
669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693
	/**
	 * This method is required by the interface ArrayAccess.
	 * @param mixed $offset the offset to check on
	 * @return boolean
	 */
	public function offsetExists($offset)
	{
		return isset($_SESSION[$offset]);
	}

	/**
	 * This method is required by the interface ArrayAccess.
	 * @param integer $offset the offset to retrieve element.
	 * @return mixed the element at the offset, null if no element is found at the offset
	 */
	public function offsetGet($offset)
	{
		return isset($_SESSION[$offset]) ? $_SESSION[$offset] : null;
	}

	/**
	 * This method is required by the interface ArrayAccess.
	 * @param integer $offset the offset to set element
	 * @param mixed $item the element value
	 */
Qiang Xue committed
694
	public function offsetSet($offset, $item)
Qiang Xue committed
695
	{
Qiang Xue committed
696
		$_SESSION[$offset] = $item;
Qiang Xue committed
697 698 699 700 701 702 703 704 705 706 707
	}

	/**
	 * This method is required by the interface ArrayAccess.
	 * @param mixed $offset the offset to unset element
	 */
	public function offsetUnset($offset)
	{
		unset($_SESSION[$offset]);
	}
}