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

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

namespace RdtClient.Data.Data
{
    public class TorrentData
    {
        private static IList<Torrent> _torrentCache;
        private static readonly SemaphoreSlim _torrentCacheLock = new(1);

        private readonly DataContext _dataContext;

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

        public async Task<IList<Torrent>> Get()
        {
            await _torrentCacheLock.WaitAsync();

            try
            {
                if (_torrentCache == null)
                {
                    _torrentCache = await _dataContext.Torrents
                                                      .AsNoTracking()
                                                      .Include(m => m.Downloads)
                                                      .ToListAsync();
                }

                return _torrentCache.OrderByDescending(m => m.Added).ToList();
            }
            finally
            {
                _torrentCacheLock.Release();
            }
        }

        public async Task<Torrent> GetById(Guid torrentId)
        {
            var dbTorrent = await _dataContext.Torrents
                                              .AsNoTracking()
                                              .Include(m => m.Downloads)
                                              .FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return null;
            }

            foreach (var file in dbTorrent.Downloads)
            {
                file.Torrent = null;
            }

            return dbTorrent;
        }

        public async Task<Torrent> GetByHash(String hash)
        {
            var dbTorrent = await _dataContext.Torrents
                                              .AsNoTracking()
                                              .Include(m => m.Downloads)
                                              .FirstOrDefaultAsync(m => m.Hash.ToLower() == hash.ToLower());

            if (dbTorrent == null)
            {
                return null;
            }

            foreach (var file in dbTorrent.Downloads)
            {
                file.Torrent = null;
            }

            return dbTorrent;
        }

        public async Task<Torrent> Add(String realDebridId,
                                       String hash,
                                       String category,
                                       TorrentDownloadAction downloadAction,
                                       TorrentFinishedAction finishedAction,
                                       Int32 downloadMinSize,
                                       String downloadManualFiles,
                                       String fileOrMagnetContents,
                                       Boolean isFile)
        {
            var torrent = new Torrent
            {
                TorrentId = Guid.NewGuid(),
                Added = DateTimeOffset.UtcNow,
                RdId = realDebridId,
                Hash = hash.ToLower(),
                Category = category,
                DownloadAction = downloadAction,
                FinishedAction = finishedAction,
                DownloadMinSize = downloadMinSize,
                DownloadManualFiles = downloadManualFiles,
                FileOrMagnet = fileOrMagnetContents,
                IsFile = isFile
            };

            await _dataContext.Torrents.AddAsync(torrent);

            await _dataContext.SaveChangesAsync();

            await VoidCache();

            return torrent;
        }

        public async Task UpdateRdData(Torrent torrent)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrent.TorrentId);

            if (dbTorrent == null)
            {
                return;
            }

            dbTorrent.RdName = torrent.RdName;
            dbTorrent.RdSize = torrent.RdSize;
            dbTorrent.RdHost = torrent.RdHost;
            dbTorrent.RdSplit = torrent.RdSplit;
            dbTorrent.RdProgress = torrent.RdProgress;
            dbTorrent.RdStatus = torrent.RdStatus;
            dbTorrent.RdStatusRaw = torrent.RdStatusRaw;
            dbTorrent.RdAdded = torrent.RdAdded;
            dbTorrent.RdEnded = torrent.RdEnded;
            dbTorrent.RdSpeed = torrent.RdSpeed;
            dbTorrent.RdSeeders = torrent.RdSeeders;

            if (torrent.Files != null)
            {
                dbTorrent.RdFiles = torrent.RdFiles;
            }

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public async Task UpdateCategory(Guid torrentId, String category)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return;
            }

            dbTorrent.Category = category;

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public async Task UpdateComplete(Guid torrentId, DateTimeOffset? datetime)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return;
            }

            dbTorrent.Completed = datetime;

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public async Task UpdateFilesSelected(Guid torrentId, DateTimeOffset datetime)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return;
            }

            dbTorrent.FilesSelected = datetime;

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public async Task UpdateRetryCount(Guid torrentId, Int32 retryCount)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return;
            }

            dbTorrent.RetryCount = retryCount;

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public async Task Delete(Guid torrentId)
        {
            var dbTorrent = await _dataContext.Torrents.FirstOrDefaultAsync(m => m.TorrentId == torrentId);

            if (dbTorrent == null)
            {
                return;
            }

            _dataContext.Torrents.Remove(dbTorrent);

            await _dataContext.SaveChangesAsync();

            await VoidCache();
        }

        public static async Task VoidCache()
        {
            await _torrentCacheLock.WaitAsync();

            try
            {
                _torrentCache = null;
            }
            finally
            {
                _torrentCacheLock.Release();
            }
        }
    }
}