SWP\Bundle\ContentBundle\File\FileDownloader

Complexity

Total Complexity

11

Size/Duplication

Total Lines	74
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	9

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	download()	0	15	2
B	retryDecider()	0	38	7
A	retryDelay()	0	6	1

<?php

declare(strict_types=1);

/*
 * This file is part of the Superdesk Web Publisher Content Bundle.
 *
 * Copyright 2019 Sourcefabric z.ú. and contributors.
 *
 * For the full copyright and license information, please see the
 * AUTHORS and LICENSE files distributed with this source code.
 *
 * @copyright 2019 Sourcefabric z.ú
 * @license http://www.superdesk.org/license
 */

namespace SWP\Bundle\ContentBundle\File;

use function rtrim;
use function sha1;
use function strlen;
use function sys_get_temp_dir;
use function pathinfo;
use GuzzleHttp\Client;
use GuzzleHttp\Exception\ConnectException;
use GuzzleHttp\Exception\RequestException;
use GuzzleHttp\Handler\CurlHandler;
use GuzzleHttp\HandlerStack;
use GuzzleHttp\Middleware;
use GuzzleHttp\Psr7\Request;
use GuzzleHttp\Psr7\Response;
use Hoa\Mime\Mime;
use Psr\Log\LoggerInterface;
use Symfony\Component\HttpFoundation\File\UploadedFile;

final class FileDownloader implements FileDownloaderInterface
{
    private $logger;

    private $retryDownloads;

    public function __construct(LoggerInterface $logger, bool $retryDownloads)
    {
        $this->logger = $logger;
        $this->retryDownloads = $retryDownloads;
    }

    public function download(string $url, string $mediaId, string $mimeType = null): UploadedFile
    {
        $pathParts = pathinfo($url);
        if (null === $mimeType) {
            $mimeType = Mime::getMimeFromExtension($pathParts['extension']);
        }

        $handlerStack = HandlerStack::create(new CurlHandler());
        $handlerStack->push(Middleware::retry($this->retryDecider(), $this->retryDelay()));
        $client = new Client(array('handler' => $handlerStack));
        $tempLocation = rtrim(sys_get_temp_dir(), '/').DIRECTORY_SEPARATOR.sha1($mediaId.date('his'));
        $client->request('GET', $url, ['sink' => $tempLocation]);

        return new UploadedFile($tempLocation, $mediaId, $mimeType, strlen($tempLocation), null, true);
    }

    private function retryDecider(): callable
    {
        return function (
            $retries,
            Request $request,
            Response $response = null,
            RequestException $exception = null
        ): bool {
            $retry = false;
            if (!$this->retryDownloads) {
                $this->logger->error(\sprintf('Retries are disabled'));

                return false;
            }

            if ($retries >= 4) {
                $this->logger->error(\sprintf('Maximum number of retires reached'));

                return false;
            }

            // Retry connection exceptions
            if ($exception instanceof ConnectException) {
                $retry = true;
            }

            // Retry on server errors
            if ($response && $response->getStatusCode() >= 400) {
                $retry = true;
            }

            if (true === $retry) {
                $this->logger->info(\sprintf('Retry downloading %s', $request->getUri()));
            }

            return $retry;
        };
    }

    private function retryDelay(): callable
    {
        return static function ($numberOfRetries): int {
            return 1000 * $numberOfRetries;
        };
    }
}