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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
<?php
namespace MediaWiki\Shell;
use GuzzleHttp\Psr7\Uri;
use GuzzleHttp\RequestOptions;
use MediaWiki\Http\HttpRequestFactory;
use RuntimeException;
use Shellbox\Client;
use Shellbox\RPC\LocalRpcClient;
use Shellbox\RPC\RpcClient;
use WebRequest;
/**
* This is a service which provides a configured client to access a remote
* Shellbox installation.
*
* @since 1.36
*/
class ShellboxClientFactory {
/** @var HttpRequestFactory */
private $requestFactory;
/** @var (string|false|null)[]|null */
private $urls;
/** @var string|null */
private $key;
/** The default request timeout, in seconds */
public const DEFAULT_TIMEOUT = 10;
/**
* @internal Use MediaWikiServices::getShellboxClientFactory()
* @param HttpRequestFactory $requestFactory The factory which will be used
* to make HTTP clients.
* @param (string|false|null)[]|null $urls The Shellbox base URL mapping
* @param string|null $key The shared secret key used for HMAC authentication
*/
public function __construct( HttpRequestFactory $requestFactory, $urls, $key ) {
$this->requestFactory = $requestFactory;
$this->urls = $urls;
$this->key = $key;
}
/**
* Test whether remote Shellbox is enabled by configuration.
*
* @param string|null $service Same as the service option for getClient.
* @return bool
*/
public function isEnabled( ?string $service = null ): bool {
return $this->getUrl( $service ) !== null;
}
/**
* Get a Shellbox client with the specified options. If remote Shellbox is
* not configured (isEnabled() returns false), an exception will be thrown.
*
* @param array $options Associative array of options:
* - timeout: The request timeout in seconds
* - service: the shellbox backend name to get the URL from the mapping
* @return Client
* @throws RuntimeException
*/
public function getClient( array $options = [] ) {
$url = $this->getUrl( $options['service'] ?? null );
if ( $url === null ) {
throw new RuntimeException( 'To use a remote shellbox to run shell commands, ' .
'$wgShellboxUrls and $wgShellboxSecretKey must be configured.' );
}
return new Client(
$this->requestFactory->createGuzzleClient( [
RequestOptions::TIMEOUT => $options['timeout'] ?? self::DEFAULT_TIMEOUT,
RequestOptions::HEADERS => [
'X-Request-Id' => WebRequest::getRequestId(),
],
RequestOptions::HTTP_ERRORS => false,
] ),
new Uri( $url ),
$this->key
);
}
/**
* Get a Shellbox RPC client with the specified options. If remote Shellbox is
* not configured (isEnabled() returns false), an exception will be thrown.
*
* @param array $options Associative array of options:
* - timeout: The request timeout in seconds
* - service: the shellbox backend name to get the URL from the mapping
* @return RpcClient
* @throws RuntimeException
*/
public function getRemoteRpcClient( array $options = [] ): RpcClient {
return $this->getClient( $options );
}
/**
* Get a Shellbox RPC client with specified options. If remote Shellbox is
* not configured (isEnabled() returns false), a local fallback is returned.
*
* @param array $options
* @return RpcClient
*/
public function getRpcClient( array $options = [] ): RpcClient {
$url = $this->getUrl( $options['service'] ?? null );
if ( $url === null ) {
return new LocalRpcClient();
}
return $this->getRemoteRpcClient( $options );
}
private function getUrl( ?string $service ): ?string {
if ( $this->urls === null || !strlen( $this->key ) ) {
return null;
}
$url = $this->urls[$service] ?? $this->urls['default'] ?? null;
if ( !is_string( $url ) ) {
return null;
}
return $url;
}
}
|
