Attempting to improve #219

We now have a new setting "Advanced Search". What this does is for each item  in the plex library we will get the ratingKey and perform another API call to get the MetaData for that ratingKey, inside the metadata it contains the "guid", the guid contains either a IMDBId or TheTVDBID. We will then use that to match the request. This does mean that we now need to capture the TvDbId for each TV Request.
This commit is contained in:
tidusjar 2016-06-09 22:13:43 +01:00
parent 5bf557658d
commit fefad77ac1
15 changed files with 311 additions and 30 deletions

View file

@ -40,5 +40,6 @@ namespace PlexRequests.Api.Interfaces
PlexAccount GetAccount(string authToken); PlexAccount GetAccount(string authToken);
PlexLibraries GetLibrarySections(string authToken, Uri plexFullHost); PlexLibraries GetLibrarySections(string authToken, Uri plexFullHost);
PlexSearch GetLibrary(string authToken, Uri plexFullHost, string libraryId); PlexSearch GetLibrary(string authToken, Uri plexFullHost, string libraryId);
PlexMetadata GetMetadata(string authToken, Uri plexFullHost, string itemId);
} }
} }

View file

@ -0,0 +1,58 @@
#region Copyright
// /************************************************************************
// Copyright (c) 2016 Jamie Rees
// File: PlexMetadata.cs
// Created By: Jamie Rees
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
// ************************************************************************/
#endregion
using System.Collections.Generic;
using System.Xml.Serialization;
namespace PlexRequests.Api.Models.Plex
{
[XmlRoot(ElementName = "MediaContainer")]
public class PlexMetadata
{
[XmlElement(ElementName= "Video")]
public Video Video { get; set; }
[XmlElement(ElementName = "Directory")]
public Directory1 Directory { get; set; }
[XmlAttribute(AttributeName = "size")]
public string Size { get; set; }
[XmlAttribute(AttributeName = "allowSync")]
public string AllowSync { get; set; }
[XmlAttribute(AttributeName = "identifier")]
public string Identifier { get; set; }
[XmlAttribute(AttributeName = "librarySectionID")]
public string LibrarySectionID { get; set; }
[XmlAttribute(AttributeName = "librarySectionTitle")]
public string LibrarySectionTitle { get; set; }
[XmlAttribute(AttributeName = "librarySectionUUID")]
public string LibrarySectionUUID { get; set; }
[XmlAttribute(AttributeName = "mediaTagPrefix")]
public string MediaTagPrefix { get; set; }
[XmlAttribute(AttributeName = "mediaTagVersion")]
public string MediaTagVersion { get; set; }
}
}

View file

@ -133,6 +133,9 @@ namespace PlexRequests.Api.Models.Plex
[XmlRoot(ElementName = "Video")] [XmlRoot(ElementName = "Video")]
public class Video public class Video
{ {
public string ProviderId { get; set; }
[XmlAttribute(AttributeName = "guid")]
public string Guid { get; set; }
[XmlElement(ElementName = "Media")] [XmlElement(ElementName = "Media")]
public List<Media> Media { get; set; } public List<Media> Media { get; set; }
[XmlElement(ElementName = "Genre")] [XmlElement(ElementName = "Genre")]
@ -241,6 +244,9 @@ namespace PlexRequests.Api.Models.Plex
[XmlRoot(ElementName = "Directory")] [XmlRoot(ElementName = "Directory")]
public class Directory1 public class Directory1
{ {
public string ProviderId { get; set; }
[XmlAttribute(AttributeName = "guid")]
public string Guid { get; set; }
[XmlElement(ElementName = "Genre")] [XmlElement(ElementName = "Genre")]
public List<Genre> Genre { get; set; } public List<Genre> Genre { get; set; }
[XmlElement(ElementName = "Role")] [XmlElement(ElementName = "Role")]
@ -311,6 +317,7 @@ namespace PlexRequests.Api.Models.Plex
[XmlRoot(ElementName = "MediaContainer")] [XmlRoot(ElementName = "MediaContainer")]
public class PlexSearch public class PlexSearch
{ {
[XmlElement(ElementName = "Directory")] [XmlElement(ElementName = "Directory")]
public List<Directory1> Directory { get; set; } public List<Directory1> Directory { get; set; }
[XmlElement(ElementName = "Video")] [XmlElement(ElementName = "Video")]

View file

@ -64,6 +64,7 @@
<Compile Include="Plex\PlexError.cs" /> <Compile Include="Plex\PlexError.cs" />
<Compile Include="Plex\PlexFriends.cs" /> <Compile Include="Plex\PlexFriends.cs" />
<Compile Include="Plex\PlexLibraries.cs" /> <Compile Include="Plex\PlexLibraries.cs" />
<Compile Include="Plex\PlexMetadata.cs" />
<Compile Include="Plex\PlexSearch.cs" /> <Compile Include="Plex\PlexSearch.cs" />
<Compile Include="Plex\PlexStatus.cs" /> <Compile Include="Plex\PlexStatus.cs" />
<Compile Include="Plex\PlexMediaType.cs" /> <Compile Include="Plex\PlexMediaType.cs" />

View file

@ -220,6 +220,36 @@ namespace PlexRequests.Api
} }
} }
public PlexMetadata GetMetadata(string authToken, Uri plexFullHost, string itemId)
{
var request = new RestRequest
{
Method = Method.GET,
Resource = "library/metadata/{itemId}"
};
request.AddUrlSegment("itemId", itemId);
AddHeaders(ref request, authToken);
try
{
var lib = RetryHandler.Execute(() => Api.ExecuteXml<PlexMetadata>(request, plexFullHost),
new[] {
TimeSpan.FromSeconds (5),
TimeSpan.FromSeconds(10),
TimeSpan.FromSeconds(30)
},
(exception, timespan) => Log.Error(exception, "Exception when calling GetMetadata for Plex, Retrying {0}", timespan));
return lib;
}
catch (Exception e)
{
Log.Error(e, "There has been a API Exception when attempting to get the Plex GetMetadata");
return new PlexMetadata();
}
}
private void AddHeaders(ref RestRequest request, string authToken) private void AddHeaders(ref RestRequest request, string authToken)
{ {
request.AddHeader("X-Plex-Token", authToken); request.AddHeader("X-Plex-Token", authToken);

View file

@ -37,6 +37,7 @@ namespace PlexRequests.Core.SettingModels
public int Port { get; set; } public int Port { get; set; }
public bool Ssl { get; set; } public bool Ssl { get; set; }
public string SubDir { get; set; } public string SubDir { get; set; }
public bool AdvancedSearch { get; set; }
[JsonIgnore] [JsonIgnore]
public Uri FullUri public Uri FullUri

View file

@ -0,0 +1,61 @@
#region Copyright
// /************************************************************************
// Copyright (c) 2016 Jamie Rees
// File: UriHelperTests.cs
// Created By: Jamie Rees
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
// ************************************************************************/
#endregion
using System;
using System.Collections.Generic;
using NUnit.Framework;
namespace PlexRequests.Helpers.Tests
{
[TestFixture]
public class PlexHelperTests
{
[TestCaseSource(nameof(PlexGuids))]
public string CreateUriWithSubDir(string guid)
{
return PlexHelper.GetProviderIdFromPlexGuid(guid);
}
private static IEnumerable<TestCaseData> PlexGuids
{
get
{
yield return new TestCaseData("com.plexapp.agents.thetvdb://269586/3/17?lang=en").Returns("269586");
yield return new TestCaseData("com.plexapp.agents.imdb://tt3300542?lang=en").Returns("tt3300542");
yield return new TestCaseData("com.plexapp.agents.thetvdb://71326/10/5?lang=en").Returns("71326");
yield return new TestCaseData("local://3450").Returns("3450");
yield return new TestCaseData("com.plexapp.agents.imdb://tt1179933?lang=en").Returns("tt1179933");
yield return new TestCaseData("com.plexapp.agents.imdb://tt0284837?lang=en").Returns("tt0284837");
yield return new TestCaseData("com.plexapp.agents.imdb://tt0076759?lang=en").Returns("tt0076759");
}
}
}
}

View file

@ -75,6 +75,7 @@
<Compile Include="HtmlRemoverTests.cs" /> <Compile Include="HtmlRemoverTests.cs" />
<Compile Include="AssemblyHelperTests.cs" /> <Compile Include="AssemblyHelperTests.cs" />
<Compile Include="Properties\AssemblyInfo.cs" /> <Compile Include="Properties\AssemblyInfo.cs" />
<Compile Include="PlexHelperTests.cs" />
<Compile Include="UriHelperTests.cs" /> <Compile Include="UriHelperTests.cs" />
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>

View file

@ -0,0 +1,47 @@
#region Copyright
// /************************************************************************
// Copyright (c) 2016 Jamie Rees
// File: PlexHelper.cs
// Created By: Jamie Rees
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
// ************************************************************************/
#endregion
using System;
namespace PlexRequests.Helpers
{
public class PlexHelper
{
public static string GetProviderIdFromPlexGuid(string guid)
{
if (string.IsNullOrEmpty(guid))
return guid;
var guidSplit = guid.Split(new[] {'/', '?'}, StringSplitOptions.RemoveEmptyEntries);
if (guidSplit.Length > 1)
{
return guidSplit[1];
}
return string.Empty;
}
}
}

View file

@ -79,6 +79,7 @@
<Compile Include="MemoryCacheProvider.cs" /> <Compile Include="MemoryCacheProvider.cs" />
<Compile Include="ObjectCopier.cs" /> <Compile Include="ObjectCopier.cs" />
<Compile Include="PasswordHasher.cs" /> <Compile Include="PasswordHasher.cs" />
<Compile Include="PlexHelper.cs" />
<Compile Include="Properties\AssemblyInfo.cs" /> <Compile Include="Properties\AssemblyInfo.cs" />
<Compile Include="SerializerSettings.cs" /> <Compile Include="SerializerSettings.cs" />
<Compile Include="StringCipher.cs" /> <Compile Include="StringCipher.cs" />

View file

@ -33,9 +33,9 @@ namespace PlexRequests.Services.Interfaces
{ {
void CheckAndUpdateAll(); void CheckAndUpdateAll();
List<PlexMovie> GetPlexMovies(); List<PlexMovie> GetPlexMovies();
bool IsMovieAvailable(PlexMovie[] plexMovies, string title, string year); bool IsMovieAvailable(PlexMovie[] plexMovies, string title, string year, string providerId = null);
List<PlexTvShow> GetPlexTvShows(); List<PlexTvShow> GetPlexTvShows();
bool IsTvShowAvailable(PlexTvShow[] plexShows, string title, string year); bool IsTvShowAvailable(PlexTvShow[] plexShows, string title, string year, string providerId = null);
List<PlexAlbum> GetPlexAlbums(); List<PlexAlbum> GetPlexAlbums();
bool IsAlbumAvailable(PlexAlbum[] plexAlbums, string title, string year, string artist); bool IsAlbumAvailable(PlexAlbum[] plexAlbums, string title, string year, string artist);
} }

View file

@ -111,10 +111,10 @@ namespace PlexRequests.Services.Jobs
switch (r.Type) switch (r.Type)
{ {
case RequestType.Movie: case RequestType.Movie:
matchResult = IsMovieAvailable(movies, r.Title, releaseDate); matchResult = IsMovieAvailable(movies, r.Title, releaseDate, r.ImdbId);
break; break;
case RequestType.TvShow: case RequestType.TvShow:
matchResult = IsTvShowAvailable(shows, r.Title, releaseDate); matchResult = IsTvShowAvailable(shows, r.Title, releaseDate, r.ProviderId.ToString());
break; break;
case RequestType.Album: case RequestType.Album:
matchResult = IsAlbumAvailable(albums, r.Title, r.ReleaseDate.Year.ToString(), r.ArtistName); matchResult = IsAlbumAvailable(albums, r.Title, r.ReleaseDate.Year.ToString(), r.ArtistName);
@ -158,19 +158,36 @@ namespace PlexRequests.Services.Jobs
foreach (var lib in movieLibs) foreach (var lib in movieLibs)
{ {
movies.AddRange(lib.Video.Select(x => new PlexMovie() // movies are in the Video list movies.AddRange(lib.Video.Select(video => new PlexMovie
{ {
Title = x.Title, ReleaseYear = video.Year,
ReleaseYear = x.Year Title = video.Title,
ProviderId = video.ProviderId,
})); }));
} }
} }
return movies; return movies;
} }
public bool IsMovieAvailable(PlexMovie[] plexMovies, string title, string year) public bool IsMovieAvailable(PlexMovie[] plexMovies, string title, string year, string providerId = null)
{ {
return plexMovies.Any(x => x.Title.Equals(title, StringComparison.CurrentCultureIgnoreCase) && x.ReleaseYear.Equals(year, StringComparison.CurrentCultureIgnoreCase)); var advanced = !string.IsNullOrEmpty(providerId);
foreach (var movie in plexMovies)
{
if (advanced)
{
if (movie.ProviderId.Equals(providerId, StringComparison.InvariantCultureIgnoreCase))
{
return true;
}
}
if (movie.Title.Equals(title, StringComparison.CurrentCultureIgnoreCase) &&
movie.ReleaseYear.Equals(year, StringComparison.CurrentCultureIgnoreCase))
{
return true;
}
}
return false;
} }
public List<PlexTvShow> GetPlexTvShows() public List<PlexTvShow> GetPlexTvShows()
@ -190,18 +207,33 @@ namespace PlexRequests.Services.Jobs
shows.AddRange(lib.Directory.Select(x => new PlexTvShow() // shows are in the directory list shows.AddRange(lib.Directory.Select(x => new PlexTvShow() // shows are in the directory list
{ {
Title = x.Title, Title = x.Title,
ReleaseYear = x.Year ReleaseYear = x.Year,
ProviderId = x.ProviderId,
})); }));
} }
} }
return shows; return shows;
} }
public bool IsTvShowAvailable(PlexTvShow[] plexShows, string title, string year) public bool IsTvShowAvailable(PlexTvShow[] plexShows, string title, string year, string providerId = null)
{ {
return plexShows.Any(x => var advanced = !string.IsNullOrEmpty(providerId);
(x.Title.Equals(title, StringComparison.CurrentCultureIgnoreCase) || x.Title.StartsWith(title, StringComparison.CurrentCultureIgnoreCase)) && foreach (var show in plexShows)
x.ReleaseYear.Equals(year, StringComparison.CurrentCultureIgnoreCase)); {
if (advanced)
{
if (show.ProviderId.Equals(providerId, StringComparison.InvariantCultureIgnoreCase))
{
return true;
}
}
if (show.Title.Equals(title, StringComparison.CurrentCultureIgnoreCase) &&
show.ReleaseYear.Equals(year, StringComparison.CurrentCultureIgnoreCase))
{
return true;
}
}
return false;
} }
public List<PlexAlbum> GetPlexAlbums() public List<PlexAlbum> GetPlexAlbums()
@ -239,12 +271,12 @@ namespace PlexRequests.Services.Jobs
private List<PlexSearch> CachedLibraries(AuthenticationSettings authSettings, PlexSettings plexSettings, bool setCache) private List<PlexSearch> CachedLibraries(AuthenticationSettings authSettings, PlexSettings plexSettings, bool setCache)
{ {
List<PlexSearch> results = new List<PlexSearch>(); var results = new List<PlexSearch>();
if (!ValidateSettings(plexSettings, authSettings)) if (!ValidateSettings(plexSettings, authSettings))
{ {
Log.Warn("The settings are not configured"); Log.Warn("The settings are not configured");
return results; // don't error out here, just let it go! return results; // don't error out here, just let it go! let it goo!!!
} }
try try
@ -252,7 +284,28 @@ namespace PlexRequests.Services.Jobs
if (setCache) if (setCache)
{ {
results = GetLibraries(authSettings, plexSettings); results = GetLibraries(authSettings, plexSettings);
if (plexSettings.AdvancedSearch)
{
for (var i = 0; i < results.Count; i++)
{
for (var j = 0; j < results[i].Directory.Count; j++)
{
var currentItem = results[i].Directory[j];
var metaData = PlexApi.GetMetadata(authSettings.PlexAuthToken, plexSettings.FullUri,
currentItem.RatingKey);
var providerId = PlexHelper.GetProviderIdFromPlexGuid(metaData.Directory.Guid);
results[i].Directory[j].ProviderId = providerId;
}
for (var j = 0; j < results[i].Video.Count; j++)
{
var currentItem = results[i].Video[j];
var metaData = PlexApi.GetMetadata(authSettings.PlexAuthToken, plexSettings.FullUri,
currentItem.RatingKey);
var providerId = PlexHelper.GetProviderIdFromPlexGuid(metaData.Video.Guid);
results[i].Video[j].ProviderId = providerId;
}
}
}
if (results != null) if (results != null)
{ {
Cache.Set(CacheKeys.PlexLibaries, results, CacheKeys.TimeFrameMinutes.SchedulerCaching); Cache.Set(CacheKeys.PlexLibaries, results, CacheKeys.TimeFrameMinutes.SchedulerCaching);

View file

@ -4,5 +4,6 @@
{ {
public string Title { get; set; } public string Title { get; set; }
public string ReleaseYear { get; set; } public string ReleaseYear { get; set; }
public string ProviderId { get; set; }
} }
} }

View file

@ -4,5 +4,6 @@
{ {
public string Title { get; set; } public string Title { get; set; }
public string ReleaseYear { get; set; } public string ReleaseYear { get; set; }
public string ProviderId { get; set; }
} }
} }

View file

@ -1,4 +1,5 @@
@using PlexRequests.UI.Helpers @using PlexRequests.UI.Helpers
@inherits Nancy.ViewEngines.Razor.NancyRazorViewBase<PlexRequests.Core.SettingModels.PlexSettings>
@Html.Partial("_Sidebar") @Html.Partial("_Sidebar")
@{ @{
int port; int port;
@ -44,6 +45,23 @@
</div> </div>
</div> </div>
<div class="form-group">
<div class="checkbox">
@if (Model.AdvancedSearch)
{
<input type="checkbox" id="AdvancedSearch" name="AdvancedSearch" checked="checked"><label for="AdvancedSearch">Use Advanced Search</label>
}
else
{
<input type="checkbox" id="AdvancedSearch" name="AdvancedSearch"><label for="AdvancedSearch">Use Advanced Search</label>
}
</div>
<small>If enabled we will be able to have a 100% accurate match, but we will be querying Plex for every single item in your library. So if you have a big library then this could take some time.</small>
</div>
<div class="form-group"> <div class="form-group">
<label for="SubDir" class="control-label">Plex SubDirectory</label> <label for="SubDir" class="control-label">Plex SubDirectory</label>
<div> <div>