Client.php 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
  1. <?php
  2. /*
  3. * This file is part of the Goutte package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Goutte;
  11. use GuzzleHttp\Client as GuzzleClient;
  12. use GuzzleHttp\ClientInterface as GuzzleClientInterface;
  13. use GuzzleHttp\Cookie\CookieJar;
  14. use GuzzleHttp\Exception\RequestException;
  15. use Psr\Http\Message\ResponseInterface;
  16. use Symfony\Component\BrowserKit\Client as BaseClient;
  17. use Symfony\Component\BrowserKit\Request;
  18. use Symfony\Component\BrowserKit\Response;
  19. /**
  20. * Client.
  21. *
  22. * @author Fabien Potencier <fabien.potencier@symfony-project.com>
  23. * @author Michael Dowling <michael@guzzlephp.org>
  24. * @author Charles Sarrazin <charles@sarraz.in>
  25. */
  26. class Client extends BaseClient
  27. {
  28. protected $client;
  29. private $headers = array();
  30. private $auth = null;
  31. public function setClient(GuzzleClientInterface $client)
  32. {
  33. $this->client = $client;
  34. return $this;
  35. }
  36. public function getClient()
  37. {
  38. if (!$this->client) {
  39. $this->client = new GuzzleClient(array('defaults' => array('allow_redirects' => false, 'cookies' => true)));
  40. }
  41. return $this->client;
  42. }
  43. public function setHeader($name, $value)
  44. {
  45. $this->headers[$name] = $value;
  46. return $this;
  47. }
  48. public function removeHeader($name)
  49. {
  50. unset($this->headers[$name]);
  51. }
  52. public function setAuth($user, $password = '', $type = 'basic')
  53. {
  54. $this->auth = array($user, $password, $type);
  55. return $this;
  56. }
  57. public function resetAuth()
  58. {
  59. $this->auth = null;
  60. return $this;
  61. }
  62. /**
  63. * @param Request $request
  64. *
  65. * @return Response
  66. */
  67. protected function doRequest($request)
  68. {
  69. $headers = array();
  70. foreach ($request->getServer() as $key => $val) {
  71. $key = strtolower(str_replace('_', '-', $key));
  72. $contentHeaders = array('content-length' => true, 'content-md5' => true, 'content-type' => true);
  73. if (0 === strpos($key, 'http-')) {
  74. $headers[substr($key, 5)] = $val;
  75. }
  76. // CONTENT_* are not prefixed with HTTP_
  77. elseif (isset($contentHeaders[$key])) {
  78. $headers[$key] = $val;
  79. }
  80. }
  81. $cookies = CookieJar::fromArray(
  82. $this->getCookieJar()->allRawValues($request->getUri()),
  83. parse_url($request->getUri(), PHP_URL_HOST)
  84. );
  85. $requestOptions = array(
  86. 'cookies' => $cookies,
  87. 'allow_redirects' => false,
  88. 'auth' => $this->auth,
  89. );
  90. if (!in_array($request->getMethod(), array('GET', 'HEAD'))) {
  91. if (null !== $content = $request->getContent()) {
  92. $requestOptions['body'] = $content;
  93. } else {
  94. if ($files = $request->getFiles()) {
  95. $requestOptions['multipart'] = [];
  96. $this->addPostFields($request->getParameters(), $requestOptions['multipart']);
  97. $this->addPostFiles($files, $requestOptions['multipart']);
  98. } else {
  99. $requestOptions['form_params'] = $request->getParameters();
  100. }
  101. }
  102. }
  103. if (!empty($headers)) {
  104. $requestOptions['headers'] = $headers;
  105. }
  106. $method = $request->getMethod();
  107. $uri = $request->getUri();
  108. foreach ($this->headers as $name => $value) {
  109. $requestOptions['headers'][$name] = $value;
  110. }
  111. // Let BrowserKit handle redirects
  112. try {
  113. $response = $this->getClient()->request($method, $uri, $requestOptions);
  114. } catch (RequestException $e) {
  115. $response = $e->getResponse();
  116. if (null === $response) {
  117. throw $e;
  118. }
  119. }
  120. return $this->createResponse($response);
  121. }
  122. protected function addPostFiles(array $files, array &$multipart, $arrayName = '')
  123. {
  124. if (empty($files)) {
  125. return;
  126. }
  127. foreach ($files as $name => $info) {
  128. if (!empty($arrayName)) {
  129. $name = $arrayName.'['.$name.']';
  130. }
  131. $file = [
  132. 'name' => $name,
  133. ];
  134. if (is_array($info)) {
  135. if (isset($info['tmp_name'])) {
  136. if ('' !== $info['tmp_name']) {
  137. $file['contents'] = fopen($info['tmp_name'], 'r');
  138. if (isset($info['name'])) {
  139. $file['filename'] = $info['name'];
  140. }
  141. } else {
  142. continue;
  143. }
  144. } else {
  145. $this->addPostFiles($info, $multipart, $name);
  146. continue;
  147. }
  148. } else {
  149. $file['contents'] = fopen($info, 'r');
  150. }
  151. $multipart[] = $file;
  152. }
  153. }
  154. public function addPostFields(array $formParams, array &$multipart, $arrayName = '')
  155. {
  156. foreach ($formParams as $name => $value) {
  157. if (!empty($arrayName)) {
  158. $name = $arrayName.'['.$name.']';
  159. }
  160. if (is_array($value)) {
  161. $this->addPostFields($value, $multipart, $name);
  162. } else {
  163. $multipart[] = [
  164. 'name' => $name,
  165. 'contents' => $value,
  166. ];
  167. }
  168. }
  169. }
  170. protected function createResponse(ResponseInterface $response)
  171. {
  172. return new Response((string) $response->getBody(), $response->getStatusCode(), $response->getHeaders());
  173. }
  174. }