rdt-client/server/RdtClient.Data/Data/DownloadData.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using Download = RdtClient.Data.Models.Data.Download;

namespace RdtClient.Data.Data
{
    public interface IDownloadData
    {
        Task<Download> GetById(Guid downloadId);
        Task<Download> Get(Guid torrentId, String path);
        Task<Download> Add(Guid torrentId, String path);
        Task UpdateUnrestrictedLink(Guid downloadId, String unrestrictedLink);
        Task UpdateDownloadStarted(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateDownloadFinished(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateUnpackingQueued(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateUnpackingStarted(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateUnpackingFinished(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateCompleted(Guid downloadId, DateTimeOffset? dateTime);
        Task UpdateError(Guid downloadId, String error);
        Task UpdateRetryCount(Guid downloadId, Int32 retryCount);
        Task DeleteForTorrent(Guid torrentId);
    }

    public class DownloadData : IDownloadData
    {
        private readonly DataContext _dataContext;

        public DownloadData(DataContext dataContext)
        {
            _dataContext = dataContext;
        }

        public async Task<IList<Download>> Get()
        {
            return await _dataContext.Downloads
                                     .AsNoTracking()
                                     .Include(m => m.Torrent)
                                     .ToListAsync();
        }

        public async Task<IList<Download>> GetForTorrent(Guid torrentId)
        {
            return await _dataContext.Downloads
                                     .AsNoTracking()
                                     .Where(m => m.TorrentId == torrentId)
                                     .ToListAsync();
        }

        public async Task<Download> GetById(Guid downloadId)
        {
            return await _dataContext.Downloads
                                     .Include(m => m.Torrent)
                                     .AsNoTracking()
                                     .FirstOrDefaultAsync(m => m.DownloadId == downloadId);
        }

        public async Task<Download> Get(Guid torrentId, String path)
        {
            return await _dataContext.Downloads.FirstOrDefaultAsync(m => m.TorrentId == torrentId && m.Path == path);
        }

        public async Task<Download> Add(Guid torrentId, String path)
        {
            var download = new Download
            {
                DownloadId = Guid.NewGuid(),
                TorrentId = torrentId,
                Path = path,
                Added = DateTimeOffset.UtcNow,
                DownloadQueued = DateTimeOffset.UtcNow,
                RetryCount = 0
            };

            await _dataContext.Downloads.AddAsync(download);

            await _dataContext.SaveChangesAsync();

            return download;
        }

        public async Task UpdateUnrestrictedLink(Guid downloadId, String unrestrictedLink)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.Link = unrestrictedLink;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateDownloadStarted(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.DownloadStarted = dateTime;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateDownloadFinished(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            if (dbDownload == null)
            {
                return;
            }
            
            dbDownload.DownloadFinished = dateTime;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateUnpackingQueued(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.UnpackingQueued = dateTime;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateUnpackingStarted(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.UnpackingStarted = dateTime;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateUnpackingFinished(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.UnpackingFinished = dateTime;

            await _dataContext.SaveChangesAsync();
        }
        
        public async Task UpdateCompleted(Guid downloadId, DateTimeOffset? dateTime)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.Completed = dateTime;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateError(Guid downloadId, String error)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.Error = error;

            await _dataContext.SaveChangesAsync();
        }

        public async Task UpdateRetryCount(Guid downloadId, Int32 retryCount)
        {
            var dbDownload = await _dataContext.Downloads
                                               .FirstOrDefaultAsync(m => m.DownloadId == downloadId);

            dbDownload.RetryCount = retryCount;

            await _dataContext.SaveChangesAsync();
        }

        public async Task DeleteForTorrent(Guid torrentId)
        {
            var downloads = await _dataContext.Downloads
                                              .Where(m => m.TorrentId == torrentId)
                                              .ToListAsync();

            _dataContext.Downloads.RemoveRange(downloads);

            await _dataContext.SaveChangesAsync();
        }
    }
}