From 6d2fac5cd05f47a267be71c9362e381d81f01b51 Mon Sep 17 00:00:00 2001
From: Taloth Saldono <Taloth@users.noreply.github.com>
Date: Sat, 31 May 2014 17:53:04 +0200
Subject: [PATCH] New: Adding new series by tvdbid: or slug: is now possible.

---
 .../MetadataSource/TraktProxy.cs              | 45 +++++++++++++++----
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/src/NzbDrone.Core/MetadataSource/TraktProxy.cs b/src/NzbDrone.Core/MetadataSource/TraktProxy.cs
index 22e11bec1..0236a4169 100644
--- a/src/NzbDrone.Core/MetadataSource/TraktProxy.cs
+++ b/src/NzbDrone.Core/MetadataSource/TraktProxy.cs
@@ -20,7 +20,7 @@ namespace NzbDrone.Core.MetadataSource
     {
         private readonly Logger _logger;
         private static readonly Regex CollapseSpaceRegex = new Regex(@"\s+", RegexOptions.Compiled);
-        private static readonly Regex InvalidSearchCharRegex = new Regex(@"(?:\*|\(|\)|'|!|@)", RegexOptions.Compiled);
+        private static readonly Regex InvalidSearchCharRegex = new Regex(@"(?:\*|\(|\)|'|!|@|\+)", RegexOptions.Compiled);
 
         public TraktProxy(Logger logger)
         {
@@ -31,13 +31,43 @@ namespace NzbDrone.Core.MetadataSource
         {
             try
             {
-                var client = BuildClient("search", "shows");
-                var restRequest = new RestRequest(GetSearchTerm(title) + "/30/seasons");
-                var response = client.ExecuteAndValidate<List<Show>>(restRequest);
+                if (title.StartsWith("tvdb:") || title.StartsWith("tvdbid:") || title.StartsWith("slug:"))
+                {
+                    try
+                    {
+                        var slug = title.Split(':')[1];
 
-                return response.Select(MapSeries)
-                    .OrderBy(v => title.LevenshteinDistanceClean(v.Title))
-                    .ToList();
+                        if (slug.IsNullOrWhiteSpace() || slug.Any(char.IsWhiteSpace))
+                        {
+                            return new List<Series>();
+                        }
+
+                        var client = BuildClient("show", "summary");
+                        var restRequest = new RestRequest(GetSearchTerm(slug) + "/extended");
+                        var response = client.ExecuteAndValidate<Show>(restRequest);
+
+                        return new List<Series> { MapSeries(response) };
+                    }
+                    catch (RestException ex)
+                    {
+                        if (ex.Response.StatusCode == HttpStatusCode.NotFound)
+                        {
+                            return new List<Series>();
+                        }
+
+                        throw;
+                    }
+                }
+                else
+                {
+                    var client = BuildClient("search", "shows");
+                    var restRequest = new RestRequest(GetSearchTerm(title) + "/30/seasons");
+                    var response = client.ExecuteAndValidate<List<Show>>(restRequest);
+
+                    return response.Select(MapSeries)
+                        .OrderBy(v => title.LevenshteinDistanceClean(v.Title))
+                        .ToList();
+                }
             }
             catch (WebException ex)
             {
@@ -172,7 +202,6 @@ namespace NzbDrone.Core.MetadataSource
             phrase = CollapseSpaceRegex.Replace(phrase, " ").Trim().ToLower();
             phrase = phrase.Trim('-');
             phrase = HttpUtility.UrlEncode(phrase);
-            
 
             return phrase;
         }