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

179 lines
6.7 KiB
C#
Raw Normal View History

2015-07-20 18:32:55 +00:00
using MediaBrowser.Common.Configuration;
using MediaBrowser.Common.Extensions;
using MediaBrowser.Controller.LiveTv;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.LiveTv;
2015-08-19 16:43:23 +00:00
using MediaBrowser.Model.Logging;
2015-07-20 18:32:55 +00:00
using MediaBrowser.Model.MediaInfo;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Model.IO;
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;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Controller.IO;
2015-10-14 02:41:46 +00:00
using MediaBrowser.Controller.MediaEncoding;
2015-09-23 01:22:52 +00:00
using MediaBrowser.Model.Serialization;
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-09-13 23:07:54 +00:00
private readonly IFileSystem _fileSystem;
2015-10-14 02:41:46 +00:00
private readonly IHttpClient _httpClient;
2016-10-18 18:35:27 +00:00
private readonly IServerApplicationHost _appHost;
2015-09-23 01:22:52 +00:00
2016-10-18 18:35:27 +00:00
public M3UTunerHost(IServerConfigurationManager config, ILogger logger, IJsonSerializer jsonSerializer, IMediaEncoder mediaEncoder, IFileSystem fileSystem, IHttpClient httpClient, IServerApplicationHost appHost)
2015-10-30 16:40:12 +00:00
: base(config, logger, jsonSerializer, mediaEncoder)
2015-07-20 18:32:55 +00:00
{
2015-09-13 23:07:54 +00:00
_fileSystem = fileSystem;
2015-09-23 01:22:52 +00:00
_httpClient = httpClient;
2016-10-18 18:35:27 +00:00
_appHost = appHost;
2015-07-20 18:32:55 +00:00
}
2015-08-19 17:58:41 +00:00
public override string Type
2015-07-20 18:32:55 +00:00
{
2015-08-19 17:58:41 +00:00
get { return "m3u"; }
2015-08-19 16:43:23 +00:00
}
2015-08-19 17:58:41 +00:00
public string Name
2015-08-19 16:43:23 +00:00
{
2015-08-19 17:58:41 +00:00
get { return "M3U Tuner"; }
2015-08-19 16:43:23 +00:00
}
2015-08-19 19:25:18 +00:00
private const string ChannelIdPrefix = "m3u_";
2015-08-19 17:58:41 +00:00
protected override async Task<IEnumerable<ChannelInfo>> GetChannelsInternal(TunerHostInfo info, CancellationToken cancellationToken)
2015-08-19 16:43:23 +00:00
{
2017-01-25 20:14:47 +00:00
return await new M3uParser(Logger, _fileSystem, _httpClient, _appHost).Parse(info.Url, ChannelIdPrefix, info.Id, !info.EnableTvgId, cancellationToken).ConfigureAwait(false);
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);
}
2016-09-25 18:39:13 +00:00
protected override async Task<LiveStream> GetChannelStream(TunerHostInfo info, string channelId, string streamId, CancellationToken cancellationToken)
2015-08-27 19:59:42 +00:00
{
var sources = await GetChannelStreamMediaSources(info, channelId, cancellationToken).ConfigureAwait(false);
2016-09-25 18:39:13 +00:00
var liveStream = new LiveStream(sources.First());
return liveStream;
2015-08-27 19:59:42 +00:00
}
public async Task Validate(TunerHostInfo info)
{
2016-10-18 18:35:27 +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
}
}
protected override bool IsValidChannelId(string channelId)
{
2016-11-17 03:58:27 +00:00
if (string.IsNullOrWhiteSpace(channelId))
{
throw new ArgumentNullException("channelId");
}
2015-08-27 19:59:42 +00:00
return channelId.StartsWith(ChannelIdPrefix, StringComparison.OrdinalIgnoreCase);
}
protected override async Task<List<MediaSourceInfo>> GetChannelStreamMediaSources(TunerHostInfo info, string channelId, CancellationToken cancellationToken)
2015-07-20 18:32:55 +00:00
{
2015-08-16 18:37:53 +00:00
var urlHash = info.Url.GetMD5().ToString("N");
2015-08-19 19:25:18 +00:00
var prefix = ChannelIdPrefix + urlHash;
if (!channelId.StartsWith(prefix, StringComparison.OrdinalIgnoreCase))
2015-08-16 18:37:53 +00:00
{
return null;
}
2015-08-19 17:58:41 +00:00
var channels = await GetChannels(info, true, cancellationToken).ConfigureAwait(false);
2015-07-20 18:32:55 +00:00
var m3uchannels = channels.Cast<M3UChannel>();
2015-08-27 19:59:42 +00:00
var channel = m3uchannels.FirstOrDefault(c => string.Equals(c.Id, channelId, StringComparison.OrdinalIgnoreCase));
2015-07-20 18:32:55 +00:00
if (channel != null)
{
var path = channel.Path;
MediaProtocol protocol = MediaProtocol.File;
2015-08-27 19:59:42 +00:00
if (path.StartsWith("http", StringComparison.OrdinalIgnoreCase))
2015-07-20 18:32:55 +00:00
{
protocol = MediaProtocol.Http;
}
2015-08-27 19:59:42 +00:00
else if (path.StartsWith("rtmp", StringComparison.OrdinalIgnoreCase))
2015-07-20 18:32:55 +00:00
{
protocol = MediaProtocol.Rtmp;
}
2015-08-27 19:59:42 +00:00
else if (path.StartsWith("rtsp", StringComparison.OrdinalIgnoreCase))
2015-07-20 18:32:55 +00:00
{
protocol = MediaProtocol.Rtsp;
}
2016-10-16 17:11:32 +00:00
else if (path.StartsWith("udp", StringComparison.OrdinalIgnoreCase))
{
protocol = MediaProtocol.Udp;
}
2017-01-18 06:05:33 +00:00
else if (path.StartsWith("rtp", StringComparison.OrdinalIgnoreCase))
{
protocol = MediaProtocol.Rtmp;
}
2015-07-20 18:32:55 +00:00
2015-08-27 19:59:42 +00:00
var mediaSource = new MediaSourceInfo
2015-07-20 18:32:55 +00:00
{
Path = channel.Path,
Protocol = protocol,
MediaStreams = new List<MediaStream>
{
new MediaStream
{
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
},
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
}
2015-07-23 23:40:54 +00:00
},
2016-12-10 01:58:52 +00:00
RequiresOpening = true,
RequiresClosing = true,
2016-05-03 04:17:57 +00:00
2016-09-25 18:39:13 +00:00
ReadAtNativeFramerate = false,
2016-09-30 02:21:59 +00:00
Id = channel.Path.GetMD5().ToString("N"),
2017-01-07 08:04:18 +00:00
IsInfiniteStream = true,
2017-01-20 17:53:48 +00:00
SupportsDirectStream = false,
IsRemote = true
2015-07-20 18:32:55 +00:00
};
2015-07-23 13:23:22 +00:00
2017-01-21 20:27:07 +00:00
mediaSource.InferTotalBitrate();
2015-08-27 19:59:42 +00:00
return new List<MediaSourceInfo> { mediaSource };
2015-07-23 13:23:22 +00:00
}
return new List<MediaSourceInfo>();
2015-07-23 23:40:54 +00:00
}
2015-10-30 16:40:12 +00:00
protected override Task<bool> IsAvailableInternal(TunerHostInfo tuner, string channelId, CancellationToken cancellationToken)
{
return Task.FromResult(true);
}
2015-07-20 18:32:55 +00:00
}
2017-01-25 20:14:47 +00:00
}