jellyfin/MediaBrowser.Controller/Resolvers/VideoResolver.cs

125 lines
3.5 KiB
C#
Raw Normal View History

using System.Collections.Generic;
using System.ComponentModel.Composition;
using System.IO;
2012-07-12 06:55:27 +00:00
using MediaBrowser.Controller.Events;
using MediaBrowser.Model.Entities;
namespace MediaBrowser.Controller.Resolvers
{
/// <summary>
/// Resolves a Path into a Video
/// </summary>
[Export(typeof(IBaseItemResolver))]
2012-07-12 06:55:27 +00:00
public class VideoResolver : BaseVideoResolver<Video>
{
}
/// <summary>
/// Resolves a Path into a Video or Video subclass
/// </summary>
2012-07-12 06:55:27 +00:00
public abstract class BaseVideoResolver<T> : BaseItemResolver<T>
where T : Video, new()
{
protected override T Resolve(ItemResolveEventArgs args)
{
// If the path is a file check for a matching extensions
2012-07-12 06:55:27 +00:00
if (!args.IsFolder)
{
if (IsVideoFile(args.Path))
{
return new T()
{
VideoType = VideoType.VideoFile,
Path = args.Path
};
}
}
else
{
// If the path is a folder, check if it's bluray or dvd
2012-07-12 06:55:27 +00:00
T item = ResolveFromFolderName(args.Path);
if (item != null)
{
return item;
}
// Also check the subfolders for bluray or dvd
2012-07-12 06:55:27 +00:00
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;
}
}
}
}