123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578 |
- <?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\Link;
- use Symfony\Component\DomCrawler\Form;
- 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 $isMainRequest = true;
- /**
- * Constructor.
- *
- * @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;
- }
- /**
- * Sets the maximum number of requests that crawler can follow.
- *
- * @param int $maxRedirects
- */
- public function setMaxRedirects($maxRedirects)
- {
- $this->maxRedirects = $maxRedirects < 0 ? -1 : $maxRedirects;
- $this->followRedirects = -1 != $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_HOST' => 'localhost',
- 'HTTP_USER_AGENT' => 'Symfony2 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.
- *
- * @param Link $link A Link instance
- *
- * @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);
- if (!empty($server['HTTP_HOST'])) {
- $uri = preg_replace('{^(https?\://)'.preg_quote($this->extractHost($uri)).'}', '${1}'.$server['HTTP_HOST'], $uri);
- }
- if (isset($server['HTTPS'])) {
- $uri = preg_replace('{^'.parse_url($uri, PHP_URL_SCHEME).'}', $server['HTTPS'] ? 'https' : 'http', $uri);
- }
- $server = array_merge($this->server, $server);
- if (!$this->history->isEmpty()) {
- $server['HTTP_REFERER'] = $this->history->current()->getUri();
- }
- $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) {
- 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)
- {
- $process = new PhpProcess($this->getScript($request), null, null);
- $process->run();
- 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()
- {
- return $this->requestFromRequest($this->history->back(), false);
- }
- /**
- * Goes forward in the browser history.
- *
- * @return Crawler
- */
- public function forward()
- {
- return $this->requestFromRequest($this->history->forward(), 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) {
- throw new \LogicException(sprintf('The maximum number (%d) of redirections was reached.', $this->maxRedirects));
- }
- }
- $request = $this->internalRequest;
- if (in_array($this->internalResponse->getStatus(), array(302, 303))) {
- $method = 'get';
- $files = array();
- $content = null;
- } else {
- $method = $request->getMethod();
- $files = $request->getFiles();
- $content = $request->getContent();
- }
- if ('get' === strtolower($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?
- if (!$uri || '#' == $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;
- }
- }
|