From f95d2cc5067463642e395a435a13a72b7abd8185 Mon Sep 17 00:00:00 2001
From: Luke Pulverenti <luke.pulverenti@gmail.com>
Date: Mon, 29 Apr 2013 21:29:04 -0400
Subject: [PATCH] added more movie filters

---
 MediaBrowser.Api/UserLibrary/ItemsService.cs  | 51 +++++++++++++++++++
 .../Library/LuceneSearchEngine.cs             | 38 +++++---------
 2 files changed, 65 insertions(+), 24 deletions(-)

diff --git a/MediaBrowser.Api/UserLibrary/ItemsService.cs b/MediaBrowser.Api/UserLibrary/ItemsService.cs
index baa47e787..45bf17397 100644
--- a/MediaBrowser.Api/UserLibrary/ItemsService.cs
+++ b/MediaBrowser.Api/UserLibrary/ItemsService.cs
@@ -1,6 +1,7 @@
 using MediaBrowser.Controller.Dto;
 using MediaBrowser.Controller.Entities;
 using MediaBrowser.Controller.Entities.Audio;
+using MediaBrowser.Controller.Entities.Movies;
 using MediaBrowser.Controller.Entities.TV;
 using MediaBrowser.Controller.Library;
 using MediaBrowser.Controller.Localization;
@@ -146,6 +147,21 @@ namespace MediaBrowser.Api.UserLibrary
         /// <value>The max offical rating.</value>
         [ApiMember(Name = "MaxOfficialRating", Description = "Optional filter by maximum official rating (PG, PG-13, TV-MA, etc).", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
         public string MaxOfficialRating { get; set; }
+
+        [ApiMember(Name = "HasThemeSong", Description = "Optional filter by items with theme songs.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
+        public bool? HasThemeSong { get; set; }
+
+        [ApiMember(Name = "HasThemeVideo", Description = "Optional filter by items with theme videos.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
+        public bool? HasThemeVideo { get; set; }
+
+        [ApiMember(Name = "HasSubtitles", Description = "Optional filter by items with subtitles.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
+        public bool? HasSubtitles { get; set; }
+
+        [ApiMember(Name = "HasSpecialFeature", Description = "Optional filter by items with special features.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
+        public bool? HasSpecialFeature { get; set; }
+
+        [ApiMember(Name = "HasTrailer", Description = "Optional filter by items with trailers.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
+        public bool? HasTrailer { get; set; }
         
         /// <summary>
         /// Gets the order by.
@@ -529,6 +545,41 @@ namespace MediaBrowser.Api.UserLibrary
                 }
             }
 
+            if (request.HasTrailer.HasValue)
+            {
+                items = items.Where(i => request.HasTrailer.Value ? i.LocalTrailers.Count > 0 : i.LocalTrailers.Count == 0);
+            }
+
+            if (request.HasThemeSong.HasValue)
+            {
+                items = items.Where(i => request.HasThemeSong.Value ? i.ThemeSongs.Count > 0 : i.ThemeSongs.Count == 0);
+            }
+
+            if (request.HasThemeVideo.HasValue)
+            {
+                items = items.Where(i => request.HasThemeVideo.Value ? i.ThemeVideos.Count > 0 : i.ThemeVideos.Count == 0);
+            }
+
+            if (request.HasSpecialFeature.HasValue)
+            {
+                items = items.OfType<Movie>().Where(i => request.HasSpecialFeature.Value ? i.SpecialFeatures.Count > 0 : i.SpecialFeatures.Count == 0);
+            }
+
+            if (request.HasSubtitles.HasValue)
+            {
+                items = items.OfType<Video>().Where(i =>
+                {
+                    if (request.HasSubtitles.Value)
+                    {
+                        return i.MediaStreams != null && i.MediaStreams.Any(m => m.Type == MediaStreamType.Subtitle);
+                    }
+                    else
+                    {
+                        return i.MediaStreams == null || i.MediaStreams.All(m => m.Type != MediaStreamType.Subtitle);
+                    }
+                });
+            }
+
             return items;
         }
 
diff --git a/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs b/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs
index 2dc4ecbb8..ef9662d0b 100644
--- a/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs
+++ b/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs
@@ -241,20 +241,15 @@ namespace MediaBrowser.Server.Implementations.Library
                 return new Tuple<string, int>(searchInput, 0);
             }
 
-            var match = Regex.Match(input, searchInput, RegexOptions.IgnoreCase);
+            var index = input.IndexOf(searchInput, StringComparison.OrdinalIgnoreCase);
 
-            if (match.Success)
+            if (index == 0)
             {
-                var index = match.Index;
-
-                if (index == 0)
-                {
-                    return new Tuple<string, int>(searchInput, 1);
-                }
-                if (index > 0)
-                {
-                    return new Tuple<string, int>(searchInput, 2);
-                }
+                return new Tuple<string, int>(searchInput, 1);
+            }
+            if (index > 0)
+            {
+                return new Tuple<string, int>(searchInput, 2);
             }
 
             var items = GetWords(input);
@@ -272,20 +267,15 @@ namespace MediaBrowser.Server.Implementations.Library
                         return new Tuple<string, int>(searchTerm, 3 + (i + 1) * (j + 1));
                     }
 
-                    match = Regex.Match(item, searchTerm, RegexOptions.IgnoreCase);
+                    index = item.IndexOf(searchTerm, StringComparison.OrdinalIgnoreCase);
 
-                    if (match.Success)
+                    if (index == 0)
                     {
-                        var index = match.Index;
-
-                        if (index == 0)
-                        {
-                            return new Tuple<string, int>(searchTerm, 4 + (i + 1) * (j + 1));
-                        }
-                        if (index > 0)
-                        {
-                            return new Tuple<string, int>(searchTerm, 5 + (i + 1) * (j + 1));
-                        }
+                        return new Tuple<string, int>(searchTerm, 4 + (i + 1) * (j + 1));
+                    }
+                    if (index > 0)
+                    {
+                        return new Tuple<string, int>(searchTerm, 5 + (i + 1) * (j + 1));
                     }
                 }
             }