danog/MadelineProto
2
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Upload files with no content-length, too

Browse Source
This commit is contained in:
Daniil Gentili 2019-06-25 15:36:09 +02:00
parent 8ab033e995
commit e1556822c0
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/danog/MadelineProto/MTProtoTools/Files.php
View File

@ -21,6 +21,7 @@ namespace danog\MadelineProto\MTProtoTools;
use Amp\ByteStream\InputStream; use Amp\ByteStream\InputStream;
use Amp\ByteStream\OutputStream; use Amp\ByteStream\OutputStream;
use Amp\ByteStream\ResourceInputStream;
use Amp\ByteStream\ResourceOutputStream; use Amp\ByteStream\ResourceOutputStream;
use Amp\ByteStream\StreamException; use Amp\ByteStream\StreamException;
use Amp\Deferred; use Amp\Deferred;
@ -41,6 +42,7 @@ use function Amp\File\open;
use function Amp\File\stat; use function Amp\File\stat;
use function Amp\File\touch; use function Amp\File\touch;
use function Amp\Promise\all; use function Amp\Promise\all;
use Amp\File\BlockingHandle;
/** /**
* Manages upload and download of files. * Manages upload and download of files.
@ -98,13 +100,23 @@ trait Files
} }
$mime = trim(explode(';', $response->getHeader('content-type') ?? 'application/octet-stream')[0]); $mime = trim(explode(';', $response->getHeader('content-type') ?? 'application/octet-stream')[0]);
$size = $response->getHeader('content-length') ?? $size; $size = $response->getHeader('content-length') ?? $size;
$stream = $response->getBody();
if (!$size) { if (!$size) {
throw new Exception('Wrong size'); $body = $stream;
$stream = new BlockingHandle(fopen('php://temp', 'r+b'), 'php://temp', 'r+b');
while (null !== $chunk = yield $body->read()) {
yield $stream->write($chunk);
}
$size = $stream->tell();
if (!$size) {
throw new Exception('Wrong size!');
}
yield $stream->seek(0);
} }
$body = $response->getBody(); return yield $this->upload_from_stream_async($stream, $size, $mime, $file_name, $cb, $encrypted);
return yield $this->upload_from_stream_async($body, $size, $mime, $file_name, $cb, $encrypted);
} }
public function upload_from_stream_async($stream, int $size, string $mime, string $file_name = '', $cb = null, bool $encrypted = false) public function upload_from_stream_async($stream, int $size, string $mime, string $file_name = '', $cb = null, bool $encrypted = false)
{ {