Jackett/src/Jackett.Common/Models/IndexerConfig/ConfigurationData.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Jackett.Common.Services.Interfaces;
using Jackett.Common.Utils;
using Newtonsoft.Json.Linq;

namespace Jackett.Common.Models.IndexerConfig
{
    public class ConfigurationData
    {
        private const string PASSWORD_REPLACEMENT = "|||%%PREVJACKPASSWD%%|||";
        protected Dictionary<string, ConfigurationItem> dynamics = new Dictionary<string, ConfigurationItem>(); // list for dynamic items

        public HiddenStringConfigurationItem CookieHeader { get; private set; } = new HiddenStringConfigurationItem(name:"CookieHeader");
        public HiddenStringConfigurationItem LastError { get; private set; } = new HiddenStringConfigurationItem(name:"LastError");
        public StringConfigurationItem SiteLink { get; private set; } = new StringConfigurationItem(name:"Site Link");

        public ConfigurationData()
        {

        }

        /// <summary>
        /// Loads all JSON values into the matching properties.
        /// </summary>
        public void LoadConfigDataValuesFromJson(JToken json, IProtectionService ps = null)
        {
            if (json == null)
                return;

            var jsonArray = (JArray)json;

            foreach (var item in GetAllConfigurationItems().Where(x => x.CanBeSavedToFile))
            {
                var jsonToken = jsonArray.FirstOrDefault(f => f.Value<string>("id") == item.ID);
                if (jsonToken == null)
                    continue;

                switch (item)
                {
                    case StringConfigurationItem stringItem:
                    {
                        if (HasPasswordValue(item))
                        {
                            var pw = ReadValueAs<string>(jsonToken);
                            if (pw != PASSWORD_REPLACEMENT)
                            {
                                stringItem.Value = ps != null ? ps.UnProtect(pw) : pw;
                            }
                        }
                        else
                        {
                            stringItem.Value = ReadValueAs<string>(jsonToken);
                        }
                        break;
                    }
                    case HiddenStringConfigurationItem hiddenStringItem:
                    {
                        hiddenStringItem.Value = ReadValueAs<string>(jsonToken);
                        break;
                    }
                    case BoolConfigurationItem boolItem:
                    {
                        boolItem.Value = ReadValueAs<bool>(jsonToken);
                        break;
                    }
                    case SingleSelectConfigurationItem singleSelectItem:
                    {
                        singleSelectItem.Value = ReadValueAs<string>(jsonToken);
                        break;
                    }
                    case MultiSelectConfigurationItem multiSelectItem:
                    {
                        var values = jsonToken.Value<JArray>("values");
                        if (values != null)
                        {
                            multiSelectItem.Values = values.Values<string>().ToArray();
                        }
                        break;
                    }
                    case PasswordConfigurationItem passwordItem:
                    {
                        var pw = ReadValueAs<string>(jsonToken);
                        if (pw != PASSWORD_REPLACEMENT)
                        {
                            passwordItem.Value = ps != null ? ps.UnProtect(pw) : pw;
                        }
                        break;
                    }
                }
            }
        }

        private T ReadValueAs<T>(JToken jToken) => jToken.Value<T>("value");

        private bool HasPasswordValue(ConfigurationItem item)
            => string.Equals(item.Name, "password", StringComparison.InvariantCultureIgnoreCase);

        public JToken ToJson(IProtectionService ps, bool forDisplay = true)
        {
            var jArray = new JArray();

            var configurationItems = GetConfigurationItems(forDisplay);
            foreach (var configurationItem in configurationItems)
            {
                JObject jObject = null;

                switch (configurationItem)
                {
                    case ConfigurationItemMaybePassword maybePassword:
                    {
                        // Remove this code and give each derived ConfigurationItem class its own ToJson method
                        // as soon as everyone is using PasswordConfigurationItem for passwords.
                        jObject = maybePassword.ToJson(ps);
                        break;
                    }
                    case BoolConfigurationItem boolItem:
                    {
                        jObject = boolItem.ToJson();
                        break;
                    }
                    case SingleSelectConfigurationItem singleSelectItem:
                    {
                        jObject = singleSelectItem.ToJson();
                        break;
                    }
                    case MultiSelectConfigurationItem multiSelectItem:
                    {
                        jObject = multiSelectItem.ToJson();
                        break;
                    }
                    case DisplayImageConfigurationItem imageItem:
                    {
                        jObject = imageItem.ToJson();
                        break;
                    }
                    case PasswordConfigurationItem passwordItem:
                    {
                        jObject = passwordItem.ToJson(forDisplay, ps);
                        break;
                    }
                }

                if (jObject != null)
                {
                    jArray.Add(jObject);
                }
            }
            return jArray;
        }

        private IEnumerable<ConfigurationItem> GetAllConfigurationItems()
        {
            var properties = GetType()
                .GetProperties()
                .Where(p => p.CanRead)
                .Where(p => p.PropertyType.IsSubclassOf(typeof(ConfigurationItem)))
                .Where(p => p.GetValue(this) != null)
                .Select(p => (ConfigurationItem)p.GetValue(this)).ToList();

            // remove/insert Site Link manualy to make sure it shows up first
            properties.Remove(SiteLink);
            properties.Insert(0, SiteLink);

            properties.AddRange(dynamics.Values);

            return properties;
        }

        private ConfigurationItem[] GetConfigurationItems(bool forDisplay)
            => GetAllConfigurationItems().Where(p => forDisplay ? p.CanBeShownToUser : p.CanBeSavedToFile).ToArray();

        public void AddDynamic(string ID, ConfigurationItem item) => dynamics[ID] = item;

        public ConfigurationItem GetDynamic(string ID) => dynamics.TryGetValue(ID, out var value) ? value : null;

        public ConfigurationItem GetDynamicByName(string Name)
            => dynamics.Values.FirstOrDefault(i => string.Equals(i.Name, Name, StringComparison.InvariantCultureIgnoreCase));

        public abstract class ConfigurationItem
        {
            public string ID => Name.Replace(" ", "").ToLower();
            public string Name { get; }
            public string ItemType { get; }

            public bool CanBeShownToUser { get; } = true;
            public bool CanBeSavedToFile { get; } = true;

            protected ConfigurationItem(string name, string itemType, bool canBeShownToUser = true, bool canBeSavedToFile = true)
            {
                Name = name; // TODO Error when name is null/empty/not unique/...
                ItemType = itemType;
                CanBeShownToUser = canBeShownToUser;
                CanBeSavedToFile = canBeSavedToFile;
            }

            protected JObject CreateJObject()
            {
                return new JObject
                {
                    ["id"] = ID,
                    ["type"] = ItemType.ToLower(),
                    ["name"] = Name
                };
            }
        }

        /// <summary>
        /// Remove this class when all passwords are configured to use the correct class: PasswordConfigurationItem.
        /// </summary>
        public abstract class ConfigurationItemMaybePassword : ConfigurationItem
        {
            public string Value { get; set; }

            protected ConfigurationItemMaybePassword(string name, string itemType, bool canBeShownToUser = true, bool canBeSavedToFile = true)
                : base(name, itemType, canBeShownToUser, canBeSavedToFile)
            {
            }

            public JObject ToJson(IProtectionService protectionService = null)
            {
                var jObject = CreateJObject();

                if (string.Equals(Name, "password", StringComparison.InvariantCultureIgnoreCase))
                {
                    var password = Value;
                    if (string.IsNullOrEmpty(password))
                        password = string.Empty;
                    else if (protectionService != null)
                        password = protectionService.Protect(password);
                    jObject["value"] = password;
                }
                else
                {
                    jObject["value"] = Value;
                }
                return jObject;
            }
        }

        public class StringConfigurationItem : ConfigurationItemMaybePassword
        {
            public StringConfigurationItem(string name)
                : base(name, itemType: "inputstring")
            {
            }
        }

        public class HiddenStringConfigurationItem : ConfigurationItemMaybePassword
        {
            public HiddenStringConfigurationItem(string name)
                : base(name, itemType: "hiddendata", canBeShownToUser: false)
            {
            }
        }

        public class DisplayInfoConfigurationItem : ConfigurationItemMaybePassword
        {
            public DisplayInfoConfigurationItem(string name, string value)
                : base(name, itemType: "displayinfo", canBeSavedToFile: false)
            {
                Value = value;
            }
        }

        public class BoolConfigurationItem : ConfigurationItem
        {
            public bool Value { get; set; }

            public BoolConfigurationItem(string name)
                : base(name, itemType: "inputbool")
            {
            }

            public JObject ToJson()
            {
                var jObject = CreateJObject();
                jObject["value"] = Value;
                return jObject;
            }
        }

        public class DisplayImageConfigurationItem : ConfigurationItem
        {
            public byte[] Value { get; set; }

            public DisplayImageConfigurationItem(string name)
                : base(name, itemType: "displayimage", canBeSavedToFile: false)
            {
            }

            public JObject ToJson()
            {
                var jObject = CreateJObject();

                var dataUri = DataUrlUtils.BytesToDataUrl(Value, "image/jpeg");
                jObject["value"] = dataUri;

                return jObject;
            }
        }

        public class SingleSelectConfigurationItem : ConfigurationItem
        {
            public string Value { get; set; }

            public Dictionary<string, string> Options { get; }

            public SingleSelectConfigurationItem(string name, Dictionary<string, string> options)
                : base(name, itemType: "inputselect") => Options = options;

            public JObject ToJson()
            {
                var jObject = CreateJObject();

                jObject["value"] = Value;
                jObject["options"] = new JObject();
                foreach (var option in Options)
                {
                    jObject["options"][option.Key] = option.Value;
                }

                return jObject;
            }
        }

        public class MultiSelectConfigurationItem : ConfigurationItem
        {
            public string[] Values { get; set; }

            public Dictionary<string, string> Options { get; }

            public MultiSelectConfigurationItem(string name, Dictionary<string, string> options)
                : base(name, itemType: "inputcheckbox") => Options = options;

            public JObject ToJson()
            {
                var jObject = CreateJObject();

                jObject["values"] = new JArray(Values);
                jObject["options"] = new JObject();
                foreach (var option in Options)
                {
                    jObject["options"][option.Key] = option.Value;
                }

                return jObject;
            }
        }

        public class PasswordConfigurationItem : ConfigurationItem
        {
            public string Value { get; set; }

            public PasswordConfigurationItem(string name)
                : base(name, itemType: "password")
            {
            }

            public JObject ToJson(bool forDisplay, IProtectionService protectionService = null)
            {
                var jObject = CreateJObject();

                var password = Value;
                if (string.IsNullOrEmpty(password))
                    password = string.Empty;
                else if (forDisplay)
                    password = PASSWORD_REPLACEMENT;
                else if (protectionService != null)
                    password = protectionService.Protect(password);
                jObject["value"] = password;

                return jObject;
            }
        }
    }
}