using System; using System.Collections.Generic; using System.Linq; using System.Text; using BTCPayServer.Payments; using BTCPayServer.Services.Rates; using NBXplorer; using Newtonsoft.Json.Linq; namespace BTCPayServer.Data { public static class StoreDataExtensions { #pragma warning disable CS0618 public static PaymentMethodId GetDefaultPaymentId(this StoreData storeData, BTCPayNetworkProvider networks) { PaymentMethodId[] paymentMethodIds = storeData.GetEnabledPaymentIds(networks); PaymentMethodId.TryParse(storeData.DefaultCrypto, out var defaultPaymentId); var chosen = paymentMethodIds.FirstOrDefault(f => f == defaultPaymentId) ?? paymentMethodIds.FirstOrDefault(f => f.CryptoCode == defaultPaymentId?.CryptoCode) ?? paymentMethodIds.FirstOrDefault(); return chosen; } public static PaymentMethodId[] GetEnabledPaymentIds(this StoreData storeData, BTCPayNetworkProvider networks) { var excludeFilter = storeData.GetStoreBlob().GetExcludedPaymentMethods(); var paymentMethodIds = storeData.GetSupportedPaymentMethods(networks).Select(p => p.PaymentId) .Where(a => !excludeFilter.Match(a)) .OrderByDescending(a => a.CryptoCode == "BTC") .ThenBy(a => a.CryptoCode) .ThenBy(a => a.PaymentType == PaymentTypes.LightningLike ? 1 : 0) .ToArray(); return paymentMethodIds; } public static void SetDefaultPaymentId(this StoreData storeData, PaymentMethodId defaultPaymentId) { storeData.DefaultCrypto = defaultPaymentId?.ToString(); } #pragma warning restore CS0618 public static StoreBlob GetStoreBlob(this StoreData storeData) { var result = storeData.StoreBlob == null ? new StoreBlob() : new Serializer(null).ToObject(Encoding.UTF8.GetString(storeData.StoreBlob)); if (result.PreferredExchange == null) result.PreferredExchange = CoinGeckoRateProvider.CoinGeckoName; if (result.Hints == null) result.Hints = new StoreBlob.StoreHints(); return result; } public static bool SetStoreBlob(this StoreData storeData, StoreBlob storeBlob) { var original = new Serializer(null).ToString(storeData.GetStoreBlob()); var newBlob = new Serializer(null).ToString(storeBlob); if (original == newBlob) return false; storeData.StoreBlob = Encoding.UTF8.GetBytes(newBlob); return true; } public static IEnumerable GetSupportedPaymentMethods(this StoreData storeData, BTCPayNetworkProvider networks) { if (storeData == null) throw new ArgumentNullException(nameof(storeData)); #pragma warning disable CS0618 bool btcReturned = false; if (!string.IsNullOrEmpty(storeData.DerivationStrategies)) { JObject strategies = JObject.Parse(storeData.DerivationStrategies); foreach (var strat in strategies.Properties()) { if (!PaymentMethodId.TryParse(strat.Name, out var paymentMethodId)) { continue; } var network = networks.GetNetwork(paymentMethodId.CryptoCode); if (network != null) { if (network == networks.BTC && paymentMethodId.PaymentType == PaymentTypes.BTCLike && btcReturned) continue; if (strat.Value.Type == JTokenType.Null) continue; yield return paymentMethodId.PaymentType.DeserializeSupportedPaymentMethod(network, strat.Value); } } } #pragma warning restore CS0618 } public static void SetSupportedPaymentMethod(this StoreData storeData, ISupportedPaymentMethod supportedPaymentMethod) { storeData.SetSupportedPaymentMethod(null, supportedPaymentMethod); } /// /// Set or remove a new supported payment method for the store /// /// The paymentMethodId /// The payment method, or null to remove public static void SetSupportedPaymentMethod(this StoreData storeData, PaymentMethodId paymentMethodId, ISupportedPaymentMethod supportedPaymentMethod) { if (supportedPaymentMethod != null && paymentMethodId != null && paymentMethodId != supportedPaymentMethod.PaymentId) { throw new InvalidOperationException("Incoherent arguments, this should never happen"); } if (supportedPaymentMethod == null && paymentMethodId == null) throw new ArgumentException($"{nameof(supportedPaymentMethod)} or {nameof(paymentMethodId)} should be specified"); if (supportedPaymentMethod != null && paymentMethodId == null) { paymentMethodId = supportedPaymentMethod.PaymentId; } #pragma warning disable CS0618 JObject strategies = string.IsNullOrEmpty(storeData.DerivationStrategies) ? new JObject() : JObject.Parse(storeData.DerivationStrategies); bool existing = false; foreach (var strat in strategies.Properties().ToList()) { var stratId = PaymentMethodId.Parse(strat.Name); if (stratId == paymentMethodId) { if (supportedPaymentMethod == null) { strat.Remove(); } else { strat.Value = PaymentMethodExtensions.Serialize(supportedPaymentMethod); } existing = true; break; } } if (!existing && supportedPaymentMethod != null) strategies.Add(new JProperty(supportedPaymentMethod.PaymentId.ToString(), PaymentMethodExtensions.Serialize(supportedPaymentMethod))); storeData.DerivationStrategies = strategies.ToString(); #pragma warning restore CS0618 } } }