|  | @@ -1,14 +1,12 @@
 | 
	
		
			
				|  |  | -using System;
 | 
	
		
			
				|  |  | -using System.Collections.Generic;
 | 
	
		
			
				|  |  |  using System.Globalization;
 | 
	
		
			
				|  |  |  using System.IO;
 | 
	
		
			
				|  |  |  using System.Linq;
 | 
	
		
			
				|  |  | -using System.Reflection;
 | 
	
		
			
				|  |  | +using System.Threading;
 | 
	
		
			
				|  |  |  using Jellyfin.Extensions;
 | 
	
		
			
				|  |  |  using MediaBrowser.Model.MediaInfo;
 | 
	
		
			
				|  |  |  using Microsoft.Extensions.Logging;
 | 
	
		
			
				|  |  |  using Nikse.SubtitleEdit.Core.Common;
 | 
	
		
			
				|  |  | -using Nikse.SubtitleEdit.Core.SubtitleFormats;
 | 
	
		
			
				|  |  | +using ILogger = Microsoft.Extensions.Logging.ILogger;
 | 
	
		
			
				|  |  |  using SubtitleFormat = Nikse.SubtitleEdit.Core.SubtitleFormats.SubtitleFormat;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  namespace MediaBrowser.MediaEncoding.Subtitles
 | 
	
	
		
			
				|  | @@ -16,57 +14,31 @@ namespace MediaBrowser.MediaEncoding.Subtitles
 | 
	
		
			
				|  |  |      /// <summary>
 | 
	
		
			
				|  |  |      /// SubStation Alpha subtitle parser.
 | 
	
		
			
				|  |  |      /// </summary>
 | 
	
		
			
				|  |  | -    public class SubtitleEditParser : ISubtitleParser
 | 
	
		
			
				|  |  | +    /// <typeparam name="T">The <see cref="SubtitleFormat" />.</typeparam>
 | 
	
		
			
				|  |  | +    public abstract class SubtitleEditParser<T> : ISubtitleParser
 | 
	
		
			
				|  |  | +        where T : SubtitleFormat, new()
 | 
	
		
			
				|  |  |      {
 | 
	
		
			
				|  |  | -        private readonly ILogger<SubtitleEditParser> _logger;
 | 
	
		
			
				|  |  | -        private readonly Dictionary<string, SubtitleFormat[]> _subtitleFormats;
 | 
	
		
			
				|  |  | +        private readonly ILogger _logger;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          /// <summary>
 | 
	
		
			
				|  |  | -        /// Initializes a new instance of the <see cref="SubtitleEditParser"/> class.
 | 
	
		
			
				|  |  | +        /// Initializes a new instance of the <see cref="SubtitleEditParser{T}"/> class.
 | 
	
		
			
				|  |  |          /// </summary>
 | 
	
		
			
				|  |  |          /// <param name="logger">The logger.</param>
 | 
	
		
			
				|  |  | -        public SubtitleEditParser(ILogger<SubtitleEditParser> logger)
 | 
	
		
			
				|  |  | +        protected SubtitleEditParser(ILogger logger)
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  |              _logger = logger;
 | 
	
		
			
				|  |  | -            _subtitleFormats = GetSubtitleFormats()
 | 
	
		
			
				|  |  | -                .Where(subtitleFormat => !string.IsNullOrEmpty(subtitleFormat.Extension))
 | 
	
		
			
				|  |  | -                .GroupBy(subtitleFormat => subtitleFormat.Extension.TrimStart('.'), StringComparer.OrdinalIgnoreCase)
 | 
	
		
			
				|  |  | -                .ToDictionary(g => g.Key, g => g.ToArray(), StringComparer.OrdinalIgnoreCase);
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          /// <inheritdoc />
 | 
	
		
			
				|  |  | -        public SubtitleTrackInfo Parse(Stream stream, string fileExtension)
 | 
	
		
			
				|  |  | +        public SubtitleTrackInfo Parse(Stream stream, CancellationToken cancellationToken)
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  |              var subtitle = new Subtitle();
 | 
	
		
			
				|  |  | +            var subRip = new T();
 | 
	
		
			
				|  |  |              var lines = stream.ReadAllLines().ToList();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            if (!_subtitleFormats.TryGetValue(fileExtension, out var subtitleFormats))
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                throw new ArgumentException($"Unsupported file extension: {fileExtension}", nameof(fileExtension));
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            foreach (var subtitleFormat in subtitleFormats)
 | 
	
		
			
				|  |  | +            subRip.LoadSubtitle(subtitle, lines, "untitled");
 | 
	
		
			
				|  |  | +            if (subRip.ErrorCount > 0)
 | 
	
		
			
				|  |  |              {
 | 
	
		
			
				|  |  | -                _logger.LogDebug(
 | 
	
		
			
				|  |  | -                    "Trying to parse '{FileExtension}' subtitle using the {SubtitleFormatParser} format parser",
 | 
	
		
			
				|  |  | -                    fileExtension,
 | 
	
		
			
				|  |  | -                    subtitleFormat.Name);
 | 
	
		
			
				|  |  | -                subtitleFormat.LoadSubtitle(subtitle, lines, fileExtension);
 | 
	
		
			
				|  |  | -                if (subtitleFormat.ErrorCount == 0)
 | 
	
		
			
				|  |  | -                {
 | 
	
		
			
				|  |  | -                    break;
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                _logger.LogError(
 | 
	
		
			
				|  |  | -                    "{ErrorCount} errors encountered while parsing '{FileExtension}' subtitle using the {SubtitleFormatParser} format parser",
 | 
	
		
			
				|  |  | -                    subtitleFormat.ErrorCount,
 | 
	
		
			
				|  |  | -                    fileExtension,
 | 
	
		
			
				|  |  | -                    subtitleFormat.Name);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            if (subtitle.Paragraphs.Count == 0)
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                throw new ArgumentException("Unsupported format: " + fileExtension);
 | 
	
		
			
				|  |  | +                _logger.LogError("{ErrorCount} errors encountered while parsing subtitle", subRip.ErrorCount);
 | 
	
		
			
				|  |  |              }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |              var trackInfo = new SubtitleTrackInfo();
 | 
	
	
		
			
				|  | @@ -85,36 +57,5 @@ namespace MediaBrowser.MediaEncoding.Subtitles
 | 
	
		
			
				|  |  |              trackInfo.TrackEvents = trackEvents;
 | 
	
		
			
				|  |  |              return trackInfo;
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <inheritdoc />
 | 
	
		
			
				|  |  | -        public bool SupportsFileExtension(string fileExtension)
 | 
	
		
			
				|  |  | -            => _subtitleFormats.ContainsKey(fileExtension);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        private IEnumerable<SubtitleFormat> GetSubtitleFormats()
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            var subtitleFormats = new List<SubtitleFormat>();
 | 
	
		
			
				|  |  | -            var assembly = typeof(SubtitleFormat).Assembly;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            foreach (var type in assembly.GetTypes())
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                if (!type.IsSubclassOf(typeof(SubtitleFormat)) || type.IsAbstract)
 | 
	
		
			
				|  |  | -                {
 | 
	
		
			
				|  |  | -                    continue;
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                try
 | 
	
		
			
				|  |  | -                {
 | 
	
		
			
				|  |  | -                    // It shouldn't be null, but the exception is caught if it is
 | 
	
		
			
				|  |  | -                    var subtitleFormat = (SubtitleFormat)Activator.CreateInstance(type, true)!;
 | 
	
		
			
				|  |  | -                    subtitleFormats.Add(subtitleFormat);
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -                catch (Exception ex)
 | 
	
		
			
				|  |  | -                {
 | 
	
		
			
				|  |  | -                    _logger.LogWarning(ex, "Failed to create instance of the subtitle format {SubtitleFormatType}", type.Name);
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            return subtitleFormats;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  }
 |