4 * This file is part of the Goutte package.
6 * (c) Fabien Potencier <fabien@symfony.com>
8 * For the full copyright and license information, please view the LICENSE
9 * file that was distributed with this source code.
14 use GuzzleHttp\Client as GuzzleClient;
15 use GuzzleHttp\ClientInterface as GuzzleClientInterface;
16 use GuzzleHttp\Cookie\CookieJar;
17 use GuzzleHttp\Exception\RequestException;
18 use Psr\Http\Message\ResponseInterface;
19 use Symfony\Component\BrowserKit\Client as BaseClient;
20 use Symfony\Component\BrowserKit\Request;
21 use Symfony\Component\BrowserKit\Response;
26 * @author Fabien Potencier <fabien.potencier@symfony-project.com>
27 * @author Michael Dowling <michael@guzzlephp.org>
28 * @author Charles Sarrazin <charles@sarraz.in>
30 class Client extends BaseClient
34 private $headers = array();
37 public function setClient(GuzzleClientInterface $client)
39 $this->client = $client;
44 public function getClient()
47 $this->client = new GuzzleClient(array('allow_redirects' => false, 'cookies' => true));
53 public function setHeader($name, $value)
55 $this->headers[strtolower($name)] = $value;
60 public function removeHeader($name)
62 unset($this->headers[strtolower($name)]);
65 public function resetHeaders()
67 $this->headers = array();
75 public function restart()
82 public function setAuth($user, $password = '', $type = 'basic')
84 $this->auth = array($user, $password, $type);
89 public function resetAuth()
97 * @param Request $request
101 protected function doRequest($request)
104 foreach ($request->getServer() as $key => $val) {
105 $key = strtolower(str_replace('_', '-', $key));
106 $contentHeaders = array('content-length' => true, 'content-md5' => true, 'content-type' => true);
107 if (0 === strpos($key, 'http-')) {
108 $headers[substr($key, 5)] = $val;
110 // CONTENT_* are not prefixed with HTTP_
111 elseif (isset($contentHeaders[$key])) {
112 $headers[$key] = $val;
116 $cookies = CookieJar::fromArray(
117 $this->getCookieJar()->allRawValues($request->getUri()),
118 parse_url($request->getUri(), PHP_URL_HOST)
121 $requestOptions = array(
122 'cookies' => $cookies,
123 'allow_redirects' => false,
124 'auth' => $this->auth,
127 if (!in_array($request->getMethod(), array('GET', 'HEAD'))) {
128 if (null !== $content = $request->getContent()) {
129 $requestOptions['body'] = $content;
131 if ($files = $request->getFiles()) {
132 $requestOptions['multipart'] = [];
134 $this->addPostFields($request->getParameters(), $requestOptions['multipart']);
135 $this->addPostFiles($files, $requestOptions['multipart']);
137 $requestOptions['form_params'] = $request->getParameters();
142 if (!empty($headers)) {
143 $requestOptions['headers'] = $headers;
146 $method = $request->getMethod();
147 $uri = $request->getUri();
149 foreach ($this->headers as $name => $value) {
150 $requestOptions['headers'][$name] = $value;
153 // Let BrowserKit handle redirects
155 $response = $this->getClient()->request($method, $uri, $requestOptions);
156 } catch (RequestException $e) {
157 $response = $e->getResponse();
158 if (null === $response) {
163 return $this->createResponse($response);
166 protected function addPostFiles(array $files, array &$multipart, $arrayName = '')
172 foreach ($files as $name => $info) {
173 if (!empty($arrayName)) {
174 $name = $arrayName.'['.$name.']';
181 if (is_array($info)) {
182 if (isset($info['tmp_name'])) {
183 if ('' !== $info['tmp_name']) {
184 $file['contents'] = fopen($info['tmp_name'], 'r');
185 if (isset($info['name'])) {
186 $file['filename'] = $info['name'];
192 $this->addPostFiles($info, $multipart, $name);
196 $file['contents'] = fopen($info, 'r');
199 $multipart[] = $file;
203 public function addPostFields(array $formParams, array &$multipart, $arrayName = '')
205 foreach ($formParams as $name => $value) {
206 if (!empty($arrayName)) {
207 $name = $arrayName.'['.$name.']';
210 if (is_array($value)) {
211 $this->addPostFields($value, $multipart, $name);
215 'contents' => $value,
221 protected function createResponse(ResponseInterface $response)
223 return new Response((string) $response->getBody(), $response->getStatusCode(), $response->getHeaders());