620 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			620 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?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;
 | |
| 
 | |
| use Symfony\Component\DomCrawler\Crawler;
 | |
| use Symfony\Component\DomCrawler\Form;
 | |
| use Symfony\Component\DomCrawler\Link;
 | |
| use Symfony\Component\Process\PhpProcess;
 | |
| 
 | |
| /**
 | |
|  * Client simulates a browser.
 | |
|  *
 | |
|  * To make the actual request, you need to implement the doRequest() method.
 | |
|  *
 | |
|  * If you want to be able to run requests in their own process (insulated flag),
 | |
|  * you need to also implement the getScript() method.
 | |
|  *
 | |
|  * @author Fabien Potencier <fabien@symfony.com>
 | |
|  */
 | |
| abstract class Client
 | |
| {
 | |
|     protected $history;
 | |
|     protected $cookieJar;
 | |
|     protected $server = array();
 | |
|     protected $internalRequest;
 | |
|     protected $request;
 | |
|     protected $internalResponse;
 | |
|     protected $response;
 | |
|     protected $crawler;
 | |
|     protected $insulated = false;
 | |
|     protected $redirect;
 | |
|     protected $followRedirects = true;
 | |
| 
 | |
|     private $maxRedirects = -1;
 | |
|     private $redirectCount = 0;
 | |
|     private $redirects = array();
 | |
|     private $isMainRequest = true;
 | |
| 
 | |
|     /**
 | |
|      * @param array     $server    The server parameters (equivalent of $_SERVER)
 | |
|      * @param History   $history   A History instance to store the browser history
 | |
|      * @param CookieJar $cookieJar A CookieJar instance to store the cookies
 | |
|      */
 | |
|     public function __construct(array $server = array(), History $history = null, CookieJar $cookieJar = null)
 | |
|     {
 | |
|         $this->setServerParameters($server);
 | |
|         $this->history = $history ?: new History();
 | |
|         $this->cookieJar = $cookieJar ?: new CookieJar();
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Sets whether to automatically follow redirects or not.
 | |
|      *
 | |
|      * @param bool $followRedirect Whether to follow redirects
 | |
|      */
 | |
|     public function followRedirects($followRedirect = true)
 | |
|     {
 | |
|         $this->followRedirects = (bool) $followRedirect;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns whether client automatically follows redirects or not.
 | |
|      *
 | |
|      * @return bool
 | |
|      */
 | |
|     public function isFollowingRedirects()
 | |
|     {
 | |
|         return $this->followRedirects;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Sets the maximum number of redirects that crawler can follow.
 | |
|      *
 | |
|      * @param int $maxRedirects
 | |
|      */
 | |
|     public function setMaxRedirects($maxRedirects)
 | |
|     {
 | |
|         $this->maxRedirects = $maxRedirects < 0 ? -1 : $maxRedirects;
 | |
|         $this->followRedirects = -1 != $this->maxRedirects;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the maximum number of redirects that crawler can follow.
 | |
|      *
 | |
|      * @return int
 | |
|      */
 | |
|     public function getMaxRedirects()
 | |
|     {
 | |
|         return $this->maxRedirects;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Sets the insulated flag.
 | |
|      *
 | |
|      * @param bool $insulated Whether to insulate the requests or not
 | |
|      *
 | |
|      * @throws \RuntimeException When Symfony Process Component is not installed
 | |
|      */
 | |
|     public function insulate($insulated = true)
 | |
|     {
 | |
|         if ($insulated && !class_exists('Symfony\\Component\\Process\\Process')) {
 | |
|             throw new \RuntimeException('Unable to isolate requests as the Symfony Process Component is not installed.');
 | |
|         }
 | |
| 
 | |
|         $this->insulated = (bool) $insulated;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Sets server parameters.
 | |
|      *
 | |
|      * @param array $server An array of server parameters
 | |
|      */
 | |
|     public function setServerParameters(array $server)
 | |
|     {
 | |
|         $this->server = array_merge(array(
 | |
|             'HTTP_USER_AGENT' => 'Symfony BrowserKit',
 | |
|         ), $server);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Sets single server parameter.
 | |
|      *
 | |
|      * @param string $key   A key of the parameter
 | |
|      * @param string $value A value of the parameter
 | |
|      */
 | |
|     public function setServerParameter($key, $value)
 | |
|     {
 | |
|         $this->server[$key] = $value;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Gets single server parameter for specified key.
 | |
|      *
 | |
|      * @param string $key     A key of the parameter to get
 | |
|      * @param string $default A default value when key is undefined
 | |
|      *
 | |
|      * @return string A value of the parameter
 | |
|      */
 | |
|     public function getServerParameter($key, $default = '')
 | |
|     {
 | |
|         return isset($this->server[$key]) ? $this->server[$key] : $default;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the History instance.
 | |
|      *
 | |
|      * @return History A History instance
 | |
|      */
 | |
|     public function getHistory()
 | |
|     {
 | |
|         return $this->history;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the CookieJar instance.
 | |
|      *
 | |
|      * @return CookieJar A CookieJar instance
 | |
|      */
 | |
|     public function getCookieJar()
 | |
|     {
 | |
|         return $this->cookieJar;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the current Crawler instance.
 | |
|      *
 | |
|      * @return Crawler|null A Crawler instance
 | |
|      */
 | |
|     public function getCrawler()
 | |
|     {
 | |
|         return $this->crawler;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the current BrowserKit Response instance.
 | |
|      *
 | |
|      * @return Response|null A BrowserKit Response instance
 | |
|      */
 | |
|     public function getInternalResponse()
 | |
|     {
 | |
|         return $this->internalResponse;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the current origin response instance.
 | |
|      *
 | |
|      * The origin response is the response instance that is returned
 | |
|      * by the code that handles requests.
 | |
|      *
 | |
|      * @return object|null A response instance
 | |
|      *
 | |
|      * @see doRequest()
 | |
|      */
 | |
|     public function getResponse()
 | |
|     {
 | |
|         return $this->response;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the current BrowserKit Request instance.
 | |
|      *
 | |
|      * @return Request|null A BrowserKit Request instance
 | |
|      */
 | |
|     public function getInternalRequest()
 | |
|     {
 | |
|         return $this->internalRequest;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Returns the current origin Request instance.
 | |
|      *
 | |
|      * The origin request is the request instance that is sent
 | |
|      * to the code that handles requests.
 | |
|      *
 | |
|      * @return object|null A Request instance
 | |
|      *
 | |
|      * @see doRequest()
 | |
|      */
 | |
|     public function getRequest()
 | |
|     {
 | |
|         return $this->request;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Clicks on a given link.
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function click(Link $link)
 | |
|     {
 | |
|         if ($link instanceof Form) {
 | |
|             return $this->submit($link);
 | |
|         }
 | |
| 
 | |
|         return $this->request($link->getMethod(), $link->getUri());
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Submits a form.
 | |
|      *
 | |
|      * @param Form  $form   A Form instance
 | |
|      * @param array $values An array of form field values
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function submit(Form $form, array $values = array())
 | |
|     {
 | |
|         $form->setValues($values);
 | |
| 
 | |
|         return $this->request($form->getMethod(), $form->getUri(), $form->getPhpValues(), $form->getPhpFiles());
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Calls a URI.
 | |
|      *
 | |
|      * @param string $method        The request method
 | |
|      * @param string $uri           The URI to fetch
 | |
|      * @param array  $parameters    The Request parameters
 | |
|      * @param array  $files         The files
 | |
|      * @param array  $server        The server parameters (HTTP headers are referenced with a HTTP_ prefix as PHP does)
 | |
|      * @param string $content       The raw body data
 | |
|      * @param bool   $changeHistory Whether to update the history or not (only used internally for back(), forward(), and reload())
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function request($method, $uri, array $parameters = array(), array $files = array(), array $server = array(), $content = null, $changeHistory = true)
 | |
|     {
 | |
|         if ($this->isMainRequest) {
 | |
|             $this->redirectCount = 0;
 | |
|         } else {
 | |
|             ++$this->redirectCount;
 | |
|         }
 | |
| 
 | |
|         $uri = $this->getAbsoluteUri($uri);
 | |
| 
 | |
|         $server = array_merge($this->server, $server);
 | |
| 
 | |
|         if (isset($server['HTTPS'])) {
 | |
|             $uri = preg_replace('{^'.parse_url($uri, PHP_URL_SCHEME).'}', $server['HTTPS'] ? 'https' : 'http', $uri);
 | |
|         }
 | |
| 
 | |
|         if (!$this->history->isEmpty()) {
 | |
|             $server['HTTP_REFERER'] = $this->history->current()->getUri();
 | |
|         }
 | |
| 
 | |
|         if (empty($server['HTTP_HOST'])) {
 | |
|             $server['HTTP_HOST'] = $this->extractHost($uri);
 | |
|         }
 | |
| 
 | |
|         $server['HTTPS'] = 'https' == parse_url($uri, PHP_URL_SCHEME);
 | |
| 
 | |
|         $this->internalRequest = new Request($uri, $method, $parameters, $files, $this->cookieJar->allValues($uri), $server, $content);
 | |
| 
 | |
|         $this->request = $this->filterRequest($this->internalRequest);
 | |
| 
 | |
|         if (true === $changeHistory) {
 | |
|             $this->history->add($this->internalRequest);
 | |
|         }
 | |
| 
 | |
|         if ($this->insulated) {
 | |
|             $this->response = $this->doRequestInProcess($this->request);
 | |
|         } else {
 | |
|             $this->response = $this->doRequest($this->request);
 | |
|         }
 | |
| 
 | |
|         $this->internalResponse = $this->filterResponse($this->response);
 | |
| 
 | |
|         $this->cookieJar->updateFromResponse($this->internalResponse, $uri);
 | |
| 
 | |
|         $status = $this->internalResponse->getStatus();
 | |
| 
 | |
|         if ($status >= 300 && $status < 400) {
 | |
|             $this->redirect = $this->internalResponse->getHeader('Location');
 | |
|         } else {
 | |
|             $this->redirect = null;
 | |
|         }
 | |
| 
 | |
|         if ($this->followRedirects && $this->redirect) {
 | |
|             $this->redirects[serialize($this->history->current())] = true;
 | |
| 
 | |
|             return $this->crawler = $this->followRedirect();
 | |
|         }
 | |
| 
 | |
|         return $this->crawler = $this->createCrawlerFromContent($this->internalRequest->getUri(), $this->internalResponse->getContent(), $this->internalResponse->getHeader('Content-Type'));
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Makes a request in another process.
 | |
|      *
 | |
|      * @param object $request An origin request instance
 | |
|      *
 | |
|      * @return object An origin response instance
 | |
|      *
 | |
|      * @throws \RuntimeException When processing returns exit code
 | |
|      */
 | |
|     protected function doRequestInProcess($request)
 | |
|     {
 | |
|         $deprecationsFile = tempnam(sys_get_temp_dir(), 'deprec');
 | |
|         putenv('SYMFONY_DEPRECATIONS_SERIALIZE='.$deprecationsFile);
 | |
|         $_ENV['SYMFONY_DEPRECATIONS_SERIALIZE'] = $deprecationsFile;
 | |
|         $process = new PhpProcess($this->getScript($request), null, null);
 | |
|         $process->run();
 | |
| 
 | |
|         if (file_exists($deprecationsFile)) {
 | |
|             $deprecations = file_get_contents($deprecationsFile);
 | |
|             unlink($deprecationsFile);
 | |
|             foreach ($deprecations ? unserialize($deprecations) : array() as $deprecation) {
 | |
|                 if ($deprecation[0]) {
 | |
|                     @trigger_error($deprecation[1], E_USER_DEPRECATED);
 | |
|                 } else {
 | |
|                     @trigger_error($deprecation[1], E_USER_DEPRECATED);
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         if (!$process->isSuccessful() || !preg_match('/^O\:\d+\:/', $process->getOutput())) {
 | |
|             throw new \RuntimeException(sprintf('OUTPUT: %s ERROR OUTPUT: %s', $process->getOutput(), $process->getErrorOutput()));
 | |
|         }
 | |
| 
 | |
|         return unserialize($process->getOutput());
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Makes a request.
 | |
|      *
 | |
|      * @param object $request An origin request instance
 | |
|      *
 | |
|      * @return object An origin response instance
 | |
|      */
 | |
|     abstract protected function doRequest($request);
 | |
| 
 | |
|     /**
 | |
|      * Returns the script to execute when the request must be insulated.
 | |
|      *
 | |
|      * @param object $request An origin request instance
 | |
|      *
 | |
|      * @throws \LogicException When this abstract class is not implemented
 | |
|      */
 | |
|     protected function getScript($request)
 | |
|     {
 | |
|         throw new \LogicException('To insulate requests, you need to override the getScript() method.');
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Filters the BrowserKit request to the origin one.
 | |
|      *
 | |
|      * @param Request $request The BrowserKit Request to filter
 | |
|      *
 | |
|      * @return object An origin request instance
 | |
|      */
 | |
|     protected function filterRequest(Request $request)
 | |
|     {
 | |
|         return $request;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Filters the origin response to the BrowserKit one.
 | |
|      *
 | |
|      * @param object $response The origin response to filter
 | |
|      *
 | |
|      * @return Response An BrowserKit Response instance
 | |
|      */
 | |
|     protected function filterResponse($response)
 | |
|     {
 | |
|         return $response;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Creates a crawler.
 | |
|      *
 | |
|      * This method returns null if the DomCrawler component is not available.
 | |
|      *
 | |
|      * @param string $uri     A URI
 | |
|      * @param string $content Content for the crawler to use
 | |
|      * @param string $type    Content type
 | |
|      *
 | |
|      * @return Crawler|null
 | |
|      */
 | |
|     protected function createCrawlerFromContent($uri, $content, $type)
 | |
|     {
 | |
|         if (!class_exists('Symfony\Component\DomCrawler\Crawler')) {
 | |
|             return;
 | |
|         }
 | |
| 
 | |
|         $crawler = new Crawler(null, $uri);
 | |
|         $crawler->addContent($content, $type);
 | |
| 
 | |
|         return $crawler;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Goes back in the browser history.
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function back()
 | |
|     {
 | |
|         do {
 | |
|             $request = $this->history->back();
 | |
|         } while (array_key_exists(serialize($request), $this->redirects));
 | |
| 
 | |
|         return $this->requestFromRequest($request, false);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Goes forward in the browser history.
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function forward()
 | |
|     {
 | |
|         do {
 | |
|             $request = $this->history->forward();
 | |
|         } while (array_key_exists(serialize($request), $this->redirects));
 | |
| 
 | |
|         return $this->requestFromRequest($request, false);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Reloads the current browser.
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     public function reload()
 | |
|     {
 | |
|         return $this->requestFromRequest($this->history->current(), false);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Follow redirects?
 | |
|      *
 | |
|      * @return Crawler
 | |
|      *
 | |
|      * @throws \LogicException If request was not a redirect
 | |
|      */
 | |
|     public function followRedirect()
 | |
|     {
 | |
|         if (empty($this->redirect)) {
 | |
|             throw new \LogicException('The request was not redirected.');
 | |
|         }
 | |
| 
 | |
|         if (-1 !== $this->maxRedirects) {
 | |
|             if ($this->redirectCount > $this->maxRedirects) {
 | |
|                 $this->redirectCount = 0;
 | |
|                 throw new \LogicException(sprintf('The maximum number (%d) of redirections was reached.', $this->maxRedirects));
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         $request = $this->internalRequest;
 | |
| 
 | |
|         if (\in_array($this->internalResponse->getStatus(), array(301, 302, 303))) {
 | |
|             $method = 'GET';
 | |
|             $files = array();
 | |
|             $content = null;
 | |
|         } else {
 | |
|             $method = $request->getMethod();
 | |
|             $files = $request->getFiles();
 | |
|             $content = $request->getContent();
 | |
|         }
 | |
| 
 | |
|         if ('GET' === strtoupper($method)) {
 | |
|             // Don't forward parameters for GET request as it should reach the redirection URI
 | |
|             $parameters = array();
 | |
|         } else {
 | |
|             $parameters = $request->getParameters();
 | |
|         }
 | |
| 
 | |
|         $server = $request->getServer();
 | |
|         $server = $this->updateServerFromUri($server, $this->redirect);
 | |
| 
 | |
|         $this->isMainRequest = false;
 | |
| 
 | |
|         $response = $this->request($method, $this->redirect, $parameters, $files, $server, $content);
 | |
| 
 | |
|         $this->isMainRequest = true;
 | |
| 
 | |
|         return $response;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Restarts the client.
 | |
|      *
 | |
|      * It flushes history and all cookies.
 | |
|      */
 | |
|     public function restart()
 | |
|     {
 | |
|         $this->cookieJar->clear();
 | |
|         $this->history->clear();
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Takes a URI and converts it to absolute if it is not already absolute.
 | |
|      *
 | |
|      * @param string $uri A URI
 | |
|      *
 | |
|      * @return string An absolute URI
 | |
|      */
 | |
|     protected function getAbsoluteUri($uri)
 | |
|     {
 | |
|         // already absolute?
 | |
|         if (0 === strpos($uri, 'http://') || 0 === strpos($uri, 'https://')) {
 | |
|             return $uri;
 | |
|         }
 | |
| 
 | |
|         if (!$this->history->isEmpty()) {
 | |
|             $currentUri = $this->history->current()->getUri();
 | |
|         } else {
 | |
|             $currentUri = sprintf('http%s://%s/',
 | |
|                 isset($this->server['HTTPS']) ? 's' : '',
 | |
|                 isset($this->server['HTTP_HOST']) ? $this->server['HTTP_HOST'] : 'localhost'
 | |
|             );
 | |
|         }
 | |
| 
 | |
|         // protocol relative URL
 | |
|         if (0 === strpos($uri, '//')) {
 | |
|             return parse_url($currentUri, PHP_URL_SCHEME).':'.$uri;
 | |
|         }
 | |
| 
 | |
|         // anchor or query string parameters?
 | |
|         if (!$uri || '#' == $uri[0] || '?' == $uri[0]) {
 | |
|             return preg_replace('/[#?].*?$/', '', $currentUri).$uri;
 | |
|         }
 | |
| 
 | |
|         if ('/' !== $uri[0]) {
 | |
|             $path = parse_url($currentUri, PHP_URL_PATH);
 | |
| 
 | |
|             if ('/' !== substr($path, -1)) {
 | |
|                 $path = substr($path, 0, strrpos($path, '/') + 1);
 | |
|             }
 | |
| 
 | |
|             $uri = $path.$uri;
 | |
|         }
 | |
| 
 | |
|         return preg_replace('#^(.*?//[^/]+)\/.*$#', '$1', $currentUri).$uri;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Makes a request from a Request object directly.
 | |
|      *
 | |
|      * @param Request $request       A Request instance
 | |
|      * @param bool    $changeHistory Whether to update the history or not (only used internally for back(), forward(), and reload())
 | |
|      *
 | |
|      * @return Crawler
 | |
|      */
 | |
|     protected function requestFromRequest(Request $request, $changeHistory = true)
 | |
|     {
 | |
|         return $this->request($request->getMethod(), $request->getUri(), $request->getParameters(), $request->getFiles(), $request->getServer(), $request->getContent(), $changeHistory);
 | |
|     }
 | |
| 
 | |
|     private function updateServerFromUri($server, $uri)
 | |
|     {
 | |
|         $server['HTTP_HOST'] = $this->extractHost($uri);
 | |
|         $scheme = parse_url($uri, PHP_URL_SCHEME);
 | |
|         $server['HTTPS'] = null === $scheme ? $server['HTTPS'] : 'https' == $scheme;
 | |
|         unset($server['HTTP_IF_NONE_MATCH'], $server['HTTP_IF_MODIFIED_SINCE']);
 | |
| 
 | |
|         return $server;
 | |
|     }
 | |
| 
 | |
|     private function extractHost($uri)
 | |
|     {
 | |
|         $host = parse_url($uri, PHP_URL_HOST);
 | |
| 
 | |
|         if ($port = parse_url($uri, PHP_URL_PORT)) {
 | |
|             return $host.':'.$port;
 | |
|         }
 | |
| 
 | |
|         return $host;
 | |
|     }
 | |
| }
 |