btcpayserver/BTCPayServer/Data/Payouts/PayoutExtensions.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using BTCPayServer.Client.Models;
using BTCPayServer.Payments;
using BTCPayServer.Payouts;
using BTCPayServer.Services;
using Microsoft.EntityFrameworkCore;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;

namespace BTCPayServer.Data
{
    public static class PayoutExtensions
    {
        public static async Task<PayoutData> GetPayout(this DbSet<PayoutData> payouts, string payoutId, string storeId, bool includePullPayment = false, bool includeStore = false)
        {
            IQueryable<PayoutData> query = payouts;
            if (includePullPayment)
                query = query.Include(p => p.PullPaymentData);
            if (includeStore)
                query = query.Include(p => p.StoreData);
            var payout = await query.Where(p => p.Id == payoutId &&
                                                p.StoreDataId == storeId).FirstOrDefaultAsync();
            if (payout is null)
                return null;
            return payout;
        }

        public static PayoutMethodId GetPayoutMethodId(this PayoutData data)
        {
            return PayoutMethodId.TryParse(data.PayoutMethodId, out var pmi) ? pmi : null;
        }

        public static string GetPayoutSource(this PayoutData data, BTCPayNetworkJsonSerializerSettings jsonSerializerSettings)
        {
            var ppBlob = data.PullPaymentData?.GetBlob();
            var payoutBlob = data.GetBlob(jsonSerializerSettings);
            return payoutBlob.Metadata?.TryGetValue("source", StringComparison.InvariantCultureIgnoreCase, out var source) is true
                ? source.Value<string>()
                : ppBlob?.Name ?? data.PullPaymentDataId;
        }

        public static PayoutBlob GetBlob(this PayoutData data, BTCPayNetworkJsonSerializerSettings serializers)
        {
            var result =  JsonConvert.DeserializeObject<PayoutBlob>(data.Blob, serializers.GetSerializer(data.GetPayoutMethodId()));
            result.Metadata ??= new JObject();
            return result;
        }
        public static void SetBlob(this PayoutData data, PayoutBlob blob, BTCPayNetworkJsonSerializerSettings serializers)
        {
            data.Blob = JsonConvert.SerializeObject(blob, serializers.GetSerializer(data.GetPayoutMethodId())).ToString();
        }

        public static JObject GetProofBlobJson(this PayoutData data)
        {
            return data?.Proof is null ? null : JObject.Parse(data.Proof);
        }
        public static void SetProofBlob(this PayoutData data, IPayoutProof blob, JsonSerializerSettings settings)
        {
            if (blob is null)
            {
                data.Proof = null;
                return;
            }

            data.SetProofBlob(settings is null
                ? JObject.FromObject(blob)
                : JObject.FromObject(blob, JsonSerializer.Create(settings)));
        }
        public static void SetProofBlob(this PayoutData data, JObject blob)
        {
            if (blob is null)
            {
                data.Proof = null;
                return;
            }
            // We only update the property if the bytes actually changed, this prevent from hammering the DB too much
            if (!JToken.DeepEquals(blob, data.Proof is null ? null : JObject.Parse(data.Proof)))
            {
                data.Proof = blob.ToString(Formatting.None);
            }
        }

        public static HashSet<PayoutMethodId> GetSupportedPayoutMethods(this PayoutMethodHandlerDictionary payoutHandlers, StoreData storeData)
        {
            return payoutHandlers.Where(handler => handler.IsSupported(storeData)).Select(p => p.PayoutMethodId).ToHashSet();
        }
    }
}