rdt-client/server/RdtClient.Service/Services/Downloaders/Aria2cDownloader.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Threading.Tasks;
using Aria2NET;
using RdtClient.Data.Models.Internal;

namespace RdtClient.Service.Services.Downloaders
{
    public class Aria2cDownloader : IDownloader
    {
        public event EventHandler<DownloadCompleteEventArgs> DownloadComplete;
        public event EventHandler<DownloadProgressEventArgs> DownloadProgress;

        private const Int32 RetryCount = 5;

        private readonly String _uri;
        private readonly String _filePath;

        private readonly Aria2NetClient _aria2NetClient;
        
        private String _gid;

        public Aria2cDownloader(String gid, String uri, String filePath, DbSettings settings)
        {
            _gid = gid;
            _uri = uri;
            _filePath = filePath;

            var httpClient = new HttpClient
            {
                Timeout = TimeSpan.FromSeconds(10)
            };

            _aria2NetClient = new Aria2NetClient(settings.Aria2cUrl, settings.Aria2cSecret, httpClient, 10);
        }
        
        public async Task<String> Download()
        {
            var path = Path.GetDirectoryName(_filePath);
            var fileName = Path.GetFileName(_filePath);

            var isAlreadyAdded = await CheckIfAdded();

            if (isAlreadyAdded)
            {
                throw new Exception($"The download link {_uri} has already been added to Aria2");
            }

            var retryCount = 0;
            while(true)
            {
                try
                {
                    if (_gid != null)
                    {
                        try
                        {
                            await _aria2NetClient.TellStatus(_gid);

                            return _gid;
                        }
                        catch
                        {
                            _gid = null;
                        }
                    }
                    
                    _gid ??= await _aria2NetClient.AddUri(new List<String>
                                                          {
                                                              _uri
                                                          },
                                                          new Dictionary<String, Object>
                                                          {
                                                              {
                                                                  "dir", path
                                                              },
                                                              {
                                                                  "out", fileName
                                                              }
                                                          });

                    await _aria2NetClient.TellStatus(_gid);

                    return _gid;
                }
                catch
                {
                    if (retryCount >= RetryCount)
                    {
                        throw;
                    }

                    await Task.Delay(retryCount * 1000);

                    retryCount++;
                }
            }
        }

        public async Task Cancel()
        {
            if (String.IsNullOrWhiteSpace(_gid))
            {
                return;
            }

            try
            {
                await _aria2NetClient.ForceRemove(_gid);
            }
            catch
            {
                // ignored
            }

            try
            {
                await _aria2NetClient.RemoveDownloadResult(_gid);
            }
            catch
            {
                // ignored
            }
        }

        public async Task Pause()
        {
            if (String.IsNullOrWhiteSpace(_gid))
            {
                return;
            }

            try
            {
                await _aria2NetClient.Pause(_gid);
            }
            catch
            {
                // ignored
            }
        }

        public async Task Resume()
        {
            if (String.IsNullOrWhiteSpace(_gid))
            {
                return;
            }

            try
            {
                await _aria2NetClient.Unpause(_gid);
            }
            catch
            {
                // ignored
            }
        }

        public async Task Update(IEnumerable<DownloadStatusResult> allDownloads)
        {
            if (_gid == null)
            {
                return;
            }

            var download = allDownloads.FirstOrDefault(m => m.Gid == _gid);

            if (download == null)
            {
                return;
            }

            if (!String.IsNullOrWhiteSpace(download.ErrorMessage) || download.Status == "error")
            {
                await Cancel();
                DownloadComplete?.Invoke(this, new DownloadCompleteEventArgs
                {
                    Error = $"{download.ErrorCode}: {download.ErrorMessage}"
                });
                return;
            }

            if (download.Status == "complete" || download.Status == "removed")
            {
                await Cancel();

                var retryCount = 0;
                while (true)
                {
                    if (retryCount >= 10)
                    {
                        DownloadComplete?.Invoke(this, new DownloadCompleteEventArgs
                        {
                            Error = $"File not found at {_filePath}"
                        });
                        break;
                    }

                    if (File.Exists(_filePath))
                    {
                        DownloadComplete?.Invoke(this, new DownloadCompleteEventArgs());
                        break;
                    }

                    await Task.Delay(1000 * retryCount);
                    retryCount++;
                }
                return;
            }

            DownloadProgress?.Invoke(this, new DownloadProgressEventArgs
            {
                BytesDone = download.CompletedLength,
                BytesTotal = download.TotalLength,
                Speed = download.DownloadSpeed
            });
        }

        private async Task<Boolean> CheckIfAdded()
        {
            var allDownloads = await _aria2NetClient.TellAll();

            foreach (var download in allDownloads.Where(m => m.Files != null))
            {
                foreach (var file in download.Files.Where(m => m.Uris != null))
                {
                    if (file.Uris.Any(uri => uri.Uri == _uri))
                    {
                        return true;
                    }
                }
            }

            return false;
        }
    }
}