jellyfin-server/MediaBrowser.Providers/Savers/MovieXmlSaver.cs

130 lines
4.5 KiB
C#
Raw Normal View History

using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
2013-06-09 16:23:06 +00:00
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Library;
2013-08-12 19:18:31 +00:00
using MediaBrowser.Controller.Persistence;
using MediaBrowser.Providers.Movies;
using System;
using System.Globalization;
using System.IO;
using System.Security;
2013-06-23 18:55:30 +00:00
using System.Text;
2013-06-09 16:23:06 +00:00
using System.Threading;
namespace MediaBrowser.Providers.Savers
2013-06-09 16:23:06 +00:00
{
/// <summary>
/// Saves movie.xml for movies, trailers and music videos
/// </summary>
public class MovieXmlSaver : IMetadataSaver
{
2013-06-23 18:55:30 +00:00
private readonly IServerConfigurationManager _config;
2013-08-12 19:18:31 +00:00
private readonly IItemRepository _itemRepository;
2013-06-23 18:55:30 +00:00
2013-08-12 19:18:31 +00:00
public MovieXmlSaver(IServerConfigurationManager config, IItemRepository itemRepository)
2013-06-23 18:55:30 +00:00
{
_config = config;
2013-08-12 19:18:31 +00:00
_itemRepository = itemRepository;
2013-06-23 18:55:30 +00:00
}
2013-06-09 16:23:06 +00:00
/// <summary>
/// Determines whether [is enabled for] [the specified item].
2013-06-09 16:23:06 +00:00
/// </summary>
/// <param name="item">The item.</param>
/// <param name="updateType">Type of the update.</param>
/// <returns><c>true</c> if [is enabled for] [the specified item]; otherwise, <c>false</c>.</returns>
public bool IsEnabledFor(BaseItem item, ItemUpdateType updateType)
2013-06-09 16:23:06 +00:00
{
var wasMetadataEdited = (updateType & ItemUpdateType.MetadataEdit) == ItemUpdateType.MetadataEdit;
var wasMetadataDownloaded = (updateType & ItemUpdateType.MetadataDownload) == ItemUpdateType.MetadataDownload;
// If new metadata has been downloaded and save local is on, OR metadata was manually edited, proceed
if ((_config.Configuration.SaveLocalMeta && (wasMetadataEdited || wasMetadataDownloaded)) || wasMetadataEdited)
2013-06-09 16:23:06 +00:00
{
var trailer = item as Trailer;
2013-06-09 16:23:06 +00:00
// Don't support local trailers
if (trailer != null)
{
return !trailer.IsLocalTrailer;
}
2013-06-09 16:23:06 +00:00
2013-07-16 18:47:05 +00:00
return item is Movie || item is MusicVideo || item is AdultVideo;
2013-06-09 16:23:06 +00:00
}
return false;
2013-06-09 16:23:06 +00:00
}
2013-07-01 16:00:20 +00:00
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
2013-06-09 16:23:06 +00:00
/// <summary>
/// Saves the specified item.
/// </summary>
/// <param name="item">The item.</param>
/// <param name="cancellationToken">The cancellation token.</param>
/// <returns>Task.</returns>
public void Save(BaseItem item, CancellationToken cancellationToken)
2013-06-09 16:23:06 +00:00
{
var builder = new StringBuilder();
builder.Append("<Title>");
2013-06-23 18:55:30 +00:00
XmlSaverHelpers.AddCommonNodes(item, builder);
2013-07-01 16:00:20 +00:00
if (item.CommunityRating.HasValue)
{
builder.Append("<IMDBrating>" + SecurityElement.Escape(item.CommunityRating.Value.ToString(UsCulture)) + "</IMDBrating>");
}
if (!string.IsNullOrEmpty(item.Overview))
{
builder.Append("<Description><![CDATA[" + item.Overview + "]]></Description>");
}
var musicVideo = item as MusicVideo;
if (musicVideo != null)
{
if (!string.IsNullOrEmpty(musicVideo.Artist))
{
builder.Append("<Artist>" + SecurityElement.Escape(musicVideo.Artist) + "</Artist>");
}
if (!string.IsNullOrEmpty(musicVideo.Album))
{
builder.Append("<Album>" + SecurityElement.Escape(musicVideo.Album) + "</Album>");
}
}
var video = (Video)item;
XmlSaverHelpers.AddMediaInfo(video, builder);
XmlSaverHelpers.AddChapters(video, builder, _itemRepository);
2013-08-12 19:18:31 +00:00
builder.Append("</Title>");
2013-06-10 01:40:53 +00:00
var xmlFilePath = GetSavePath(item);
2013-06-10 01:40:53 +00:00
2013-07-01 16:00:20 +00:00
XmlSaverHelpers.Save(builder, xmlFilePath, new[]
{
"IMDBrating",
2013-07-12 19:56:40 +00:00
"Description",
"Artist",
"Album"
2013-07-01 16:00:20 +00:00
});
// Set last refreshed so that the provider doesn't trigger after the file save
MovieProviderFromXml.Current.SetLastRefreshed(item, DateTime.UtcNow);
}
public string GetSavePath(BaseItem item)
{
var video = (Video)item;
return video.IsInMixedFolder ?
Path.ChangeExtension(item.Path, ".xml") :
Path.Combine(item.MetaLocation, "movie.xml");
}
2013-06-09 16:23:06 +00:00
}
}