87 lines
4.4 KiB
PHP
87 lines
4.4 KiB
PHP
<?php
|
|
|
|
namespace UglyRobot\Infinite_Uploads\Aws;
|
|
|
|
use UglyRobot\Infinite_Uploads\Aws\Signature\SignatureV4;
|
|
use UglyRobot\Infinite_Uploads\Aws\Endpoint\EndpointProvider;
|
|
use UglyRobot\Infinite_Uploads\GuzzleHttp\Psr7\Uri;
|
|
use UglyRobot\Infinite_Uploads\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 array query params that are not on the operation's model to add before signing */
|
|
private $extraQueryParams;
|
|
/** @var string */
|
|
private $serviceName;
|
|
/** @var string */
|
|
private $presignParam;
|
|
/** @var bool */
|
|
private $requireDifferentRegion;
|
|
public function __construct(array $options, callable $endpointProvider, \UglyRobot\Infinite_Uploads\Aws\AwsClientInterface $client, callable $nextHandler)
|
|
{
|
|
$this->endpointProvider = $endpointProvider;
|
|
$this->client = $client;
|
|
$this->nextHandler = $nextHandler;
|
|
$this->commandPool = $options['operations'];
|
|
$this->serviceName = $options['service'];
|
|
$this->presignParam = !empty($options['presign_param']) ? $options['presign_param'] : 'PresignedUrl';
|
|
$this->extraQueryParams = !empty($options['extra_query_params']) ? $options['extra_query_params'] : [];
|
|
$this->requireDifferentRegion = !empty($options['require_different_region']);
|
|
}
|
|
public static function wrap(\UglyRobot\Infinite_Uploads\Aws\AwsClientInterface $client, callable $endpointProvider, array $options = [])
|
|
{
|
|
return function (callable $handler) use($endpointProvider, $client, $options) {
|
|
$f = new \UglyRobot\Infinite_Uploads\Aws\PresignUrlMiddleware($options, $endpointProvider, $client, $handler);
|
|
return $f;
|
|
};
|
|
}
|
|
public function __invoke(\UglyRobot\Infinite_Uploads\Aws\CommandInterface $cmd, \UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface $request = null)
|
|
{
|
|
if (in_array($cmd->getName(), $this->commandPool) && !isset($cmd->{'__skip' . $cmd->getName()})) {
|
|
$cmd['DestinationRegion'] = $this->client->getRegion();
|
|
if (!empty($cmd['SourceRegion']) && !empty($cmd[$this->presignParam])) {
|
|
goto nexthandler;
|
|
}
|
|
if (!$this->requireDifferentRegion || !empty($cmd['SourceRegion']) && $cmd['SourceRegion'] !== $cmd['DestinationRegion']) {
|
|
$cmd[$this->presignParam] = $this->createPresignedUrl($this->client, $cmd);
|
|
}
|
|
}
|
|
nexthandler:
|
|
$nextHandler = $this->nextHandler;
|
|
return $nextHandler($cmd, $request);
|
|
}
|
|
private function createPresignedUrl(\UglyRobot\Infinite_Uploads\Aws\AwsClientInterface $client, \UglyRobot\Infinite_Uploads\Aws\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 = \UglyRobot\Infinite_Uploads\Aws\serialize($newCmd);
|
|
// Create the new endpoint for the target endpoint.
|
|
$endpoint = \UglyRobot\Infinite_Uploads\Aws\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 \UglyRobot\Infinite_Uploads\GuzzleHttp\Psr7\Uri($endpoint))->getHost());
|
|
$request = $request->withUri($uri);
|
|
// Create a presigned URL for our generated request.
|
|
$signer = new \UglyRobot\Infinite_Uploads\Aws\Signature\SignatureV4($this->serviceName, $cmd['SourceRegion']);
|
|
$currentQueryParams = (string) $request->getBody();
|
|
$paramsToAdd = false;
|
|
if (!empty($this->extraQueryParams[$cmdName])) {
|
|
foreach ($this->extraQueryParams[$cmdName] as $param) {
|
|
if (!strpos($currentQueryParams, $param)) {
|
|
$paramsToAdd = "&{$param}={$cmd[$param]}";
|
|
}
|
|
}
|
|
}
|
|
return (string) $signer->presign(\UglyRobot\Infinite_Uploads\Aws\Signature\SignatureV4::convertPostToGet($request, $paramsToAdd ?: ""), $client->getCredentials()->wait(), '+1 hour')->getUri();
|
|
}
|
|
}
|