Skip to content

Commit

Permalink
Got mostly everything working for #254 Ready for testing
Browse files Browse the repository at this point in the history
  • Loading branch information
tidusjar committed Aug 7, 2016
1 parent 131a99b commit 0699d77
Show file tree
Hide file tree
Showing 7 changed files with 83 additions and 39 deletions.
3 changes: 2 additions & 1 deletion PlexRequests.Api.Interfaces/ISonarrApi.cs
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@ public interface ISonarrApi
List<SonarrProfile> GetProfiles(string apiKey, Uri baseUrl);

SonarrAddSeries AddSeries(int tvdbId, string title, int qualityId, bool seasonFolders, string rootPath,
int seasonCount, int[] seasons, string apiKey, Uri baseUrl, bool monitored = true);
int seasonCount, int[] seasons, string apiKey, Uri baseUrl, bool monitor = true,
bool searchForMissingEpisodes = false);

SystemStatus SystemStatus(string apiKey, Uri baseUrl);

Expand Down
17 changes: 13 additions & 4 deletions PlexRequests.Api/SonarrApi.cs
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ public List<SonarrProfile> GetProfiles(string apiKey, Uri baseUrl)
return obj;
}

public SonarrAddSeries AddSeries(int tvdbId, string title, int qualityId, bool seasonFolders, string rootPath, int seasonCount, int[] seasons, string apiKey, Uri baseUrl, bool monitor = true)
public SonarrAddSeries AddSeries(int tvdbId, string title, int qualityId, bool seasonFolders, string rootPath, int seasonCount, int[] seasons, string apiKey, Uri baseUrl, bool monitor = true, bool searchForMissingEpisodes = false)
{
Log.Debug("Adding series {0}", title);
Log.Debug("Seasons = {0}, out of {1} seasons", seasons.DumpJson(), seasonCount);
Expand All @@ -86,6 +86,16 @@ public SonarrAddSeries AddSeries(int tvdbId, string title, int qualityId, bool s
monitored = monitor
};

if (!searchForMissingEpisodes)
{
options.addOptions = new AddOptions
{
searchForMissingEpisodes = false,
ignoreEpisodesWithFiles = true,
ignoreEpisodesWithoutFiles = true
};
}

for (var i = 1; i <= seasonCount; i++)
{
var season = new Season
Expand All @@ -107,9 +117,8 @@ public SonarrAddSeries AddSeries(int tvdbId, string title, int qualityId, bool s
try
{
var policy = RetryHandler.RetryAndWaitPolicy((exception, timespan) => Log.Error(exception, "Exception when calling AddSeries for Sonarr, Retrying {0}", timespan), new TimeSpan[] {
TimeSpan.FromSeconds (2),
TimeSpan.FromSeconds(5),
TimeSpan.FromSeconds(10)
TimeSpan.FromSeconds (1),
TimeSpan.FromSeconds(2),
});

result = policy.Execute(() => Api.ExecuteJson<SonarrAddSeries>(request, baseUrl));
Expand Down
25 changes: 18 additions & 7 deletions PlexRequests.Helpers/PlexHelper.cs
Original file line number Diff line number Diff line change
Expand Up @@ -26,17 +26,20 @@
#endregion

using System;
using NLog;

namespace PlexRequests.Helpers
{
public class PlexHelper
{

private static Logger Log = LogManager.GetCurrentClassLogger();
public static string GetProviderIdFromPlexGuid(string guid)
{
if (string.IsNullOrEmpty(guid))
return guid;

var guidSplit = guid.Split(new[] {'/', '?'}, StringSplitOptions.RemoveEmptyEntries);
var guidSplit = guid.Split(new[] { '/', '?' }, StringSplitOptions.RemoveEmptyEntries);
if (guidSplit.Length > 1)
{
return guidSplit[1];
Expand All @@ -50,15 +53,23 @@ public static EpisodeModelHelper GetSeasonsAndEpisodesFromPlexGuid(string guid)
//guid="com.plexapp.agents.thetvdb://269586/2/8?lang=en"
if (string.IsNullOrEmpty(guid))
return null;
try
{
var guidSplit = guid.Split(new[] { '/', '?' }, StringSplitOptions.RemoveEmptyEntries);
if (guidSplit.Length > 2)
{
ep.ProviderId = guidSplit[1];
ep.SeasonNumber = int.Parse(guidSplit[2]);
ep.EpisodeNumber = int.Parse(guidSplit[3]);
}
return ep;

var guidSplit = guid.Split(new[] { '/', '?' }, StringSplitOptions.RemoveEmptyEntries);
if (guidSplit.Length > 2)
}
catch (Exception e)
{
ep.ProviderId = guidSplit[1];
ep.SeasonNumber = int.Parse(guidSplit[2]);
ep.EpisodeNumber = int.Parse(guidSplit[3]);
Log.Error(e);
return ep;
}
return ep;
}
}

Expand Down
22 changes: 21 additions & 1 deletion PlexRequests.Store/RequestedModel.cs
Original file line number Diff line number Diff line change
Expand Up @@ -108,9 +108,29 @@ public enum IssueState
Other = 4, // Provide a message
}

public class EpisodesModel
public class EpisodesModel : IEquatable<EpisodesModel>
{
public int SeasonNumber { get; set; }
public int EpisodeNumber { get; set; }
public bool Equals(EpisodesModel other)
{
// Check whether the compared object is null.
if (ReferenceEquals(other, null)) return false;

//Check whether the compared object references the same data.
if (ReferenceEquals(this, other)) return true;

//Check whether the properties are equal.
return SeasonNumber.Equals(other.SeasonNumber) && EpisodeNumber.Equals(other.EpisodeNumber);
}

public override int GetHashCode()
{
var hashSeason = SeasonNumber.GetHashCode();
var hashEp = EpisodeNumber.GetHashCode();

//Calculate the hash code.
return hashSeason + hashEp;
}
}
}
1 change: 1 addition & 0 deletions PlexRequests.UI/Content/requests.js
Original file line number Diff line number Diff line change
Expand Up @@ -592,6 +592,7 @@ function tvLoad() {
context.episodes = tvObject;
var html = searchTemplate(context);
$tvl.append(html);
tvObject = new Array();

});

Expand Down
36 changes: 15 additions & 21 deletions PlexRequests.UI/Helpers/TvSender.cs
Original file line number Diff line number Diff line change
Expand Up @@ -75,14 +75,15 @@ public async Task<SonarrAddSeries> SendToSonarr(SonarrSettings sonarrSettings, R
var seriesTask = GetSonarrSeries(sonarrSettings, model.ProviderId);

if (episodeRequest)
{
{
// Does series exist?
var series = await seriesTask;
if (series != null)
{
// Series Exists
// Request the episodes in the existing series
await RequestEpisodesWithExistingSeries(model, series, sonarrSettings);
return new SonarrAddSeries {title = series.title};
}
else
{
Expand All @@ -91,29 +92,22 @@ public async Task<SonarrAddSeries> SendToSonarr(SonarrSettings sonarrSettings, R
sonarrSettings.SeasonFolders, sonarrSettings.RootPath, 0, new int[0], sonarrSettings.ApiKey,
sonarrSettings.FullUri, false));

if (string.IsNullOrEmpty(addResult?.title))

// Get the series that was just added
series = await GetSonarrSeries(sonarrSettings, model.ProviderId);
series.monitored = false; // Un-monitor the series

// Un-monitor all seasons
foreach (var season in series.seasons)
{
var sw = new Stopwatch();
sw.Start();
while (series == null)
{
await Task.Delay(TimeSpan.FromSeconds(5));
series = await GetSonarrSeries(sonarrSettings, model.ProviderId);

// Check how long we have been doing this for
if (sw.Elapsed > TimeSpan.FromSeconds(30))
{
// 30 seconds is a long time, it's not going to work.
throw new ApiRequestException("Sonarr still didn't have the series added after 30 seconds.");
}
}
sw.Stop();

// Update the series, Since we cannot add as unmonitoed due to the following bug: https://github.com/Sonarr/Sonarr/issues/1404
SonarrApi.UpdateSeries(series, sonarrSettings.ApiKey, sonarrSettings.FullUri);
season.monitored = false;
}

// We now have the series in Sonarr
// Update the series, Since we cannot add as un-monitored due to the following bug: https://github.com/Sonarr/Sonarr/issues/1404
SonarrApi.UpdateSeries(series, sonarrSettings.ApiKey, sonarrSettings.FullUri);


// We now have the series in Sonarr, update it to request the episodes.
await RequestEpisodesWithExistingSeries(model, series, sonarrSettings);

return addResult;
Expand Down
18 changes: 13 additions & 5 deletions PlexRequests.UI/Modules/SearchModule.cs
Original file line number Diff line number Diff line change
Expand Up @@ -585,8 +585,16 @@ private async Task<Response> RequestTvShow(int showId, string seasons)
if (episodeRequest)
{
difference = GetListDifferences(existingRequest.Episodes, episodeModel.Episodes).ToList();
if (!difference.Any())
if (difference.Any())
{
existingRequest.Episodes = episodeModel.Episodes
.Select(r =>
new EpisodesModel
{
SeasonNumber = r.SeasonNumber,
EpisodeNumber = r.EpisodeNumber
}).ToList();

return await AddUserToRequest(existingRequest, settings, fullShowName);
}
// We have an episode that has not yet been requested, let's continue
Expand Down Expand Up @@ -730,7 +738,7 @@ private async Task<Response> RequestTvShow(int showId, string seasons)
private async Task<Response> AddUserToRequest(RequestedModel existingRequest, PlexRequestSettings settings, string fullShowName)
{
// check if the current user is already marked as a requester for this show, if not, add them
if (!existingRequest.UserHasRequested(Username))
if (!existingRequest.UserHasRequested(Username) || existingRequest.Episodes.Any())
{
existingRequest.RequestedUsers.Add(Username);
await RequestService.UpdateRequestAsync(existingRequest);
Expand Down Expand Up @@ -1092,17 +1100,17 @@ await RequestLimitRepo.InsertAsync(new RequestLimit
return Response.AsJson(new JsonResponseModel { Result = true, Message = message });
}

private IEnumerable<Store.EpisodesModel> GetListDifferences(IEnumerable<Store.EpisodesModel> model, IEnumerable<Models.EpisodesModel> request)
private IEnumerable<Store.EpisodesModel> GetListDifferences(IEnumerable<EpisodesModel> existing, IEnumerable<Models.EpisodesModel> request)
{
var newRequest = request
.Select(r =>
new Store.EpisodesModel
new EpisodesModel
{
SeasonNumber = r.SeasonNumber,
EpisodeNumber = r.EpisodeNumber
}).ToList();

return newRequest.Except(model);
return newRequest.Except(existing);
}
}
}

0 comments on commit 0699d77

Please sign in to comment.