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