[Summary view] [Print] [Text view]

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