mirror of https://github.com/ErsatzTV/ErsatzTV.git
Browse Source
* properly track discontinuity sequences with fmp4 * update dependenciespull/2549/head
Jason Dove
10 months ago
committed by
GitHub
15 changed files with 611 additions and 435 deletions
@ -1,5 +0,0 @@
@@ -1,5 +0,0 @@
|
||||
using ErsatzTV.Core; |
||||
|
||||
namespace ErsatzTV.Application.Streaming; |
||||
|
||||
public record GetLastPtsTime(string ChannelNumber) : IRequest<Either<BaseError, PtsTime>>; |
||||
@ -1,170 +0,0 @@
@@ -1,170 +0,0 @@
|
||||
using System.Text; |
||||
using Bugsnag; |
||||
using CliWrap; |
||||
using ErsatzTV.Core; |
||||
using ErsatzTV.Core.Domain; |
||||
using ErsatzTV.Core.FFmpeg; |
||||
using ErsatzTV.Core.Interfaces.FFmpeg; |
||||
using ErsatzTV.Core.Interfaces.Metadata; |
||||
using ErsatzTV.Core.Interfaces.Repositories; |
||||
using Microsoft.Extensions.Logging; |
||||
using Newtonsoft.Json; |
||||
|
||||
namespace ErsatzTV.Application.Streaming; |
||||
|
||||
public class GetLastPtsTimeHandler : IRequestHandler<GetLastPtsTime, Either<BaseError, PtsTime>> |
||||
{ |
||||
private readonly IClient _client; |
||||
private readonly IConfigElementRepository _configElementRepository; |
||||
private readonly ILocalFileSystem _localFileSystem; |
||||
private readonly ILogger<GetLastPtsTimeHandler> _logger; |
||||
private readonly ITempFilePool _tempFilePool; |
||||
|
||||
public GetLastPtsTimeHandler( |
||||
IClient client, |
||||
ILocalFileSystem localFileSystem, |
||||
ITempFilePool tempFilePool, |
||||
IConfigElementRepository configElementRepository, |
||||
ILogger<GetLastPtsTimeHandler> logger) |
||||
{ |
||||
_client = client; |
||||
_localFileSystem = localFileSystem; |
||||
_tempFilePool = tempFilePool; |
||||
_configElementRepository = configElementRepository; |
||||
_logger = logger; |
||||
} |
||||
|
||||
public async Task<Either<BaseError, PtsTime>> Handle( |
||||
GetLastPtsTime request, |
||||
CancellationToken cancellationToken) |
||||
{ |
||||
Validation<BaseError, RequestParameters> validation = await Validate(request, cancellationToken); |
||||
return await validation.Match( |
||||
parameters => Handle(parameters, cancellationToken), |
||||
error => Task.FromResult<Either<BaseError, PtsTime>>(error.Join())); |
||||
} |
||||
|
||||
private async Task<Validation<BaseError, RequestParameters>> Validate( |
||||
GetLastPtsTime request, |
||||
CancellationToken cancellationToken) => |
||||
await ValidateFFprobePath(cancellationToken) |
||||
.MapT(ffprobePath => new RequestParameters(request.ChannelNumber, ffprobePath)); |
||||
|
||||
private async Task<Either<BaseError, PtsTime>> Handle( |
||||
RequestParameters parameters, |
||||
CancellationToken cancellationToken) |
||||
{ |
||||
Option<FileInfo> maybeLastSegment = GetLastSegment(parameters.ChannelNumber); |
||||
foreach (FileInfo segment in maybeLastSegment) |
||||
{ |
||||
return await GetPts(parameters, segment, cancellationToken).IfNoneAsync(PtsTime.Zero); |
||||
} |
||||
|
||||
return BaseError.New($"Failed to determine last pts duration for channel {parameters.ChannelNumber}"); |
||||
} |
||||
|
||||
private async Task<Option<PtsTime>> GetPts( |
||||
RequestParameters parameters, |
||||
FileInfo segment, |
||||
CancellationToken cancellationToken) |
||||
{ |
||||
string[] argumentList = |
||||
{ |
||||
"-v", "0", |
||||
"-show_entries", |
||||
"packet=pts,duration", |
||||
"-of", "compact=p=0:nk=1", |
||||
segment.FullName |
||||
}; |
||||
|
||||
string lastLine = string.Empty; |
||||
Action<string> replaceLine = s => |
||||
{ |
||||
if (!string.IsNullOrWhiteSpace(s)) |
||||
{ |
||||
lastLine = s.Trim(); |
||||
} |
||||
}; |
||||
|
||||
CommandResult probe = await Cli.Wrap(parameters.FFprobePath) |
||||
.WithArguments(argumentList) |
||||
.WithValidation(CommandResultValidation.None) |
||||
.WithStandardOutputPipe(PipeTarget.ToDelegate(replaceLine)) |
||||
.ExecuteAsync(cancellationToken); |
||||
|
||||
if (probe.ExitCode != 0) |
||||
{ |
||||
return Option<PtsTime>.None; |
||||
} |
||||
|
||||
try |
||||
{ |
||||
return PtsTime.From(lastLine); |
||||
} |
||||
catch (Exception ex) |
||||
{ |
||||
_client.Notify(ex); |
||||
await SaveTroubleshootingData(parameters.ChannelNumber, lastLine); |
||||
} |
||||
|
||||
return Option<PtsTime>.None; |
||||
} |
||||
|
||||
private static Option<FileInfo> GetLastSegment(string channelNumber) |
||||
{ |
||||
var directory = new DirectoryInfo(Path.Combine(FileSystemLayout.TranscodeFolder, channelNumber)); |
||||
return Optional(directory.GetFiles("*.ts").OrderByDescending(f => f.Name).FirstOrDefault()); |
||||
} |
||||
|
||||
private Task<Validation<BaseError, string>> ValidateFFprobePath(CancellationToken cancellationToken) => |
||||
_configElementRepository.GetValue<string>(ConfigElementKey.FFprobePath, cancellationToken) |
||||
.FilterT(File.Exists) |
||||
.Map(ffprobePath => ffprobePath.ToValidation<BaseError>("FFprobe path does not exist on the file system")); |
||||
|
||||
private async Task SaveTroubleshootingData(string channelNumber, string output) |
||||
{ |
||||
try |
||||
{ |
||||
var directory = new DirectoryInfo(Path.Combine(FileSystemLayout.TranscodeFolder, channelNumber)); |
||||
FileInfo[] allFiles = directory.GetFiles(); |
||||
|
||||
string playlistFileName = Path.Combine(FileSystemLayout.TranscodeFolder, channelNumber, "live.m3u8"); |
||||
string playlistContents = string.Empty; |
||||
if (_localFileSystem.FileExists(playlistFileName)) |
||||
{ |
||||
playlistContents = await File.ReadAllTextAsync(playlistFileName); |
||||
} |
||||
|
||||
var data = new TroubleshootingData(allFiles, playlistContents, output); |
||||
string serialized = data.Serialize(); |
||||
|
||||
string file = _tempFilePool.GetNextTempFile(TempFileCategory.BadTranscodeFolder); |
||||
await File.WriteAllTextAsync(file, serialized); |
||||
|
||||
_logger.LogWarning("Transcode folder is in bad state; troubleshooting info saved to {File}", file); |
||||
} |
||||
catch (Exception ex) |
||||
{ |
||||
_client.Notify(ex); |
||||
} |
||||
} |
||||
|
||||
private sealed record RequestParameters(string ChannelNumber, string FFprobePath); |
||||
|
||||
private sealed record TroubleshootingData(IEnumerable<FileInfo> Files, string Playlist, string ProbeOutput) |
||||
{ |
||||
public string Serialize() |
||||
{ |
||||
var data = new InternalData( |
||||
Files.Map(f => new FileData(f.FullName, f.Length, f.LastWriteTimeUtc)).ToList(), |
||||
Convert.ToBase64String(Encoding.UTF8.GetBytes(Playlist)), |
||||
Convert.ToBase64String(Encoding.UTF8.GetBytes(ProbeOutput))); |
||||
|
||||
return JsonConvert.SerializeObject(data); |
||||
} |
||||
|
||||
private sealed record FileData(string FileName, long Bytes, DateTime LastWriteTimeUtc); |
||||
|
||||
private sealed record InternalData(List<FileData> Files, string EncodedPlaylist, string EncodedProbeOutput); |
||||
} |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in new issue