From b50f78e5da6f3fdfc59e577ca61b88771da7d211 Mon Sep 17 00:00:00 2001
From: LukePulverenti Luke Pulverenti luke pulverenti
 <LukePulverenti Luke Pulverenti luke.pulverenti@gmail.com>
Date: Thu, 12 Jul 2012 02:55:27 -0400
Subject: Initial check-in

---
 MediaBrowser.Controller/Resolvers/VideoResolver.cs | 114 +++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 MediaBrowser.Controller/Resolvers/VideoResolver.cs

(limited to 'MediaBrowser.Controller/Resolvers/VideoResolver.cs')

diff --git a/MediaBrowser.Controller/Resolvers/VideoResolver.cs b/MediaBrowser.Controller/Resolvers/VideoResolver.cs
new file mode 100644
index 000000000..ba51dab87
--- /dev/null
+++ b/MediaBrowser.Controller/Resolvers/VideoResolver.cs
@@ -0,0 +1,114 @@
+using System.IO;
+using MediaBrowser.Controller.Events;
+using MediaBrowser.Model.Entities;
+using System.Linq;
+using System.Collections.Generic;
+
+namespace MediaBrowser.Controller.Resolvers
+{
+    public class VideoResolver : BaseVideoResolver<Video>
+    {
+    }
+
+    public abstract class BaseVideoResolver<T> : BaseItemResolver<T>
+        where T : Video, new()
+    {
+        protected override T Resolve(ItemResolveEventArgs args)
+        {
+            if (!args.IsFolder)
+            {
+                if (IsVideoFile(args.Path))
+                {
+                    return new T()
+                    {
+                        VideoType = VideoType.VideoFile,
+                        Path = args.Path
+                    };
+                }
+            }
+
+            else
+            {
+                T item = ResolveFromFolderName(args.Path);
+
+                if (item != null)
+                {
+                    return item;
+                }
+
+                foreach (KeyValuePair<string, FileAttributes> folder in args.FileSystemChildren)
+                {
+                    if (!folder.Value.HasFlag(FileAttributes.Directory))
+                    {
+                        continue;
+                    }
+
+                    item = ResolveFromFolderName(folder.Key);
+
+                    if (item != null)
+                    {
+                        return item;
+                    }
+                }
+            }
+
+            return null;
+        }
+
+        private T ResolveFromFolderName(string folder)
+        {
+            if (folder.IndexOf("video_ts", System.StringComparison.OrdinalIgnoreCase) != -1)
+            {
+                return new T()
+                {
+                    VideoType = VideoType.DVD,
+                    Path = Path.GetDirectoryName(folder)
+                };
+            }
+            if (folder.IndexOf("bdmv", System.StringComparison.OrdinalIgnoreCase) != -1)
+            {
+                return new T()
+                {
+                    VideoType = VideoType.BluRay,
+                    Path = Path.GetDirectoryName(folder)
+                };
+            }
+
+            return null;
+        }
+
+        private static bool IsVideoFile(string path)
+        {
+            string extension = Path.GetExtension(path).ToLower();
+
+            switch (extension)
+            {
+                case ".mkv":
+                case ".m2ts":
+                case ".iso":
+                case ".ts":
+                case ".rmvb":
+                case ".mov":
+                case ".avi":
+                case ".mpg":
+                case ".mpeg":
+                case ".wmv":
+                case ".mp4":
+                case ".divx":
+                case ".dvr-ms":
+                case ".wtv":
+                case ".ogm":
+                case ".ogv":
+                case ".asf":
+                case ".m4v":
+                case ".flv":
+                case ".f4v":
+                case ".3gp":
+                    return true;
+
+                default:
+                    return false;
+            }
+        }
+    }
+}
-- 
cgit v1.2.3