FileDownloader.cs

Go to the documentation of this file.

using System;
using System.Net.Http;
using System.Net.Http.Headers;
 
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api;
using Tgstation.Server.Common.Http;
 
namespace Tgstation.Server.Host.IO
{
    public sealed class FileDownloader : IFileDownloader
    {
        readonly IAbstractHttpClientFactory httpClientFactory;
 
        readonly ILogger<FileDownloader> logger;
 
        public FileDownloader(IAbstractHttpClientFactory httpClientFactory, ILogger<FileDownloader> logger)
        {
            this.httpClientFactory = httpClientFactory ?? throw new ArgumentNullException(nameof(httpClientFactory));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }
 
        public IFileStreamProvider DownloadFile(Uri url, string? bearerToken)
        {
            ArgumentNullException.ThrowIfNull(url);
 
            logger.LogDebug("Starting download of {url}...", url);
            var httpClient = httpClientFactory.CreateClient();
            try
            {
                var request = new HttpRequestMessage(
                    HttpMethod.Get,
                    url);
                try
                {
                    if (bearerToken != null)
                        request.Headers.Authorization = new AuthenticationHeaderValue(ApiHeaders.BearerAuthenticationScheme, bearerToken);
 
                    return new RequestFileStreamProvider(httpClient, request);
                }
                catch
                {
                    request.Dispose();
                    throw;
                }
            }
            catch
            {
                httpClient.Dispose();
                throw;
            }
        }
    }
}