The move!

src/Ombi.Core/Engine/TvRequestEngine.cs

@@ -0,0 +1,253 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Linq;
+using System.Security.Principal;
+using System.Threading.Tasks;
+using AutoMapper;
+using Hangfire;
+using Ombi.Api.TvMaze;
+using Ombi.Core.Models.Requests;
+using Ombi.Core.Models.Search;
+using Ombi.Store.Entities;
+using Ombi.Helpers;
+using Ombi.Notifications;
+using Ombi.Notifications.Models;
+
+namespace Ombi.Core.Engine
+{
+    public class TvRequestEngine : BaseMediaEngine, ITvRequestEngine
+    {
+        public TvRequestEngine(ITvMazeApi tvApi, IRequestServiceMain requestService, IPrincipal user, INotificationService notificationService, IMapper map) : base(user, requestService)
+        {
+            TvApi = tvApi;
+            NotificationService = notificationService;
+            Mapper = map;
+        }
+        private INotificationService NotificationService { get; }
+        private ITvMazeApi TvApi { get; }
+        private IMapper Mapper { get; }
+
+        public async Task<RequestEngineResult> RequestTvShow(SearchTvShowViewModel tv)
+        {
+
+            var showInfo = await TvApi.ShowLookupByTheTvDbId(tv.Id);
+            DateTime.TryParse(showInfo.premiered, out DateTime firstAir);
+            
+            // For some reason the poster path is always http
+            var posterPath = showInfo.image?.medium.Replace("http:", "https:");
+            var model = new TvRequestModel
+            {
+                Id = tv.Id,
+                Type = RequestType.TvShow,
+                Overview = showInfo.summary.RemoveHtml(),
+                PosterPath = posterPath,
+                Title = showInfo.name,
+                ReleaseDate = firstAir,
+                Status = showInfo.status,
+                RequestedDate = DateTime.UtcNow,
+                Approved = false,
+                RequestedUsers = new List<string> { Username },
+                Issues = IssueState.None,
+                ImdbId = showInfo.externals?.imdb ?? string.Empty,
+                TvDbId = tv.Id.ToString(),
+                ProviderId = tv.Id,
+                RequestAll = tv.RequestAll
+            };
+
+            var episodes = await TvApi.EpisodeLookup(showInfo.id);
+
+            foreach (var e in episodes)
+            {
+                var season = model.SeasonRequests.FirstOrDefault(x => x.SeasonNumber == e.season);
+                season?.Episodes.Add(new EpisodesRequested
+                {
+                    Url = e.url,
+                    Title = e.name,
+                    AirDate = DateTime.Parse(e.airstamp),
+                    EpisodeNumber = e.number,
+                });
+            }
+
+            if (tv.LatestSeason)
+            {
+                var latest = showInfo.Season.OrderBy(x => x.SeasonNumber).FirstOrDefault();
+                foreach (var modelSeasonRequest in model.SeasonRequests)
+                {
+                    if (modelSeasonRequest.SeasonNumber == latest.SeasonNumber)
+                    {
+                        foreach (var episodesRequested in modelSeasonRequest.Episodes)
+                        {
+                            episodesRequested.Requested = true;
+                        }
+                    }
+                }
+            }
+            if (tv.FirstSeason)
+            {
+                var first = showInfo.Season.OrderByDescending(x => x.SeasonNumber).FirstOrDefault();
+                foreach (var modelSeasonRequest in model.SeasonRequests)
+                {
+                    if (modelSeasonRequest.SeasonNumber == first.SeasonNumber)
+                    {
+                        foreach (var episodesRequested in modelSeasonRequest.Episodes)
+                        {
+                            episodesRequested.Requested = true;
+                        }
+                    }
+                }
+            }
+
+
+            var existingRequest = await TvRequestService.CheckRequestAsync(model.Id);
+            if (existingRequest != null)
+            {
+                return await AddExistingRequest(model, existingRequest);
+            }
+
+            // This is a new request
+            return await AddRequest(model);
+        }
+
+        public async Task<IEnumerable<TvRequestModel>> GetTvRequests(int count, int position)
+        {
+            var allRequests = await TvRequestService.GetAllAsync(count, position);
+            return allRequests;
+        }
+        public async Task<IEnumerable<TvRequestModel>> SearchTvRequest(string search)
+        {
+            var allRequests = await TvRequestService.GetAllAsync();
+            var results = allRequests.Where(x => x.Title.Contains(search, CompareOptions.IgnoreCase));
+            return results;
+        }
+        public async Task<TvRequestModel> UpdateTvRequest(TvRequestModel request)
+        {
+            var allRequests = await TvRequestService.GetAllAsync();
+            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.AdminNote = request.AdminNote;
+            results.ImdbId = request.ImdbId;
+            results.IssueId = request.IssueId;
+            results.Issues = request.Issues;
+            results.OtherMessage = request.OtherMessage;
+            results.Overview = request.Overview;
+            results.PosterPath = request.PosterPath;
+            results.RequestedUsers = request.RequestedUsers?.ToList() ?? new List<string>();
+
+            var model = TvRequestService.UpdateRequest(results);
+            return model;
+        }
+
+        public async Task RemoveTvRequest(int requestId)
+        {
+            await TvRequestService.DeleteRequestAsync(requestId);
+        }
+
+        private async Task<RequestEngineResult> AddExistingRequest(TvRequestModel newRequest, TvRequestModel existingRequest)
+        {
+            var episodeDifference = new List<SeasonRequestModel>();
+            if (existingRequest.HasChildRequests)
+            {
+                // Let's check if this has already been requested as a child!
+                foreach (var children in existingRequest.ChildRequests)
+                {
+                    var difference = GetListDifferences(children.SeasonRequests, newRequest.SeasonRequests).ToList();
+                    if (difference.Any())
+                    {
+                        episodeDifference = difference;
+                    }
+                }
+            }
+
+            if (episodeDifference.Any())
+            {
+                // This is where there are some episodes that have been requested, but this list contains the 'new' requests
+                newRequest.SeasonRequests = episodeDifference;
+            }
+
+            if (!existingRequest.HasChildRequests)
+            {
+                // So this is the first child request, we will want to convert the original request to a child
+                var originalRequest = Mapper.Map<TvRequestModel>(existingRequest);
+                existingRequest.ChildRequests.Add(originalRequest);
+                existingRequest.RequestedUsers.Clear();
+                existingRequest.Approved = false;
+                existingRequest.Available = false;
+            }
+
+            existingRequest.ChildRequests.Add(newRequest);
+
+            TvRequestService.UpdateRequest(existingRequest);
+
+            if (ShouldAutoApprove(RequestType.TvShow))
+            {
+                // TODO Auto Approval Code
+            }
+            return await AfterRequest(newRequest);
+        }
+
+        private IEnumerable<SeasonRequestModel> GetListDifferences(IEnumerable<SeasonRequestModel> existing, IEnumerable<SeasonRequestModel> request)
+        {
+            var newRequest = request
+                .Select(r =>
+                    new SeasonRequestModel
+                    {
+                        SeasonNumber = r.SeasonNumber,
+                        Episodes = r.Episodes
+                    }).ToList();
+
+            return newRequest.Except(existing);
+        }
+
+        private async Task<RequestEngineResult> AddRequest(TvRequestModel model)
+        {
+            await TvRequestService.AddRequestAsync(model);
+
+           return await AfterRequest(model);
+        }
+
+        private async Task<RequestEngineResult> AfterRequest(BaseRequestModel model)
+        {
+            if (ShouldSendNotification(model.Type))
+            {
+                var notificationModel = new NotificationModel
+                {
+                    Title = model.Title,
+                    User = Username,
+                    DateTime = DateTime.Now,
+                    NotificationType = NotificationType.NewRequest,
+                    RequestType = model.Type,
+                    ImgSrc = model.PosterPath
+                };
+
+                BackgroundJob.Enqueue(() => NotificationService.Publish(notificationModel).Wait());
+            }
+
+            //var limit = await RequestLimitRepo.GetAllAsync();
+            //var usersLimit = limit.FirstOrDefault(x => x.Username == Username && x.RequestType == model.Type);
+            //if (usersLimit == null)
+            //{
+            //    await RequestLimitRepo.InsertAsync(new RequestLimit
+            //    {
+            //        Username = Username,
+            //        RequestType = model.Type,
+            //        FirstRequestDate = DateTime.UtcNow,
+            //        RequestCount = 1
+            //    });
+            //}
+            //else
+            //{
+            //    usersLimit.RequestCount++;
+            //    await RequestLimitRepo.UpdateAsync(usersLimit);
+            //}
+
+            return new RequestEngineResult { RequestAdded = true };
+        }
+
+
+    }
+}