Create interface for providing store id to plugins (btcpayserver#3910)

woutersamaey · Jun 29, 2022 · b8f1c0d · b8f1c0d
1 parent ed1f249
commit b8f1c0d
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 0 deletions.
diff --git a/BTCPayServer.Abstractions/Contracts/IScopeProvider.cs b/BTCPayServer.Abstractions/Contracts/IScopeProvider.cs
@@ -0,0 +1,7 @@
+#nullable enable
+namespace BTCPayServer.Abstractions.Contracts;
+
+public interface IScopeProvider
+{
+    string? GetCurrentStoreId();
+}
diff --git a/BTCPayServer/Extensions.cs b/BTCPayServer/Extensions.cs
@@ -22,6 +22,7 @@
 using BTCPayServer.Models.StoreViewModels;
 using BTCPayServer.Payments;
 using BTCPayServer.Payments.Bitcoin;
+using BTCPayServer.Security;
 using BTCPayServer.Services.Invoices;
 using BTCPayServer.Services.Wallets;
 using Microsoft.AspNetCore.Http;
@@ -257,6 +258,11 @@ private static void SetCurrentStoreId(this HttpContext ctx, string storeId)
             }
         }
 
+        public static string GetCurrentStoreId(this HttpContext ctx)
+        {
+            return ctx.GetImplicitStoreId() ?? ctx.GetUserPrefsCookie()?.CurrentStoreId;
+        }
+
         public static StoreData GetStoreData(this HttpContext ctx)
         {
             return ctx.Items.TryGet("BTCPAY.STOREDATA") as StoreData;

diff --git a/BTCPayServer/Hosting/BTCPayServerServices.cs b/BTCPayServer/Hosting/BTCPayServerServices.cs
@@ -98,6 +98,7 @@ public static IServiceCollection AddBTCPayServer(this IServiceCollection service
             if (configuration.SupportChain("yec") || configuration.SupportChain("zec"))
                 services.AddZcashLike();
 #endif
+            services.AddScoped<IScopeProvider, ScopeProvider>();
             services.TryAddSingleton<SettingsRepository>();
             services.TryAddSingleton<ISettingsRepository>(provider => provider.GetService<SettingsRepository>());
             services.TryAddSingleton<IStoreRepository>(provider => provider.GetService<StoreRepository>());

diff --git a/BTCPayServer/Services/Stores/ScopeProvider.cs b/BTCPayServer/Services/Stores/ScopeProvider.cs
@@ -0,0 +1,19 @@
+#nullable enable
+using BTCPayServer.Abstractions.Contracts;
+using Microsoft.AspNetCore.Http;
+
+namespace BTCPayServer.Services.Stores;
+
+public class ScopeProvider : IScopeProvider
+{
+    private readonly IHttpContextAccessor _httpContextAccessor;
+
+    public ScopeProvider(IHttpContextAccessor httpContextAccessor)
+    {
+        _httpContextAccessor = httpContextAccessor;
+    }
+    public string? GetCurrentStoreId()
+    { 
+        return _httpContextAccessor.HttpContext.GetStoreData()?.Id;
+    }
+}