mirror of
https://github.com/jellyfin/jellyfin.git
synced 2025-07-09 03:04:24 -04:00
feat: make subtitleeditparser generic
This commit is contained in:
parent
84a1674f39
commit
1db748399c
@ -1,19 +0,0 @@
|
|||||||
using Microsoft.Extensions.Logging;
|
|
||||||
using Nikse.SubtitleEdit.Core.SubtitleFormats;
|
|
||||||
|
|
||||||
namespace MediaBrowser.MediaEncoding.Subtitles
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// Advanced SubStation Alpha subtitle parser.
|
|
||||||
/// </summary>
|
|
||||||
public class AssParser : SubtitleEditParser<AdvancedSubStationAlpha>
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// Initializes a new instance of the <see cref="AssParser"/> class.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="logger">The logger.</param>
|
|
||||||
public AssParser(ILogger logger) : base(logger)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,7 +1,6 @@
|
|||||||
#pragma warning disable CS1591
|
#pragma warning disable CS1591
|
||||||
|
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.Threading;
|
|
||||||
using MediaBrowser.Model.MediaInfo;
|
using MediaBrowser.Model.MediaInfo;
|
||||||
|
|
||||||
namespace MediaBrowser.MediaEncoding.Subtitles
|
namespace MediaBrowser.MediaEncoding.Subtitles
|
||||||
@ -12,8 +11,8 @@ namespace MediaBrowser.MediaEncoding.Subtitles
|
|||||||
/// Parses the specified stream.
|
/// Parses the specified stream.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="stream">The stream.</param>
|
/// <param name="stream">The stream.</param>
|
||||||
/// <param name="cancellationToken">The cancellation token.</param>
|
/// <param name="fileExtension">The file extension.</param>
|
||||||
/// <returns>SubtitleTrackInfo.</returns>
|
/// <returns>SubtitleTrackInfo.</returns>
|
||||||
SubtitleTrackInfo Parse(Stream stream, CancellationToken cancellationToken);
|
SubtitleTrackInfo Parse(Stream stream, string fileExtension);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,19 +0,0 @@
|
|||||||
using Microsoft.Extensions.Logging;
|
|
||||||
using Nikse.SubtitleEdit.Core.SubtitleFormats;
|
|
||||||
|
|
||||||
namespace MediaBrowser.MediaEncoding.Subtitles
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// SubRip subtitle parser.
|
|
||||||
/// </summary>
|
|
||||||
public class SrtParser : SubtitleEditParser<SubRip>
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// Initializes a new instance of the <see cref="SrtParser"/> class.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="logger">The logger.</param>
|
|
||||||
public SrtParser(ILogger logger) : base(logger)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,19 +0,0 @@
|
|||||||
using Microsoft.Extensions.Logging;
|
|
||||||
using Nikse.SubtitleEdit.Core.SubtitleFormats;
|
|
||||||
|
|
||||||
namespace MediaBrowser.MediaEncoding.Subtitles
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// SubStation Alpha subtitle parser.
|
|
||||||
/// </summary>
|
|
||||||
public class SsaParser : SubtitleEditParser<SubStationAlpha>
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// Initializes a new instance of the <see cref="SsaParser"/> class.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="logger">The logger.</param>
|
|
||||||
public SsaParser(ILogger logger) : base(logger)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,7 +1,7 @@
|
|||||||
|
using System;
|
||||||
using System.Globalization;
|
using System.Globalization;
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using System.Threading;
|
|
||||||
using Jellyfin.Extensions;
|
using Jellyfin.Extensions;
|
||||||
using MediaBrowser.Model.MediaInfo;
|
using MediaBrowser.Model.MediaInfo;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
@ -14,31 +14,34 @@ namespace MediaBrowser.MediaEncoding.Subtitles
|
|||||||
/// <summary>
|
/// <summary>
|
||||||
/// SubStation Alpha subtitle parser.
|
/// SubStation Alpha subtitle parser.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <typeparam name="T">The <see cref="SubtitleFormat" />.</typeparam>
|
public class SubtitleEditParser : ISubtitleParser
|
||||||
public abstract class SubtitleEditParser<T> : ISubtitleParser
|
|
||||||
where T : SubtitleFormat, new()
|
|
||||||
{
|
{
|
||||||
private readonly ILogger _logger;
|
private readonly ILogger _logger;
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Initializes a new instance of the <see cref="SubtitleEditParser{T}"/> class.
|
/// Initializes a new instance of the <see cref="SubtitleEditParser"/> class.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="logger">The logger.</param>
|
/// <param name="logger">The logger.</param>
|
||||||
protected SubtitleEditParser(ILogger logger)
|
public SubtitleEditParser(ILogger logger)
|
||||||
{
|
{
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <inheritdoc />
|
/// <inheritdoc />
|
||||||
public SubtitleTrackInfo Parse(Stream stream, CancellationToken cancellationToken)
|
public SubtitleTrackInfo Parse(Stream stream, string fileExtension)
|
||||||
{
|
{
|
||||||
var subtitle = new Subtitle();
|
var subtitleFormat = SubtitleFormat.AllSubtitleFormats.FirstOrDefault(asf => asf.Extension.Equals(fileExtension, StringComparison.OrdinalIgnoreCase));
|
||||||
var subRip = new T();
|
if (subtitleFormat == null)
|
||||||
var lines = stream.ReadAllLines().ToList();
|
|
||||||
subRip.LoadSubtitle(subtitle, lines, "untitled");
|
|
||||||
if (subRip.ErrorCount > 0)
|
|
||||||
{
|
{
|
||||||
_logger.LogError("{ErrorCount} errors encountered while parsing subtitle", subRip.ErrorCount);
|
throw new ArgumentException("Unsupported format: " + fileExtension);
|
||||||
|
}
|
||||||
|
|
||||||
|
var lines = stream.ReadAllLines().ToList();
|
||||||
|
var subtitle = new Subtitle();
|
||||||
|
subtitleFormat.LoadSubtitle(subtitle, lines, fileExtension);
|
||||||
|
if (subtitleFormat.ErrorCount > 0)
|
||||||
|
{
|
||||||
|
_logger.LogError("{ErrorCount} errors encountered while parsing subtitle", subtitleFormat.ErrorCount);
|
||||||
}
|
}
|
||||||
|
|
||||||
var trackInfo = new SubtitleTrackInfo();
|
var trackInfo = new SubtitleTrackInfo();
|
||||||
|
@ -74,7 +74,7 @@ namespace MediaBrowser.MediaEncoding.Subtitles
|
|||||||
try
|
try
|
||||||
{
|
{
|
||||||
var reader = GetReader(inputFormat);
|
var reader = GetReader(inputFormat);
|
||||||
var trackInfo = reader.Parse(stream, cancellationToken);
|
var trackInfo = reader.Parse(stream, $".{inputFormat}");
|
||||||
|
|
||||||
FilterEvents(trackInfo, startTimeTicks, endTimeTicks, preserveOriginalTimestamps);
|
FilterEvents(trackInfo, startTimeTicks, endTimeTicks, preserveOriginalTimestamps);
|
||||||
|
|
||||||
@ -249,26 +249,8 @@ namespace MediaBrowser.MediaEncoding.Subtitles
|
|||||||
|
|
||||||
private bool TryGetReader(string format, [NotNullWhen(true)] out ISubtitleParser? value)
|
private bool TryGetReader(string format, [NotNullWhen(true)] out ISubtitleParser? value)
|
||||||
{
|
{
|
||||||
if (string.Equals(format, SubtitleFormat.SRT, StringComparison.OrdinalIgnoreCase))
|
value = new SubtitleEditParser(_logger);
|
||||||
{
|
return true;
|
||||||
value = new SrtParser(_logger);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (string.Equals(format, SubtitleFormat.SSA, StringComparison.OrdinalIgnoreCase))
|
|
||||||
{
|
|
||||||
value = new SsaParser(_logger);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (string.Equals(format, SubtitleFormat.ASS, StringComparison.OrdinalIgnoreCase))
|
|
||||||
{
|
|
||||||
value = new AssParser(_logger);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
value = null;
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private ISubtitleParser GetReader(string format)
|
private ISubtitleParser GetReader(string format)
|
||||||
|
@ -15,7 +15,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
using (var stream = File.OpenRead("Test Data/example.ass"))
|
using (var stream = File.OpenRead("Test Data/example.ass"))
|
||||||
{
|
{
|
||||||
var parsed = new AssParser(new NullLogger<AssParser>()).Parse(stream, CancellationToken.None);
|
var parsed = new SubtitleEditParser(new NullLogger<SubtitleEditParser>()).Parse(stream, ".ass");
|
||||||
Assert.Single(parsed.TrackEvents);
|
Assert.Single(parsed.TrackEvents);
|
||||||
var trackEvent = parsed.TrackEvents[0];
|
var trackEvent = parsed.TrackEvents[0];
|
||||||
|
|
||||||
|
@ -15,7 +15,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
using (var stream = File.OpenRead("Test Data/example.srt"))
|
using (var stream = File.OpenRead("Test Data/example.srt"))
|
||||||
{
|
{
|
||||||
var parsed = new SrtParser(new NullLogger<SrtParser>()).Parse(stream, CancellationToken.None);
|
var parsed = new SubtitleEditParser(new NullLogger<SubtitleEditParser>()).Parse(stream, ".srt");
|
||||||
Assert.Equal(2, parsed.TrackEvents.Count);
|
Assert.Equal(2, parsed.TrackEvents.Count);
|
||||||
|
|
||||||
var trackEvent1 = parsed.TrackEvents[0];
|
var trackEvent1 = parsed.TrackEvents[0];
|
||||||
@ -37,7 +37,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
using (var stream = File.OpenRead("Test Data/example2.srt"))
|
using (var stream = File.OpenRead("Test Data/example2.srt"))
|
||||||
{
|
{
|
||||||
var parsed = new SrtParser(new NullLogger<SrtParser>()).Parse(stream, CancellationToken.None);
|
var parsed = new SubtitleEditParser(new NullLogger<SubtitleEditParser>()).Parse(stream, ".srt");
|
||||||
Assert.Equal(2, parsed.TrackEvents.Count);
|
Assert.Equal(2, parsed.TrackEvents.Count);
|
||||||
|
|
||||||
var trackEvent1 = parsed.TrackEvents[0];
|
var trackEvent1 = parsed.TrackEvents[0];
|
||||||
|
@ -13,7 +13,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
public class SsaParserTests
|
public class SsaParserTests
|
||||||
{
|
{
|
||||||
private readonly SsaParser _parser = new SsaParser(new NullLogger<AssParser>());
|
private readonly SubtitleEditParser _parser = new SubtitleEditParser(new NullLogger<SubtitleEditParser>());
|
||||||
|
|
||||||
[Theory]
|
[Theory]
|
||||||
[MemberData(nameof(Parse_MultipleDialogues_TestData))]
|
[MemberData(nameof(Parse_MultipleDialogues_TestData))]
|
||||||
@ -21,7 +21,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
using (Stream stream = new MemoryStream(Encoding.UTF8.GetBytes(ssa)))
|
using (Stream stream = new MemoryStream(Encoding.UTF8.GetBytes(ssa)))
|
||||||
{
|
{
|
||||||
SubtitleTrackInfo subtitleTrackInfo = _parser.Parse(stream, CancellationToken.None);
|
SubtitleTrackInfo subtitleTrackInfo = _parser.Parse(stream, ".ssa");
|
||||||
|
|
||||||
Assert.Equal(expectedSubtitleTrackEvents.Count, subtitleTrackInfo.TrackEvents.Count);
|
Assert.Equal(expectedSubtitleTrackEvents.Count, subtitleTrackInfo.TrackEvents.Count);
|
||||||
|
|
||||||
@ -76,7 +76,7 @@ namespace Jellyfin.MediaEncoding.Subtitles.Tests
|
|||||||
{
|
{
|
||||||
using (var stream = File.OpenRead("Test Data/example.ssa"))
|
using (var stream = File.OpenRead("Test Data/example.ssa"))
|
||||||
{
|
{
|
||||||
var parsed = _parser.Parse(stream, CancellationToken.None);
|
var parsed = _parser.Parse(stream, ".ssa");
|
||||||
Assert.Single(parsed.TrackEvents);
|
Assert.Single(parsed.TrackEvents);
|
||||||
var trackEvent = parsed.TrackEvents[0];
|
var trackEvent = parsed.TrackEvents[0];
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user