| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475 |
- <?php
- namespace Aws\S3;
- use Aws\CommandInterface;
- use 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(CommandInterface $command, 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(
- RequestInterface $request,
- 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 = S3Client::encodeKey(rawurldecode($path));
- }
- return $request->withUri($uri->withPath($path));
- }
- }
|