aboutsummaryrefslogtreecommitdiffstats
path: root/includes/shell/ShellboxHttpClient.php
blob: 5ceac4f87baa9316cc6287a95dfbd5b6524d9f55 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
<?php

namespace MediaWiki\Shell;

use MediaWiki\Http\HttpRequestFactory;
use Psr\Http\Message\RequestInterface;
use Shellbox\GuzzleHttpClient;

/**
 * The MediaWiki-specific implementation of a Shellbox HTTP client
 */
class ShellboxHttpClient extends GuzzleHttpClient {
	/** @var HttpRequestFactory */
	private $requestFactory;
	/** @var int|float Timeout in seconds */
	private $timeout;

	/**
	 * @param HttpRequestFactory $requestFactory
	 * @param int|float $timeout
	 */
	public function __construct( HttpRequestFactory $requestFactory, $timeout ) {
		$this->requestFactory = $requestFactory;
		$this->timeout = $timeout;
	}

	protected function modifyRequest( RequestInterface $request ): RequestInterface {
		return $request
			->withHeader( 'X-Request-Id', \WebRequest::getRequestId() );
	}

	protected function createClient( RequestInterface $request ) {
		return $this->requestFactory->createGuzzleClient( [
			'timeout' => $this->timeout
		] );
	}
}