From b3e03a648d8e56d0e24c4000a60e42c551d73cc8 Mon Sep 17 00:00:00 2001
From: Devin Buhl <devin.kray@gmail.com>
Date: Wed, 18 Jan 2017 14:53:29 -0500
Subject: [PATCH] Add IMDb ID to file naming

---
 src/NzbDrone.Core/Organizer/FileNameBuilder.cs            | 8 ++++++++
 src/NzbDrone.Core/Organizer/FileNameSampleService.cs      | 3 ++-
 .../MediaManagement/Naming/NamingViewTemplate.hbs         | 2 ++
 .../Naming/Partials/ImdbIdNamingPartial.hbs               | 1 +
 4 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 src/UI/Settings/MediaManagement/Naming/Partials/ImdbIdNamingPartial.hbs

diff --git a/src/NzbDrone.Core/Organizer/FileNameBuilder.cs b/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
index 68f2e9102..fe3a8acc8 100644
--- a/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
+++ b/src/NzbDrone.Core/Organizer/FileNameBuilder.cs
@@ -161,6 +161,7 @@ namespace NzbDrone.Core.Organizer
 
             AddMovieTokens(tokenHandlers, movie);
             AddReleaseDateTokens(tokenHandlers, movie.Year); //In case we want to separate the year
+            AddImdbIdTokens(tokenHandlers, movie.ImdbId);
             AddQualityTokens(tokenHandlers, movie, movieFile);
             AddMediaInfoTokens(tokenHandlers, movieFile);
             AddMovieFileTokens(tokenHandlers, movieFile);
@@ -301,6 +302,7 @@ namespace NzbDrone.Core.Organizer
 
             AddMovieTokens(tokenHandlers, movie);
             AddReleaseDateTokens(tokenHandlers, movie.Year);
+            AddImdbIdTokens(tokenHandlers, movie.ImdbId);
 
             return CleanFolderName(ReplaceTokens(namingConfig.MovieFolderFormat, tokenHandlers, namingConfig));
         }
@@ -477,6 +479,11 @@ namespace NzbDrone.Core.Organizer
             tokenHandlers["{Release Year}"] = m => string.Format("{0}", releaseYear.ToString()); //Do I need m.CustomFormat?
         }
 
+        private void AddImdbIdTokens(Dictionary<string, Func<TokenMatch, string>> tokenHandlers, string imdbId)
+        {
+            tokenHandlers["{IMDb Id}"] = m => $"{imdbId}";
+        }
+
         private void AddSeasonTokens(Dictionary<string, Func<TokenMatch, string>> tokenHandlers, int seasonNumber)
         {
             tokenHandlers["{Season}"] = m => seasonNumber.ToString(m.CustomFormat);
@@ -508,6 +515,7 @@ namespace NzbDrone.Core.Organizer
         {
             tokenHandlers["{Original Title}"] = m => GetOriginalTitle(episodeFile);
             tokenHandlers["{Original Filename}"] = m => GetOriginalFileName(episodeFile);
+            //tokenHandlers["{IMDb Id}"] = m => 
             tokenHandlers["{Release Group}"] = m => episodeFile.ReleaseGroup ?? m.DefaultValue("Sonarr");
         }
 
diff --git a/src/NzbDrone.Core/Organizer/FileNameSampleService.cs b/src/NzbDrone.Core/Organizer/FileNameSampleService.cs
index 654179b52..c8f574326 100644
--- a/src/NzbDrone.Core/Organizer/FileNameSampleService.cs
+++ b/src/NzbDrone.Core/Organizer/FileNameSampleService.cs
@@ -47,7 +47,8 @@ namespace NzbDrone.Core.Organizer
             _movie = new Movie
             {
                 Title = "Movie Title",
-                Year = 2010
+                Year = 2010,
+                ImdbId = "tt0066921"
             };
 
             _standardSeries = new Series
diff --git a/src/UI/Settings/MediaManagement/Naming/NamingViewTemplate.hbs b/src/UI/Settings/MediaManagement/Naming/NamingViewTemplate.hbs
index 615f95b1a..2dbba5a3f 100644
--- a/src/UI/Settings/MediaManagement/Naming/NamingViewTemplate.hbs
+++ b/src/UI/Settings/MediaManagement/Naming/NamingViewTemplate.hbs
@@ -72,6 +72,7 @@
                             {{> MediaInfoNamingPartial}}
                             {{> ReleaseGroupNamingPartial}}
                             {{> OriginalTitleNamingPartial}}
+                            {{> ImdbIdNamingPartial}}
                             {{> SeparatorNamingPartial}}
                         </ul>
                     </div>
@@ -161,6 +162,7 @@
                     <ul class="dropdown-menu">
                         {{> MovieTitleNamingPartial}}
                         {{> ReleaseYearNamingPartial}}
+                        {{> ImdbIdNamingPartial}}
                     </ul>
                 </div>
             </div>
diff --git a/src/UI/Settings/MediaManagement/Naming/Partials/ImdbIdNamingPartial.hbs b/src/UI/Settings/MediaManagement/Naming/Partials/ImdbIdNamingPartial.hbs
new file mode 100644
index 000000000..9c0686d42
--- /dev/null
+++ b/src/UI/Settings/MediaManagement/Naming/Partials/ImdbIdNamingPartial.hbs
@@ -0,0 +1 @@
+<li><a href="#" data-token="IMDb Id">IMDb Id</a></li>
\ No newline at end of file