149 lines
3.6 KiB
PHP
149 lines
3.6 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\HttpClient\Internal;
|
|
|
|
use Amp\ByteStream\InputStream;
|
|
use Amp\ByteStream\ResourceInputStream;
|
|
use Amp\Http\Client\RequestBody;
|
|
use Amp\Promise;
|
|
use Amp\Success;
|
|
use Symfony\Component\HttpClient\Exception\TransportException;
|
|
|
|
/**
|
|
* @author Nicolas Grekas <p@tchwork.com>
|
|
*
|
|
* @internal
|
|
*/
|
|
class AmpBody implements RequestBody, InputStream
|
|
{
|
|
private ResourceInputStream|\Closure|string $body;
|
|
private array $info;
|
|
private ?int $offset = 0;
|
|
private int $length = -1;
|
|
private ?int $uploaded = null;
|
|
|
|
/**
|
|
* @param \Closure|resource|string $body
|
|
*/
|
|
public function __construct(
|
|
$body,
|
|
&$info,
|
|
private \Closure $onProgress,
|
|
) {
|
|
$this->info = &$info;
|
|
|
|
if (\is_resource($body)) {
|
|
$this->offset = ftell($body);
|
|
$this->length = fstat($body)['size'];
|
|
$this->body = new ResourceInputStream($body);
|
|
} elseif (\is_string($body)) {
|
|
$this->length = \strlen($body);
|
|
$this->body = $body;
|
|
} else {
|
|
$this->body = $body;
|
|
}
|
|
}
|
|
|
|
public function createBodyStream(): InputStream
|
|
{
|
|
if (null !== $this->uploaded) {
|
|
$this->uploaded = null;
|
|
|
|
if (\is_string($this->body)) {
|
|
$this->offset = 0;
|
|
} elseif ($this->body instanceof ResourceInputStream) {
|
|
fseek($this->body->getResource(), $this->offset);
|
|
}
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function getHeaders(): Promise
|
|
{
|
|
return new Success([]);
|
|
}
|
|
|
|
public function getBodyLength(): Promise
|
|
{
|
|
return new Success($this->length - $this->offset);
|
|
}
|
|
|
|
public function read(): Promise
|
|
{
|
|
$this->info['size_upload'] += $this->uploaded;
|
|
$this->uploaded = 0;
|
|
($this->onProgress)();
|
|
|
|
$chunk = $this->doRead();
|
|
$chunk->onResolve(function ($e, $data) {
|
|
if (null !== $data) {
|
|
$this->uploaded = \strlen($data);
|
|
} else {
|
|
$this->info['upload_content_length'] = $this->info['size_upload'];
|
|
}
|
|
});
|
|
|
|
return $chunk;
|
|
}
|
|
|
|
public static function rewind(RequestBody $body): RequestBody
|
|
{
|
|
if (!$body instanceof self) {
|
|
return $body;
|
|
}
|
|
|
|
$body->uploaded = null;
|
|
|
|
if ($body->body instanceof ResourceInputStream) {
|
|
fseek($body->body->getResource(), $body->offset);
|
|
|
|
return new $body($body->body, $body->info, $body->onProgress);
|
|
}
|
|
|
|
if (\is_string($body->body)) {
|
|
$body->offset = 0;
|
|
}
|
|
|
|
return $body;
|
|
}
|
|
|
|
private function doRead(): Promise
|
|
{
|
|
if ($this->body instanceof ResourceInputStream) {
|
|
return $this->body->read();
|
|
}
|
|
|
|
if (null === $this->offset || !$this->length) {
|
|
return new Success();
|
|
}
|
|
|
|
if (\is_string($this->body)) {
|
|
$this->offset = null;
|
|
|
|
return new Success($this->body);
|
|
}
|
|
|
|
if ('' === $data = ($this->body)(16372)) {
|
|
$this->offset = null;
|
|
|
|
return new Success();
|
|
}
|
|
|
|
if (!\is_string($data)) {
|
|
throw new TransportException(sprintf('Return value of the "body" option callback must be string, "%s" returned.', get_debug_type($data)));
|
|
}
|
|
|
|
return new Success($data);
|
|
}
|
|
}
|