aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Providers/BoxSets/BoxSetMetadataService.cs
blob: 062519f9d5bec8c85d6fd0e890d445dcdb4d1d68 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Localization;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
using MediaBrowser.Providers.Manager;
using System;
using System.Collections.Generic;
using System.Linq;

namespace MediaBrowser.Providers.BoxSets
{
    public class BoxSetMetadataService : MetadataService<BoxSet, BoxSetInfo>
    {
        private readonly ILocalizationManager _iLocalizationManager;

        public BoxSetMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IProviderRepository providerRepo, IFileSystem fileSystem, IUserDataManager userDataManager, ILocalizationManager iLocalizationManager) : base(serverConfigurationManager, logger, providerManager, providerRepo, fileSystem, userDataManager)
        {
            _iLocalizationManager = iLocalizationManager;
        }

        /// <summary>
        /// Merges the specified source.
        /// </summary>
        /// <param name="source">The source.</param>
        /// <param name="target">The target.</param>
        /// <param name="lockedFields">The locked fields.</param>
        /// <param name="replaceData">if set to <c>true</c> [replace data].</param>
        /// <param name="mergeMetadataSettings">if set to <c>true</c> [merge metadata settings].</param>
        protected override void MergeData(BoxSet source, BoxSet target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
        {
            ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);

            if (mergeMetadataSettings)
            {
                var list = source.LinkedChildren.Where(i => i.Type != LinkedChildType.Manual).ToList();

                list.AddRange(target.LinkedChildren.Where(i => i.Type == LinkedChildType.Manual));

                target.LinkedChildren = list;
            }

            if (replaceData || target.Shares.Count == 0)
            {
                target.Shares = source.Shares;
            }
        }

        protected override ItemUpdateType BeforeSave(BoxSet item)
        {
            var updateType = base.BeforeSave(item);

            if (!item.LockedFields.Contains(MetadataFields.OfficialRating))
            {
                var currentOfficialRating = item.OfficialRating;

                // Gather all possible ratings
                var ratings = item.RecursiveChildren
                    .Concat(item.GetLinkedChildren())
                    .Where(i => i is Movie || i is Series)
                    .Select(i => i.OfficialRating)
                    .Where(i => !string.IsNullOrEmpty(i))
                    .Distinct(StringComparer.OrdinalIgnoreCase)
                    .Select(i => new Tuple<string, int?>(i, _iLocalizationManager.GetRatingLevel(i)))
                    .OrderBy(i => i.Item2 ?? 1000)
                    .Select(i => i.Item1);

                item.OfficialRating = ratings.FirstOrDefault() ?? item.OfficialRating;

                if (!string.Equals(currentOfficialRating ?? string.Empty, item.OfficialRating ?? string.Empty,
                    StringComparison.OrdinalIgnoreCase))
                {
                    updateType = updateType | ItemUpdateType.MetadataDownload;
                }
            }

            return updateType;
        }
    }
}