PresignUrlMiddleware.php
3.13 KB
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
<?php
namespace NF_FU_VENDOR\Aws;
use NF_FU_VENDOR\Aws\Signature\SignatureV4;
use NF_FU_VENDOR\Aws\Endpoint\EndpointProvider;
use NF_FU_VENDOR\GuzzleHttp\Psr7\Uri;
use NF_FU_VENDOR\Psr\Http\Message\RequestInterface;
/**
* @internal Adds computed values to service operations that need presigned url.
*/
class PresignUrlMiddleware
{
private $client;
private $endpointProvider;
private $nextHandler;
/** @var array names of operations that require presign url */
private $commandPool;
/** @var string */
private $serviceName;
/** @var string */
private $presignParam;
/** @var bool */
private $requireDifferentRegion;
public function __construct(array $options, callable $endpointProvider, AwsClientInterface $client, callable $nextHandler)
{
$this->endpointProvider = $endpointProvider;
$this->client = $client;
$this->nextHandler = $nextHandler;
$this->commandPool = $options['operations'];
$this->serviceName = $options['service'];
$this->presignParam = $options['presign_param'];
$this->requireDifferentRegion = !empty($options['require_different_region']);
}
public static function wrap(AwsClientInterface $client, callable $endpointProvider, array $options = [])
{
return function (callable $handler) use($endpointProvider, $client, $options) {
$f = new PresignUrlMiddleware($options, $endpointProvider, $client, $handler);
return $f;
};
}
public function __invoke(CommandInterface $cmd, RequestInterface $request = null)
{
if (\in_array($cmd->getName(), $this->commandPool) && !isset($cmd->{'__skip' . $cmd->getName()})) {
$cmd['DestinationRegion'] = $this->client->getRegion();
if (!$this->requireDifferentRegion || !empty($cmd['SourceRegion']) && $cmd['SourceRegion'] !== $cmd['DestinationRegion']) {
$cmd[$this->presignParam] = $this->createPresignedUrl($this->client, $cmd);
}
}
$f = $this->nextHandler;
return $f($cmd, $request);
}
private function createPresignedUrl(AwsClientInterface $client, CommandInterface $cmd)
{
$cmdName = $cmd->getName();
$newCmd = $client->getCommand($cmdName, $cmd->toArray());
// Avoid infinite recursion by flagging the new command.
$newCmd->{'__skip' . $cmdName} = \true;
// Serialize a request for the operation.
$request = \NF_FU_VENDOR\Aws\serialize($newCmd);
// Create the new endpoint for the target endpoint.
$endpoint = EndpointProvider::resolve($this->endpointProvider, ['region' => $cmd['SourceRegion'], 'service' => $this->serviceName])['endpoint'];
// Set the request to hit the target endpoint.
$uri = $request->getUri()->withHost((new Uri($endpoint))->getHost());
$request = $request->withUri($uri);
// Create a presigned URL for our generated request.
$signer = new SignatureV4($this->serviceName, $cmd['SourceRegion']);
return (string) $signer->presign(SignatureV4::convertPostToGet($request), $client->getCredentials()->wait(), '+1 hour')->getUri();
}
}