mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +02:00
Add Argument*Exceptions now use proper nameof operators. Added exception messages to quite a few Argument*Exceptions. Fixed rethorwing to be proper syntax. Added a ton of null checkes. (This is only a start, there are about 500 places that need proper null handling) Added some TODOs to log certain exceptions. Fix sln again. Fixed all AssemblyInfo's and added proper copyright (where I could find them) We live in *current year*. Fixed the use of braces. Fixed a ton of properties, and made a fair amount of functions static that should be and can be static. Made more Methods that should be static static. You can now use static to find bad functions! Removed unused variable. And added one more proper XML comment.
388 lines
14 KiB
C#
388 lines
14 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Common.Net;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.MediaEncoding;
|
|
using MediaBrowser.Model.Configuration;
|
|
using MediaBrowser.Model.Diagnostics;
|
|
using MediaBrowser.Model.Dto;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.LiveTv;
|
|
using Microsoft.Extensions.Logging;
|
|
using MediaBrowser.Model.Serialization;
|
|
using MediaBrowser.Common.Configuration;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Model.Reflection;
|
|
|
|
namespace Emby.Server.Implementations.LiveTv.EmbyTV
|
|
{
|
|
public class EncodedRecorder : IRecorder
|
|
{
|
|
private readonly ILogger _logger;
|
|
private readonly IFileSystem _fileSystem;
|
|
private readonly IHttpClient _httpClient;
|
|
private readonly IMediaEncoder _mediaEncoder;
|
|
private readonly IServerApplicationPaths _appPaths;
|
|
private bool _hasExited;
|
|
private Stream _logFileStream;
|
|
private string _targetPath;
|
|
private IProcess _process;
|
|
private readonly IProcessFactory _processFactory;
|
|
private readonly IJsonSerializer _json;
|
|
private readonly TaskCompletionSource<bool> _taskCompletionSource = new TaskCompletionSource<bool>();
|
|
private readonly IServerConfigurationManager _config;
|
|
private readonly IAssemblyInfo _assemblyInfo;
|
|
|
|
public EncodedRecorder(ILogger logger, IFileSystem fileSystem, IMediaEncoder mediaEncoder, IServerApplicationPaths appPaths, IJsonSerializer json, IHttpClient httpClient, IProcessFactory processFactory, IServerConfigurationManager config, IAssemblyInfo assemblyInfo)
|
|
{
|
|
_logger = logger;
|
|
_fileSystem = fileSystem;
|
|
_mediaEncoder = mediaEncoder;
|
|
_appPaths = appPaths;
|
|
_json = json;
|
|
_httpClient = httpClient;
|
|
_processFactory = processFactory;
|
|
_config = config;
|
|
_assemblyInfo = assemblyInfo;
|
|
}
|
|
|
|
private static bool CopySubtitles => false;
|
|
|
|
public string GetOutputPath(MediaSourceInfo mediaSource, string targetFile)
|
|
{
|
|
return Path.ChangeExtension(targetFile, ".ts");
|
|
}
|
|
|
|
public async Task Record(IDirectStreamProvider directStreamProvider, MediaSourceInfo mediaSource, string targetFile, TimeSpan duration, Action onStarted, CancellationToken cancellationToken)
|
|
{
|
|
// The media source is infinite so we need to handle stopping ourselves
|
|
var durationToken = new CancellationTokenSource(duration);
|
|
cancellationToken = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, durationToken.Token).Token;
|
|
|
|
await RecordFromFile(mediaSource, mediaSource.Path, targetFile, duration, onStarted, cancellationToken).ConfigureAwait(false);
|
|
|
|
_logger.LogInformation("Recording completed to file {0}", targetFile);
|
|
}
|
|
|
|
private EncodingOptions GetEncodingOptions()
|
|
{
|
|
return _config.GetConfiguration<EncodingOptions>("encoding");
|
|
}
|
|
|
|
private Task RecordFromFile(MediaSourceInfo mediaSource, string inputFile, string targetFile, TimeSpan duration, Action onStarted, CancellationToken cancellationToken)
|
|
{
|
|
_targetPath = targetFile;
|
|
_fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(targetFile));
|
|
|
|
var process = _processFactory.Create(new ProcessOptions
|
|
{
|
|
CreateNoWindow = true,
|
|
UseShellExecute = false,
|
|
|
|
// Must consume both stdout and stderr or deadlocks may occur
|
|
//RedirectStandardOutput = true,
|
|
RedirectStandardError = true,
|
|
RedirectStandardInput = true,
|
|
|
|
FileName = _mediaEncoder.EncoderPath,
|
|
Arguments = GetCommandLineArgs(mediaSource, inputFile, targetFile, duration),
|
|
|
|
IsHidden = true,
|
|
ErrorDialog = false,
|
|
EnableRaisingEvents = true
|
|
});
|
|
|
|
_process = process;
|
|
|
|
var commandLineLogMessage = process.StartInfo.FileName + " " + process.StartInfo.Arguments;
|
|
_logger.LogInformation(commandLineLogMessage);
|
|
|
|
var logFilePath = Path.Combine(_appPaths.LogDirectoryPath, "record-transcode-" + Guid.NewGuid() + ".txt");
|
|
_fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(logFilePath));
|
|
|
|
// FFMpeg writes debug/error info to stderr. This is useful when debugging so let's put it in the log directory.
|
|
_logFileStream = _fileSystem.GetFileStream(logFilePath, FileOpenMode.Create, FileAccessMode.Write, FileShareMode.Read, true);
|
|
|
|
var commandLineLogMessageBytes = Encoding.UTF8.GetBytes(_json.SerializeToString(mediaSource) + Environment.NewLine + Environment.NewLine + commandLineLogMessage + Environment.NewLine + Environment.NewLine);
|
|
_logFileStream.Write(commandLineLogMessageBytes, 0, commandLineLogMessageBytes.Length);
|
|
|
|
process.Exited += (sender, args) => OnFfMpegProcessExited(process, inputFile);
|
|
|
|
process.Start();
|
|
|
|
cancellationToken.Register(Stop);
|
|
|
|
// MUST read both stdout and stderr asynchronously or a deadlock may occurr
|
|
//process.BeginOutputReadLine();
|
|
|
|
onStarted();
|
|
|
|
// Important - don't await the log task or we won't be able to kill ffmpeg when the user stops playback
|
|
StartStreamingLog(process.StandardError.BaseStream, _logFileStream);
|
|
|
|
_logger.LogInformation("ffmpeg recording process started for {0}", _targetPath);
|
|
|
|
return _taskCompletionSource.Task;
|
|
}
|
|
|
|
private string GetCommandLineArgs(MediaSourceInfo mediaSource, string inputTempFile, string targetFile, TimeSpan duration)
|
|
{
|
|
string videoArgs;
|
|
if (EncodeVideo(mediaSource))
|
|
{
|
|
var maxBitrate = 25000000;
|
|
videoArgs = string.Format(
|
|
"-codec:v:0 libx264 -force_key_frames \"expr:gte(t,n_forced*5)\" {0} -pix_fmt yuv420p -preset superfast -crf 23 -b:v {1} -maxrate {1} -bufsize ({1}*2) -vsync -1 -profile:v high -level 41",
|
|
GetOutputSizeParam(),
|
|
maxBitrate.ToString(CultureInfo.InvariantCulture));
|
|
}
|
|
else
|
|
{
|
|
videoArgs = "-codec:v:0 copy";
|
|
}
|
|
|
|
videoArgs += " -fflags +genpts";
|
|
|
|
var durationParam = " -t " + _mediaEncoder.GetTimeParameter(duration.Ticks);
|
|
durationParam = string.Empty;
|
|
|
|
var flags = new List<string>();
|
|
if (mediaSource.IgnoreDts)
|
|
{
|
|
flags.Add("+igndts");
|
|
}
|
|
if (mediaSource.IgnoreIndex)
|
|
{
|
|
flags.Add("+ignidx");
|
|
}
|
|
if (mediaSource.GenPtsInput)
|
|
{
|
|
flags.Add("+genpts");
|
|
}
|
|
|
|
var inputModifier = "-async 1 -vsync -1";
|
|
|
|
if (flags.Count > 0)
|
|
{
|
|
inputModifier += " -fflags " + string.Join("", flags.ToArray());
|
|
}
|
|
|
|
var videoStream = mediaSource.VideoStream;
|
|
string videoDecoder = null;
|
|
|
|
if (!string.IsNullOrEmpty(videoDecoder))
|
|
{
|
|
inputModifier += " " + videoDecoder;
|
|
}
|
|
|
|
if (mediaSource.ReadAtNativeFramerate)
|
|
{
|
|
inputModifier += " -re";
|
|
}
|
|
|
|
if (mediaSource.RequiresLooping)
|
|
{
|
|
inputModifier += " -stream_loop -1 -reconnect_at_eof 1 -reconnect_streamed 1 -reconnect_delay_max 2";
|
|
}
|
|
|
|
var analyzeDurationSeconds = 5;
|
|
var analyzeDuration = " -analyzeduration " +
|
|
(analyzeDurationSeconds * 1000000).ToString(CultureInfo.InvariantCulture);
|
|
inputModifier += analyzeDuration;
|
|
|
|
var subtitleArgs = CopySubtitles ? " -codec:s copy" : " -sn";
|
|
|
|
//var outputParam = string.Equals(Path.GetExtension(targetFile), ".mp4", StringComparison.OrdinalIgnoreCase) ?
|
|
// " -f mp4 -movflags frag_keyframe+empty_moov" :
|
|
// string.Empty;
|
|
|
|
var outputParam = string.Empty;
|
|
|
|
var commandLineArgs = string.Format("-i \"{0}\"{5} {2} -map_metadata -1 -threads 0 {3}{4}{6} -y \"{1}\"",
|
|
inputTempFile,
|
|
targetFile,
|
|
videoArgs,
|
|
GetAudioArgs(mediaSource),
|
|
subtitleArgs,
|
|
durationParam,
|
|
outputParam);
|
|
|
|
return inputModifier + " " + commandLineArgs;
|
|
}
|
|
|
|
private static string GetAudioArgs(MediaSourceInfo mediaSource)
|
|
{
|
|
var mediaStreams = mediaSource.MediaStreams ?? new List<MediaStream>();
|
|
var inputAudioCodec = mediaStreams.Where(i => i.Type == MediaStreamType.Audio).Select(i => i.Codec).FirstOrDefault() ?? string.Empty;
|
|
|
|
return "-codec:a:0 copy";
|
|
|
|
//var audioChannels = 2;
|
|
//var audioStream = mediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Audio);
|
|
//if (audioStream != null)
|
|
//{
|
|
// audioChannels = audioStream.Channels ?? audioChannels;
|
|
//}
|
|
//return "-codec:a:0 aac -strict experimental -ab 320000";
|
|
}
|
|
|
|
private static bool EncodeVideo(MediaSourceInfo mediaSource)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
protected string GetOutputSizeParam()
|
|
{
|
|
var filters = new List<string>();
|
|
|
|
filters.Add("yadif=0:-1:0");
|
|
|
|
var output = string.Empty;
|
|
|
|
if (filters.Count > 0)
|
|
{
|
|
output += string.Format(" -vf \"{0}\"", string.Join(",", filters.ToArray()));
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
private void Stop()
|
|
{
|
|
if (!_hasExited)
|
|
{
|
|
try
|
|
{
|
|
_logger.LogInformation("Stopping ffmpeg recording process for {path}", _targetPath);
|
|
|
|
//process.Kill();
|
|
_process.StandardInput.WriteLine("q");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error stopping recording transcoding job for {path}", _targetPath);
|
|
}
|
|
|
|
if (_hasExited)
|
|
{
|
|
return;
|
|
}
|
|
|
|
try
|
|
{
|
|
_logger.LogInformation("Calling recording process.WaitForExit for {path}", _targetPath);
|
|
|
|
if (_process.WaitForExit(10000))
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error waiting for recording process to exit for {path}", _targetPath);
|
|
}
|
|
|
|
if (_hasExited)
|
|
{
|
|
return;
|
|
}
|
|
|
|
try
|
|
{
|
|
_logger.LogInformation("Killing ffmpeg recording process for {path}", _targetPath);
|
|
|
|
_process.Kill();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error killing recording transcoding job for {path}", _targetPath);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Processes the exited.
|
|
/// </summary>
|
|
private void OnFfMpegProcessExited(IProcess process, string inputFile)
|
|
{
|
|
_hasExited = true;
|
|
|
|
DisposeLogStream();
|
|
|
|
try
|
|
{
|
|
var exitCode = process.ExitCode;
|
|
|
|
_logger.LogInformation("FFMpeg recording exited with code {ExitCode} for {path}", exitCode, _targetPath);
|
|
|
|
if (exitCode == 0)
|
|
{
|
|
_taskCompletionSource.TrySetResult(true);
|
|
}
|
|
else
|
|
{
|
|
_taskCompletionSource.TrySetException(new Exception(string.Format("Recording for {path} failed. Exit code {ExitCode}", _targetPath, exitCode)));
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
_logger.LogError("FFMpeg recording exited with an error for {path}.", _targetPath);
|
|
_taskCompletionSource.TrySetException(new Exception(string.Format("Recording for {path} failed", _targetPath)));
|
|
}
|
|
}
|
|
|
|
private void DisposeLogStream()
|
|
{
|
|
if (_logFileStream != null)
|
|
{
|
|
try
|
|
{
|
|
_logFileStream.Dispose();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error disposing recording log stream");
|
|
}
|
|
|
|
_logFileStream = null;
|
|
}
|
|
}
|
|
|
|
private async void StartStreamingLog(Stream source, Stream target)
|
|
{
|
|
try
|
|
{
|
|
using (var reader = new StreamReader(source))
|
|
{
|
|
while (!reader.EndOfStream)
|
|
{
|
|
var line = await reader.ReadLineAsync().ConfigureAwait(false);
|
|
|
|
var bytes = Encoding.UTF8.GetBytes(Environment.NewLine + line);
|
|
|
|
await target.WriteAsync(bytes, 0, bytes.Length).ConfigureAwait(false);
|
|
await target.FlushAsync().ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
catch (ObjectDisposedException)
|
|
{
|
|
// TODO Investigate and properly fix.
|
|
// Don't spam the log. This doesn't seem to throw in windows, but sometimes under linux
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Error reading ffmpeg recording log");
|
|
}
|
|
}
|
|
}
|
|
}
|