[f9c482b] | 1 | <?php
|
---|
| 2 |
|
---|
| 3 | namespace GuzzleHttp;
|
---|
| 4 |
|
---|
| 5 | use GuzzleHttp\Promise\PromiseInterface;
|
---|
| 6 | use Psr\Http\Message\RequestInterface;
|
---|
| 7 |
|
---|
| 8 | /**
|
---|
| 9 | * Prepares requests that contain a body, adding the Content-Length,
|
---|
| 10 | * Content-Type, and Expect headers.
|
---|
| 11 | *
|
---|
| 12 | * @final
|
---|
| 13 | */
|
---|
| 14 | class PrepareBodyMiddleware
|
---|
| 15 | {
|
---|
| 16 | /**
|
---|
| 17 | * @var callable(RequestInterface, array): PromiseInterface
|
---|
| 18 | */
|
---|
| 19 | private $nextHandler;
|
---|
| 20 |
|
---|
| 21 | /**
|
---|
| 22 | * @param callable(RequestInterface, array): PromiseInterface $nextHandler Next handler to invoke.
|
---|
| 23 | */
|
---|
| 24 | public function __construct(callable $nextHandler)
|
---|
| 25 | {
|
---|
| 26 | $this->nextHandler = $nextHandler;
|
---|
| 27 | }
|
---|
| 28 |
|
---|
| 29 | public function __invoke(RequestInterface $request, array $options): PromiseInterface
|
---|
| 30 | {
|
---|
| 31 | $fn = $this->nextHandler;
|
---|
| 32 |
|
---|
| 33 | // Don't do anything if the request has no body.
|
---|
| 34 | if ($request->getBody()->getSize() === 0) {
|
---|
| 35 | return $fn($request, $options);
|
---|
| 36 | }
|
---|
| 37 |
|
---|
| 38 | $modify = [];
|
---|
| 39 |
|
---|
| 40 | // Add a default content-type if possible.
|
---|
| 41 | if (!$request->hasHeader('Content-Type')) {
|
---|
| 42 | if ($uri = $request->getBody()->getMetadata('uri')) {
|
---|
| 43 | if (is_string($uri) && $type = Psr7\MimeType::fromFilename($uri)) {
|
---|
| 44 | $modify['set_headers']['Content-Type'] = $type;
|
---|
| 45 | }
|
---|
| 46 | }
|
---|
| 47 | }
|
---|
| 48 |
|
---|
| 49 | // Add a default content-length or transfer-encoding header.
|
---|
| 50 | if (!$request->hasHeader('Content-Length')
|
---|
| 51 | && !$request->hasHeader('Transfer-Encoding')
|
---|
| 52 | ) {
|
---|
| 53 | $size = $request->getBody()->getSize();
|
---|
| 54 | if ($size !== null) {
|
---|
| 55 | $modify['set_headers']['Content-Length'] = $size;
|
---|
| 56 | } else {
|
---|
| 57 | $modify['set_headers']['Transfer-Encoding'] = 'chunked';
|
---|
| 58 | }
|
---|
| 59 | }
|
---|
| 60 |
|
---|
| 61 | // Add the expect header if needed.
|
---|
| 62 | $this->addExpectHeader($request, $options, $modify);
|
---|
| 63 |
|
---|
| 64 | return $fn(Psr7\Utils::modifyRequest($request, $modify), $options);
|
---|
| 65 | }
|
---|
| 66 |
|
---|
| 67 | /**
|
---|
| 68 | * Add expect header
|
---|
| 69 | */
|
---|
| 70 | private function addExpectHeader(RequestInterface $request, array $options, array &$modify): void
|
---|
| 71 | {
|
---|
| 72 | // Determine if the Expect header should be used
|
---|
| 73 | if ($request->hasHeader('Expect')) {
|
---|
| 74 | return;
|
---|
| 75 | }
|
---|
| 76 |
|
---|
| 77 | $expect = $options['expect'] ?? null;
|
---|
| 78 |
|
---|
| 79 | // Return if disabled or using HTTP/1.0
|
---|
| 80 | if ($expect === false || $request->getProtocolVersion() === '1.0') {
|
---|
| 81 | return;
|
---|
| 82 | }
|
---|
| 83 |
|
---|
| 84 | // The expect header is unconditionally enabled
|
---|
| 85 | if ($expect === true) {
|
---|
| 86 | $modify['set_headers']['Expect'] = '100-Continue';
|
---|
| 87 |
|
---|
| 88 | return;
|
---|
| 89 | }
|
---|
| 90 |
|
---|
| 91 | // By default, send the expect header when the payload is > 1mb
|
---|
| 92 | if ($expect === null) {
|
---|
| 93 | $expect = 1048576;
|
---|
| 94 | }
|
---|
| 95 |
|
---|
| 96 | // Always add if the body cannot be rewound, the size cannot be
|
---|
| 97 | // determined, or the size is greater than the cutoff threshold
|
---|
| 98 | $body = $request->getBody();
|
---|
| 99 | $size = $body->getSize();
|
---|
| 100 |
|
---|
| 101 | if ($size === null || $size >= (int) $expect || !$body->isSeekable()) {
|
---|
| 102 | $modify['set_headers']['Expect'] = '100-Continue';
|
---|
| 103 | }
|
---|
| 104 | }
|
---|
| 105 | }
|
---|