jellyfin-server/MediaBrowser.Providers/MediaInfo/SubtitleResolver.cs

195 lines
6.9 KiB
C#
Raw Normal View History

using System;
2014-05-16 19:16:29 +00:00
using System.Collections.Generic;
using System.IO;
using System.Linq;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Extensions;
2016-10-24 02:45:23 +00:00
using MediaBrowser.Model.Globalization;
using MediaBrowser.Model.IO;
2014-05-16 19:16:29 +00:00
namespace MediaBrowser.Providers.MediaInfo
{
public class SubtitleResolver
{
private readonly ILocalizationManager _localization;
2014-07-26 17:30:15 +00:00
private readonly IFileSystem _fileSystem;
2014-05-16 19:16:29 +00:00
private string[] SubtitleExtensions = new[]
{
".srt",
".ssa",
".ass",
".sub",
".smi",
".sami",
".vtt"
};
2014-07-26 17:30:15 +00:00
public SubtitleResolver(ILocalizationManager localization, IFileSystem fileSystem)
2014-05-16 19:16:29 +00:00
{
_localization = localization;
2014-07-26 17:30:15 +00:00
_fileSystem = fileSystem;
2014-05-16 19:16:29 +00:00
}
public List<MediaStream> GetExternalSubtitleStreams(Video video,
2014-05-16 19:16:29 +00:00
int startIndex,
IDirectoryService directoryService,
bool clearCache)
{
var streams = new List<MediaStream>();
2018-09-12 17:26:21 +00:00
if (!video.IsFileProtocol)
{
return streams;
}
AddExternalSubtitleStreams(streams, video.ContainingFolderPath, video.Path, startIndex, directoryService, clearCache);
startIndex += streams.Count;
try
{
2018-09-12 17:26:21 +00:00
AddExternalSubtitleStreams(streams, video.GetInternalMetadataPath(), video.Path, startIndex, directoryService, clearCache);
}
catch (IOException)
{
}
return streams;
}
public List<string> GetExternalSubtitleFiles(Video video,
IDirectoryService directoryService,
bool clearCache)
{
var list = new List<string>();
2018-09-12 17:26:21 +00:00
if (!video.IsFileProtocol)
{
return list;
}
var streams = GetExternalSubtitleStreams(video, 0, directoryService, clearCache);
foreach (var stream in streams)
{
list.Add(stream.Path);
}
return list;
}
2018-09-12 17:26:21 +00:00
private void AddExternalSubtitleStreams(List<MediaStream> streams, string folder,
string videoPath,
int startIndex,
IDirectoryService directoryService,
bool clearCache)
2018-09-12 17:26:21 +00:00
{
var files = directoryService.GetFilePaths(folder, clearCache).OrderBy(i => i).ToArray();
AddExternalSubtitleStreams(streams, videoPath, startIndex, files);
}
public void AddExternalSubtitleStreams(List<MediaStream> streams,
string videoPath,
int startIndex,
string[] files)
{
var videoFileNameWithoutExtension = Path.GetFileNameWithoutExtension(videoPath);
videoFileNameWithoutExtension = NormalizeFilenameForSubtitleComparison(videoFileNameWithoutExtension);
2014-05-16 19:16:29 +00:00
2017-03-29 06:26:48 +00:00
foreach (var fullName in files)
2014-05-16 19:16:29 +00:00
{
var extension = Path.GetExtension(fullName);
if (!SubtitleExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase))
{
continue;
}
var fileNameWithoutExtension = Path.GetFileNameWithoutExtension(fullName);
fileNameWithoutExtension = NormalizeFilenameForSubtitleComparison(fileNameWithoutExtension);
2014-05-16 19:16:29 +00:00
if (!string.Equals(videoFileNameWithoutExtension, fileNameWithoutExtension, StringComparison.OrdinalIgnoreCase) &&
!fileNameWithoutExtension.StartsWith(videoFileNameWithoutExtension + ".", StringComparison.OrdinalIgnoreCase))
{
continue;
}
2019-01-27 11:03:43 +00:00
var codec = Path.GetExtension(fullName).ToLowerInvariant().TrimStart('.');
2014-05-16 19:16:29 +00:00
2016-11-08 18:44:23 +00:00
if (string.Equals(codec, "txt", StringComparison.OrdinalIgnoreCase))
{
codec = "srt";
}
2014-05-16 19:16:29 +00:00
// If the subtitle file matches the video file name
if (string.Equals(videoFileNameWithoutExtension, fileNameWithoutExtension, StringComparison.OrdinalIgnoreCase))
{
streams.Add(new MediaStream
{
Index = startIndex++,
Type = MediaStreamType.Subtitle,
IsExternal = true,
Path = fullName,
Codec = codec
});
}
else if (fileNameWithoutExtension.StartsWith(videoFileNameWithoutExtension + ".", StringComparison.OrdinalIgnoreCase))
{
var isForced = fullName.IndexOf(".forced.", StringComparison.OrdinalIgnoreCase) != -1 ||
fullName.IndexOf(".foreign.", StringComparison.OrdinalIgnoreCase) != -1;
var isDefault = fullName.IndexOf(".default.", StringComparison.OrdinalIgnoreCase) != -1;
2014-05-16 19:16:29 +00:00
// Support xbmc naming conventions - 300.spanish.srt
var language = fileNameWithoutExtension
.Replace(".forced", string.Empty, StringComparison.OrdinalIgnoreCase)
.Replace(".foreign", string.Empty, StringComparison.OrdinalIgnoreCase)
2018-09-12 17:26:21 +00:00
.Replace(".default", string.Empty, StringComparison.OrdinalIgnoreCase)
2014-05-16 19:16:29 +00:00
.Split('.')
.LastOrDefault();
// Try to translate to three character code
// Be flexible and check against both the full and three character versions
2018-09-12 17:26:21 +00:00
var culture = _localization.FindLanguageInfo(language);
2014-05-16 19:16:29 +00:00
if (culture != null)
{
language = culture.ThreeLetterISOLanguageName;
}
streams.Add(new MediaStream
{
Index = startIndex++,
Type = MediaStreamType.Subtitle,
IsExternal = true,
Path = fullName,
Codec = codec,
Language = language,
IsForced = isForced,
IsDefault = isDefault
2014-05-16 19:16:29 +00:00
});
}
}
}
private string NormalizeFilenameForSubtitleComparison(string filename)
{
// Try to account for sloppy file naming
filename = filename.Replace("_", string.Empty);
filename = filename.Replace(" ", string.Empty);
2016-07-21 02:55:35 +00:00
// can't normalize this due to languages such as pt-br
//filename = filename.Replace("-", string.Empty);
//filename = filename.Replace(".", string.Empty);
return filename;
}
2014-05-16 19:16:29 +00:00
}
}