namespace Microsoft.WindowsAzure.Storage.Table
{
    using System;
    using System.Collections.Generic;
    using System.Text.Json;

    /// <summary>
    /// Useful method for working with <see cref="TableEntity">table entities</see>.
    /// </summary>
    /// <remarks>Source: https://bitbucket.org/snippets/just_dmitry/Lrg7gB/entitypropertyextensionscs.</remarks>
    public static class EntityPropertyExtensions
    {
        public static readonly JsonSerializerOptions JsonOptions = new JsonSerializerOptions
        {
            AllowTrailingCommas = true,
            Encoder = System.Text.Encodings.Web.JavaScriptEncoder.Create(System.Text.Unicode.UnicodeRanges.All),
            IgnoreNullValues = true,
            IgnoreReadOnlyProperties = true,
            PropertyNameCaseInsensitive = true,
            ReadCommentHandling = JsonCommentHandling.Skip,
            WriteIndented = false,
        };

        public static EntityProperty? TryGet(this IDictionary<string, EntityProperty> properties, string name)
        {
            properties = properties ?? throw new ArgumentNullException(nameof(properties));

            properties.TryGetValue(name, out var prop);
            return prop;
        }

        public static void Set(this IDictionary<string, EntityProperty> properties, string name, string value)
        {
            properties = properties ?? throw new ArgumentNullException(nameof(properties));

            if (!string.IsNullOrEmpty(value))
            {
                properties[name] = new EntityProperty(value);
            }
        }

        public static T? TryGetDeserialized<T>(this IDictionary<string, EntityProperty> properties, string name)
            where T : class
        {
            properties = properties ?? throw new ArgumentNullException(nameof(properties));

            if (!properties.TryGetValue(name, out var prop))
            {
                return default;
            }

            var val = prop.StringValue;

            if (string.IsNullOrEmpty(val))
            {
                return default;
            }

            return JsonSerializer.Deserialize<T>(val, JsonOptions);
        }

        public static void SetSerialized<T>(this IDictionary<string, EntityProperty> properties, string name, T? value)
            where T : class
        {
            properties = properties ?? throw new ArgumentNullException(nameof(properties));

            if (value != null)
            {
                properties.Add(name, new EntityProperty(JsonSerializer.Serialize(value, JsonOptions)));
            }
        }
    }
}