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

169 lines
6.3 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MediaBrowser.Common.Configuration;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Library;
2014-02-20 16:37:41 +00:00
using MediaBrowser.Controller.MediaEncoding;
2013-12-06 03:39:44 +00:00
using MediaBrowser.Controller.Persistence;
using MediaBrowser.Controller.Providers;
2015-04-05 15:01:57 +00:00
using MediaBrowser.Model.Dlna;
using MediaBrowser.Model.Dto;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.MediaInfo;
2014-02-09 21:11:11 +00:00
using MediaBrowser.Model.Serialization;
2013-02-21 01:33:05 +00:00
2013-06-09 16:47:28 +00:00
namespace MediaBrowser.Providers.MediaInfo
2013-02-21 01:33:05 +00:00
{
class FFProbeAudioInfo
2013-02-21 01:33:05 +00:00
{
private readonly IMediaEncoder _mediaEncoder;
2013-12-06 03:39:44 +00:00
private readonly IItemRepository _itemRepo;
2014-02-09 21:11:11 +00:00
private readonly IApplicationPaths _appPaths;
private readonly IJsonSerializer _json;
2015-06-28 17:00:36 +00:00
private readonly ILibraryManager _libraryManager;
2018-09-12 17:26:21 +00:00
private readonly IMediaSourceManager _mediaSourceManager;
2013-12-06 03:39:44 +00:00
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
2018-09-12 17:26:21 +00:00
public FFProbeAudioInfo(IMediaSourceManager mediaSourceManager, IMediaEncoder mediaEncoder, IItemRepository itemRepo, IApplicationPaths appPaths, IJsonSerializer json, ILibraryManager libraryManager)
2013-03-02 17:59:15 +00:00
{
_mediaEncoder = mediaEncoder;
2013-12-06 03:39:44 +00:00
_itemRepo = itemRepo;
2014-02-09 21:11:11 +00:00
_appPaths = appPaths;
_json = json;
2015-06-28 17:00:36 +00:00
_libraryManager = libraryManager;
2018-09-12 17:26:21 +00:00
_mediaSourceManager = mediaSourceManager;
2013-03-02 17:59:15 +00:00
}
2018-09-12 17:26:21 +00:00
public async Task<ItemUpdateType> Probe<T>(T item, MetadataRefreshOptions options,
CancellationToken cancellationToken)
where T : Audio
2013-06-18 19:16:27 +00:00
{
2018-09-12 17:26:21 +00:00
var path = item.Path;
var protocol = item.PathProtocol ?? MediaProtocol.File;
2013-06-18 19:16:27 +00:00
2018-09-12 17:26:21 +00:00
if (!item.IsShortcut || options.EnableRemoteContentProbe)
{
if (item.IsShortcut)
{
path = item.ShortcutPath;
protocol = _mediaSourceManager.GetPathProtocol(path);
}
2013-06-18 19:16:27 +00:00
2018-09-12 17:26:21 +00:00
var result = await _mediaEncoder.GetMediaInfo(new MediaInfoRequest
{
MediaType = DlnaProfileType.Audio,
MediaSource = new MediaSourceInfo
{
Path = path,
Protocol = protocol
}
2014-03-26 21:05:31 +00:00
2018-09-12 17:26:21 +00:00
}, cancellationToken).ConfigureAwait(false);
2018-09-12 17:26:21 +00:00
cancellationToken.ThrowIfCancellationRequested();
2014-02-09 21:11:11 +00:00
2018-09-12 17:26:21 +00:00
Fetch(item, cancellationToken, result);
}
2014-02-09 21:11:11 +00:00
2018-09-12 17:26:21 +00:00
return ItemUpdateType.MetadataImport;
2013-06-18 19:16:27 +00:00
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Fetches the specified audio.
/// </summary>
/// <param name="audio">The audio.</param>
/// <param name="cancellationToken">The cancellation token.</param>
2015-04-04 19:35:29 +00:00
/// <param name="mediaInfo">The media information.</param>
2013-02-21 01:33:05 +00:00
/// <returns>Task.</returns>
protected void Fetch(Audio audio, CancellationToken cancellationToken, Model.MediaInfo.MediaInfo mediaInfo)
2013-02-21 01:33:05 +00:00
{
2014-04-22 17:25:54 +00:00
var mediaStreams = mediaInfo.MediaStreams;
2013-02-21 01:33:05 +00:00
2017-08-04 20:29:34 +00:00
audio.Container = mediaInfo.Container;
2015-04-04 19:35:29 +00:00
audio.TotalBitrate = mediaInfo.Bitrate;
2013-12-06 03:39:44 +00:00
2015-04-04 19:35:29 +00:00
audio.RunTimeTicks = mediaInfo.RunTimeTicks;
audio.Size = mediaInfo.Size;
//var extension = (Path.GetExtension(audio.Path) ?? string.Empty).TrimStart('.');
2017-08-04 20:29:34 +00:00
//audio.Container = extension;
2014-01-11 23:07:56 +00:00
FetchDataFromTags(audio, mediaInfo);
2013-12-06 03:39:44 +00:00
_itemRepo.SaveMediaStreams(audio.Id, mediaStreams, cancellationToken);
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Fetches data from the tags dictionary
/// </summary>
/// <param name="audio">The audio.</param>
2015-04-04 19:35:29 +00:00
/// <param name="data">The data.</param>
private void FetchDataFromTags(Audio audio, Model.MediaInfo.MediaInfo data)
2013-02-21 01:33:05 +00:00
{
// Only set Name if title was found in the dictionary
2016-02-18 02:55:15 +00:00
if (!string.IsNullOrEmpty(data.Name))
2013-02-21 01:33:05 +00:00
{
2016-02-18 02:55:15 +00:00
audio.Name = data.Name;
2013-02-21 01:33:05 +00:00
}
2017-10-22 21:38:03 +00:00
if (audio.SupportsPeople && !audio.LockedFields.Contains(MetadataFields.Cast))
2013-02-21 01:33:05 +00:00
{
2015-06-28 17:00:36 +00:00
var people = new List<PersonInfo>();
2013-08-29 21:00:27 +00:00
2015-04-04 19:35:29 +00:00
foreach (var person in data.People)
2013-08-04 00:59:23 +00:00
{
2015-06-28 17:00:36 +00:00
PeopleHelper.AddPerson(people, new PersonInfo
2013-02-21 01:33:05 +00:00
{
2015-04-04 19:35:29 +00:00
Name = person.Name,
Type = person.Type,
Role = person.Role
});
2015-01-27 22:45:59 +00:00
}
2015-06-28 17:00:36 +00:00
_libraryManager.UpdatePeople(audio, people);
2015-03-13 01:55:22 +00:00
}
2014-06-23 16:05:19 +00:00
2015-04-04 19:35:29 +00:00
audio.Album = data.Album;
2017-08-24 19:52:19 +00:00
audio.Artists = data.Artists;
2015-04-04 19:35:29 +00:00
audio.AlbumArtists = data.AlbumArtists;
audio.IndexNumber = data.IndexNumber;
2015-04-09 17:30:18 +00:00
audio.ParentIndexNumber = data.ParentIndexNumber;
2015-04-04 19:35:29 +00:00
audio.ProductionYear = data.ProductionYear;
audio.PremiereDate = data.PremiereDate;
2013-02-21 01:33:05 +00:00
// If we don't have a ProductionYear try and get it from PremiereDate
if (audio.PremiereDate.HasValue && !audio.ProductionYear.HasValue)
{
2013-04-12 18:22:40 +00:00
audio.ProductionYear = audio.PremiereDate.Value.ToLocalTime().Year;
2013-02-21 01:33:05 +00:00
}
2013-08-04 00:59:23 +00:00
if (!audio.LockedFields.Contains(MetadataFields.Genres))
2013-02-21 01:33:05 +00:00
{
2018-09-12 17:26:21 +00:00
audio.Genres = Array.Empty<string>();
2015-04-04 19:35:29 +00:00
foreach (var genre in data.Genres)
{
2013-09-05 22:59:07 +00:00
audio.AddGenre(genre);
}
2013-02-21 01:33:05 +00:00
}
2015-04-04 19:35:29 +00:00
if (!audio.LockedFields.Contains(MetadataFields.Studios))
2013-02-21 01:33:05 +00:00
{
audio.SetStudios(data.Studios);
2013-02-21 01:33:05 +00:00
}
2015-04-04 19:35:29 +00:00
audio.SetProviderId(MetadataProviders.MusicBrainzAlbumArtist, data.GetProviderId(MetadataProviders.MusicBrainzAlbumArtist));
audio.SetProviderId(MetadataProviders.MusicBrainzArtist, data.GetProviderId(MetadataProviders.MusicBrainzArtist));
audio.SetProviderId(MetadataProviders.MusicBrainzAlbum, data.GetProviderId(MetadataProviders.MusicBrainzAlbum));
audio.SetProviderId(MetadataProviders.MusicBrainzReleaseGroup, data.GetProviderId(MetadataProviders.MusicBrainzReleaseGroup));
audio.SetProviderId(MetadataProviders.MusicBrainzTrack, data.GetProviderId(MetadataProviders.MusicBrainzTrack));
2013-02-21 01:33:05 +00:00
}
}
2013-02-21 01:33:05 +00:00
}