64 lines
2.1 KiB
PHP
64 lines
2.1 KiB
PHP
<?php
|
|
|
|
namespace UglyRobot\Infinite_Uploads\Aws\S3;
|
|
|
|
use UglyRobot\Infinite_Uploads\Aws\CommandInterface;
|
|
use UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface;
|
|
/**
|
|
* Used to update the host used for S3 requests in the case of using a
|
|
* "bucket endpoint" or CNAME bucket.
|
|
*
|
|
* IMPORTANT: this middleware must be added after the "build" step.
|
|
*
|
|
* @internal
|
|
*/
|
|
class BucketEndpointMiddleware
|
|
{
|
|
private static $exclusions = ['GetBucketLocation' => true];
|
|
private $nextHandler;
|
|
/**
|
|
* Create a middleware wrapper function.
|
|
*
|
|
* @return callable
|
|
*/
|
|
public static function wrap()
|
|
{
|
|
return function (callable $handler) {
|
|
return new self($handler);
|
|
};
|
|
}
|
|
public function __construct(callable $nextHandler)
|
|
{
|
|
$this->nextHandler = $nextHandler;
|
|
}
|
|
public function __invoke(\UglyRobot\Infinite_Uploads\Aws\CommandInterface $command, \UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface $request)
|
|
{
|
|
$nextHandler = $this->nextHandler;
|
|
$bucket = $command['Bucket'];
|
|
if ($bucket && !isset(self::$exclusions[$command->getName()])) {
|
|
$request = $this->modifyRequest($request, $command);
|
|
}
|
|
return $nextHandler($command, $request);
|
|
}
|
|
private function removeBucketFromPath($path, $bucket)
|
|
{
|
|
$len = strlen($bucket) + 1;
|
|
if (substr($path, 0, $len) === "/{$bucket}") {
|
|
$path = substr($path, $len);
|
|
}
|
|
return $path ?: '/';
|
|
}
|
|
private function modifyRequest(\UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface $request, \UglyRobot\Infinite_Uploads\Aws\CommandInterface $command)
|
|
{
|
|
$uri = $request->getUri();
|
|
$path = $uri->getPath();
|
|
$bucket = $command['Bucket'];
|
|
$path = $this->removeBucketFromPath($path, $bucket);
|
|
// Modify the Key to make sure the key is encoded, but slashes are not.
|
|
if ($command['Key']) {
|
|
$path = \UglyRobot\Infinite_Uploads\Aws\S3\S3Client::encodeKey(rawurldecode($path));
|
|
}
|
|
return $request->withUri($uri->withPath($path));
|
|
}
|
|
}
|