using System;

namespace SKIT.FlurlHttpClient.Wechat.TenpayV3
{
    using SKIT.FlurlHttpClient.Primitives;
    using SKIT.FlurlHttpClient.Wechat.TenpayV3.Constants;

    public static class WechatTenpayClientEventDecryptionExtensions
    {
        /// <summary>
        /// <para>反序列化得到 <see cref="WechatTenpayEvent"/> 对象。</para>
        /// </summary>
        /// <param name="client"></param>
        /// <param name="webhookJson"></param>
        /// <returns></returns>
        public static WechatTenpayEvent DeserializeEvent(this WechatTenpayClient client, string webhookJson)
        {
            if (client is null) throw new ArgumentNullException(nameof(client));
            if (webhookJson is null) throw new ArgumentNullException(webhookJson);

            return client.JsonSerializer.Deserialize<WechatTenpayEvent>(webhookJson);
        }

        /// <summary>
        /// 返回序列化并解密事件数据中被加密的通知数据。
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="client"></param>
        /// <param name="webhookEvent"></param>
        /// <returns></returns>
        public static T DecryptEventResource<T>(this WechatTenpayClient client, WechatTenpayEvent webhookEvent)
            where T : WechatTenpayEvent.Types.IDecryptedResource, new()
        {
            if (client is null) throw new ArgumentNullException(nameof(client));
            if (webhookEvent is null) throw new ArgumentNullException(nameof(webhookEvent));

            return DecryptEventResource<T>(client, webhookEvent.Resource);
        }

        /// <summary>
        /// 返回序列化并解密事件数据中被加密的通知数据。
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="client"></param>
        /// <param name="webhookEventResource"></param>
        /// <returns></returns>
        public static T DecryptEventResource<T>(this WechatTenpayClient client, WechatTenpayEvent.Types.Resource webhookEventResource)
            where T : WechatTenpayEvent.Types.IDecryptedResource, new()
        {
            if (client is null) throw new ArgumentNullException(nameof(client));
            if (webhookEventResource is null) throw new ArgumentNullException(nameof(webhookEventResource));

            string plainJson;
            switch (webhookEventResource.Algorithm)
            {
                case EncryptionAlgorithms.AEAD_AES_256_GCM:
                    {
                        try
                        {
                            plainJson = Utilities.AESUtility.DecryptWithGCM(
                                encodingKey: new EncodedString(client.Credentials.MerchantV3Secret, EncodingKinds.Literal),
                                encodingNonce: new EncodedString(webhookEventResource.Nonce, EncodingKinds.Literal),
                                encodingAssociatedData: new EncodedString(webhookEventResource.AssociatedData, EncodingKinds.Literal),
                                encodingCipher: new EncodedString(webhookEventResource.CipherText, EncodingKinds.Base64)
                            )!;
                        }
                        catch (Exception ex)
                        {
                            throw new WechatTenpayException("Failed to decrypt event resource data. Please see the inner exception for more details.", ex);
                        }
                    }
                    break;

                case EncryptionAlgorithms.AEAD_SM4_128_GCM:
                    {
                        try
                        {
                            // REF: https://pay.weixin.qq.com/doc/v3/merchant/4012076194
                            // 由于 SM4 密钥长度的限制，密钥由 APIv3 密钥通过国密 SM3 Hash 计算生成。SM4 密钥取其摘要（256bit）的前 128bit。
                            byte[] secretBytes = Utilities.SM3Utility.Hash(EncodedString.FromLiteralString(client.Credentials.MerchantV3Secret));
                            byte[] keyBytes = new byte[16];
                            Array.Copy(secretBytes, keyBytes, keyBytes.Length);

                            byte[] plainBytes = Utilities.SM4Utility.DecryptWithGCM(
                                keyBytes: keyBytes,
                                nonceBytes: EncodedString.FromLiteralString(webhookEventResource.Nonce),
                                associatedDataBytes: EncodedString.FromLiteralString(webhookEventResource.AssociatedData),
                                cipherBytes: EncodedString.FromBase64String(webhookEventResource.CipherText)
                            );
                            plainJson = EncodedString.ToLiteralString(plainBytes).Value!;
                        }
                        catch (Exception ex)
                        {
                            throw new WechatTenpayException("Failed to decrypt event resource data. Please see the inner exception for more details.", ex);
                        }
                    }
                    break;

                default:
                    {
                        throw new WechatTenpayException($"Failed to decrypt event resource data. Unsupported encryption algorithm: \"{webhookEventResource.Algorithm}\".");
                    }
            }

            return client.JsonSerializer.Deserialize<T>(plainJson);
        }
    }
}