add photo album

1 files changed, 9 insertions, 17 deletions

diff --git a/MediaBrowser.Server.Implementations/Library/Resolvers/PhotoResolver.cs b/MediaBrowser.Server.Implementations/Library/Resolvers/PhotoResolver.cs
index cba7aba9a..60e7edfdd 100644
--- a/MediaBrowser.Server.Implementations/Library/Resolvers/PhotoResolver.cs
+++ b/MediaBrowser.Server.Implementations/Library/Resolvers/PhotoResolver.cs
@@ -1,24 +1,14 @@
-using MediaBrowser.Controller;
-using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Entities;
 using MediaBrowser.Controller.Library;
+using MediaBrowser.Model.Entities;
 using System;
+using System.IO;
 using System.Linq;
 
 namespace MediaBrowser.Server.Implementations.Library.Resolvers
 {
     public class PhotoResolver : ItemResolver<Photo>
     {
-        private readonly IServerApplicationPaths _applicationPaths;
-
-        /// <summary>
-        /// Initializes a new instance of the <see cref="PhotoResolver" /> class.
-        /// </summary>
-        /// <param name="applicationPaths">The application paths.</param>
-        public PhotoResolver(IServerApplicationPaths applicationPaths)
-        {
-            _applicationPaths = applicationPaths;
-        }
-
         /// <summary>
         /// Resolves the specified args.
         /// </summary>
@@ -27,7 +17,7 @@ namespace MediaBrowser.Server.Implementations.Library.Resolvers
         protected override Photo Resolve(ItemResolveArgs args)
         {
             // Must be an image file within a photo collection
-            if (!args.IsDirectory && IsImageFile(args.Path) && string.Equals(args.GetCollectionType(), "photos", StringComparison.OrdinalIgnoreCase))
+            if (!args.IsDirectory && IsImageFile(args.Path) && string.Equals(args.GetCollectionType(), CollectionType.Photos, StringComparison.OrdinalIgnoreCase))
             {
                 return new Photo
                 {
@@ -39,10 +29,12 @@ namespace MediaBrowser.Server.Implementations.Library.Resolvers
         }
 
         protected static string[] ImageExtensions = { ".tiff", ".jpeg", ".jpg", ".png", ".aiff" };
-        protected bool IsImageFile(string path)
+        internal static bool IsImageFile(string path)
         {
-            return !path.EndsWith("folder.jpg", StringComparison.OrdinalIgnoreCase)
-                && ImageExtensions.Any(p => path.EndsWith(p, StringComparison.OrdinalIgnoreCase));
+            var filename = Path.GetFileName(path);
+
+            return !string.Equals(filename, "folder.jpg", StringComparison.OrdinalIgnoreCase)
+                && ImageExtensions.Contains(Path.GetExtension(path) ?? string.Empty, StringComparer.OrdinalIgnoreCase);
         }
 
     }