using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Linq; using System.Net.Http; using System.Threading.Tasks; using BTCPayServer.Rating; using ExchangeSharp; using Microsoft.Extensions.Caching.Memory; using Microsoft.Extensions.Options; using static BTCPayServer.Services.Rates.RateProviderFactory; namespace BTCPayServer.Services.Rates { public class ExchangeException { public Exception Exception { get; set; } public string ExchangeName { get; set; } } public class RateResult { public List ExchangeExceptions { get; set; } = new List(); public string Rule { get; set; } public string EvaluatedRule { get; set; } public HashSet Errors { get; set; } public BidAsk BidAsk { get; set; } public bool Cached { get; internal set; } } public class RateFetcher { private readonly RateProviderFactory _rateProviderFactory; public RateFetcher(RateProviderFactory rateProviderFactory) { _rateProviderFactory = rateProviderFactory; } public RateProviderFactory RateProviderFactory => _rateProviderFactory; public async Task FetchRate(CurrencyPair pair, RateRules rules) { return await FetchRates(new HashSet(new[] { pair }), rules).First().Value; } public Dictionary> FetchRates(HashSet pairs, RateRules rules) { if (rules == null) throw new ArgumentNullException(nameof(rules)); var fetchingRates = new Dictionary>(); var fetchingExchanges = new Dictionary>(); var consolidatedRates = new ExchangeRates(); foreach (var i in pairs.Select(p => (Pair: p, RateRule: rules.GetRuleFor(p)))) { var dependentQueries = new List>(); foreach (var requiredExchange in i.RateRule.ExchangeRates) { if (!fetchingExchanges.TryGetValue(requiredExchange.Exchange, out var fetching)) { fetching = _rateProviderFactory.QueryRates(requiredExchange.Exchange); fetchingExchanges.Add(requiredExchange.Exchange, fetching); } dependentQueries.Add(fetching); } fetchingRates.Add(i.Pair, GetRuleValue(dependentQueries, i.RateRule)); } return fetchingRates; } private async Task GetRuleValue(List> dependentQueries, RateRule rateRule) { var result = new RateResult(); result.Cached = true; foreach (var queryAsync in dependentQueries) { var query = await queryAsync; if (!query.CachedResult) result.Cached = false; result.ExchangeExceptions.AddRange(query.Exceptions); foreach (var rule in query.ExchangeRates) { rateRule.ExchangeRates.SetRate(rule.Exchange, rule.CurrencyPair, rule.BidAsk); } } rateRule.Reevaluate(); result.BidAsk = rateRule.BidAsk; result.Errors = rateRule.Errors; result.EvaluatedRule = rateRule.ToString(true); result.Rule = rateRule.ToString(false); return result; } } }