mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
283 lines
10 KiB
C#
283 lines
10 KiB
C#
using MediaBrowser.Common.Extensions;
|
|
using MediaBrowser.Common.MediaInfo;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
using MediaBrowser.Controller.Persistence;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.Serialization;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Providers.MediaInfo
|
|
{
|
|
/// <summary>
|
|
/// Extracts audio information using ffprobe
|
|
/// </summary>
|
|
public class FFProbeAudioInfoProvider : BaseFFProbeProvider<Audio>
|
|
{
|
|
private readonly IItemRepository _itemRepo;
|
|
|
|
public FFProbeAudioInfoProvider(ILogManager logManager, IServerConfigurationManager configurationManager, IMediaEncoder mediaEncoder, IJsonSerializer jsonSerializer, IItemRepository itemRepo)
|
|
: base(logManager, configurationManager, mediaEncoder, jsonSerializer)
|
|
{
|
|
_itemRepo = itemRepo;
|
|
}
|
|
|
|
public override async Task<bool> FetchAsync(BaseItem item, bool force, BaseProviderInfo providerInfo, CancellationToken cancellationToken)
|
|
{
|
|
var myItem = (Audio)item;
|
|
|
|
OnPreFetch(myItem, null);
|
|
|
|
var result = await GetMediaInfo(item, null, cancellationToken).ConfigureAwait(false);
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
NormalizeFFProbeResult(result);
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
await Fetch(myItem, cancellationToken, result).ConfigureAwait(false);
|
|
|
|
SetLastRefreshed(item, DateTime.UtcNow, providerInfo);
|
|
|
|
return true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Fetches the specified audio.
|
|
/// </summary>
|
|
/// <param name="audio">The audio.</param>
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
/// <param name="data">The data.</param>
|
|
/// <returns>Task.</returns>
|
|
protected Task Fetch(Audio audio, CancellationToken cancellationToken, MediaInfoResult data)
|
|
{
|
|
var internalStreams = data.streams ?? new MediaStreamInfo[] { };
|
|
|
|
var mediaStreams = internalStreams.Select(s => GetMediaStream(s, data.format))
|
|
.Where(i => i != null)
|
|
.ToList();
|
|
|
|
audio.HasEmbeddedImage = mediaStreams.Any(i => i.Type == MediaStreamType.Video);
|
|
|
|
// Get the first audio stream
|
|
var stream = internalStreams.FirstOrDefault(s => s.codec_type.Equals("audio", StringComparison.OrdinalIgnoreCase));
|
|
|
|
if (stream != null)
|
|
{
|
|
// Get duration from stream properties
|
|
var duration = stream.duration;
|
|
|
|
// If it's not there go into format properties
|
|
if (string.IsNullOrEmpty(duration))
|
|
{
|
|
duration = data.format.duration;
|
|
}
|
|
|
|
// If we got something, parse it
|
|
if (!string.IsNullOrEmpty(duration))
|
|
{
|
|
audio.RunTimeTicks = TimeSpan.FromSeconds(double.Parse(duration, UsCulture)).Ticks;
|
|
}
|
|
}
|
|
|
|
if (data.format.tags != null)
|
|
{
|
|
FetchDataFromTags(audio, data.format.tags);
|
|
}
|
|
|
|
return _itemRepo.SaveMediaStreams(audio.Id, mediaStreams, cancellationToken);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Fetches data from the tags dictionary
|
|
/// </summary>
|
|
/// <param name="audio">The audio.</param>
|
|
/// <param name="tags">The tags.</param>
|
|
private void FetchDataFromTags(Audio audio, Dictionary<string, string> tags)
|
|
{
|
|
var title = GetDictionaryValue(tags, "title");
|
|
|
|
// Only set Name if title was found in the dictionary
|
|
if (!string.IsNullOrEmpty(title))
|
|
{
|
|
audio.Name = title;
|
|
}
|
|
|
|
if (!audio.LockedFields.Contains(MetadataFields.Cast))
|
|
{
|
|
audio.People.Clear();
|
|
|
|
var composer = GetDictionaryValue(tags, "composer");
|
|
|
|
if (!string.IsNullOrWhiteSpace(composer))
|
|
{
|
|
foreach (var person in Split(composer))
|
|
{
|
|
audio.AddPerson(new PersonInfo { Name = person, Type = PersonType.Composer });
|
|
}
|
|
}
|
|
}
|
|
|
|
audio.Album = GetDictionaryValue(tags, "album");
|
|
|
|
var artist = GetDictionaryValue(tags, "artist");
|
|
|
|
if (string.IsNullOrWhiteSpace(artist))
|
|
{
|
|
audio.Artists.Clear();
|
|
}
|
|
else
|
|
{
|
|
audio.Artists = SplitArtists(artist)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
|
|
}
|
|
|
|
// Several different forms of albumartist
|
|
audio.AlbumArtist = GetDictionaryValue(tags, "albumartist") ?? GetDictionaryValue(tags, "album artist") ?? GetDictionaryValue(tags, "album_artist");
|
|
|
|
// Track number
|
|
audio.IndexNumber = GetDictionaryDiscValue(tags, "track");
|
|
|
|
// Disc number
|
|
audio.ParentIndexNumber = GetDictionaryDiscValue(tags, "disc");
|
|
|
|
audio.ProductionYear = GetDictionaryNumericValue(tags, "date");
|
|
|
|
// Several different forms of retaildate
|
|
audio.PremiereDate = GetDictionaryDateTime(tags, "retaildate") ?? GetDictionaryDateTime(tags, "retail date") ?? GetDictionaryDateTime(tags, "retail_date");
|
|
|
|
// If we don't have a ProductionYear try and get it from PremiereDate
|
|
if (audio.PremiereDate.HasValue && !audio.ProductionYear.HasValue)
|
|
{
|
|
audio.ProductionYear = audio.PremiereDate.Value.ToLocalTime().Year;
|
|
}
|
|
|
|
if (!audio.LockedFields.Contains(MetadataFields.Genres))
|
|
{
|
|
FetchGenres(audio, tags);
|
|
}
|
|
|
|
if (!audio.LockedFields.Contains(MetadataFields.Studios))
|
|
{
|
|
audio.Studios.Clear();
|
|
|
|
// There's several values in tags may or may not be present
|
|
FetchStudios(audio, tags, "organization");
|
|
FetchStudios(audio, tags, "ensemble");
|
|
FetchStudios(audio, tags, "publisher");
|
|
}
|
|
}
|
|
|
|
private readonly char[] _nameDelimiters = new[] { '/', '|', ';', '\\' };
|
|
|
|
/// <summary>
|
|
/// Splits the specified val.
|
|
/// </summary>
|
|
/// <param name="val">The val.</param>
|
|
/// <returns>System.String[][].</returns>
|
|
private IEnumerable<string> Split(string val)
|
|
{
|
|
// Only use the comma as a delimeter if there are no slashes or pipes.
|
|
// We want to be careful not to split names that have commas in them
|
|
var delimeter = _nameDelimiters.Any(i => val.IndexOf(i) != -1) ? _nameDelimiters : new[] { ',' };
|
|
|
|
return val.Split(delimeter, StringSplitOptions.RemoveEmptyEntries)
|
|
.Where(i => !string.IsNullOrWhiteSpace(i))
|
|
.Select(i => i.Trim());
|
|
}
|
|
|
|
private const string ArtistReplaceValue = " | ";
|
|
|
|
private IEnumerable<string> SplitArtists(string val)
|
|
{
|
|
val = val.Replace(" featuring ", ArtistReplaceValue, StringComparison.OrdinalIgnoreCase)
|
|
.Replace(" feat. ", ArtistReplaceValue, StringComparison.OrdinalIgnoreCase);
|
|
|
|
// Only use the comma as a delimeter if there are no slashes or pipes.
|
|
// We want to be careful not to split names that have commas in them
|
|
var delimeter = _nameDelimiters;
|
|
|
|
return val.Split(delimeter, StringSplitOptions.RemoveEmptyEntries)
|
|
.Where(i => !string.IsNullOrWhiteSpace(i))
|
|
.Select(i => i.Trim());
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the studios from the tags collection
|
|
/// </summary>
|
|
/// <param name="audio">The audio.</param>
|
|
/// <param name="tags">The tags.</param>
|
|
/// <param name="tagName">Name of the tag.</param>
|
|
private void FetchStudios(Audio audio, Dictionary<string, string> tags, string tagName)
|
|
{
|
|
var val = GetDictionaryValue(tags, tagName);
|
|
|
|
if (!string.IsNullOrEmpty(val))
|
|
{
|
|
// Sometimes the artist name is listed here, account for that
|
|
var studios = Split(val).Where(i => !audio.HasArtist(i));
|
|
|
|
foreach (var studio in studios)
|
|
{
|
|
audio.AddStudio(studio);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the genres from the tags collection
|
|
/// </summary>
|
|
/// <param name="audio">The audio.</param>
|
|
/// <param name="tags">The tags.</param>
|
|
private void FetchGenres(Audio audio, Dictionary<string, string> tags)
|
|
{
|
|
var val = GetDictionaryValue(tags, "genre");
|
|
|
|
if (!string.IsNullOrEmpty(val))
|
|
{
|
|
audio.Genres.Clear();
|
|
|
|
foreach (var genre in Split(val))
|
|
{
|
|
audio.AddGenre(genre);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the disc number, which is sometimes can be in the form of '1', or '1/3'
|
|
/// </summary>
|
|
/// <param name="tags">The tags.</param>
|
|
/// <param name="tagName">Name of the tag.</param>
|
|
/// <returns>System.Nullable{System.Int32}.</returns>
|
|
private int? GetDictionaryDiscValue(Dictionary<string, string> tags, string tagName)
|
|
{
|
|
var disc = GetDictionaryValue(tags, tagName);
|
|
|
|
if (!string.IsNullOrEmpty(disc))
|
|
{
|
|
disc = disc.Split('/')[0];
|
|
|
|
int num;
|
|
|
|
if (int.TryParse(disc, out num))
|
|
{
|
|
return num;
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|
|
|
|
}
|