MediaBrowser.Controller/Dto/DtoOptions.cs


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

#pragma warning disable CS1591

using System;
using System.Collections.Generic;
using System.Linq;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Querying;

namespace MediaBrowser.Controller.Dto
{
    public class DtoOptions
    {
        private static readonly ItemFields[] DefaultExcludedFields = new[]
        {
            ItemFields.SeasonUserData,
            ItemFields.RefreshState
        };

        private static readonly ImageType[] AllImageTypes = Enum.GetValues<ImageType>();

        private static readonly ItemFields[] AllItemFields = Enum.GetValues<ItemFields>()
            .Except(DefaultExcludedFields)
            .ToArray();

        public DtoOptions()
            : this(true)
        {
        }

        public DtoOptions(bool allFields)
        {
            ImageTypeLimit = int.MaxValue;
            EnableImages = true;
            EnableUserData = true;
            AddCurrentProgram = true;

            Fields = allFields ? AllItemFields : [];
            ImageTypes = AllImageTypes;
        }

        public IReadOnlyList<ItemFields> Fields { get; set; }

        public IReadOnlyList<ImageType> ImageTypes { get; set; }

        public int ImageTypeLimit { get; set; }

        public bool EnableImages { get; set; }

        public bool AddProgramRecordingInfo { get; set; }

        public bool EnableUserData { get; set; }

        public bool AddCurrentProgram { get; set; }

        public bool ContainsField(ItemFields field)
            => Fields.Contains(field);

        public int GetImageLimit(ImageType type)
        {
            if (EnableImages && ImageTypes.Contains(type))
            {
                return ImageTypeLimit;
            }

            return 0;
        }
    }
}