Ombi/src/Ombi.Core/Rule/Rules/Request/ExistingMovieRequestRule.cs

using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using Ombi.Core.Rule.Interfaces;
using Ombi.Helpers;
using Ombi.Store.Entities;
using Ombi.Store.Entities.Requests;
using Ombi.Core.Engine;
using Ombi.Store.Repository.Requests;
using Ombi.Core.Services;
using Ombi.Settings.Settings.Models;

namespace Ombi.Core.Rule.Rules.Request
{
    public class ExistingMovieRequestRule : BaseRequestRule, IRules<BaseRequest>
    {
        private readonly IFeatureService _featureService;

        public ExistingMovieRequestRule(IMovieRequestRepository movie, IFeatureService featureService)
        {
            Movie = movie;
            _featureService = featureService;
        }

        private IMovieRequestRepository Movie { get; }

        /// <summary>
        /// We check if the request exists, if it does then we don't want to re-request it.
        /// </summary>
        /// <param name="obj">The object.</param>
        /// <returns></returns>
        public async Task<RuleResult> Execute(BaseRequest obj)
        {
            if (obj.RequestType == RequestType.Movie)
            {
                var movie = (MovieRequests) obj;
                var movieRequests = Movie.GetAll();
                var found = false;
                var existing = await movieRequests.FirstOrDefaultAsync(x => x.TheMovieDbId == movie.TheMovieDbId);
                if (existing != null) // Do we already have a request for this?
                {
                    found = await Check4KRequests(movie, existing);
                }

                if (!found && movie.ImdbId.HasValue())
                {
                   // Let's check imdbid
                   existing = await movieRequests.FirstOrDefaultAsync(x =>
                       x.ImdbId == movie.ImdbId);
                   if (existing != null)
                   {
                       found = await Check4KRequests(movie, existing);
                   }
                }
                if (found)
                {
                    return Fail(ErrorCode.AlreadyRequested, $"\"{obj.Title}\" has already been requested");
                }
            }
            return Success();
        }

        private async Task<bool> Check4KRequests(MovieRequests movie, MovieRequests existing)
        {
            var featureEnabled = await _featureService.FeatureEnabled(FeatureNames.Movie4KRequests);
            if (movie.Is4kRequest && existing.Has4KRequest && featureEnabled)
            {
               return true;
            }
            if (!movie.Is4kRequest && !existing.Has4KRequest || !featureEnabled)
            {
                return true;
            }

            return false;
        }
    }
}