jellyfin/Emby.Server.Implementations/LiveTv/TunerHosts/M3UTunerHost.cs

202 lines
7.8 KiB
C#
Raw Normal View History

using System;
2015-07-20 18:32:55 +00:00
using System.Collections.Generic;
using System.IO;
2015-07-20 18:32:55 +00:00
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MediaBrowser.Common.Extensions;
2015-09-23 01:22:52 +00:00
using MediaBrowser.Common.Net;
2016-10-18 18:35:27 +00:00
using MediaBrowser.Controller;
2016-09-25 18:39:13 +00:00
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.LiveTv;
2015-10-14 02:41:46 +00:00
using MediaBrowser.Controller.MediaEncoding;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.IO;
using MediaBrowser.Model.LiveTv;
using MediaBrowser.Model.MediaInfo;
2015-09-23 01:22:52 +00:00
using MediaBrowser.Model.Serialization;
2017-05-22 04:54:02 +00:00
using MediaBrowser.Model.System;
using Microsoft.Extensions.Logging;
2015-07-20 18:32:55 +00:00
2016-11-03 23:35:19 +00:00
namespace Emby.Server.Implementations.LiveTv.TunerHosts
2015-07-20 18:32:55 +00:00
{
2016-02-19 06:20:18 +00:00
public class M3UTunerHost : BaseTunerHost, ITunerHost, IConfigurableTunerHost
2015-07-20 18:32:55 +00:00
{
2015-10-14 02:41:46 +00:00
private readonly IHttpClient _httpClient;
2016-10-18 18:35:27 +00:00
private readonly IServerApplicationHost _appHost;
2017-05-22 04:54:02 +00:00
private readonly IEnvironmentInfo _environment;
2017-08-20 19:10:00 +00:00
private readonly INetworkManager _networkManager;
2018-09-12 17:26:21 +00:00
private readonly IMediaSourceManager _mediaSourceManager;
2015-09-23 01:22:52 +00:00
2018-09-12 17:26:21 +00:00
public M3UTunerHost(IServerConfigurationManager config, IMediaSourceManager mediaSourceManager, ILogger logger, IJsonSerializer jsonSerializer, IMediaEncoder mediaEncoder, IFileSystem fileSystem, IHttpClient httpClient, IServerApplicationHost appHost, IEnvironmentInfo environment, INetworkManager networkManager) : base(config, logger, jsonSerializer, mediaEncoder, fileSystem)
2015-07-20 18:32:55 +00:00
{
2015-09-23 01:22:52 +00:00
_httpClient = httpClient;
2016-10-18 18:35:27 +00:00
_appHost = appHost;
2017-05-22 04:54:02 +00:00
_environment = environment;
2017-08-20 19:10:00 +00:00
_networkManager = networkManager;
2018-09-12 17:26:21 +00:00
_mediaSourceManager = mediaSourceManager;
2015-07-20 18:32:55 +00:00
}
public override string Type => "m3u";
2015-08-19 16:43:23 +00:00
public virtual string Name => "M3U Tuner";
2015-08-19 16:43:23 +00:00
2017-07-30 18:02:25 +00:00
private string GetFullChannelIdPrefix(TunerHostInfo info)
{
return ChannelIdPrefix + info.Url.GetMD5().ToString("N");
}
2017-02-23 19:13:26 +00:00
protected override async Task<List<ChannelInfo>> GetChannelsInternal(TunerHostInfo info, CancellationToken cancellationToken)
2015-08-19 16:43:23 +00:00
{
2017-07-30 18:02:25 +00:00
var channelIdPrefix = GetFullChannelIdPrefix(info);
var result = await new M3uParser(Logger, FileSystem, _httpClient, _appHost).Parse(info.Url, channelIdPrefix, info.Id, cancellationToken).ConfigureAwait(false);
2017-02-23 19:13:26 +00:00
return result.Cast<ChannelInfo>().ToList();
2015-07-20 18:32:55 +00:00
}
2015-08-19 16:43:23 +00:00
public Task<List<LiveTvTunerInfo>> GetTunerInfos(CancellationToken cancellationToken)
2015-07-20 18:32:55 +00:00
{
2016-02-19 06:20:18 +00:00
var list = GetTunerHosts()
2015-08-19 16:43:23 +00:00
.Select(i => new LiveTvTunerInfo()
2015-07-20 18:32:55 +00:00
{
Name = Name,
SourceType = Type,
Status = LiveTvTunerStatus.Available,
2015-08-19 16:43:23 +00:00
Id = i.Url.GetMD5().ToString("N"),
Url = i.Url
})
.ToList();
2015-08-19 19:25:18 +00:00
2015-07-20 18:32:55 +00:00
return Task.FromResult(list);
}
2018-09-12 17:26:21 +00:00
private string[] _disallowedSharedStreamExtensions = new string[]
2017-11-29 20:50:18 +00:00
{
".mkv",
".mp4",
".m3u8",
".mpd"
};
2018-09-12 17:26:21 +00:00
protected override async Task<ILiveStream> GetChannelStream(TunerHostInfo info, ChannelInfo channelInfo, string streamId, List<ILiveStream> currentLiveStreams, CancellationToken cancellationToken)
2015-08-27 19:59:42 +00:00
{
var tunerCount = info.TunerCount;
if (tunerCount > 0)
{
2018-09-12 17:26:21 +00:00
var tunerHostId = info.Id;
var liveStreams = currentLiveStreams.Where(i => string.Equals(i.TunerHostId, tunerHostId, StringComparison.OrdinalIgnoreCase)).ToList();
2018-09-12 17:26:21 +00:00
if (liveStreams.Count >= tunerCount)
{
2018-09-12 17:26:21 +00:00
throw new LiveTvConflictException("M3U simultaneous stream limit has been reached.");
}
}
2018-09-12 17:26:21 +00:00
var sources = await GetChannelStreamMediaSources(info, channelInfo, cancellationToken).ConfigureAwait(false);
2015-08-27 19:59:42 +00:00
2017-11-14 07:41:21 +00:00
var mediaSource = sources.First();
if (mediaSource.Protocol == MediaProtocol.Http && !mediaSource.RequiresLooping)
{
2017-11-29 20:50:18 +00:00
var extension = Path.GetExtension(mediaSource.Path) ?? string.Empty;
if (!_disallowedSharedStreamExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase))
{
2018-09-12 17:26:21 +00:00
return new SharedHttpStream(mediaSource, info, streamId, FileSystem, _httpClient, Logger, Config.ApplicationPaths, _appHost);
2017-11-29 20:50:18 +00:00
}
2017-11-14 07:41:21 +00:00
}
2018-09-12 17:26:21 +00:00
return new LiveStream(mediaSource, info, FileSystem, Logger, Config.ApplicationPaths);
2015-08-27 19:59:42 +00:00
}
public async Task Validate(TunerHostInfo info)
{
2017-07-05 18:30:12 +00:00
using (var stream = await new M3uParser(Logger, FileSystem, _httpClient, _appHost).GetListingsStream(info.Url, CancellationToken.None).ConfigureAwait(false))
2015-08-27 19:59:42 +00:00
{
2015-09-23 01:22:52 +00:00
2015-08-27 19:59:42 +00:00
}
}
2018-09-12 17:26:21 +00:00
protected override Task<List<MediaSourceInfo>> GetChannelStreamMediaSources(TunerHostInfo info, ChannelInfo channelInfo, CancellationToken cancellationToken)
2015-07-20 18:32:55 +00:00
{
2018-09-12 17:26:21 +00:00
return Task.FromResult(new List<MediaSourceInfo> { CreateMediaSourceInfo(info, channelInfo) });
2017-08-20 19:10:00 +00:00
}
protected virtual MediaSourceInfo CreateMediaSourceInfo(TunerHostInfo info, ChannelInfo channel)
{
var path = channel.Path;
2018-09-12 17:26:21 +00:00
var supportsDirectPlay = !info.EnableStreamLooping && info.TunerCount == 0;
var supportsDirectStream = !info.EnableStreamLooping;
var protocol = _mediaSourceManager.GetPathProtocol(path);
2017-08-20 19:10:00 +00:00
var isRemote = true;
if (Uri.TryCreate(path, UriKind.Absolute, out var uri))
2017-08-20 19:10:00 +00:00
{
isRemote = !_networkManager.IsInLocalNetwork(uri.Host);
}
2015-07-20 18:32:55 +00:00
2018-09-12 17:26:21 +00:00
var httpHeaders = new Dictionary<string, string>();
if (protocol == MediaProtocol.Http)
{
// Use user-defined user-agent. If there isn't one, make it look like a browser.
httpHeaders["User-Agent"] = string.IsNullOrWhiteSpace(info.UserAgent) ?
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/64.0.3282.85 Safari/537.36" :
info.UserAgent;
}
2017-11-21 22:14:56 +00:00
2017-08-20 19:10:00 +00:00
var mediaSource = new MediaSourceInfo
{
Path = path,
Protocol = protocol,
MediaStreams = new List<MediaStream>
2015-07-20 18:32:55 +00:00
{
2017-08-20 19:10:00 +00:00
new MediaStream
2015-07-20 18:32:55 +00:00
{
2017-08-20 19:10:00 +00:00
Type = MediaStreamType.Video,
// Set the index to -1 because we don't know the exact index of the video stream within the container
Index = -1,
IsInterlaced = true
2015-07-23 23:40:54 +00:00
},
2017-08-20 19:10:00 +00:00
new MediaStream
{
Type = MediaStreamType.Audio,
// Set the index to -1 because we don't know the exact index of the audio stream within the container
Index = -1
}
},
RequiresOpening = true,
RequiresClosing = true,
RequiresLooping = info.EnableStreamLooping,
2016-09-25 18:39:13 +00:00
2017-08-20 19:10:00 +00:00
ReadAtNativeFramerate = false,
2017-07-15 20:19:58 +00:00
2017-08-20 19:10:00 +00:00
Id = channel.Path.GetMD5().ToString("N"),
IsInfiniteStream = true,
IsRemote = isRemote,
2015-07-23 13:23:22 +00:00
2017-11-21 22:14:56 +00:00
IgnoreDts = true,
2018-09-12 17:26:21 +00:00
SupportsDirectPlay = supportsDirectPlay,
SupportsDirectStream = supportsDirectStream,
RequiredHttpHeaders = httpHeaders
2017-08-20 19:10:00 +00:00
};
2017-01-21 20:27:07 +00:00
2017-08-20 19:10:00 +00:00
mediaSource.InferTotalBitrate();
return mediaSource;
2015-07-23 23:40:54 +00:00
}
2015-10-30 16:40:12 +00:00
2017-03-13 04:49:10 +00:00
public Task<List<TunerHostInfo>> DiscoverDevices(int discoveryDurationMs, CancellationToken cancellationToken)
2017-03-13 04:08:49 +00:00
{
return Task.FromResult(new List<TunerHostInfo>());
}
2015-07-20 18:32:55 +00:00
}
}