src/Ombi.Core/Engine/BaseMediaEngine.cs

Rama develop · réplica de lectura del repositorio original.

using Ombi.Core.Engine.Interfaces;
using Ombi.Core.Models.Requests;
using Ombi.Helpers;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Principal;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using Ombi.Core.Rule.Interfaces;
using Ombi.Store.Entities.Requests;
using Ombi.Store.Repository.Requests;
using Ombi.Core.Authentication;
using Ombi.Core.Settings;
using Ombi.Settings.Settings.Models;
using Ombi.Store.Entities;
using Ombi.Store.Repository;
using Ombi.Api.External.ExternalApis.TheMovieDb.Models;
using Ombi.Core.Helpers;

namespace Ombi.Core.Engine
{
    public abstract class BaseMediaEngine : BaseEngine
    {
        private long _cacheTime;
        private Dictionary<int, MovieRequests> _dbMovies;
        private Dictionary<int, TvRequests> _dbTv;

        protected BaseMediaEngine(ICurrentUser identity, IRequestServiceMain requestService,
            IRuleEvaluator rules, OmbiUserManager um, ICacheService cache, ISettingsService<OmbiSettings> ombiSettings, IRepository<RequestSubscription> sub) : base(identity, um, rules)
        {
            RequestService = requestService;
            Cache = cache;
            OmbiSettings = ombiSettings;
            _subscriptionRepository = sub;
        }
        private int _resultLimit;
        public int ResultLimit
        {
            get => _resultLimit > 0 ? _resultLimit : 10;
            set => _resultLimit = value;
        }
        protected IRequestServiceMain RequestService { get; }
        protected IMovieRequestRepository MovieRepository => RequestService.MovieRequestService;
        protected ITvRequestRepository TvRepository => RequestService.TvRequestService;
        protected IMusicRequestRepository MusicRepository => RequestService.MusicRequestRepository;
        protected readonly ICacheService Cache;
        protected readonly ISettingsService<OmbiSettings> OmbiSettings;
        protected readonly IRepository<RequestSubscription> _subscriptionRepository;

        private bool _demo = DemoSingleton.Instance.Demo;

        private const long CacheExpiryTicks = 30 * TimeSpan.TicksPerSecond; // 30 seconds

        protected async Task<Dictionary<int, MovieRequests>> GetMovieRequests()
        {
            var now = DateTime.Now.Ticks;
            if (_dbMovies == null || now - _cacheTime > CacheExpiryTicks)
            {
                var allResults = await MovieRepository.GetAll().ToListAsync();

                var distinctResults = allResults.DistinctBy(x => x.TheMovieDbId);
                _dbMovies = distinctResults.ToDictionary(x => x.TheMovieDbId);
                _cacheTime = now;
            }
            return _dbMovies;
        }

        protected async Task<Dictionary<int, TvRequests>> GetTvRequests()
        {
            var now = DateTime.Now.Ticks;
            if (_dbTv == null || now - _cacheTime > CacheExpiryTicks)
            {
                var allResults = await TvRepository.Get().ToListAsync();

                var distinctResults = allResults.DistinctBy(x => x.ExternalProviderId);
                _dbTv = distinctResults.ToDictionary(x => x.ExternalProviderId);
                _cacheTime = now;
            }
            return _dbTv;
        }

        protected async Task<RequestEngineResult> CheckCanManageRequest(BaseRequest request) {
            var errorResult = new RequestEngineResult {
                Result = false,
                ErrorCode = ErrorCode.NoPermissions
            };
            var successResult = new RequestEngineResult { Result = true };
            
            // Admins can always manage requests
            var isAdmin = await IsInRole(OmbiRoles.PowerUser) || await IsInRole(OmbiRoles.Admin);
            if (isAdmin) {
                return successResult;
            }

            // Users with 'ManageOwnRequests' can only manage their own requests
            var canManageOwnRequests = await IsInRole(OmbiRoles.ManageOwnRequests);
            if (!canManageOwnRequests) {
                return errorResult;
            }
            var isRequestedBySameUser = ( await GetUser() ).Id == request.RequestedUser?.Id;
            if (isRequestedBySameUser) {
                return successResult;
            }

            return errorResult;
        }

        public RequestCountModel RequestCount()
        {
            var movieQuery = MovieRepository.GetAll();
            var tvQuery = TvRepository.Get();

            var pendingMovies = movieQuery.Count(x => !x.Approved && !x.Available && !(x.Denied ?? false));
            var approvedMovies = movieQuery.Count(x => x.Approved && !x.Available);
            var availableMovies = movieQuery.Count(x => x.Available);
            var deniedMovies = movieQuery.Count(x => x.Denied ?? false);

            var pendingTv = 0;
            var approvedTv = 0;
            var availableTv = 0;
            var deniedTv = 0;
            foreach (var tv in tvQuery)
            {
                foreach (var child in tv.ChildRequests)
                {
                    if (!child.Approved && !child.Available && !(child.Denied ?? false))
                    {
                        pendingTv++;
                    }
                    if (child.Approved && !child.Available)
                    {
                        approvedTv++;
                    }
                    if (child.Available)
                    {
                        availableTv++;
                    }
                    if (child.Denied ?? false)
                    {
                        deniedTv++;
                    }
                }
            }

            return new RequestCountModel
            {
                Approved = approvedTv + approvedMovies,
                Available = availableTv + availableMovies,
                Pending = pendingMovies + pendingTv,
                Denied = deniedMovies + deniedTv
            };
        }

        protected static IQueryable<T> FilterByRequestedUser<T>(IQueryable<T> requests, string requestedByUserId, bool isAdmin) where T : BaseRequest
        {
            if (!isAdmin || string.IsNullOrEmpty(requestedByUserId))
            {
                return requests;
            }

            return requests.Where(x => x.RequestedUserId == requestedByUserId);
        }

        protected async Task<HideResult> HideFromOtherUsers()
        {
            var user = await GetUser();
            if (await IsInRole(OmbiRoles.Admin) || await IsInRole(OmbiRoles.PowerUser) || user.IsSystemUser)
            {
                return new HideResult
                {
                    UserId = user.Id,
                    IsAdmin = true
                };
            }
            var settings = await Cache.GetOrAddAsync(CacheKeys.OmbiSettings, () => OmbiSettings.GetSettingsAsync());
            var result = new HideResult
            {
                Hide = settings.HideRequestsUsers,
                UserId = user.Id
            };
            return result;
        }

        public async Task SubscribeToRequest(int requestId, RequestType type)
        {
            var user = await GetUser();
            var existingSub = await _subscriptionRepository.GetAll().FirstOrDefaultAsync(x =>
                x.UserId == user.Id && x.RequestId == requestId && x.RequestType == type);
            if (existingSub != null)
            {
                return;
            }
            var sub = new RequestSubscription
            {
                UserId = user.Id,
                RequestId = requestId,
                RequestType = type
            };

            await _subscriptionRepository.Add(sub);
        }

        public async Task UnSubscribeRequest(int requestId, RequestType type)
        {
            var user = await GetUser();
            var existingSub = await _subscriptionRepository.GetAll().FirstOrDefaultAsync(x =>
                x.UserId == user.Id && x.RequestId == requestId && x.RequestType == type);
            if (existingSub != null)
            {
                await _subscriptionRepository.Delete(existingSub);
            }
        }

        protected async Task<string> DefaultLanguageCode(string currentCode)
        {
            if (currentCode.HasValue())
            {
                return currentCode;
            }
            var user = await GetUser();
            if (user == null)
            {
                return "en";
            }

            if (string.IsNullOrEmpty(user.Language))
            {
                var s = await GetOmbiSettings();
                return s.DefaultLanguageCode;
            }

            return user.Language;
        }

        protected async Task<List<StreamData>> GetUserWatchProvider(WatchProviders providers)
        {
            var user = await GetUser();
            return WatchProviderParser.GetUserWatchProviders(providers, user);
        }

        private OmbiSettings ombiSettings;
        protected async Task<OmbiSettings> GetOmbiSettings()
        {
            return ombiSettings ?? (ombiSettings = await OmbiSettings.GetSettingsAsync());
        }

        protected bool DemoCheck(string title)
        {
            if (!title.HasValue())
            {
                return false;
            }
            if (_demo)
            {
                if (ExcludedDemo.ExcludedContent.Any(x => title.Contains(x, System.Globalization.CompareOptions.OrdinalIgnoreCase)))
                {
                    return true;
                }
                return false;
            }
            return false;
        }

        public class HideResult
        {
            public bool Hide { get; set; }
            public string UserId { get; set; }
            public bool IsAdmin { get; set; }
        }
    }
}