68 lines
2.7 KiB
PHP
68 lines
2.7 KiB
PHP
<?php
|
|
|
|
namespace UglyRobot\Infinite_Uploads\Aws;
|
|
|
|
use UglyRobot\Infinite_Uploads\Aws\Api\Service;
|
|
use UglyRobot\Infinite_Uploads\Aws\Exception\IncalculablePayloadException;
|
|
use UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface;
|
|
/**
|
|
* @internal
|
|
*/
|
|
class StreamRequestPayloadMiddleware
|
|
{
|
|
private $nextHandler;
|
|
private $service;
|
|
/**
|
|
* Create a middleware wrapper function
|
|
*
|
|
* @param Service $service
|
|
* @return \Closure
|
|
*/
|
|
public static function wrap(\UglyRobot\Infinite_Uploads\Aws\Api\Service $service)
|
|
{
|
|
return function (callable $handler) use($service) {
|
|
return new self($handler, $service);
|
|
};
|
|
}
|
|
public function __construct(callable $nextHandler, \UglyRobot\Infinite_Uploads\Aws\Api\Service $service)
|
|
{
|
|
$this->nextHandler = $nextHandler;
|
|
$this->service = $service;
|
|
}
|
|
public function __invoke(\UglyRobot\Infinite_Uploads\Aws\CommandInterface $command, \UglyRobot\Infinite_Uploads\Psr\Http\Message\RequestInterface $request)
|
|
{
|
|
$nextHandler = $this->nextHandler;
|
|
$operation = $this->service->getOperation($command->getName());
|
|
$contentLength = $request->getHeader('content-length');
|
|
$hasStreaming = false;
|
|
$requiresLength = false;
|
|
// Check if any present input member is a stream and requires the
|
|
// content length
|
|
foreach ($operation->getInput()->getMembers() as $name => $member) {
|
|
if (!empty($member['streaming']) && isset($command[$name])) {
|
|
$hasStreaming = true;
|
|
if (!empty($member['requiresLength'])) {
|
|
$requiresLength = true;
|
|
}
|
|
}
|
|
}
|
|
if ($hasStreaming) {
|
|
// Add 'transfer-encoding' header if payload size not required to
|
|
// to be calculated and not already known
|
|
if (empty($requiresLength) && empty($contentLength) && isset($operation['authtype']) && $operation['authtype'] == 'v4-unsigned-body') {
|
|
$request = $request->withHeader('transfer-encoding', 'chunked');
|
|
// Otherwise, make sure 'content-length' header is added
|
|
} else {
|
|
if (empty($contentLength)) {
|
|
$size = $request->getBody()->getSize();
|
|
if (is_null($size)) {
|
|
throw new \UglyRobot\Infinite_Uploads\Aws\Exception\IncalculablePayloadException('Payload' . ' content length is required and can not be' . ' calculated.');
|
|
}
|
|
$request = $request->withHeader('content-length', $size);
|
|
}
|
|
}
|
|
}
|
|
return $nextHandler($command, $request);
|
|
}
|
|
}
|