| | | 1 | | #nullable disable |
| | | 2 | | |
| | | 3 | | #pragma warning disable CS1591 |
| | | 4 | | |
| | | 5 | | using System; |
| | | 6 | | using System.Collections.Concurrent; |
| | | 7 | | using System.Collections.Generic; |
| | | 8 | | using System.Globalization; |
| | | 9 | | using System.IO; |
| | | 10 | | using System.Linq; |
| | | 11 | | using System.Text.Json; |
| | | 12 | | using System.Threading; |
| | | 13 | | using System.Threading.Tasks; |
| | | 14 | | using AsyncKeyedLock; |
| | | 15 | | using Jellyfin.Data.Entities; |
| | | 16 | | using Jellyfin.Data.Enums; |
| | | 17 | | using Jellyfin.Extensions; |
| | | 18 | | using Jellyfin.Extensions.Json; |
| | | 19 | | using MediaBrowser.Common.Configuration; |
| | | 20 | | using MediaBrowser.Common.Extensions; |
| | | 21 | | using MediaBrowser.Controller; |
| | | 22 | | using MediaBrowser.Controller.Entities; |
| | | 23 | | using MediaBrowser.Controller.Library; |
| | | 24 | | using MediaBrowser.Controller.LiveTv; |
| | | 25 | | using MediaBrowser.Controller.MediaEncoding; |
| | | 26 | | using MediaBrowser.Controller.Persistence; |
| | | 27 | | using MediaBrowser.Controller.Providers; |
| | | 28 | | using MediaBrowser.Model.Dlna; |
| | | 29 | | using MediaBrowser.Model.Dto; |
| | | 30 | | using MediaBrowser.Model.Entities; |
| | | 31 | | using MediaBrowser.Model.Globalization; |
| | | 32 | | using MediaBrowser.Model.IO; |
| | | 33 | | using MediaBrowser.Model.MediaInfo; |
| | | 34 | | using Microsoft.Extensions.Logging; |
| | | 35 | | |
| | | 36 | | namespace Emby.Server.Implementations.Library |
| | | 37 | | { |
| | | 38 | | public class MediaSourceManager : IMediaSourceManager, IDisposable |
| | | 39 | | { |
| | | 40 | | // Do not use a pipe here because Roku http requests to the server will fail, without any explicit error message |
| | | 41 | | private const char LiveStreamIdDelimeter = '_'; |
| | | 42 | | |
| | | 43 | | private readonly IServerApplicationHost _appHost; |
| | | 44 | | private readonly IItemRepository _itemRepo; |
| | | 45 | | private readonly IUserManager _userManager; |
| | | 46 | | private readonly ILibraryManager _libraryManager; |
| | | 47 | | private readonly IFileSystem _fileSystem; |
| | | 48 | | private readonly ILogger<MediaSourceManager> _logger; |
| | | 49 | | private readonly IUserDataManager _userDataManager; |
| | | 50 | | private readonly IMediaEncoder _mediaEncoder; |
| | | 51 | | private readonly ILocalizationManager _localizationManager; |
| | | 52 | | private readonly IApplicationPaths _appPaths; |
| | | 53 | | private readonly IDirectoryService _directoryService; |
| | | 54 | | |
| | 28 | 55 | | private readonly ConcurrentDictionary<string, ILiveStream> _openStreams = new ConcurrentDictionary<string, ILive |
| | 28 | 56 | | private readonly AsyncNonKeyedLocker _liveStreamLocker = new(1); |
| | 28 | 57 | | private readonly JsonSerializerOptions _jsonOptions = JsonDefaults.Options; |
| | | 58 | | |
| | | 59 | | private IMediaSourceProvider[] _providers; |
| | | 60 | | |
| | | 61 | | public MediaSourceManager( |
| | | 62 | | IServerApplicationHost appHost, |
| | | 63 | | IItemRepository itemRepo, |
| | | 64 | | IApplicationPaths applicationPaths, |
| | | 65 | | ILocalizationManager localizationManager, |
| | | 66 | | IUserManager userManager, |
| | | 67 | | ILibraryManager libraryManager, |
| | | 68 | | ILogger<MediaSourceManager> logger, |
| | | 69 | | IFileSystem fileSystem, |
| | | 70 | | IUserDataManager userDataManager, |
| | | 71 | | IMediaEncoder mediaEncoder, |
| | | 72 | | IDirectoryService directoryService) |
| | | 73 | | { |
| | 28 | 74 | | _appHost = appHost; |
| | 28 | 75 | | _itemRepo = itemRepo; |
| | 28 | 76 | | _userManager = userManager; |
| | 28 | 77 | | _libraryManager = libraryManager; |
| | 28 | 78 | | _logger = logger; |
| | 28 | 79 | | _fileSystem = fileSystem; |
| | 28 | 80 | | _userDataManager = userDataManager; |
| | 28 | 81 | | _mediaEncoder = mediaEncoder; |
| | 28 | 82 | | _localizationManager = localizationManager; |
| | 28 | 83 | | _appPaths = applicationPaths; |
| | 28 | 84 | | _directoryService = directoryService; |
| | 28 | 85 | | } |
| | | 86 | | |
| | | 87 | | public void AddParts(IEnumerable<IMediaSourceProvider> providers) |
| | | 88 | | { |
| | 22 | 89 | | _providers = providers.ToArray(); |
| | 22 | 90 | | } |
| | | 91 | | |
| | | 92 | | public List<MediaStream> GetMediaStreams(MediaStreamQuery query) |
| | | 93 | | { |
| | 0 | 94 | | var list = _itemRepo.GetMediaStreams(query); |
| | | 95 | | |
| | 0 | 96 | | foreach (var stream in list) |
| | | 97 | | { |
| | 0 | 98 | | stream.SupportsExternalStream = StreamSupportsExternalStream(stream); |
| | | 99 | | } |
| | | 100 | | |
| | 0 | 101 | | return list; |
| | | 102 | | } |
| | | 103 | | |
| | | 104 | | private static bool StreamSupportsExternalStream(MediaStream stream) |
| | | 105 | | { |
| | 0 | 106 | | if (stream.IsExternal) |
| | | 107 | | { |
| | 0 | 108 | | return true; |
| | | 109 | | } |
| | | 110 | | |
| | 0 | 111 | | if (stream.IsTextSubtitleStream) |
| | | 112 | | { |
| | 0 | 113 | | return true; |
| | | 114 | | } |
| | | 115 | | |
| | 0 | 116 | | if (stream.IsPgsSubtitleStream) |
| | | 117 | | { |
| | 0 | 118 | | return true; |
| | | 119 | | } |
| | | 120 | | |
| | 0 | 121 | | return false; |
| | | 122 | | } |
| | | 123 | | |
| | | 124 | | public List<MediaStream> GetMediaStreams(Guid itemId) |
| | | 125 | | { |
| | 0 | 126 | | var list = GetMediaStreams(new MediaStreamQuery |
| | 0 | 127 | | { |
| | 0 | 128 | | ItemId = itemId |
| | 0 | 129 | | }); |
| | | 130 | | |
| | 0 | 131 | | return GetMediaStreamsForItem(list); |
| | | 132 | | } |
| | | 133 | | |
| | | 134 | | private List<MediaStream> GetMediaStreamsForItem(List<MediaStream> streams) |
| | | 135 | | { |
| | 0 | 136 | | foreach (var stream in streams) |
| | | 137 | | { |
| | 0 | 138 | | if (stream.Type == MediaStreamType.Subtitle) |
| | | 139 | | { |
| | 0 | 140 | | stream.SupportsExternalStream = StreamSupportsExternalStream(stream); |
| | | 141 | | } |
| | | 142 | | } |
| | | 143 | | |
| | 0 | 144 | | return streams; |
| | | 145 | | } |
| | | 146 | | |
| | | 147 | | /// <inheritdoc /> |
| | | 148 | | public List<MediaAttachment> GetMediaAttachments(MediaAttachmentQuery query) |
| | | 149 | | { |
| | 0 | 150 | | return _itemRepo.GetMediaAttachments(query); |
| | | 151 | | } |
| | | 152 | | |
| | | 153 | | /// <inheritdoc /> |
| | | 154 | | public List<MediaAttachment> GetMediaAttachments(Guid itemId) |
| | | 155 | | { |
| | 0 | 156 | | return GetMediaAttachments(new MediaAttachmentQuery |
| | 0 | 157 | | { |
| | 0 | 158 | | ItemId = itemId |
| | 0 | 159 | | }); |
| | | 160 | | } |
| | | 161 | | |
| | | 162 | | public async Task<List<MediaSourceInfo>> GetPlaybackMediaSources(BaseItem item, User user, bool allowMediaProbe, |
| | | 163 | | { |
| | | 164 | | var mediaSources = GetStaticMediaSources(item, enablePathSubstitution, user); |
| | | 165 | | |
| | | 166 | | // If file is strm or main media stream is missing, force a metadata refresh with remote probing |
| | | 167 | | if (allowMediaProbe && mediaSources[0].Type != MediaSourceType.Placeholder |
| | | 168 | | && (item.Path.EndsWith(".strm", StringComparison.OrdinalIgnoreCase) |
| | | 169 | | || (item.MediaType == MediaType.Video && mediaSources[0].MediaStreams.All(i => i.Type != MediaStream |
| | | 170 | | || (item.MediaType == MediaType.Audio && mediaSources[0].MediaStreams.All(i => i.Type != MediaStream |
| | | 171 | | { |
| | | 172 | | await item.RefreshMetadata( |
| | | 173 | | new MetadataRefreshOptions(_directoryService) |
| | | 174 | | { |
| | | 175 | | EnableRemoteContentProbe = true, |
| | | 176 | | MetadataRefreshMode = MetadataRefreshMode.FullRefresh |
| | | 177 | | }, |
| | | 178 | | cancellationToken).ConfigureAwait(false); |
| | | 179 | | |
| | | 180 | | mediaSources = GetStaticMediaSources(item, enablePathSubstitution, user); |
| | | 181 | | } |
| | | 182 | | |
| | | 183 | | var dynamicMediaSources = await GetDynamicMediaSources(item, cancellationToken).ConfigureAwait(false); |
| | | 184 | | |
| | | 185 | | var list = new List<MediaSourceInfo>(); |
| | | 186 | | |
| | | 187 | | list.AddRange(mediaSources); |
| | | 188 | | |
| | | 189 | | foreach (var source in dynamicMediaSources) |
| | | 190 | | { |
| | | 191 | | // Validate that this is actually possible |
| | | 192 | | if (source.SupportsDirectStream) |
| | | 193 | | { |
| | | 194 | | source.SupportsDirectStream = SupportsDirectStream(source.Path, source.Protocol); |
| | | 195 | | } |
| | | 196 | | |
| | | 197 | | if (user is not null) |
| | | 198 | | { |
| | | 199 | | SetDefaultAudioAndSubtitleStreamIndices(item, source, user); |
| | | 200 | | |
| | | 201 | | if (item.MediaType == MediaType.Audio) |
| | | 202 | | { |
| | | 203 | | source.SupportsTranscoding = user.HasPermission(PermissionKind.EnableAudioPlaybackTranscoding); |
| | | 204 | | } |
| | | 205 | | else if (item.MediaType == MediaType.Video) |
| | | 206 | | { |
| | | 207 | | source.SupportsTranscoding = user.HasPermission(PermissionKind.EnableVideoPlaybackTranscoding); |
| | | 208 | | source.SupportsDirectStream = user.HasPermission(PermissionKind.EnablePlaybackRemuxing); |
| | | 209 | | } |
| | | 210 | | } |
| | | 211 | | |
| | | 212 | | list.Add(source); |
| | | 213 | | } |
| | | 214 | | |
| | | 215 | | return SortMediaSources(list); |
| | | 216 | | } |
| | | 217 | | |
| | | 218 | | /// <inheritdoc />> |
| | | 219 | | public MediaProtocol GetPathProtocol(string path) |
| | | 220 | | { |
| | 1669 | 221 | | if (path.StartsWith("Rtsp", StringComparison.OrdinalIgnoreCase)) |
| | | 222 | | { |
| | 1 | 223 | | return MediaProtocol.Rtsp; |
| | | 224 | | } |
| | | 225 | | |
| | 1668 | 226 | | if (path.StartsWith("Rtmp", StringComparison.OrdinalIgnoreCase)) |
| | | 227 | | { |
| | 0 | 228 | | return MediaProtocol.Rtmp; |
| | | 229 | | } |
| | | 230 | | |
| | 1668 | 231 | | if (path.StartsWith("Http", StringComparison.OrdinalIgnoreCase)) |
| | | 232 | | { |
| | 2 | 233 | | return MediaProtocol.Http; |
| | | 234 | | } |
| | | 235 | | |
| | 1666 | 236 | | if (path.StartsWith("rtp", StringComparison.OrdinalIgnoreCase)) |
| | | 237 | | { |
| | 0 | 238 | | return MediaProtocol.Rtp; |
| | | 239 | | } |
| | | 240 | | |
| | 1666 | 241 | | if (path.StartsWith("ftp", StringComparison.OrdinalIgnoreCase)) |
| | | 242 | | { |
| | 0 | 243 | | return MediaProtocol.Ftp; |
| | | 244 | | } |
| | | 245 | | |
| | 1666 | 246 | | if (path.StartsWith("udp", StringComparison.OrdinalIgnoreCase)) |
| | | 247 | | { |
| | 0 | 248 | | return MediaProtocol.Udp; |
| | | 249 | | } |
| | | 250 | | |
| | 1666 | 251 | | return _fileSystem.IsPathFile(path) ? MediaProtocol.File : MediaProtocol.Http; |
| | | 252 | | } |
| | | 253 | | |
| | | 254 | | public bool SupportsDirectStream(string path, MediaProtocol protocol) |
| | | 255 | | { |
| | 0 | 256 | | if (protocol == MediaProtocol.File) |
| | | 257 | | { |
| | 0 | 258 | | return true; |
| | | 259 | | } |
| | | 260 | | |
| | 0 | 261 | | if (protocol == MediaProtocol.Http) |
| | | 262 | | { |
| | 0 | 263 | | if (path is not null) |
| | | 264 | | { |
| | 0 | 265 | | if (path.Contains(".m3u", StringComparison.OrdinalIgnoreCase)) |
| | | 266 | | { |
| | 0 | 267 | | return false; |
| | | 268 | | } |
| | | 269 | | |
| | 0 | 270 | | return true; |
| | | 271 | | } |
| | | 272 | | } |
| | | 273 | | |
| | 0 | 274 | | return false; |
| | | 275 | | } |
| | | 276 | | |
| | | 277 | | private async Task<IEnumerable<MediaSourceInfo>> GetDynamicMediaSources(BaseItem item, CancellationToken cancell |
| | | 278 | | { |
| | | 279 | | var tasks = _providers.Select(i => GetDynamicMediaSources(item, i, cancellationToken)); |
| | | 280 | | var results = await Task.WhenAll(tasks).ConfigureAwait(false); |
| | | 281 | | |
| | | 282 | | return results.SelectMany(i => i); |
| | | 283 | | } |
| | | 284 | | |
| | | 285 | | private async Task<IEnumerable<MediaSourceInfo>> GetDynamicMediaSources(BaseItem item, IMediaSourceProvider prov |
| | | 286 | | { |
| | | 287 | | try |
| | | 288 | | { |
| | | 289 | | var sources = await provider.GetMediaSources(item, cancellationToken).ConfigureAwait(false); |
| | | 290 | | var list = sources.ToList(); |
| | | 291 | | |
| | | 292 | | foreach (var mediaSource in list) |
| | | 293 | | { |
| | | 294 | | mediaSource.InferTotalBitrate(); |
| | | 295 | | |
| | | 296 | | SetKeyProperties(provider, mediaSource); |
| | | 297 | | } |
| | | 298 | | |
| | | 299 | | return list; |
| | | 300 | | } |
| | | 301 | | catch (Exception ex) |
| | | 302 | | { |
| | | 303 | | _logger.LogError(ex, "Error getting media sources"); |
| | | 304 | | return []; |
| | | 305 | | } |
| | | 306 | | } |
| | | 307 | | |
| | | 308 | | private static void SetKeyProperties(IMediaSourceProvider provider, MediaSourceInfo mediaSource) |
| | | 309 | | { |
| | 0 | 310 | | var prefix = provider.GetType().FullName.GetMD5().ToString("N", CultureInfo.InvariantCulture) + LiveStreamId |
| | | 311 | | |
| | 0 | 312 | | if (!string.IsNullOrEmpty(mediaSource.OpenToken) && !mediaSource.OpenToken.StartsWith(prefix, StringComparis |
| | | 313 | | { |
| | 0 | 314 | | mediaSource.OpenToken = prefix + mediaSource.OpenToken; |
| | | 315 | | } |
| | | 316 | | |
| | 0 | 317 | | if (!string.IsNullOrEmpty(mediaSource.LiveStreamId) && !mediaSource.LiveStreamId.StartsWith(prefix, StringCo |
| | | 318 | | { |
| | 0 | 319 | | mediaSource.LiveStreamId = prefix + mediaSource.LiveStreamId; |
| | | 320 | | } |
| | 0 | 321 | | } |
| | | 322 | | |
| | | 323 | | public async Task<MediaSourceInfo> GetMediaSource(BaseItem item, string mediaSourceId, string liveStreamId, bool |
| | | 324 | | { |
| | | 325 | | if (!string.IsNullOrEmpty(liveStreamId)) |
| | | 326 | | { |
| | | 327 | | return await GetLiveStream(liveStreamId, cancellationToken).ConfigureAwait(false); |
| | | 328 | | } |
| | | 329 | | |
| | | 330 | | var sources = await GetPlaybackMediaSources(item, null, false, enablePathSubstitution, cancellationToken).Co |
| | | 331 | | |
| | | 332 | | return sources.FirstOrDefault(i => string.Equals(i.Id, mediaSourceId, StringComparison.OrdinalIgnoreCase)); |
| | | 333 | | } |
| | | 334 | | |
| | | 335 | | public List<MediaSourceInfo> GetStaticMediaSources(BaseItem item, bool enablePathSubstitution, User user = null) |
| | | 336 | | { |
| | 0 | 337 | | ArgumentNullException.ThrowIfNull(item); |
| | | 338 | | |
| | 0 | 339 | | var hasMediaSources = (IHasMediaSources)item; |
| | | 340 | | |
| | 0 | 341 | | var sources = hasMediaSources.GetMediaSources(enablePathSubstitution); |
| | | 342 | | |
| | 0 | 343 | | if (user is not null) |
| | | 344 | | { |
| | 0 | 345 | | foreach (var source in sources) |
| | | 346 | | { |
| | 0 | 347 | | SetDefaultAudioAndSubtitleStreamIndices(item, source, user); |
| | | 348 | | |
| | 0 | 349 | | if (item.MediaType == MediaType.Audio) |
| | | 350 | | { |
| | 0 | 351 | | source.SupportsTranscoding = user.HasPermission(PermissionKind.EnableAudioPlaybackTranscoding); |
| | | 352 | | } |
| | 0 | 353 | | else if (item.MediaType == MediaType.Video) |
| | | 354 | | { |
| | 0 | 355 | | source.SupportsTranscoding = user.HasPermission(PermissionKind.EnableVideoPlaybackTranscoding); |
| | 0 | 356 | | source.SupportsDirectStream = user.HasPermission(PermissionKind.EnablePlaybackRemuxing); |
| | | 357 | | } |
| | | 358 | | } |
| | | 359 | | } |
| | | 360 | | |
| | 0 | 361 | | return sources; |
| | | 362 | | } |
| | | 363 | | |
| | | 364 | | private IReadOnlyList<string> NormalizeLanguage(string language) |
| | | 365 | | { |
| | 0 | 366 | | if (string.IsNullOrEmpty(language)) |
| | | 367 | | { |
| | 0 | 368 | | return []; |
| | | 369 | | } |
| | | 370 | | |
| | 0 | 371 | | var culture = _localizationManager.FindLanguageInfo(language); |
| | 0 | 372 | | if (culture is not null) |
| | | 373 | | { |
| | 0 | 374 | | return culture.ThreeLetterISOLanguageNames; |
| | | 375 | | } |
| | | 376 | | |
| | 0 | 377 | | return [language]; |
| | | 378 | | } |
| | | 379 | | |
| | | 380 | | private void SetDefaultSubtitleStreamIndex(MediaSourceInfo source, UserItemData userData, User user, bool allowR |
| | | 381 | | { |
| | 0 | 382 | | if (userData is not null |
| | 0 | 383 | | && userData.SubtitleStreamIndex.HasValue |
| | 0 | 384 | | && user.RememberSubtitleSelections |
| | 0 | 385 | | && user.SubtitleMode != SubtitlePlaybackMode.None |
| | 0 | 386 | | && allowRememberingSelection) |
| | | 387 | | { |
| | 0 | 388 | | var index = userData.SubtitleStreamIndex.Value; |
| | | 389 | | // Make sure the saved index is still valid |
| | 0 | 390 | | if (index == -1 || source.MediaStreams.Any(i => i.Type == MediaStreamType.Subtitle && i.Index == index)) |
| | | 391 | | { |
| | 0 | 392 | | source.DefaultSubtitleStreamIndex = index; |
| | 0 | 393 | | return; |
| | | 394 | | } |
| | | 395 | | } |
| | | 396 | | |
| | 0 | 397 | | var preferredSubs = NormalizeLanguage(user.SubtitleLanguagePreference); |
| | | 398 | | |
| | 0 | 399 | | var defaultAudioIndex = source.DefaultAudioStreamIndex; |
| | 0 | 400 | | var audioLanguage = defaultAudioIndex is null |
| | 0 | 401 | | ? null |
| | 0 | 402 | | : source.MediaStreams.Where(i => i.Type == MediaStreamType.Audio && i.Index == defaultAudioIndex).Select |
| | | 403 | | |
| | 0 | 404 | | source.DefaultSubtitleStreamIndex = MediaStreamSelector.GetDefaultSubtitleStreamIndex( |
| | 0 | 405 | | source.MediaStreams, |
| | 0 | 406 | | preferredSubs, |
| | 0 | 407 | | user.SubtitleMode, |
| | 0 | 408 | | audioLanguage); |
| | | 409 | | |
| | 0 | 410 | | MediaStreamSelector.SetSubtitleStreamScores(source.MediaStreams, preferredSubs, user.SubtitleMode, audioLang |
| | 0 | 411 | | } |
| | | 412 | | |
| | | 413 | | private void SetDefaultAudioStreamIndex(MediaSourceInfo source, UserItemData userData, User user, bool allowReme |
| | | 414 | | { |
| | 0 | 415 | | if (userData is not null && userData.AudioStreamIndex.HasValue && user.RememberAudioSelections && allowRemem |
| | | 416 | | { |
| | 0 | 417 | | var index = userData.AudioStreamIndex.Value; |
| | | 418 | | // Make sure the saved index is still valid |
| | 0 | 419 | | if (source.MediaStreams.Any(i => i.Type == MediaStreamType.Audio && i.Index == index)) |
| | | 420 | | { |
| | 0 | 421 | | source.DefaultAudioStreamIndex = index; |
| | 0 | 422 | | return; |
| | | 423 | | } |
| | | 424 | | } |
| | | 425 | | |
| | 0 | 426 | | var preferredAudio = NormalizeLanguage(user.AudioLanguagePreference); |
| | | 427 | | |
| | 0 | 428 | | source.DefaultAudioStreamIndex = MediaStreamSelector.GetDefaultAudioStreamIndex(source.MediaStreams, preferr |
| | 0 | 429 | | } |
| | | 430 | | |
| | | 431 | | public void SetDefaultAudioAndSubtitleStreamIndices(BaseItem item, MediaSourceInfo source, User user) |
| | | 432 | | { |
| | | 433 | | // Item would only be null if the app didn't supply ItemId as part of the live stream open request |
| | 0 | 434 | | var mediaType = item?.MediaType ?? MediaType.Video; |
| | | 435 | | |
| | 0 | 436 | | if (mediaType == MediaType.Video) |
| | | 437 | | { |
| | 0 | 438 | | var userData = item is null ? null : _userDataManager.GetUserData(user, item); |
| | | 439 | | |
| | 0 | 440 | | var allowRememberingSelection = item is null || item.EnableRememberingTrackSelections; |
| | | 441 | | |
| | 0 | 442 | | SetDefaultAudioStreamIndex(source, userData, user, allowRememberingSelection); |
| | 0 | 443 | | SetDefaultSubtitleStreamIndex(source, userData, user, allowRememberingSelection); |
| | | 444 | | } |
| | 0 | 445 | | else if (mediaType == MediaType.Audio) |
| | | 446 | | { |
| | 0 | 447 | | var audio = source.MediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Audio); |
| | | 448 | | |
| | 0 | 449 | | if (audio is not null) |
| | | 450 | | { |
| | 0 | 451 | | source.DefaultAudioStreamIndex = audio.Index; |
| | | 452 | | } |
| | | 453 | | } |
| | 0 | 454 | | } |
| | | 455 | | |
| | | 456 | | private static List<MediaSourceInfo> SortMediaSources(IEnumerable<MediaSourceInfo> sources) |
| | | 457 | | { |
| | 0 | 458 | | return sources.OrderBy(i => |
| | 0 | 459 | | { |
| | 0 | 460 | | if (i.VideoType.HasValue && i.VideoType.Value == VideoType.VideoFile) |
| | 0 | 461 | | { |
| | 0 | 462 | | return 0; |
| | 0 | 463 | | } |
| | 0 | 464 | | |
| | 0 | 465 | | return 1; |
| | 0 | 466 | | }).ThenBy(i => i.Video3DFormat.HasValue ? 1 : 0) |
| | 0 | 467 | | .ThenByDescending(i => |
| | 0 | 468 | | { |
| | 0 | 469 | | var stream = i.VideoStream; |
| | 0 | 470 | | |
| | 0 | 471 | | return stream?.Width ?? 0; |
| | 0 | 472 | | }) |
| | 0 | 473 | | .Where(i => i.Type != MediaSourceType.Placeholder) |
| | 0 | 474 | | .ToList(); |
| | | 475 | | } |
| | | 476 | | |
| | | 477 | | public async Task<Tuple<LiveStreamResponse, IDirectStreamProvider>> OpenLiveStreamInternal(LiveStreamRequest req |
| | | 478 | | { |
| | | 479 | | MediaSourceInfo mediaSource; |
| | | 480 | | ILiveStream liveStream; |
| | | 481 | | |
| | | 482 | | using (await _liveStreamLocker.LockAsync(cancellationToken).ConfigureAwait(false)) |
| | | 483 | | { |
| | | 484 | | var (provider, keyId) = GetProvider(request.OpenToken); |
| | | 485 | | |
| | | 486 | | var currentLiveStreams = _openStreams.Values.ToList(); |
| | | 487 | | |
| | | 488 | | liveStream = await provider.OpenMediaSource(keyId, currentLiveStreams, cancellationToken).ConfigureAwait |
| | | 489 | | |
| | | 490 | | mediaSource = liveStream.MediaSource; |
| | | 491 | | |
| | | 492 | | // Validate that this is actually possible |
| | | 493 | | if (mediaSource.SupportsDirectStream) |
| | | 494 | | { |
| | | 495 | | mediaSource.SupportsDirectStream = SupportsDirectStream(mediaSource.Path, mediaSource.Protocol); |
| | | 496 | | } |
| | | 497 | | |
| | | 498 | | SetKeyProperties(provider, mediaSource); |
| | | 499 | | |
| | | 500 | | _openStreams[mediaSource.LiveStreamId] = liveStream; |
| | | 501 | | } |
| | | 502 | | |
| | | 503 | | try |
| | | 504 | | { |
| | | 505 | | if (mediaSource.MediaStreams.Any(i => i.Index != -1) || !mediaSource.SupportsProbing) |
| | | 506 | | { |
| | | 507 | | AddMediaInfo(mediaSource); |
| | | 508 | | } |
| | | 509 | | else |
| | | 510 | | { |
| | | 511 | | // hack - these two values were taken from LiveTVMediaSourceProvider |
| | | 512 | | string cacheKey = request.OpenToken; |
| | | 513 | | |
| | | 514 | | await new LiveStreamHelper(_mediaEncoder, _logger, _appPaths) |
| | | 515 | | .AddMediaInfoWithProbe(mediaSource, false, cacheKey, true, cancellationToken) |
| | | 516 | | .ConfigureAwait(false); |
| | | 517 | | } |
| | | 518 | | } |
| | | 519 | | catch (Exception ex) |
| | | 520 | | { |
| | | 521 | | _logger.LogError(ex, "Error probing live tv stream"); |
| | | 522 | | AddMediaInfo(mediaSource); |
| | | 523 | | } |
| | | 524 | | |
| | | 525 | | // TODO: @bond Fix |
| | | 526 | | var json = JsonSerializer.SerializeToUtf8Bytes(mediaSource, _jsonOptions); |
| | | 527 | | _logger.LogInformation("Live stream opened: {@MediaSource}", mediaSource); |
| | | 528 | | var clone = JsonSerializer.Deserialize<MediaSourceInfo>(json, _jsonOptions); |
| | | 529 | | |
| | | 530 | | if (!request.UserId.IsEmpty()) |
| | | 531 | | { |
| | | 532 | | var user = _userManager.GetUserById(request.UserId); |
| | | 533 | | var item = request.ItemId.IsEmpty() |
| | | 534 | | ? null |
| | | 535 | | : _libraryManager.GetItemById(request.ItemId); |
| | | 536 | | SetDefaultAudioAndSubtitleStreamIndices(item, clone, user); |
| | | 537 | | } |
| | | 538 | | |
| | | 539 | | return new Tuple<LiveStreamResponse, IDirectStreamProvider>(new LiveStreamResponse(clone), liveStream as IDi |
| | | 540 | | } |
| | | 541 | | |
| | | 542 | | private static void AddMediaInfo(MediaSourceInfo mediaSource) |
| | | 543 | | { |
| | 0 | 544 | | mediaSource.DefaultSubtitleStreamIndex = null; |
| | | 545 | | |
| | | 546 | | // Null this out so that it will be treated like a live stream |
| | 0 | 547 | | if (mediaSource.IsInfiniteStream) |
| | | 548 | | { |
| | 0 | 549 | | mediaSource.RunTimeTicks = null; |
| | | 550 | | } |
| | | 551 | | |
| | 0 | 552 | | var audioStream = mediaSource.MediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Audio); |
| | | 553 | | |
| | 0 | 554 | | if (audioStream is null || audioStream.Index == -1) |
| | | 555 | | { |
| | 0 | 556 | | mediaSource.DefaultAudioStreamIndex = null; |
| | | 557 | | } |
| | | 558 | | else |
| | | 559 | | { |
| | 0 | 560 | | mediaSource.DefaultAudioStreamIndex = audioStream.Index; |
| | | 561 | | } |
| | | 562 | | |
| | 0 | 563 | | var videoStream = mediaSource.MediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Video); |
| | 0 | 564 | | if (videoStream is not null) |
| | | 565 | | { |
| | 0 | 566 | | if (!videoStream.BitRate.HasValue) |
| | | 567 | | { |
| | 0 | 568 | | var width = videoStream.Width ?? 1920; |
| | | 569 | | |
| | 0 | 570 | | if (width >= 3000) |
| | | 571 | | { |
| | 0 | 572 | | videoStream.BitRate = 30000000; |
| | | 573 | | } |
| | 0 | 574 | | else if (width >= 1900) |
| | | 575 | | { |
| | 0 | 576 | | videoStream.BitRate = 20000000; |
| | | 577 | | } |
| | 0 | 578 | | else if (width >= 1200) |
| | | 579 | | { |
| | 0 | 580 | | videoStream.BitRate = 8000000; |
| | | 581 | | } |
| | 0 | 582 | | else if (width >= 700) |
| | | 583 | | { |
| | 0 | 584 | | videoStream.BitRate = 2000000; |
| | | 585 | | } |
| | | 586 | | } |
| | | 587 | | } |
| | | 588 | | |
| | | 589 | | // Try to estimate this |
| | 0 | 590 | | mediaSource.InferTotalBitrate(); |
| | 0 | 591 | | } |
| | | 592 | | |
| | | 593 | | public async Task<LiveStreamResponse> OpenLiveStream(LiveStreamRequest request, CancellationToken cancellationTo |
| | | 594 | | { |
| | | 595 | | var result = await OpenLiveStreamInternal(request, cancellationToken).ConfigureAwait(false); |
| | | 596 | | return result.Item1; |
| | | 597 | | } |
| | | 598 | | |
| | | 599 | | public async Task<MediaSourceInfo> GetLiveStreamMediaInfo(string id, CancellationToken cancellationToken) |
| | | 600 | | { |
| | | 601 | | // TODO probably shouldn't throw here but it is kept for "backwards compatibility" |
| | | 602 | | var liveStreamInfo = GetLiveStreamInfo(id) ?? throw new ResourceNotFoundException(); |
| | | 603 | | |
| | | 604 | | var mediaSource = liveStreamInfo.MediaSource; |
| | | 605 | | |
| | | 606 | | if (liveStreamInfo is IDirectStreamProvider) |
| | | 607 | | { |
| | | 608 | | var info = await _mediaEncoder.GetMediaInfo( |
| | | 609 | | new MediaInfoRequest |
| | | 610 | | { |
| | | 611 | | MediaSource = mediaSource, |
| | | 612 | | ExtractChapters = false, |
| | | 613 | | MediaType = DlnaProfileType.Video |
| | | 614 | | }, |
| | | 615 | | cancellationToken).ConfigureAwait(false); |
| | | 616 | | |
| | | 617 | | mediaSource.MediaStreams = info.MediaStreams; |
| | | 618 | | mediaSource.Container = info.Container; |
| | | 619 | | mediaSource.Bitrate = info.Bitrate; |
| | | 620 | | } |
| | | 621 | | |
| | | 622 | | return mediaSource; |
| | | 623 | | } |
| | | 624 | | |
| | | 625 | | public async Task AddMediaInfoWithProbe(MediaSourceInfo mediaSource, bool isAudio, string cacheKey, bool addProb |
| | | 626 | | { |
| | | 627 | | var originalRuntime = mediaSource.RunTimeTicks; |
| | | 628 | | |
| | | 629 | | var now = DateTime.UtcNow; |
| | | 630 | | |
| | | 631 | | MediaInfo mediaInfo = null; |
| | | 632 | | var cacheFilePath = string.IsNullOrEmpty(cacheKey) ? null : Path.Combine(_appPaths.CachePath, "mediainfo", c |
| | | 633 | | |
| | | 634 | | if (!string.IsNullOrEmpty(cacheKey)) |
| | | 635 | | { |
| | | 636 | | FileStream jsonStream = AsyncFile.OpenRead(cacheFilePath); |
| | | 637 | | try |
| | | 638 | | { |
| | | 639 | | mediaInfo = await JsonSerializer.DeserializeAsync<MediaInfo>(jsonStream, _jsonOptions, cancellationT |
| | | 640 | | } |
| | | 641 | | catch (Exception ex) |
| | | 642 | | { |
| | | 643 | | _logger.LogDebug(ex, "_jsonSerializer.DeserializeFromFile threw an exception."); |
| | | 644 | | } |
| | | 645 | | finally |
| | | 646 | | { |
| | | 647 | | await jsonStream.DisposeAsync().ConfigureAwait(false); |
| | | 648 | | } |
| | | 649 | | } |
| | | 650 | | |
| | | 651 | | if (mediaInfo is null) |
| | | 652 | | { |
| | | 653 | | if (addProbeDelay) |
| | | 654 | | { |
| | | 655 | | var delayMs = mediaSource.AnalyzeDurationMs ?? 0; |
| | | 656 | | delayMs = Math.Max(3000, delayMs); |
| | | 657 | | await Task.Delay(delayMs, cancellationToken).ConfigureAwait(false); |
| | | 658 | | } |
| | | 659 | | |
| | | 660 | | if (isLiveStream) |
| | | 661 | | { |
| | | 662 | | mediaSource.AnalyzeDurationMs = 3000; |
| | | 663 | | } |
| | | 664 | | |
| | | 665 | | mediaInfo = await _mediaEncoder.GetMediaInfo( |
| | | 666 | | new MediaInfoRequest |
| | | 667 | | { |
| | | 668 | | MediaSource = mediaSource, |
| | | 669 | | MediaType = isAudio ? DlnaProfileType.Audio : DlnaProfileType.Video, |
| | | 670 | | ExtractChapters = false |
| | | 671 | | }, |
| | | 672 | | cancellationToken).ConfigureAwait(false); |
| | | 673 | | |
| | | 674 | | if (cacheFilePath is not null) |
| | | 675 | | { |
| | | 676 | | Directory.CreateDirectory(Path.GetDirectoryName(cacheFilePath)); |
| | | 677 | | FileStream createStream = File.Create(cacheFilePath); |
| | | 678 | | await using (createStream.ConfigureAwait(false)) |
| | | 679 | | { |
| | | 680 | | await JsonSerializer.SerializeAsync(createStream, mediaInfo, _jsonOptions, cancellationToken).Co |
| | | 681 | | } |
| | | 682 | | |
| | | 683 | | // _logger.LogDebug("Saved media info to {0}", cacheFilePath); |
| | | 684 | | } |
| | | 685 | | } |
| | | 686 | | |
| | | 687 | | var mediaStreams = mediaInfo.MediaStreams; |
| | | 688 | | |
| | | 689 | | if (isLiveStream && !string.IsNullOrEmpty(cacheKey)) |
| | | 690 | | { |
| | | 691 | | var newList = new List<MediaStream>(); |
| | | 692 | | newList.AddRange(mediaStreams.Where(i => i.Type == MediaStreamType.Video).Take(1)); |
| | | 693 | | newList.AddRange(mediaStreams.Where(i => i.Type == MediaStreamType.Audio).Take(1)); |
| | | 694 | | |
| | | 695 | | foreach (var stream in newList) |
| | | 696 | | { |
| | | 697 | | stream.Index = -1; |
| | | 698 | | stream.Language = null; |
| | | 699 | | } |
| | | 700 | | |
| | | 701 | | mediaStreams = newList; |
| | | 702 | | } |
| | | 703 | | |
| | | 704 | | _logger.LogInformation("Live tv media info probe took {0} seconds", (DateTime.UtcNow - now).TotalSeconds.ToS |
| | | 705 | | |
| | | 706 | | mediaSource.Bitrate = mediaInfo.Bitrate; |
| | | 707 | | mediaSource.Container = mediaInfo.Container; |
| | | 708 | | mediaSource.Formats = mediaInfo.Formats; |
| | | 709 | | mediaSource.MediaStreams = mediaStreams; |
| | | 710 | | mediaSource.RunTimeTicks = mediaInfo.RunTimeTicks; |
| | | 711 | | mediaSource.Size = mediaInfo.Size; |
| | | 712 | | mediaSource.Timestamp = mediaInfo.Timestamp; |
| | | 713 | | mediaSource.Video3DFormat = mediaInfo.Video3DFormat; |
| | | 714 | | mediaSource.VideoType = mediaInfo.VideoType; |
| | | 715 | | |
| | | 716 | | mediaSource.DefaultSubtitleStreamIndex = null; |
| | | 717 | | |
| | | 718 | | if (isLiveStream) |
| | | 719 | | { |
| | | 720 | | // Null this out so that it will be treated like a live stream |
| | | 721 | | if (!originalRuntime.HasValue) |
| | | 722 | | { |
| | | 723 | | mediaSource.RunTimeTicks = null; |
| | | 724 | | } |
| | | 725 | | } |
| | | 726 | | |
| | | 727 | | var audioStream = mediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Audio); |
| | | 728 | | |
| | | 729 | | if (audioStream is null || audioStream.Index == -1) |
| | | 730 | | { |
| | | 731 | | mediaSource.DefaultAudioStreamIndex = null; |
| | | 732 | | } |
| | | 733 | | else |
| | | 734 | | { |
| | | 735 | | mediaSource.DefaultAudioStreamIndex = audioStream.Index; |
| | | 736 | | } |
| | | 737 | | |
| | | 738 | | var videoStream = mediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Video); |
| | | 739 | | if (videoStream is not null) |
| | | 740 | | { |
| | | 741 | | if (!videoStream.BitRate.HasValue) |
| | | 742 | | { |
| | | 743 | | var width = videoStream.Width ?? 1920; |
| | | 744 | | |
| | | 745 | | if (width >= 3000) |
| | | 746 | | { |
| | | 747 | | videoStream.BitRate = 30000000; |
| | | 748 | | } |
| | | 749 | | else if (width >= 1900) |
| | | 750 | | { |
| | | 751 | | videoStream.BitRate = 20000000; |
| | | 752 | | } |
| | | 753 | | else if (width >= 1200) |
| | | 754 | | { |
| | | 755 | | videoStream.BitRate = 8000000; |
| | | 756 | | } |
| | | 757 | | else if (width >= 700) |
| | | 758 | | { |
| | | 759 | | videoStream.BitRate = 2000000; |
| | | 760 | | } |
| | | 761 | | } |
| | | 762 | | |
| | | 763 | | // This is coming up false and preventing stream copy |
| | | 764 | | videoStream.IsAVC = null; |
| | | 765 | | } |
| | | 766 | | |
| | | 767 | | if (isLiveStream) |
| | | 768 | | { |
| | | 769 | | mediaSource.AnalyzeDurationMs = 3000; |
| | | 770 | | } |
| | | 771 | | |
| | | 772 | | // Try to estimate this |
| | | 773 | | mediaSource.InferTotalBitrate(true); |
| | | 774 | | } |
| | | 775 | | |
| | | 776 | | public Task<Tuple<MediaSourceInfo, IDirectStreamProvider>> GetLiveStreamWithDirectStreamProvider(string id, Canc |
| | | 777 | | { |
| | 0 | 778 | | ArgumentException.ThrowIfNullOrEmpty(id); |
| | | 779 | | |
| | | 780 | | // TODO probably shouldn't throw here but it is kept for "backwards compatibility" |
| | 0 | 781 | | var info = GetLiveStreamInfo(id) ?? throw new ResourceNotFoundException(); |
| | 0 | 782 | | return Task.FromResult(new Tuple<MediaSourceInfo, IDirectStreamProvider>(info.MediaSource, info as IDirectSt |
| | | 783 | | } |
| | | 784 | | |
| | | 785 | | public ILiveStream GetLiveStreamInfo(string id) |
| | | 786 | | { |
| | 0 | 787 | | ArgumentException.ThrowIfNullOrEmpty(id); |
| | | 788 | | |
| | 0 | 789 | | if (_openStreams.TryGetValue(id, out ILiveStream info)) |
| | | 790 | | { |
| | 0 | 791 | | return info; |
| | | 792 | | } |
| | | 793 | | |
| | 0 | 794 | | return null; |
| | | 795 | | } |
| | | 796 | | |
| | | 797 | | /// <inheritdoc /> |
| | | 798 | | public ILiveStream GetLiveStreamInfoByUniqueId(string uniqueId) |
| | | 799 | | { |
| | 0 | 800 | | return _openStreams.Values.FirstOrDefault(stream => string.Equals(uniqueId, stream?.UniqueId, StringComparis |
| | | 801 | | } |
| | | 802 | | |
| | | 803 | | public async Task<MediaSourceInfo> GetLiveStream(string id, CancellationToken cancellationToken) |
| | | 804 | | { |
| | | 805 | | var result = await GetLiveStreamWithDirectStreamProvider(id, cancellationToken).ConfigureAwait(false); |
| | | 806 | | return result.Item1; |
| | | 807 | | } |
| | | 808 | | |
| | | 809 | | public async Task<List<MediaSourceInfo>> GetRecordingStreamMediaSources(ActiveRecordingInfo info, CancellationTo |
| | | 810 | | { |
| | | 811 | | var stream = new MediaSourceInfo |
| | | 812 | | { |
| | | 813 | | EncoderPath = _appHost.GetApiUrlForLocalAccess() + "/LiveTv/LiveRecordings/" + info.Id + "/stream", |
| | | 814 | | EncoderProtocol = MediaProtocol.Http, |
| | | 815 | | Path = info.Path, |
| | | 816 | | Protocol = MediaProtocol.File, |
| | | 817 | | Id = info.Id, |
| | | 818 | | SupportsDirectPlay = false, |
| | | 819 | | SupportsDirectStream = true, |
| | | 820 | | SupportsTranscoding = true, |
| | | 821 | | IsInfiniteStream = true, |
| | | 822 | | RequiresOpening = false, |
| | | 823 | | RequiresClosing = false, |
| | | 824 | | BufferMs = 0, |
| | | 825 | | IgnoreDts = true, |
| | | 826 | | IgnoreIndex = true |
| | | 827 | | }; |
| | | 828 | | |
| | | 829 | | await new LiveStreamHelper(_mediaEncoder, _logger, _appPaths) |
| | | 830 | | .AddMediaInfoWithProbe(stream, false, false, cancellationToken).ConfigureAwait(false); |
| | | 831 | | |
| | | 832 | | return new List<MediaSourceInfo> |
| | | 833 | | { |
| | | 834 | | stream |
| | | 835 | | }; |
| | | 836 | | } |
| | | 837 | | |
| | | 838 | | public async Task CloseLiveStream(string id) |
| | | 839 | | { |
| | | 840 | | ArgumentException.ThrowIfNullOrEmpty(id); |
| | | 841 | | |
| | | 842 | | using (await _liveStreamLocker.LockAsync().ConfigureAwait(false)) |
| | | 843 | | { |
| | | 844 | | if (_openStreams.TryGetValue(id, out ILiveStream liveStream)) |
| | | 845 | | { |
| | | 846 | | liveStream.ConsumerCount--; |
| | | 847 | | |
| | | 848 | | _logger.LogInformation("Live stream {0} consumer count is now {1}", liveStream.OriginalStreamId, liv |
| | | 849 | | |
| | | 850 | | if (liveStream.ConsumerCount <= 0) |
| | | 851 | | { |
| | | 852 | | _openStreams.TryRemove(id, out _); |
| | | 853 | | |
| | | 854 | | _logger.LogInformation("Closing live stream {0}", id); |
| | | 855 | | |
| | | 856 | | await liveStream.Close().ConfigureAwait(false); |
| | | 857 | | _logger.LogInformation("Live stream {0} closed successfully", id); |
| | | 858 | | } |
| | | 859 | | } |
| | | 860 | | } |
| | | 861 | | } |
| | | 862 | | |
| | | 863 | | private (IMediaSourceProvider MediaSourceProvider, string KeyId) GetProvider(string key) |
| | | 864 | | { |
| | 0 | 865 | | ArgumentException.ThrowIfNullOrEmpty(key); |
| | | 866 | | |
| | 0 | 867 | | var keys = key.Split(LiveStreamIdDelimeter, 2); |
| | | 868 | | |
| | 0 | 869 | | var provider = _providers.FirstOrDefault(i => string.Equals(i.GetType().FullName.GetMD5().ToString("N", Cult |
| | | 870 | | |
| | 0 | 871 | | var splitIndex = key.IndexOf(LiveStreamIdDelimeter, StringComparison.Ordinal); |
| | 0 | 872 | | var keyId = key.Substring(splitIndex + 1); |
| | | 873 | | |
| | 0 | 874 | | return (provider, keyId); |
| | | 875 | | } |
| | | 876 | | |
| | | 877 | | /// <inheritdoc /> |
| | | 878 | | public void Dispose() |
| | | 879 | | { |
| | 22 | 880 | | Dispose(true); |
| | 22 | 881 | | GC.SuppressFinalize(this); |
| | 22 | 882 | | } |
| | | 883 | | |
| | | 884 | | /// <summary> |
| | | 885 | | /// Releases unmanaged and - optionally - managed resources. |
| | | 886 | | /// </summary> |
| | | 887 | | /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release o |
| | | 888 | | protected virtual void Dispose(bool dispose) |
| | | 889 | | { |
| | 22 | 890 | | if (dispose) |
| | | 891 | | { |
| | 44 | 892 | | foreach (var key in _openStreams.Keys.ToList()) |
| | | 893 | | { |
| | 0 | 894 | | CloseLiveStream(key).GetAwaiter().GetResult(); |
| | | 895 | | } |
| | | 896 | | |
| | 22 | 897 | | _liveStreamLocker.Dispose(); |
| | | 898 | | } |
| | 22 | 899 | | } |
| | | 900 | | } |
| | | 901 | | } |