jellyfin/MediaBrowser.XbmcMetadata/Savers/MovieNfoSaver.cs

125 lines
3.8 KiB
C#
Raw Normal View History

using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
2015-05-15 15:46:20 +00:00
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2014-07-12 02:31:08 +00:00
using MediaBrowser.Model.Logging;
using System.Collections.Generic;
using System.IO;
2015-05-15 15:46:20 +00:00
using System.Linq;
2014-07-09 00:46:11 +00:00
using System.Xml;
2015-10-04 04:23:11 +00:00
using CommonIO;
namespace MediaBrowser.XbmcMetadata.Savers
{
2014-07-09 00:46:11 +00:00
public class MovieNfoSaver : BaseNfoSaver
{
2014-07-12 02:31:08 +00:00
public MovieNfoSaver(IFileSystem fileSystem, IServerConfigurationManager configurationManager, ILibraryManager libraryManager, IUserManager userManager, IUserDataManager userDataManager, ILogger logger) : base(fileSystem, configurationManager, libraryManager, userManager, userDataManager, logger)
{
}
2014-10-22 04:42:26 +00:00
protected override string GetLocalSavePath(IHasMetadata item)
{
2015-05-15 15:46:20 +00:00
return GetMovieSavePaths(new ItemInfo(item), FileSystem).FirstOrDefault();
}
2015-05-15 15:46:20 +00:00
public static List<string> GetMovieSavePaths(ItemInfo item, IFileSystem fileSystem)
{
2015-05-15 15:46:20 +00:00
var list = new List<string>();
2015-10-08 16:22:14 +00:00
if (item.VideoType == VideoType.Dvd && !item.IsPlaceHolder)
{
var path = item.ContainingFolderPath;
2015-05-21 20:53:14 +00:00
list.Add(Path.Combine(path, "VIDEO_TS", "VIDEO_TS.nfo"));
}
2015-05-15 15:46:20 +00:00
if (item.VideoType == VideoType.Dvd || item.VideoType == VideoType.BluRay || item.VideoType == VideoType.HdDvd)
{
var path = item.ContainingFolderPath;
list.Add(Path.Combine(path, Path.GetFileName(path) + ".nfo"));
}
else
{
list.Add(Path.ChangeExtension(item.Path, ".nfo"));
}
return list;
}
2014-07-09 00:46:11 +00:00
protected override string GetRootElementName(IHasMetadata item)
{
2014-07-09 00:46:11 +00:00
return item is MusicVideo ? "musicvideo" : "movie";
}
2014-07-09 00:46:11 +00:00
public override bool IsEnabledFor(IHasMetadata item, ItemUpdateType updateType)
{
if (!item.SupportsLocalMetadata)
{
return false;
}
2014-07-09 00:46:11 +00:00
var video = item as Video;
2014-07-09 00:46:11 +00:00
// Check parent for null to avoid running this against things like video backdrops
if (video != null && !(item is Episode) && !video.IsOwnedItem)
{
return updateType >= MinimumUpdateType;
2014-07-09 00:46:11 +00:00
}
2014-07-09 00:46:11 +00:00
return false;
}
2014-07-09 00:46:11 +00:00
protected override void WriteCustomElements(IHasMetadata item, XmlWriter writer)
{
var imdb = item.GetProviderId(MetadataProviders.Imdb);
if (!string.IsNullOrEmpty(imdb))
{
2014-07-09 00:46:11 +00:00
writer.WriteElementString("id", imdb);
}
var musicVideo = item as MusicVideo;
if (musicVideo != null)
{
2014-10-20 20:23:40 +00:00
foreach (var artist in musicVideo.Artists)
{
2014-10-20 20:23:40 +00:00
writer.WriteElementString("artist", artist);
}
if (!string.IsNullOrEmpty(musicVideo.Album))
{
2014-07-09 00:46:11 +00:00
writer.WriteElementString("album", musicVideo.Album);
}
}
var movie = item as Movie;
if (movie != null)
{
if (!string.IsNullOrEmpty(movie.TmdbCollectionName))
{
2014-07-09 00:46:11 +00:00
writer.WriteElementString("set", movie.TmdbCollectionName);
}
}
2014-07-09 00:46:11 +00:00
}
2014-07-09 00:46:11 +00:00
protected override List<string> GetTagsUsed()
{
var list = new List<string>
{
"album",
"artist",
"set",
"id"
2014-07-09 00:46:11 +00:00
};
2014-07-09 00:46:11 +00:00
return list;
}
}
}