diff options
Diffstat (limited to 'MediaBrowser.Dlna/ContentDirectory/ControlHandler.cs')
| -rw-r--r-- | MediaBrowser.Dlna/ContentDirectory/ControlHandler.cs | 393 |
1 files changed, 393 insertions, 0 deletions
diff --git a/MediaBrowser.Dlna/ContentDirectory/ControlHandler.cs b/MediaBrowser.Dlna/ContentDirectory/ControlHandler.cs new file mode 100644 index 000000000..0e4f0d8f9 --- /dev/null +++ b/MediaBrowser.Dlna/ContentDirectory/ControlHandler.cs @@ -0,0 +1,393 @@ +using MediaBrowser.Common.Extensions; +using MediaBrowser.Controller.Configuration; +using MediaBrowser.Controller.Drawing; +using MediaBrowser.Controller.Dto; +using MediaBrowser.Controller.Entities; +using MediaBrowser.Controller.Entities.Audio; +using MediaBrowser.Controller.Entities.Movies; +using MediaBrowser.Controller.Entities.TV; +using MediaBrowser.Controller.Library; +using MediaBrowser.Dlna.Didl; +using MediaBrowser.Dlna.Server; +using MediaBrowser.Dlna.Service; +using MediaBrowser.Model.Dlna; +using MediaBrowser.Model.Entities; +using MediaBrowser.Model.Logging; +using MediaBrowser.Model.Querying; +using System; +using System.Collections.Generic; +using System.Globalization; +using System.Linq; +using System.Text; +using System.Threading; +using System.Xml; + +namespace MediaBrowser.Dlna.ContentDirectory +{ + public class ControlHandler : BaseControlHandler + { + private readonly ILibraryManager _libraryManager; + private readonly IUserDataManager _userDataManager; + private readonly User _user; + + private const string NS_DC = "http://purl.org/dc/elements/1.1/"; + private const string NS_DIDL = "urn:schemas-upnp-org:metadata-1-0/DIDL-Lite/"; + private const string NS_DLNA = "urn:schemas-dlna-org:metadata-1-0/"; + private const string NS_UPNP = "urn:schemas-upnp-org:metadata-1-0/upnp/"; + + private readonly int _systemUpdateId; + private readonly CultureInfo _usCulture = new CultureInfo("en-US"); + + private readonly DidlBuilder _didlBuilder; + + private readonly DeviceProfile _profile; + + public ControlHandler(ILogger logger, ILibraryManager libraryManager, DeviceProfile profile, string serverAddress, IDtoService dtoService, IImageProcessor imageProcessor, IUserDataManager userDataManager, User user, int systemUpdateId, IServerConfigurationManager config) + : base(config, logger) + { + _libraryManager = libraryManager; + _userDataManager = userDataManager; + _user = user; + _systemUpdateId = systemUpdateId; + _profile = profile; + + _didlBuilder = new DidlBuilder(profile, imageProcessor, serverAddress, dtoService); + } + + protected override IEnumerable<KeyValuePair<string, string>> GetResult(string methodName, Headers methodParams) + { + var deviceId = "test"; + + var user = _user; + + if (string.Equals(methodName, "GetSearchCapabilities", StringComparison.OrdinalIgnoreCase)) + return HandleGetSearchCapabilities(); + + if (string.Equals(methodName, "GetSortCapabilities", StringComparison.OrdinalIgnoreCase)) + return HandleGetSortCapabilities(); + + if (string.Equals(methodName, "GetSystemUpdateID", StringComparison.OrdinalIgnoreCase)) + return HandleGetSystemUpdateID(); + + if (string.Equals(methodName, "Browse", StringComparison.OrdinalIgnoreCase)) + return HandleBrowse(methodParams, user, deviceId); + + if (string.Equals(methodName, "X_GetFeatureList", StringComparison.OrdinalIgnoreCase)) + return HandleXGetFeatureList(); + + if (string.Equals(methodName, "X_SetBookmark", StringComparison.OrdinalIgnoreCase)) + return HandleXSetBookmark(methodParams, user); + + if (string.Equals(methodName, "Search", StringComparison.OrdinalIgnoreCase)) + return HandleSearch(methodParams, user, deviceId); + + throw new ResourceNotFoundException("Unexpected control request name: " + methodName); + } + + private IEnumerable<KeyValuePair<string, string>> HandleXSetBookmark(IDictionary<string, string> sparams, User user) + { + var id = sparams["ObjectID"]; + + var item = GetItemFromObjectId(id, user); + + var newbookmark = int.Parse(sparams["PosSecond"], _usCulture); + + var userdata = _userDataManager.GetUserData(user.Id, item.GetUserDataKey()); + + userdata.PlaybackPositionTicks = TimeSpan.FromSeconds(newbookmark).Ticks; + + _userDataManager.SaveUserData(user.Id, item, userdata, UserDataSaveReason.TogglePlayed, + CancellationToken.None); + + return new Headers(); + } + + private IEnumerable<KeyValuePair<string, string>> HandleGetSearchCapabilities() + { + return new Headers { { "SearchCaps", string.Empty } }; + } + + private IEnumerable<KeyValuePair<string, string>> HandleGetSortCapabilities() + { + return new Headers { { "SortCaps", string.Empty } }; + } + + private IEnumerable<KeyValuePair<string, string>> HandleGetSystemUpdateID() + { + var headers = new Headers(true); + headers.Add("Id", _systemUpdateId.ToString(_usCulture)); + return headers; + } + + private IEnumerable<KeyValuePair<string, string>> HandleXGetFeatureList() + { + return new Headers { { "FeatureList", GetFeatureListXml() } }; + } + + private string GetFeatureListXml() + { + var builder = new StringBuilder(); + + builder.Append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>"); + builder.Append("<Features xmlns=\"urn:schemas-upnp-org:av:avs\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:schemaLocation=\"urn:schemas-upnp-org:av:avs http://www.upnp.org/schemas/av/avs.xsd\">"); + + builder.Append("<Feature name=\"samsung.com_BASICVIEW\" version=\"1\">"); + builder.Append("<container id=\"I\" type=\"object.item.imageItem\"/>"); + builder.Append("<container id=\"A\" type=\"object.item.audioItem\"/>"); + builder.Append("<container id=\"V\" type=\"object.item.videoItem\"/>"); + builder.Append("</Feature>"); + + builder.Append("</Features>"); + + return builder.ToString(); + } + + private IEnumerable<KeyValuePair<string, string>> HandleBrowse(Headers sparams, User user, string deviceId) + { + var id = sparams["ObjectID"]; + var flag = sparams["BrowseFlag"]; + var filter = new Filter(sparams.GetValueOrDefault("Filter", "*")); + var sortCriteria = new SortCriteria(sparams.GetValueOrDefault("SortCriteria", "")); + + var provided = 0; + var requested = 0; + var start = 0; + + if (sparams.ContainsKey("RequestedCount") && int.TryParse(sparams["RequestedCount"], out requested) && requested <= 0) + { + requested = 0; + } + if (sparams.ContainsKey("StartingIndex") && int.TryParse(sparams["StartingIndex"], out start) && start <= 0) + { + start = 0; + } + + //var root = GetItem(id) as IMediaFolder; + var result = new XmlDocument(); + + var didl = result.CreateElement(string.Empty, "DIDL-Lite", NS_DIDL); + didl.SetAttribute("xmlns:dc", NS_DC); + didl.SetAttribute("xmlns:dlna", NS_DLNA); + didl.SetAttribute("xmlns:upnp", NS_UPNP); + //didl.SetAttribute("xmlns:sec", NS_SEC); + result.AppendChild(didl); + + var folder = (Folder)GetItemFromObjectId(id, user); + + var children = GetChildrenSorted(folder, user, sortCriteria).ToList(); + + var totalCount = children.Count; + + if (string.Equals(flag, "BrowseMetadata")) + { + result.DocumentElement.AppendChild(_didlBuilder.GetFolderElement(result, folder, children.Count, filter)); + provided++; + } + else + { + if (start > 0) + { + children = children.Skip(start).ToList(); + } + if (requested > 0) + { + children = children.Take(requested).ToList(); + } + + provided = children.Count; + + foreach (var i in children) + { + if (i.IsFolder) + { + var f = (Folder)i; + var childCount = GetChildrenSorted(f, user, sortCriteria).Count(); + + result.DocumentElement.AppendChild(_didlBuilder.GetFolderElement(result, f, childCount, filter)); + } + else + { + result.DocumentElement.AppendChild(_didlBuilder.GetItemElement(result, i, deviceId, filter)); + } + } + } + + var resXML = result.OuterXml; + + return new List<KeyValuePair<string, string>> + { + new KeyValuePair<string,string>("Result", resXML), + new KeyValuePair<string,string>("NumberReturned", provided.ToString(_usCulture)), + new KeyValuePair<string,string>("TotalMatches", totalCount.ToString(_usCulture)), + new KeyValuePair<string,string>("UpdateID", _systemUpdateId.ToString(_usCulture)) + }; + } + + private IEnumerable<KeyValuePair<string, string>> HandleSearch(Headers sparams, User user, string deviceId) + { + var searchCriteria = new SearchCriteria(sparams.GetValueOrDefault("SearchCriteria", "")); + var sortCriteria = new SortCriteria(sparams.GetValueOrDefault("SortCriteria", "")); + var filter = new Filter(sparams.GetValueOrDefault("Filter", "*")); + + // sort example: dc:title, dc:date + + var requested = 0; + var start = 0; + + if (sparams.ContainsKey("RequestedCount") && int.TryParse(sparams["RequestedCount"], out requested) && requested <= 0) + { + requested = 0; + } + if (sparams.ContainsKey("StartingIndex") && int.TryParse(sparams["StartingIndex"], out start) && start <= 0) + { + start = 0; + } + + //var root = GetItem(id) as IMediaFolder; + var result = new XmlDocument(); + + var didl = result.CreateElement(string.Empty, "DIDL-Lite", NS_DIDL); + didl.SetAttribute("xmlns:dc", NS_DC); + didl.SetAttribute("xmlns:dlna", NS_DLNA); + didl.SetAttribute("xmlns:upnp", NS_UPNP); + + foreach (var att in _profile.XmlRootAttributes) + { + didl.SetAttribute(att.Name, att.Value); + } + + result.AppendChild(didl); + + var folder = (Folder)GetItemFromObjectId(sparams["ContainerID"], user); + + var children = GetChildrenSorted(folder, user, searchCriteria, sortCriteria).ToList(); + + var totalCount = children.Count; + + if (start > 0) + { + children = children.Skip(start).ToList(); + } + if (requested > 0) + { + children = children.Take(requested).ToList(); + } + + var provided = children.Count; + + foreach (var i in children) + { + if (i.IsFolder) + { + var f = (Folder)i; + var childCount = GetChildrenSorted(f, user, searchCriteria, sortCriteria).Count(); + + result.DocumentElement.AppendChild(_didlBuilder.GetFolderElement(result, f, childCount, filter)); + } + else + { + result.DocumentElement.AppendChild(_didlBuilder.GetItemElement(result, i, deviceId, filter)); + } + } + + var resXML = result.OuterXml; + + return new List<KeyValuePair<string, string>> + { + new KeyValuePair<string,string>("Result", resXML), + new KeyValuePair<string,string>("NumberReturned", provided.ToString(_usCulture)), + new KeyValuePair<string,string>("TotalMatches", totalCount.ToString(_usCulture)), + new KeyValuePair<string,string>("UpdateID", _systemUpdateId.ToString(_usCulture)) + }; + } + + private IEnumerable<BaseItem> GetChildrenSorted(Folder folder, User user, SearchCriteria search, SortCriteria sort) + { + if (search.SearchType == SearchType.Unknown) + { + return GetChildrenSorted(folder, user, sort); + } + + var items = folder.GetRecursiveChildren(user); + items = FilterUnsupportedContent(items); + + if (search.SearchType == SearchType.Audio) + { + items = items.OfType<Audio>(); + } + else if (search.SearchType == SearchType.Video) + { + items = items.OfType<Video>(); + } + else if (search.SearchType == SearchType.Image) + { + items = items.OfType<Photo>(); + } + else if (search.SearchType == SearchType.Playlist) + { + } + + return SortItems(items, user, sort); + } + + private IEnumerable<BaseItem> GetChildrenSorted(Folder folder, User user, SortCriteria sort) + { + var items = folder.GetChildren(user, true); + + items = FilterUnsupportedContent(items); + + if (folder is Series || folder is Season || folder is BoxSet) + { + return items; + } + + return SortItems(items, user, sort); + } + + private IEnumerable<BaseItem> SortItems(IEnumerable<BaseItem> items, User user, SortCriteria sort) + { + return _libraryManager.Sort(items, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending); + } + + private IEnumerable<BaseItem> FilterUnsupportedContent(IEnumerable<BaseItem> items) + { + return items.Where(i => + { + // Unplayable + // TODO: Display and prevent playback with restricted flag? + if (i.LocationType == LocationType.Virtual) + { + return false; + } + + // Unplayable + // TODO: Display and prevent playback with restricted flag? + var supportsPlaceHolder = i as ISupportsPlaceHolders; + if (supportsPlaceHolder != null && supportsPlaceHolder.IsPlaceHolder) + { + return false; + } + + // Upnp renderers won't understand these + // TODO: Display and prevent playback with restricted flag? + if (i is Game || i is Book) + { + return false; + } + + return true; + }); + } + + private BaseItem GetItemFromObjectId(string id, User user) + { + return string.IsNullOrWhiteSpace(id) || string.Equals(id, "0", StringComparison.OrdinalIgnoreCase) + + // Samsung sometimes uses 1 as root + || string.Equals(id, "1", StringComparison.OrdinalIgnoreCase) + + ? user.RootFolder + : _libraryManager.GetItemById(new Guid(id)); + } + } +} |