using Ombi.Api.TheMovieDb; using Ombi.Core.Models.Requests; using Ombi.Helpers; using Ombi.Store.Entities; using System; using System.Collections.Generic; using System.Globalization; using System.Linq; using System.Security.Principal; using System.Threading.Tasks; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Logging; using Ombi.Api.TheMovieDb.Models; using Ombi.Core.Authentication; using Ombi.Core.Engine.Interfaces; using Ombi.Core.Rule.Interfaces; using Ombi.Core.Settings; using Ombi.Settings.Settings.Models; using Ombi.Store.Entities.Requests; using Ombi.Store.Repository; namespace Ombi.Core.Engine { public class MovieRequestEngine : BaseMediaEngine, IMovieRequestEngine { public MovieRequestEngine(IMovieDbApi movieApi, IRequestServiceMain requestService, IPrincipal user, INotificationHelper helper, IRuleEvaluator r, IMovieSender sender, ILogger log, OmbiUserManager manager, IRepository rl, ICacheService cache, ISettingsService ombiSettings) : base(user, requestService, r, manager, cache, ombiSettings) { MovieApi = movieApi; NotificationHelper = helper; Sender = sender; Logger = log; _requestLog = rl; } private IMovieDbApi MovieApi { get; } private INotificationHelper NotificationHelper { get; } private IMovieSender Sender { get; } private ILogger Logger { get; } private readonly IRepository _requestLog; /// /// Requests the movie. /// /// The model. /// public async Task RequestMovie(MovieRequestViewModel model) { var movieInfo = await MovieApi.GetMovieInformationWithExtraInfo(model.TheMovieDbId); if (movieInfo == null || movieInfo.Id == 0) { return new RequestEngineResult { Result = false, Message = "There was an issue adding this movie!", ErrorMessage = $"TheMovieDb didn't have any information for ID {model.TheMovieDbId}" }; } var fullMovieName = $"{movieInfo.Title}{(!string.IsNullOrEmpty(movieInfo.ReleaseDate) ? $" ({DateTime.Parse(movieInfo.ReleaseDate).Year})" : string.Empty)}"; var userDetails = await GetUser(); var requestModel = new MovieRequests { TheMovieDbId = movieInfo.Id, RequestType = RequestType.Movie, Overview = movieInfo.Overview, ImdbId = movieInfo.ImdbId, PosterPath = PosterPathHelper.FixPosterPath(movieInfo.PosterPath), Title = movieInfo.Title, ReleaseDate = !string.IsNullOrEmpty(movieInfo.ReleaseDate) ? DateTime.Parse(movieInfo.ReleaseDate) : DateTime.MinValue, Status = movieInfo.Status, RequestedDate = DateTime.UtcNow, Approved = false, RequestedUserId = userDetails.Id, Background = movieInfo.BackdropPath }; var usDates = movieInfo.ReleaseDates?.Results?.FirstOrDefault(x => x.IsoCode == "US"); requestModel.DigitalReleaseDate = usDates?.ReleaseDate?.FirstOrDefault(x => x.Type == ReleaseDateType.Digital)?.ReleaseDate; var ruleResults = (await RunRequestRules(requestModel)).ToList(); if (ruleResults.Any(x => !x.Success)) { return new RequestEngineResult { ErrorMessage = ruleResults.FirstOrDefault(x => x.Message.HasValue()).Message }; } if (requestModel.Approved) // The rules have auto approved this { var requestEngineResult = await AddMovieRequest(requestModel, fullMovieName); if (requestEngineResult.Result) { var result = await ApproveMovie(requestModel); if (result.IsError) { Logger.LogWarning("Tried auto sending movie but failed. Message: {0}", result.Message); return new RequestEngineResult { Message = result.Message, ErrorMessage = result.Message, Result = false }; } return requestEngineResult; } // If there are no providers then it's successful but movie has not been sent } return await AddMovieRequest(requestModel, fullMovieName); } /// /// Gets the requests. /// /// The count. /// The position. /// public async Task> GetRequests(int count, int position) { var shouldHide = await HideFromOtherUsers(); List allRequests; if (shouldHide.Hide) { allRequests = await MovieRepository.GetWithUser(shouldHide.UserId).Skip(position).Take(count).ToListAsync(); } else { allRequests = await MovieRepository.GetWithUser().Skip(position).Take(count).ToListAsync(); } allRequests.ForEach(x => { x.PosterPath = PosterPathHelper.FixPosterPath(x.PosterPath); }); return allRequests; } /// /// Gets the requests. /// /// public async Task> GetRequests() { var shouldHide = await HideFromOtherUsers(); List allRequests; if (shouldHide.Hide) { allRequests = await MovieRepository.GetWithUser(shouldHide.UserId).ToListAsync(); } else { allRequests = await MovieRepository.GetWithUser().ToListAsync(); } return allRequests; } /// /// Searches the movie request. /// /// The search. /// public async Task> SearchMovieRequest(string search) { var shouldHide = await HideFromOtherUsers(); List allRequests; if (shouldHide.Hide) { allRequests = await MovieRepository.GetWithUser(shouldHide.UserId).ToListAsync(); } else { allRequests = await MovieRepository.GetWithUser().ToListAsync(); } var results = allRequests.Where(x => x.Title.Contains(search, CompareOptions.IgnoreCase)).ToList(); results.ForEach(x => { x.PosterPath = PosterPathHelper.FixPosterPath(x.PosterPath); }); return results; } public async Task ApproveMovieById(int requestId) { var request = await MovieRepository.Find(requestId); return await ApproveMovie(request); } public async Task DenyMovieById(int modelId) { var request = await MovieRepository.Find(modelId); if (request == null) { return new RequestEngineResult { ErrorMessage = "Request does not exist" }; } request.Denied = true; // We are denying a request NotificationHelper.Notify(request, NotificationType.RequestDeclined); await MovieRepository.Update(request); return new RequestEngineResult { Message = "Request successfully deleted", }; } public async Task ApproveMovie(MovieRequests request) { if (request == null) { return new RequestEngineResult { ErrorMessage = "Request does not exist" }; } request.Approved = true; request.Denied = false; await MovieRepository.Update(request); var canNotify = await RunSpecificRule(request, SpecificRules.CanSendNotification); if (canNotify.Success) { NotificationHelper.Notify(request, NotificationType.RequestApproved); } if (request.Approved) { var result = await Sender.Send(request); if (result.Success && result.Sent) { return new RequestEngineResult { Result = true }; } if (!result.Success) { Logger.LogWarning("Tried auto sending movie but failed. Message: {0}", result.Message); return new RequestEngineResult { Message = result.Message, ErrorMessage = result.Message, Result = false }; } // If there are no providers then it's successful but movie has not been sent } return new RequestEngineResult { Result = true }; } /// /// Updates the movie request. /// /// The request. /// public async Task UpdateMovieRequest(MovieRequests request) { var allRequests = await MovieRepository.GetWithUser().ToListAsync(); var results = allRequests.FirstOrDefault(x => x.Id == request.Id); results.Approved = request.Approved; results.Available = request.Available; results.Denied = request.Denied; results.DeniedReason = request.DeniedReason; results.ImdbId = request.ImdbId; results.IssueId = request.IssueId; results.Issues = request.Issues; results.Overview = request.Overview; results.PosterPath = PosterPathHelper.FixPosterPath(request.PosterPath); results.QualityOverride = request.QualityOverride; results.RootPathOverride = request.RootPathOverride; await MovieRepository.Update(results); return results; } /// /// Removes the movie request. /// /// The request identifier. /// public async Task RemoveMovieRequest(int requestId) { var request = await MovieRepository.GetAll().FirstOrDefaultAsync(x => x.Id == requestId); await MovieRepository.Delete(request); } public async Task UserHasRequest(string userId) { return await MovieRepository.GetAll().AnyAsync(x => x.RequestedUserId == userId); } public async Task MarkUnavailable(int modelId) { var request = await MovieRepository.Find(modelId); if (request == null) { return new RequestEngineResult { ErrorMessage = "Request does not exist" }; } request.Available = false; await MovieRepository.Update(request); return new RequestEngineResult { Message = "Request is now unavailable", Result = true }; } public async Task MarkAvailable(int modelId) { var request = await MovieRepository.Find(modelId); if (request == null) { return new RequestEngineResult { ErrorMessage = "Request does not exist" }; } request.Available = true; NotificationHelper.Notify(request, NotificationType.RequestAvailable); await MovieRepository.Update(request); return new RequestEngineResult { Message = "Request is now available", Result = true }; } private async Task AddMovieRequest(MovieRequests model, string movieName) { await MovieRepository.Add(model); var result = await RunSpecificRule(model, SpecificRules.CanSendNotification); if (result.Success) { NotificationHelper.NewRequest(model); } await _requestLog.Add(new RequestLog { UserId = (await GetUser()).Id, RequestDate = DateTime.UtcNow, RequestId = model.Id, RequestType = RequestType.Movie, }); return new RequestEngineResult { Result = true, Message = $"{movieName} has been successfully added!" }; } public async Task> Filter(FilterViewModel vm) { var shouldHide = await HideFromOtherUsers(); var requests = shouldHide.Hide ? MovieRepository.GetWithUser(shouldHide.UserId) : MovieRepository.GetWithUser(); switch (vm.AvailabilityFilter) { case FilterType.None: break; case FilterType.Available: requests = requests.Where(x => x.Available); break; case FilterType.NotAvailable: requests = requests.Where(x => !x.Available); break; default: throw new ArgumentOutOfRangeException(); } switch (vm.StatusFilter) { case FilterType.None: break; case FilterType.Approved: requests = requests.Where(x => x.Approved); break; case FilterType.Processing: requests = requests.Where(x => x.Approved && !x.Available); break; case FilterType.PendingApproval: requests = requests.Where(x => !x.Approved && !x.Available && !(x.Denied ?? false)); break; default: throw new ArgumentOutOfRangeException(); } return requests; } } }