using System; using AutoMapper; using System.Collections.Generic; using System.Linq; using System.Security.Principal; using System.Threading.Tasks; using Ombi.Core.Rule.Interfaces; using Ombi.Store.Repository.Requests; using Ombi.Core.Authentication; using Ombi.Helpers; using Ombi.Settings.Settings.Models; using Ombi.Store.Entities; using Ombi.Api.Trakt; using Ombi.Api.TvMaze; using Ombi.Core.Models.Requests; using Ombi.Core.Models.Search; using Ombi.Core.Models.Search.V2; using Ombi.Core.Settings; using Ombi.Store.Repository; using TraktSharp.Entities; using Microsoft.EntityFrameworkCore; using System.Threading; using Ombi.Api.TheMovieDb; using Ombi.Api.TheMovieDb.Models; namespace Ombi.Core.Engine.V2 { public class TvSearchEngineV2 : BaseMediaEngine, ITVSearchEngineV2 { private readonly ITvMazeApi _tvMaze; private readonly IMapper _mapper; private readonly ITraktApi _traktApi; private readonly IMovieDbApi _movieApi; private readonly ISettingsService _customization; public TvSearchEngineV2(IPrincipal identity, IRequestServiceMain service, ITvMazeApi tvMaze, IMapper mapper, ITraktApi trakt, IRuleEvaluator r, OmbiUserManager um, ICacheService memCache, ISettingsService s, IRepository sub, IMovieDbApi movieApi, ISettingsService customization) : base(identity, service, r, um, memCache, s, sub) { _tvMaze = tvMaze; _mapper = mapper; _traktApi = trakt; _movieApi = movieApi; _customization = customization; } public async Task GetShowByRequest(int requestId, CancellationToken token) { var request = await RequestService.TvRequestService.Get().FirstOrDefaultAsync(x => x.Id == requestId); return await GetShowInformation(request.ExternalProviderId.ToString(), token); // TODO } public async Task GetShowInformation(string tvdbid, CancellationToken token) { var show = await Cache.GetOrAdd(nameof(GetShowInformation) + tvdbid, async () => await _movieApi.GetTVInfo(tvdbid), DateTime.Now.AddHours(12)); if (show == null || show.name == null) { // We don't have enough information return null; } var mapped = _mapper.Map(show); foreach (var tvSeason in show.seasons.Where(x => x.season_number != 0)) // skip the first season { var seasonEpisodes = (await _movieApi.GetSeasonEpisodes(show.id, tvSeason.season_number, token)); foreach (var episode in seasonEpisodes.episodes) { var season = mapped.SeasonRequests.FirstOrDefault(x => x.SeasonNumber == episode.season_number); if (season == null) { var newSeason = new SeasonRequests { SeasonNumber = episode.season_number, Overview = tvSeason.overview, Episodes = new List() }; newSeason.Episodes.Add(new EpisodeRequests { //Url = episode...ToHttpsUrl(), Title = episode.name, AirDate = episode.air_date.HasValue() ? DateTime.Parse(episode.air_date) : DateTime.MinValue, EpisodeNumber = episode.episode_number, }); mapped.SeasonRequests.Add(newSeason); } else { // We already have the season, so just add the episode season.Episodes.Add(new EpisodeRequests { //Url = e.url.ToHttpsUrl(), Title = episode.name, AirDate = episode.air_date.HasValue() ? DateTime.Parse(episode.air_date) : DateTime.MinValue, EpisodeNumber = episode.episode_number, }); } } } return await ProcessResult(mapped); } public async Task> Popular(int currentlyLoaded, int amountToLoad) { var langCode = await DefaultLanguageCode(null); var pages = PaginationHelper.GetNextPages(currentlyLoaded, amountToLoad, ResultLimit); var results = new List(); foreach (var pagesToLoad in pages) { var apiResult = await Cache.GetOrAdd(nameof(Popular) + langCode + pagesToLoad.Page, async () => await _movieApi.PopularTv(langCode, pagesToLoad.Page), DateTime.Now.AddHours(12)); results.AddRange(apiResult.Skip(pagesToLoad.Skip).Take(pagesToLoad.Take)); } var processed = ProcessResults(results); return await processed; } public async Task> Anticipated(int currentlyLoaded, int amountToLoad) { var langCode = await DefaultLanguageCode(null); var pages = PaginationHelper.GetNextPages(currentlyLoaded, amountToLoad, ResultLimit); var results = new List(); foreach (var pagesToLoad in pages) { var apiResult = await Cache.GetOrAdd(nameof(Anticipated) + langCode + pagesToLoad.Page, async () => await _movieApi.UpcomingTv(langCode, pagesToLoad.Page), DateTime.Now.AddHours(12)); results.AddRange(apiResult.Skip(pagesToLoad.Skip).Take(pagesToLoad.Take)); } var processed = ProcessResults(results); return await processed; } public async Task> Trending(int currentlyLoaded, int amountToLoad) { var langCode = await DefaultLanguageCode(null); var pages = PaginationHelper.GetNextPages(currentlyLoaded, amountToLoad, ResultLimit); var results = new List(); foreach (var pagesToLoad in pages) { var apiResult = await Cache.GetOrAdd(nameof(Trending) + langCode + pagesToLoad.Page, async () => await _movieApi.TopRatedTv(langCode, pagesToLoad.Page), DateTime.Now.AddHours(12)); results.AddRange(apiResult.Skip(pagesToLoad.Skip).Take(pagesToLoad.Take)); } var processed = ProcessResults(results); return await processed; } public async Task> GetStreamInformation(int movieDbId, CancellationToken cancellationToken) { var providers = await _movieApi.GetTvWatchProviders(movieDbId, cancellationToken); var results = await GetUserWatchProvider(providers); var data = new List(); foreach (var result in results) { data.Add(new StreamingData { Logo = result.logo_path, Order = result.display_priority, StreamingProvider = result.provider_name }); } return data; } private async Task> ProcessResults(IEnumerable items) { var retVal = new List(); var settings = await _customization.GetSettingsAsync(); foreach (var tvMazeSearch in items) { var result = await ProcessResult(tvMazeSearch); if (result == null || settings.HideAvailableFromDiscover && result.Available) { continue; } retVal.Add(result); } return retVal; } private async Task ProcessResult(T tvMazeSearch) { var item = _mapper.Map(tvMazeSearch); await RunSearchRules(item); return item; } private async Task ProcessResult(SearchFullInfoTvShowViewModel item) { item.TheMovieDbId = item.Id.ToString(); var oldModel = _mapper.Map(item); await RunSearchRules(oldModel); item.Available = oldModel.Available; item.FullyAvailable = oldModel.FullyAvailable; item.PartlyAvailable = oldModel.PartlyAvailable; item.Requested = oldModel.Requested; item.Available = oldModel.Available; item.Approved = oldModel.Approved; item.SeasonRequests = oldModel.SeasonRequests; item.RequestId = oldModel.RequestId; if (!string.IsNullOrEmpty(item.Images?.Medium)) { item.Images.Medium = item.Images.Medium.ToHttpsUrl(); } return item; //return await GetExtraInfo(showInfoTask, item); } private async Task GetExtraInfo(Task showInfoTask, SearchFullInfoTvShowViewModel model) { if (showInfoTask != null) { var result = await showInfoTask; if (result == null) { return model; } model.Trailer = result.Trailer?.AbsoluteUri.ToHttpsUrl() ?? string.Empty; model.Certification = result.Certification; model.Homepage = result.Homepage?.AbsoluteUri.ToHttpsUrl() ?? string.Empty; } return model; } } }