Plugin: Add IGlobalCheckoutModelExtension

jessmzp · Dec 9, 2024 · 4f63f08 · 4f63f08
1 parent 1214367
commit 4f63f08
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/BTCPayServer/Controllers/UIInvoiceController.UI.cs b/BTCPayServer/Controllers/UIInvoiceController.UI.cs
@@ -936,10 +936,15 @@ _ when PaymentTypes.LN.GetPaymentMethodId(_NetworkProvider.DefaultNetwork.Crypto
             var expiration = TimeSpan.FromSeconds(model.ExpirationSeconds);
             model.TimeLeft = expiration.PrettyPrint();
 
-            if (_paymentModelExtensions.TryGetValue(paymentMethodId, out var extension) &&
-                    _handlers.TryGetValue(paymentMethodId, out var h))
+            if (_handlers.TryGetValue(paymentMethodId, out var h))
             {
-                extension.ModifyCheckoutModel(new CheckoutModelContext(model, store, storeBlob, invoice, Url, prompt, h));
+                var ctx = new CheckoutModelContext(model, store, storeBlob, invoice, Url, prompt, h);
+                if (_paymentModelExtensions.TryGetValue(paymentMethodId, out var extension))
+                {
+                    extension.ModifyCheckoutModel(ctx);
+                }
+                foreach (var global in GlobalCheckoutModelExtensions)
+                    global.ModifyCheckoutModel(ctx);
             }
             return model;
         }

diff --git a/BTCPayServer/Controllers/UIInvoiceController.cs b/BTCPayServer/Controllers/UIInvoiceController.cs
@@ -69,6 +69,7 @@ public partial class UIInvoiceController : Controller
         private readonly UriResolver _uriResolver;
 
         public WebhookSender WebhookNotificationManager { get; }
+        public IEnumerable<IGlobalCheckoutModelExtension> GlobalCheckoutModelExtensions { get; }
         public IStringLocalizer StringLocalizer { get; }
 
         public UIInvoiceController(
@@ -99,6 +100,7 @@ public UIInvoiceController(
             IAuthorizationService authorizationService,
             TransactionLinkProviders transactionLinkProviders,
             Dictionary<PaymentMethodId, ICheckoutModelExtension> paymentModelExtensions,
+            IEnumerable<IGlobalCheckoutModelExtension> globalCheckoutModelExtensions,
             IStringLocalizer stringLocalizer,
             PrettyNameProvider prettyName)
         {
@@ -124,6 +126,7 @@ public UIInvoiceController(
             _authorizationService = authorizationService;
             _transactionLinkProviders = transactionLinkProviders;
             _paymentModelExtensions = paymentModelExtensions;
+            GlobalCheckoutModelExtensions = globalCheckoutModelExtensions;
             _prettyName = prettyName;
             _fileService = fileService;
             _uriResolver = uriResolver;

diff --git a/BTCPayServer/Payments/IGlobalCheckoutModelExtension.cs b/BTCPayServer/Payments/IGlobalCheckoutModelExtension.cs
@@ -0,0 +1,10 @@
+namespace BTCPayServer.Payments
+{
+    /// <summary>
+    /// <see cref="ModifyCheckoutModel"/> will always run when showing the checkout page
+    /// </summary>
+    public interface IGlobalCheckoutModelExtension
+    {
+        void ModifyCheckoutModel(CheckoutModelContext context);
+    }
+}