jellyfin/Emby.Server.Implementations/LiveTv/TunerHosts/HdHomerun/HdHomerunHttpStream.cs

144 lines
5.5 KiB
C#
Raw Normal View History

2016-09-25 18:39:13 +00:00
using System;
using System.IO;
using System.Threading;
using System.Threading.Tasks;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
2016-09-25 18:39:13 +00:00
using MediaBrowser.Common.Net;
using MediaBrowser.Controller;
using MediaBrowser.Controller.LiveTv;
2016-10-05 07:15:29 +00:00
using MediaBrowser.Controller.Library;
2016-09-25 18:39:13 +00:00
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Logging;
using MediaBrowser.Model.MediaInfo;
2016-11-03 23:35:19 +00:00
namespace Emby.Server.Implementations.LiveTv.TunerHosts.HdHomerun
2016-09-25 18:39:13 +00:00
{
2017-03-03 05:53:21 +00:00
public class HdHomerunHttpStream : LiveStream, IDirectStreamProvider
2016-09-25 18:39:13 +00:00
{
private readonly ILogger _logger;
private readonly IHttpClient _httpClient;
private readonly IFileSystem _fileSystem;
private readonly IServerApplicationPaths _appPaths;
private readonly IServerApplicationHost _appHost;
private readonly CancellationTokenSource _liveStreamCancellationTokenSource = new CancellationTokenSource();
2016-09-27 17:51:01 +00:00
private readonly TaskCompletionSource<bool> _liveStreamTaskCompletionSource = new TaskCompletionSource<bool>();
2016-10-07 15:08:13 +00:00
private readonly MulticastStream _multicastStream;
2016-09-25 18:39:13 +00:00
2017-03-03 05:53:21 +00:00
public HdHomerunHttpStream(MediaSourceInfo mediaSource, string originalStreamId, IFileSystem fileSystem, IHttpClient httpClient, ILogger logger, IServerApplicationPaths appPaths, IServerApplicationHost appHost)
2016-09-25 18:39:13 +00:00
: base(mediaSource)
{
_fileSystem = fileSystem;
_httpClient = httpClient;
_logger = logger;
_appPaths = appPaths;
_appHost = appHost;
2016-10-07 15:08:13 +00:00
OriginalStreamId = originalStreamId;
_multicastStream = new MulticastStream(_logger);
2016-09-25 18:39:13 +00:00
}
2016-09-29 12:55:49 +00:00
protected override async Task OpenInternal(CancellationToken openCancellationToken)
2016-09-25 18:39:13 +00:00
{
_liveStreamCancellationTokenSource.Token.ThrowIfCancellationRequested();
var mediaSource = OriginalMediaSource;
var url = mediaSource.Path;
2016-10-07 15:08:13 +00:00
_logger.Info("Opening HDHR Live stream from {0}", url);
2016-09-25 18:39:13 +00:00
var taskCompletionSource = new TaskCompletionSource<bool>();
2016-10-07 15:08:13 +00:00
StartStreaming(url, taskCompletionSource, _liveStreamCancellationTokenSource.Token);
2016-09-25 18:39:13 +00:00
2016-09-30 06:50:06 +00:00
//OpenedMediaSource.Protocol = MediaProtocol.File;
//OpenedMediaSource.Path = tempFile;
//OpenedMediaSource.ReadAtNativeFramerate = true;
2016-09-25 18:39:13 +00:00
2016-10-19 20:31:46 +00:00
OpenedMediaSource.Path = _appHost.GetLocalApiUrl("127.0.0.1") + "/LiveTv/LiveStreamFiles/" + UniqueId + "/stream.ts";
2016-09-29 12:55:49 +00:00
OpenedMediaSource.Protocol = MediaProtocol.Http;
OpenedMediaSource.SupportsDirectPlay = false;
OpenedMediaSource.SupportsDirectStream = true;
OpenedMediaSource.SupportsTranscoding = true;
2016-09-30 06:50:06 +00:00
await taskCompletionSource.Task.ConfigureAwait(false);
//await Task.Delay(5000).ConfigureAwait(false);
2016-09-25 18:39:13 +00:00
}
public override Task Close()
{
2016-09-29 12:55:49 +00:00
_logger.Info("Closing HDHR live stream");
2016-09-25 18:39:13 +00:00
_liveStreamCancellationTokenSource.Cancel();
2016-09-27 17:51:01 +00:00
return _liveStreamTaskCompletionSource.Task;
2016-09-25 18:39:13 +00:00
}
2016-10-07 15:08:13 +00:00
private async Task StartStreaming(string url, TaskCompletionSource<bool> openTaskCompletionSource, CancellationToken cancellationToken)
2016-09-25 18:39:13 +00:00
{
await Task.Run(async () =>
{
2016-10-07 15:08:13 +00:00
var isFirstAttempt = true;
2016-09-25 18:39:13 +00:00
2016-10-07 15:08:13 +00:00
while (!cancellationToken.IsCancellationRequested)
{
try
2016-09-25 18:39:13 +00:00
{
2016-10-07 15:08:13 +00:00
using (var response = await _httpClient.SendAsync(new HttpRequestOptions
2016-09-25 18:39:13 +00:00
{
2016-10-07 15:08:13 +00:00
Url = url,
CancellationToken = cancellationToken,
2016-11-20 23:48:52 +00:00
BufferContent = false,
// Increase a little bit
TimeoutMs = 30000
2016-10-07 15:08:13 +00:00
}, "GET").ConfigureAwait(false))
{
_logger.Info("Opened HDHR stream from {0}", url);
2016-09-25 18:39:13 +00:00
2016-10-07 15:08:13 +00:00
if (!cancellationToken.IsCancellationRequested)
2016-09-25 18:39:13 +00:00
{
2016-10-07 15:08:13 +00:00
_logger.Info("Beginning multicastStream.CopyUntilCancelled");
2016-09-25 18:39:13 +00:00
2016-10-07 15:08:13 +00:00
Action onStarted = null;
if (isFirstAttempt)
2016-09-25 18:39:13 +00:00
{
2016-10-07 15:08:13 +00:00
onStarted = () => openTaskCompletionSource.TrySetResult(true);
2016-09-25 18:39:13 +00:00
}
2016-10-07 15:08:13 +00:00
await _multicastStream.CopyUntilCancelled(response.Content, onStarted, cancellationToken).ConfigureAwait(false);
}
2016-09-25 18:39:13 +00:00
}
}
2016-10-07 15:08:13 +00:00
catch (OperationCanceledException)
2016-10-05 07:15:29 +00:00
{
2016-10-07 15:08:13 +00:00
break;
2016-10-05 07:15:29 +00:00
}
catch (Exception ex)
{
2016-10-07 15:08:13 +00:00
if (isFirstAttempt)
{
_logger.ErrorException("Error opening live stream:", ex);
openTaskCompletionSource.TrySetException(ex);
break;
}
2016-10-06 18:55:01 +00:00
2016-10-07 15:08:13 +00:00
_logger.ErrorException("Error copying live stream, will reopen", ex);
2016-10-05 07:15:29 +00:00
}
2016-10-07 15:08:13 +00:00
isFirstAttempt = false;
2016-10-05 07:15:29 +00:00
}
2016-10-07 15:08:13 +00:00
_liveStreamTaskCompletionSource.TrySetResult(true);
2016-10-03 06:28:45 +00:00
2016-10-07 15:08:13 +00:00
}).ConfigureAwait(false);
2016-10-03 06:28:45 +00:00
}
2016-10-07 15:08:13 +00:00
public Task CopyToAsync(Stream stream, CancellationToken cancellationToken)
2016-09-25 18:39:13 +00:00
{
2016-10-07 15:08:13 +00:00
return _multicastStream.CopyToAsync(stream);
2016-09-25 18:39:13 +00:00
}
}
}