40 lines
1.1 KiB
PHP
40 lines
1.1 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Http\Client\Common\Plugin;
|
|
|
|
use Http\Client\Common\Plugin;
|
|
use Http\Message\Encoding\ChunkStream;
|
|
use Http\Promise\Promise;
|
|
use Psr\Http\Message\RequestInterface;
|
|
|
|
/**
|
|
* Allow to set the correct content length header on the request or to transfer it as a chunk if not possible.
|
|
*
|
|
* @author Joel Wurtz <joel.wurtz@gmail.com>
|
|
*/
|
|
final class ContentLengthPlugin implements Plugin
|
|
{
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function handleRequest(RequestInterface $request, callable $next, callable $first): Promise
|
|
{
|
|
if (!$request->hasHeader('Content-Length')) {
|
|
$stream = $request->getBody();
|
|
|
|
// Cannot determine the size so we use a chunk stream
|
|
if (null === $stream->getSize()) {
|
|
$stream = new ChunkStream($stream);
|
|
$request = $request->withBody($stream);
|
|
$request = $request->withAddedHeader('Transfer-Encoding', 'chunked');
|
|
} else {
|
|
$request = $request->withHeader('Content-Length', (string) $stream->getSize());
|
|
}
|
|
}
|
|
|
|
return $next($request);
|
|
}
|
|
}
|