< Summary - Jellyfin

Information
Class: Jellyfin.LiveTv.IO.EncodedRecorder
Assembly: Jellyfin.LiveTv
File(s): /srv/git/jellyfin/src/Jellyfin.LiveTv/IO/EncodedRecorder.cs
Line coverage
0%
Covered lines: 0
Uncovered lines: 98
Coverable lines: 98
Total lines: 334
Line coverage: 0%
Branch coverage
0%
Covered branches: 0
Total branches: 34
Branch coverage: 0%
Method coverage

Feature is only available for sponsors

Upgrade to PRO version

Coverage history

Coverage history 0 25 50 75 100

Metrics

MethodBranch coverage Crap Score Cyclomatic complexity Line coverage
.ctor(...)100%210%
get_CopySubtitles()100%210%
GetOutputPath(...)100%210%
GetCommandLineArgs(...)0%210140%
GetAudioArgs(...)100%210%
GetOutputSizeParam()100%210%
Stop()0%7280%
OnFfMpegProcessExited(...)0%2040%
Dispose()100%210%
Dispose(...)0%7280%

File(s)

/srv/git/jellyfin/src/Jellyfin.LiveTv/IO/EncodedRecorder.cs

#LineLine coverage
 1#nullable disable
 2
 3#pragma warning disable CS1591
 4
 5using System;
 6using System.Collections.Generic;
 7using System.Diagnostics;
 8using System.Globalization;
 9using System.IO;
 10using System.Text;
 11using System.Text.Json;
 12using System.Threading;
 13using System.Threading.Tasks;
 14using Jellyfin.Extensions;
 15using Jellyfin.Extensions.Json;
 16using MediaBrowser.Common;
 17using MediaBrowser.Common.Configuration;
 18using MediaBrowser.Controller;
 19using MediaBrowser.Controller.Configuration;
 20using MediaBrowser.Controller.Library;
 21using MediaBrowser.Controller.MediaEncoding;
 22using MediaBrowser.Model.Dto;
 23using MediaBrowser.Model.IO;
 24using Microsoft.Extensions.Logging;
 25
 26namespace Jellyfin.LiveTv.IO
 27{
 28    public class EncodedRecorder : IRecorder
 29    {
 30        private readonly ILogger _logger;
 31        private readonly IMediaEncoder _mediaEncoder;
 32        private readonly IServerApplicationPaths _appPaths;
 033        private readonly TaskCompletionSource<bool> _taskCompletionSource = new TaskCompletionSource<bool>(TaskCreationO
 34        private readonly IServerConfigurationManager _serverConfigurationManager;
 035        private readonly JsonSerializerOptions _jsonOptions = JsonDefaults.Options;
 36        private bool _hasExited;
 37        private FileStream _logFileStream;
 38        private string _targetPath;
 39        private Process _process;
 40        private bool _disposed;
 41
 42        public EncodedRecorder(
 43            ILogger logger,
 44            IMediaEncoder mediaEncoder,
 45            IServerApplicationPaths appPaths,
 46            IServerConfigurationManager serverConfigurationManager)
 47        {
 048            _logger = logger;
 049            _mediaEncoder = mediaEncoder;
 050            _appPaths = appPaths;
 051            _serverConfigurationManager = serverConfigurationManager;
 052        }
 53
 054        private static bool CopySubtitles => false;
 55
 56        public string GetOutputPath(MediaSourceInfo mediaSource, string targetFile)
 57        {
 058            return Path.ChangeExtension(targetFile, ".ts");
 59        }
 60
 61        public async Task Record(IDirectStreamProvider directStreamProvider, MediaSourceInfo mediaSource, string targetF
 62        {
 63            // The media source is infinite so we need to handle stopping ourselves
 64            using var durationToken = new CancellationTokenSource(duration);
 65            using var cancellationTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, durat
 66
 67            await RecordFromFile(mediaSource, mediaSource.Path, targetFile, onStarted, cancellationTokenSource.Token).Co
 68
 69            _logger.LogInformation("Recording completed to file {Path}", targetFile);
 70        }
 71
 72        private async Task RecordFromFile(MediaSourceInfo mediaSource, string inputFile, string targetFile, Action onSta
 73        {
 74            _targetPath = targetFile;
 75            Directory.CreateDirectory(Path.GetDirectoryName(targetFile));
 76
 77            var processStartInfo = new ProcessStartInfo
 78            {
 79                CreateNoWindow = true,
 80                UseShellExecute = false,
 81
 82                RedirectStandardError = true,
 83                RedirectStandardInput = true,
 84
 85                FileName = _mediaEncoder.EncoderPath,
 86                Arguments = GetCommandLineArgs(mediaSource, inputFile, targetFile),
 87
 88                WindowStyle = ProcessWindowStyle.Hidden,
 89                ErrorDialog = false
 90            };
 91
 92            _logger.LogInformation("{Filename} {Arguments}", processStartInfo.FileName, processStartInfo.Arguments);
 93
 94            var logFilePath = Path.Combine(_appPaths.LogDirectoryPath, "record-transcode-" + Guid.NewGuid() + ".txt");
 95            Directory.CreateDirectory(Path.GetDirectoryName(logFilePath));
 96
 97            // FFMpeg writes debug/error info to stderr. This is useful when debugging so let's put it in the log direct
 98            _logFileStream = new FileStream(logFilePath, FileMode.CreateNew, FileAccess.Write, FileShare.Read, IODefault
 99
 100            await JsonSerializer.SerializeAsync(_logFileStream, mediaSource, _jsonOptions, cancellationToken).ConfigureA
 101            await _logFileStream.WriteAsync(Encoding.UTF8.GetBytes(Environment.NewLine + Environment.NewLine + processSt
 102
 103            _process = new Process
 104            {
 105                StartInfo = processStartInfo,
 106                EnableRaisingEvents = true
 107            };
 108            _process.Exited += (_, _) => OnFfMpegProcessExited(_process);
 109
 110            _process.Start();
 111
 112            cancellationToken.Register(Stop);
 113
 114            onStarted();
 115
 116            // Important - don't await the log task or we won't be able to kill ffmpeg when the user stops playback
 117            _ = StartStreamingLog(_process.StandardError.BaseStream, _logFileStream);
 118
 119            _logger.LogInformation("ffmpeg recording process started for {Path}", _targetPath);
 120
 121            // Block until ffmpeg exits
 122            await _taskCompletionSource.Task.ConfigureAwait(false);
 123        }
 124
 125        private string GetCommandLineArgs(MediaSourceInfo mediaSource, string inputTempFile, string targetFile)
 126        {
 0127            string videoArgs = "-codec:v:0 copy -fflags +genpts";
 128
 0129            var flags = new List<string>();
 0130            if (mediaSource.IgnoreDts)
 131            {
 0132                flags.Add("+igndts");
 133            }
 134
 0135            if (mediaSource.IgnoreIndex)
 136            {
 0137                flags.Add("+ignidx");
 138            }
 139
 0140            if (mediaSource.GenPtsInput)
 141            {
 0142                flags.Add("+genpts");
 143            }
 144
 0145            var inputModifier = "-async 1";
 146
 0147            if (flags.Count > 0)
 148            {
 0149                inputModifier += " -fflags " + string.Join(string.Empty, flags);
 150            }
 151
 0152            if (mediaSource.ReadAtNativeFramerate)
 153            {
 0154                inputModifier += " -re";
 155            }
 156
 0157            if (mediaSource.RequiresLooping)
 158            {
 0159                inputModifier += " -stream_loop -1 -reconnect_at_eof 1 -reconnect_streamed 1 -reconnect_delay_max 2";
 160            }
 161
 0162            var analyzeDurationSeconds = 5;
 0163            var analyzeDuration = " -analyzeduration " +
 0164                  (analyzeDurationSeconds * 1000000).ToString(CultureInfo.InvariantCulture);
 0165            inputModifier += analyzeDuration;
 166
 0167            var subtitleArgs = CopySubtitles ? " -codec:s copy" : " -sn";
 168
 169            // var outputParam = string.Equals(Path.GetExtension(targetFile), ".mp4", StringComparison.OrdinalIgnoreCase
 170            //    " -f mp4 -movflags frag_keyframe+empty_moov" :
 171            //    string.Empty;
 172
 0173            var outputParam = string.Empty;
 174
 0175            var threads = EncodingHelper.GetNumberOfThreads(null, _serverConfigurationManager.GetEncodingOptions(), null
 0176            var commandLineArgs = string.Format(
 0177                CultureInfo.InvariantCulture,
 0178                "-i \"{0}\" {2} -map_metadata -1 -threads {6} {3}{4}{5} -y \"{1}\"",
 0179                inputTempFile,
 0180                targetFile.Replace("\"", "\\\"", StringComparison.Ordinal), // Escape quotes in filename
 0181                videoArgs,
 0182                GetAudioArgs(mediaSource),
 0183                subtitleArgs,
 0184                outputParam,
 0185                threads);
 186
 0187            return inputModifier + " " + commandLineArgs;
 188        }
 189
 190        private static string GetAudioArgs(MediaSourceInfo mediaSource)
 191        {
 0192            return "-codec:a:0 copy";
 193        }
 194
 195        protected string GetOutputSizeParam()
 0196            => "-vf \"yadif=0:-1:0\"";
 197
 198        private void Stop()
 199        {
 0200            if (!_hasExited)
 201            {
 202                try
 203                {
 0204                    _logger.LogInformation("Stopping ffmpeg recording process for {Path}", _targetPath);
 205
 0206                    _process.StandardInput.WriteLine("q");
 0207                }
 0208                catch (Exception ex)
 209                {
 0210                    _logger.LogError(ex, "Error stopping recording transcoding job for {Path}", _targetPath);
 0211                }
 212
 0213                if (_hasExited)
 214                {
 0215                    return;
 216                }
 217
 218                try
 219                {
 0220                    _logger.LogInformation("Calling recording process.WaitForExit for {Path}", _targetPath);
 221
 0222                    if (_process.WaitForExit(10000))
 223                    {
 0224                        return;
 225                    }
 0226                }
 0227                catch (Exception ex)
 228                {
 0229                    _logger.LogError(ex, "Error waiting for recording process to exit for {Path}", _targetPath);
 0230                }
 231
 0232                if (_hasExited)
 233                {
 0234                    return;
 235                }
 236
 237                try
 238                {
 0239                    _logger.LogInformation("Killing ffmpeg recording process for {Path}", _targetPath);
 240
 0241                    _process.Kill();
 0242                }
 0243                catch (Exception ex)
 244                {
 0245                    _logger.LogError(ex, "Error killing recording transcoding job for {Path}", _targetPath);
 0246                }
 247            }
 0248        }
 249
 250        /// <summary>
 251        /// Processes the exited.
 252        /// </summary>
 253        private void OnFfMpegProcessExited(Process process)
 254        {
 0255            using (process)
 256            {
 0257                _hasExited = true;
 258
 0259                _logFileStream?.Dispose();
 0260                _logFileStream = null;
 261
 0262                var exitCode = process.ExitCode;
 263
 0264                _logger.LogInformation("FFMpeg recording exited with code {ExitCode} for {Path}", exitCode, _targetPath)
 265
 0266                if (exitCode == 0)
 267                {
 0268                    _taskCompletionSource.TrySetResult(true);
 269                }
 270                else
 271                {
 0272                    _taskCompletionSource.TrySetException(
 0273                        new FfmpegException(
 0274                            string.Format(
 0275                                CultureInfo.InvariantCulture,
 0276                                "Recording for {0} failed. Exit code {1}",
 0277                                _targetPath,
 0278                                exitCode)));
 279                }
 0280            }
 0281        }
 282
 283        private async Task StartStreamingLog(Stream source, FileStream target)
 284        {
 285            try
 286            {
 287                using (var reader = new StreamReader(source))
 288                {
 289                    await foreach (var line in reader.ReadAllLinesAsync().ConfigureAwait(false))
 290                    {
 291                        var bytes = Encoding.UTF8.GetBytes(Environment.NewLine + line);
 292
 293                        await target.WriteAsync(bytes.AsMemory()).ConfigureAwait(false);
 294                        await target.FlushAsync().ConfigureAwait(false);
 295                    }
 296                }
 297            }
 298            catch (Exception ex)
 299            {
 300                _logger.LogError(ex, "Error reading ffmpeg recording log");
 301            }
 302        }
 303
 304        /// <inheritdoc />
 305        public void Dispose()
 306        {
 0307            Dispose(true);
 0308            GC.SuppressFinalize(this);
 0309        }
 310
 311        /// <summary>
 312        /// Releases unmanaged and optionally managed resources.
 313        /// </summary>
 314        /// <param name="disposing"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release
 315        protected virtual void Dispose(bool disposing)
 316        {
 0317            if (_disposed)
 318            {
 0319                return;
 320            }
 321
 0322            if (disposing)
 323            {
 0324                _logFileStream?.Dispose();
 0325                _process?.Dispose();
 326            }
 327
 0328            _logFileStream = null;
 0329            _process = null;
 330
 0331            _disposed = true;
 0332        }
 333    }
 334}
Generated by: ReportGenerator 5.4.5.0
4/3/2025 - 12:11:21 AM
GitHub | reportgenerator.io

Methods/Properties

.ctor(Microsoft.Extensions.Logging.ILogger,MediaBrowser.Controller.MediaEncoding.IMediaEncoder,MediaBrowser.Controller.IServerApplicationPaths,MediaBrowser.Controller.Configuration.IServerConfigurationManager)
get_CopySubtitles()
GetOutputPath(MediaBrowser.Model.Dto.MediaSourceInfo,System.String)
GetCommandLineArgs(MediaBrowser.Model.Dto.MediaSourceInfo,System.String,System.String)
GetAudioArgs(MediaBrowser.Model.Dto.MediaSourceInfo)
GetOutputSizeParam()
Stop()
OnFfMpegProcessExited(System.Diagnostics.Process)
Dispose()
Dispose(System.Boolean)