| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239 | <?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();    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();        foreach ($this->cookieJar as $cookieDomain => $pathCookies) {            if ($cookieDomain && $domain) {                $cookieDomain = '.'.ltrim($cookieDomain, '.');                if ($cookieDomain !== substr('.'.$domain, -\strlen($cookieDomain))) {                    continue;                }            }            foreach ($pathCookies as $cookiePath => $namedCookies) {                if (0 !== strpos($path, $cookiePath)) {                    continue;                }                if (isset($namedCookies[$name])) {                    return $namedCookies[$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]);                    }                }            }        }    }}
 |