btcpayserver/BTCPayServer/Payments/Lightning/LNCheckoutModelExtension.cs

#nullable enable
using BTCPayServer.Models.InvoicingModels;
using BTCPayServer.Payments.Bitcoin;
using BTCPayServer.Services.Invoices;
using BTCPayServer.Services;
using Org.BouncyCastle.Crypto.Modes.Gcm;
using System.Collections.Generic;
using System.Linq;
using Microsoft.Extensions.Localization;

namespace BTCPayServer.Payments.Lightning
{
    public class LNCheckoutModelExtension : ICheckoutModelExtension
    {
        public const string CheckoutBodyComponentName = "LightningCheckoutBody";
        private readonly DisplayFormatter _displayFormatter;
        IPaymentLinkExtension _PaymentLinkExtension;
        private readonly bool isBTC;

        public LNCheckoutModelExtension(
            PaymentMethodId paymentMethodId,
            BTCPayNetwork network,
            DisplayFormatter displayFormatter,
            IEnumerable<IPaymentLinkExtension> paymentLinkExtensions,
            IStringLocalizer stringLocalizer,
            PaymentMethodHandlerDictionary handlers)
        {
            Network = network;
            _displayFormatter = displayFormatter;
            StringLocalizer = stringLocalizer;
            Handlers = handlers;
            PaymentMethodId = paymentMethodId;
            _PaymentLinkExtension = paymentLinkExtensions.Single(p => p.PaymentMethodId == PaymentMethodId);
            isBTC = PaymentTypes.LN.GetPaymentMethodId("BTC") == paymentMethodId;
        }

        public BTCPayNetwork Network { get; }
        public IStringLocalizer StringLocalizer { get; }
        public PaymentMethodHandlerDictionary Handlers { get; }
        public PaymentMethodId PaymentMethodId { get; }

        public string DisplayName => isBTC ? StringLocalizer["Lightning"] : StringLocalizer["Lightning ({0})", Network.DisplayName];

        public string Image => Network.LightningImagePath;
        public string Badge => "⚡";
        public void ModifyCheckoutModel(CheckoutModelContext context)
        {
            if (context is not { Handler: LightningLikePaymentHandler handler })
                return;
            var paymentPrompt = context.InvoiceEntity.GetPaymentPrompt(PaymentMethodId);
            if (paymentPrompt is null)
                return;
            context.Model.CheckoutBodyComponentName = CheckoutBodyComponentName;
            context.Model.InvoiceBitcoinUrl = _PaymentLinkExtension.GetPaymentLink(context.Prompt, context.UrlHelper);
            if (context.Model.InvoiceBitcoinUrl is not null)
                context.Model.InvoiceBitcoinUrlQR = $"lightning:{context.Model.InvoiceBitcoinUrl.ToUpperInvariant()?.Substring("LIGHTNING:".Length)}";
            context.Model.PeerInfo = handler.ParsePaymentPromptDetails(paymentPrompt.Details).NodeInfo;
            if (context.StoreBlob.LightningAmountInSatoshi && isBTC)
            {
                BitcoinCheckoutModelExtension.PreparePaymentModelForAmountInSats(context.Model, paymentPrompt.Rate, _displayFormatter);
            }
        }
    }
}