123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256 |
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Component\BrowserKit;
- /**
- * CookieJar.
- *
- * @author Fabien Potencier <fabien@symfony.com>
- */
- class CookieJar
- {
- protected $cookieJar = array();
- /**
- * Sets a cookie.
- *
- * @param Cookie $cookie A Cookie instance
- */
- public function set(Cookie $cookie)
- {
- $this->cookieJar[$cookie->getDomain()][$cookie->getPath()][$cookie->getName()] = $cookie;
- }
- /**
- * Gets a cookie by name.
- *
- * You should never use an empty domain, but if you do so,
- * this method returns the first cookie for the given name/path
- * (this behavior ensures a BC behavior with previous versions of
- * Symfony).
- *
- * @param string $name The cookie name
- * @param string $path The cookie path
- * @param string $domain The cookie domain
- *
- * @return Cookie|null A Cookie instance or null if the cookie does not exist
- */
- public function get($name, $path = '/', $domain = null)
- {
- $this->flushExpiredCookies();
- if (!empty($domain)) {
- foreach ($this->cookieJar as $cookieDomain => $pathCookies) {
- if ($cookieDomain) {
- $cookieDomain = '.'.ltrim($cookieDomain, '.');
- if ($cookieDomain != substr('.'.$domain, -strlen($cookieDomain))) {
- continue;
- }
- }
- foreach ($pathCookies as $cookiePath => $namedCookies) {
- if ($cookiePath != substr($path, 0, strlen($cookiePath))) {
- continue;
- }
- if (isset($namedCookies[$name])) {
- return $namedCookies[$name];
- }
- }
- }
- return;
- }
- // avoid relying on this behavior that is mainly here for BC reasons
- foreach ($this->cookieJar as $cookies) {
- if (isset($cookies[$path][$name])) {
- return $cookies[$path][$name];
- }
- }
- }
- /**
- * Removes a cookie by name.
- *
- * You should never use an empty domain, but if you do so,
- * all cookies for the given name/path expire (this behavior
- * ensures a BC behavior with previous versions of Symfony).
- *
- * @param string $name The cookie name
- * @param string $path The cookie path
- * @param string $domain The cookie domain
- */
- public function expire($name, $path = '/', $domain = null)
- {
- if (null === $path) {
- $path = '/';
- }
- if (empty($domain)) {
- // an empty domain means any domain
- // this should never happen but it allows for a better BC
- $domains = array_keys($this->cookieJar);
- } else {
- $domains = array($domain);
- }
- foreach ($domains as $domain) {
- unset($this->cookieJar[$domain][$path][$name]);
- if (empty($this->cookieJar[$domain][$path])) {
- unset($this->cookieJar[$domain][$path]);
- if (empty($this->cookieJar[$domain])) {
- unset($this->cookieJar[$domain]);
- }
- }
- }
- }
- /**
- * Removes all the cookies from the jar.
- */
- public function clear()
- {
- $this->cookieJar = array();
- }
- /**
- * Updates the cookie jar from a response Set-Cookie headers.
- *
- * @param array $setCookies Set-Cookie headers from an HTTP response
- * @param string $uri The base URL
- */
- public function updateFromSetCookie(array $setCookies, $uri = null)
- {
- $cookies = array();
- foreach ($setCookies as $cookie) {
- foreach (explode(',', $cookie) as $i => $part) {
- if (0 === $i || preg_match('/^(?P<token>\s*[0-9A-Za-z!#\$%\&\'\*\+\-\.^_`\|~]+)=/', $part)) {
- $cookies[] = ltrim($part);
- } else {
- $cookies[count($cookies) - 1] .= ','.$part;
- }
- }
- }
- foreach ($cookies as $cookie) {
- try {
- $this->set(Cookie::fromString($cookie, $uri));
- } catch (\InvalidArgumentException $e) {
- // invalid cookies are just ignored
- }
- }
- }
- /**
- * Updates the cookie jar from a Response object.
- *
- * @param Response $response A Response object
- * @param string $uri The base URL
- */
- public function updateFromResponse(Response $response, $uri = null)
- {
- $this->updateFromSetCookie($response->getHeader('Set-Cookie', false), $uri);
- }
- /**
- * Returns not yet expired cookies.
- *
- * @return Cookie[] An array of cookies
- */
- public function all()
- {
- $this->flushExpiredCookies();
- $flattenedCookies = array();
- foreach ($this->cookieJar as $path) {
- foreach ($path as $cookies) {
- foreach ($cookies as $cookie) {
- $flattenedCookies[] = $cookie;
- }
- }
- }
- return $flattenedCookies;
- }
- /**
- * Returns not yet expired cookie values for the given URI.
- *
- * @param string $uri A URI
- * @param bool $returnsRawValue Returns raw value or urldecoded value
- *
- * @return array An array of cookie values
- */
- public function allValues($uri, $returnsRawValue = false)
- {
- $this->flushExpiredCookies();
- $parts = array_replace(array('path' => '/'), parse_url($uri));
- $cookies = array();
- foreach ($this->cookieJar as $domain => $pathCookies) {
- if ($domain) {
- $domain = '.'.ltrim($domain, '.');
- if ($domain != substr('.'.$parts['host'], -strlen($domain))) {
- continue;
- }
- }
- foreach ($pathCookies as $path => $namedCookies) {
- if ($path != substr($parts['path'], 0, strlen($path))) {
- continue;
- }
- foreach ($namedCookies as $cookie) {
- if ($cookie->isSecure() && 'https' != $parts['scheme']) {
- continue;
- }
- $cookies[$cookie->getName()] = $returnsRawValue ? $cookie->getRawValue() : $cookie->getValue();
- }
- }
- }
- return $cookies;
- }
- /**
- * Returns not yet expired raw cookie values for the given URI.
- *
- * @param string $uri A URI
- *
- * @return array An array of cookie values
- */
- public function allRawValues($uri)
- {
- return $this->allValues($uri, true);
- }
- /**
- * Removes all expired cookies.
- */
- public function flushExpiredCookies()
- {
- foreach ($this->cookieJar as $domain => $pathCookies) {
- foreach ($pathCookies as $path => $namedCookies) {
- foreach ($namedCookies as $name => $cookie) {
- if ($cookie->isExpired()) {
- unset($this->cookieJar[$domain][$path][$name]);
- }
- }
- }
- }
- }
- }
|