mirror of
https://github.com/btcpayserver/btcpayserver.git
synced 2024-11-20 10:40:29 +01:00
522 lines
20 KiB
C#
522 lines
20 KiB
C#
using BTCPayServer.Authentication;
|
|
using BTCPayServer.Data;
|
|
using BTCPayServer.Models;
|
|
using BTCPayServer.Models.StoreViewModels;
|
|
using BTCPayServer.Services.Stores;
|
|
using BTCPayServer.Services.Wallets;
|
|
using Microsoft.AspNetCore.Authentication.Cookies;
|
|
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.Mvc.Rendering;
|
|
using NBitcoin;
|
|
using NBitcoin.DataEncoders;
|
|
using NBitpayClient;
|
|
using NBXplorer.DerivationStrategy;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace BTCPayServer.Controllers
|
|
{
|
|
[Route("stores")]
|
|
[Authorize(AuthenticationSchemes = "Identity.Application")]
|
|
[Authorize(Policy = "CanAccessStore")]
|
|
[AutoValidateAntiforgeryToken]
|
|
public class StoresController : Controller
|
|
{
|
|
public StoresController(
|
|
StoreRepository repo,
|
|
TokenRepository tokenRepo,
|
|
UserManager<ApplicationUser> userManager,
|
|
AccessTokenController tokenController,
|
|
BTCPayWallet wallet,
|
|
BTCPayNetworkProvider networkProvider,
|
|
ExplorerClientProvider explorerProvider,
|
|
IHostingEnvironment env)
|
|
{
|
|
_Repo = repo;
|
|
_TokenRepository = tokenRepo;
|
|
_UserManager = userManager;
|
|
_TokenController = tokenController;
|
|
_Wallet = wallet;
|
|
_Env = env;
|
|
_NetworkProvider = networkProvider;
|
|
_ExplorerProvider = explorerProvider;
|
|
}
|
|
BTCPayNetworkProvider _NetworkProvider;
|
|
private ExplorerClientProvider _ExplorerProvider;
|
|
BTCPayWallet _Wallet;
|
|
AccessTokenController _TokenController;
|
|
StoreRepository _Repo;
|
|
TokenRepository _TokenRepository;
|
|
UserManager<ApplicationUser> _UserManager;
|
|
IHostingEnvironment _Env;
|
|
|
|
[TempData]
|
|
public string StatusMessage
|
|
{
|
|
get; set;
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("create")]
|
|
public IActionResult CreateStore()
|
|
{
|
|
return View();
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("create")]
|
|
public async Task<IActionResult> CreateStore(CreateStoreViewModel vm)
|
|
{
|
|
if (!ModelState.IsValid)
|
|
{
|
|
return View(vm);
|
|
}
|
|
var store = await _Repo.CreateStore(GetUserId(), vm.Name);
|
|
CreatedStoreId = store.Id;
|
|
StatusMessage = "Store successfully created";
|
|
return RedirectToAction(nameof(ListStores));
|
|
}
|
|
|
|
public string CreatedStoreId
|
|
{
|
|
get; set;
|
|
}
|
|
|
|
[HttpGet]
|
|
public async Task<IActionResult> ListStores()
|
|
{
|
|
StoresViewModel result = new StoresViewModel();
|
|
result.StatusMessage = StatusMessage;
|
|
var stores = await _Repo.GetStoresByUserId(GetUserId());
|
|
var balances = stores
|
|
.Select(s => s.GetDerivationStrategies(_NetworkProvider)
|
|
.Select(async ss => (await _Wallet.GetBalance(ss)).ToString() + " " + ss.Network.CryptoCode))
|
|
.ToArray();
|
|
|
|
await Task.WhenAll(balances.SelectMany(_ => _));
|
|
for (int i = 0; i < stores.Length; i++)
|
|
{
|
|
var store = stores[i];
|
|
result.Stores.Add(new StoresViewModel.StoreViewModel()
|
|
{
|
|
Id = store.Id,
|
|
Name = store.StoreName,
|
|
WebSite = store.StoreWebsite,
|
|
Balances = balances[i].Select(t => t.Result).ToArray()
|
|
});
|
|
}
|
|
return View(result);
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("{storeId}/delete")]
|
|
public async Task<IActionResult> DeleteStore(string storeId)
|
|
{
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
return View("Confirm", new ConfirmModel()
|
|
{
|
|
Title = "Delete store " + store.StoreName,
|
|
Description = "This store will still be accessible to users sharing it",
|
|
Action = "Delete"
|
|
});
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("{storeId}/delete")]
|
|
public async Task<IActionResult> DeleteStorePost(string storeId)
|
|
{
|
|
var userId = GetUserId();
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
await _Repo.RemoveStore(storeId, userId);
|
|
StatusMessage = "Store removed successfully";
|
|
return RedirectToAction(nameof(ListStores));
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("{storeId}")]
|
|
public async Task<IActionResult> UpdateStore(string storeId)
|
|
{
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
|
|
var storeBlob = store.GetStoreBlob();
|
|
var vm = new StoreViewModel();
|
|
vm.Id = store.Id;
|
|
vm.StoreName = store.StoreName;
|
|
vm.StoreWebsite = store.StoreWebsite;
|
|
vm.NetworkFee = !storeBlob.NetworkFeeDisabled;
|
|
vm.SpeedPolicy = store.SpeedPolicy;
|
|
AddDerivationSchemes(store, vm);
|
|
vm.StatusMessage = StatusMessage;
|
|
vm.MonitoringExpiration = storeBlob.MonitoringExpiration;
|
|
return View(vm);
|
|
}
|
|
|
|
private void AddDerivationSchemes(StoreData store, StoreViewModel vm)
|
|
{
|
|
var strategies = store
|
|
.GetDerivationStrategies(_NetworkProvider)
|
|
.ToDictionary(s => s.Network.CryptoCode);
|
|
foreach (var explorerProvider in _ExplorerProvider.GetAll())
|
|
{
|
|
if (strategies.TryGetValue(explorerProvider.Item1.CryptoCode, out DerivationStrategy strat))
|
|
{
|
|
vm.DerivationSchemes.Add(new StoreViewModel.DerivationScheme()
|
|
{
|
|
Crypto = explorerProvider.Item1.CryptoCode,
|
|
Value = strat.DerivationStrategyBase.ToString()
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("{storeId}/derivations")]
|
|
public async Task<IActionResult> AddDerivationScheme(string storeId, string selectedScheme = null)
|
|
{
|
|
selectedScheme = selectedScheme ?? "BTC";
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
DerivationSchemeViewModel vm = new DerivationSchemeViewModel();
|
|
vm.SetCryptoCurrencies(_ExplorerProvider, selectedScheme);
|
|
return View(vm);
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("{storeId}/derivations")]
|
|
public async Task<IActionResult> AddDerivationScheme(string storeId, DerivationSchemeViewModel vm, string command, string selectedScheme = null)
|
|
{
|
|
selectedScheme = selectedScheme ?? "BTC";
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
|
|
var network = vm.CryptoCurrency == null ? null : _ExplorerProvider.GetNetwork(vm.CryptoCurrency);
|
|
vm.SetCryptoCurrencies(_ExplorerProvider, selectedScheme);
|
|
if (network == null)
|
|
{
|
|
ModelState.AddModelError(nameof(vm.CryptoCurrency), "Invalid network");
|
|
return View(vm);
|
|
}
|
|
|
|
if (command == "Save")
|
|
{
|
|
try
|
|
{
|
|
if (!string.IsNullOrEmpty(vm.DerivationScheme))
|
|
{
|
|
var strategy = ParseDerivationStrategy(vm.DerivationScheme, vm.DerivationSchemeFormat, network);
|
|
await _Wallet.TrackAsync(strategy);
|
|
vm.DerivationScheme = strategy.ToString();
|
|
}
|
|
store.SetDerivationStrategy(network, vm.DerivationScheme);
|
|
}
|
|
catch
|
|
{
|
|
ModelState.AddModelError(nameof(vm.DerivationScheme), "Invalid Derivation Scheme");
|
|
return View(vm);
|
|
}
|
|
|
|
await _Repo.UpdateStore(store);
|
|
StatusMessage = $"Derivation scheme for {vm.CryptoCurrency} has been modified.";
|
|
return RedirectToAction(nameof(UpdateStore), new { storeId = storeId });
|
|
}
|
|
else
|
|
{
|
|
if (!string.IsNullOrEmpty(vm.DerivationScheme))
|
|
{
|
|
try
|
|
{
|
|
var scheme = ParseDerivationStrategy(vm.DerivationScheme, vm.DerivationSchemeFormat, network);
|
|
var line = scheme.DerivationStrategyBase.GetLineFor(DerivationFeature.Deposit);
|
|
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
var address = line.Derive((uint)i);
|
|
vm.AddressSamples.Add((line.Path.Derive((uint)i).ToString(), address.ScriptPubKey.GetDestinationAddress(scheme.Network.NBitcoinNetwork).ToString()));
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
ModelState.AddModelError(nameof(vm.DerivationScheme), "Invalid Derivation Scheme");
|
|
}
|
|
}
|
|
return View(vm);
|
|
}
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("{storeId}")]
|
|
public async Task<IActionResult> UpdateStore(string storeId, StoreViewModel model)
|
|
{
|
|
if (!ModelState.IsValid)
|
|
{
|
|
return View(model);
|
|
}
|
|
var store = await _Repo.FindStore(storeId, GetUserId());
|
|
if (store == null)
|
|
return NotFound();
|
|
AddDerivationSchemes(store, model);
|
|
|
|
bool needUpdate = false;
|
|
if (store.SpeedPolicy != model.SpeedPolicy)
|
|
{
|
|
needUpdate = true;
|
|
store.SpeedPolicy = model.SpeedPolicy;
|
|
}
|
|
if (store.StoreName != model.StoreName)
|
|
{
|
|
needUpdate = true;
|
|
store.StoreName = model.StoreName;
|
|
}
|
|
if (store.StoreWebsite != model.StoreWebsite)
|
|
{
|
|
needUpdate = true;
|
|
store.StoreWebsite = model.StoreWebsite;
|
|
}
|
|
|
|
var blob = store.GetStoreBlob();
|
|
blob.NetworkFeeDisabled = !model.NetworkFee;
|
|
blob.MonitoringExpiration = model.MonitoringExpiration;
|
|
|
|
if (store.SetStoreBlob(blob))
|
|
{
|
|
needUpdate = true;
|
|
}
|
|
|
|
if (needUpdate)
|
|
{
|
|
await _Repo.UpdateStore(store);
|
|
StatusMessage = "Store successfully updated";
|
|
}
|
|
|
|
return RedirectToAction(nameof(UpdateStore), new
|
|
{
|
|
storeId = storeId
|
|
});
|
|
}
|
|
|
|
private DerivationStrategy ParseDerivationStrategy(string derivationScheme, string format, BTCPayNetwork network)
|
|
{
|
|
if (format == "Electrum")
|
|
{
|
|
//Unsupported Electrum
|
|
//var p2wsh_p2sh = 0x295b43fU;
|
|
//var p2wsh = 0x2aa7ed3U;
|
|
Dictionary<uint, string[]> electrumMapping = new Dictionary<uint, string[]>();
|
|
//Source https://github.com/spesmilo/electrum/blob/9edffd17542de5773e7284a8c8a2673c766bb3c3/lib/bitcoin.py
|
|
var standard = 0x0488b21eU;
|
|
electrumMapping.Add(standard, new[] { "legacy" });
|
|
var p2wpkh_p2sh = 0x049d7cb2U;
|
|
electrumMapping.Add(p2wpkh_p2sh, new string[] { "p2sh" });
|
|
var p2wpkh = 0x4b24746U;
|
|
electrumMapping.Add(p2wpkh, new string[] { });
|
|
|
|
var data = Encoders.Base58Check.DecodeData(derivationScheme);
|
|
if (data.Length < 4)
|
|
throw new FormatException("data.Length < 4");
|
|
var prefix = Utils.ToUInt32(data, false);
|
|
if (!electrumMapping.TryGetValue(prefix, out string[] labels))
|
|
throw new FormatException("!electrumMapping.TryGetValue(prefix, out string[] labels)");
|
|
var standardPrefix = Utils.ToBytes(network.NBitcoinNetwork == Network.Main ? 0x0488b21eU : 0x043587cf, false);
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
data[i] = standardPrefix[i];
|
|
|
|
derivationScheme = new BitcoinExtPubKey(Encoders.Base58Check.EncodeData(data), network.NBitcoinNetwork).ToString();
|
|
foreach (var label in labels)
|
|
{
|
|
derivationScheme = derivationScheme + $"-[{label}]";
|
|
}
|
|
}
|
|
|
|
return DerivationStrategy.Parse(new DerivationStrategyFactory(network.NBitcoinNetwork).Parse(derivationScheme).ToString(), network);
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("{storeId}/Tokens")]
|
|
public async Task<IActionResult> ListTokens(string storeId)
|
|
{
|
|
var model = new TokensViewModel();
|
|
var tokens = await _TokenRepository.GetTokensByStoreIdAsync(storeId);
|
|
model.StatusMessage = StatusMessage;
|
|
model.Tokens = tokens.Select(t => new TokenViewModel()
|
|
{
|
|
Facade = t.Facade,
|
|
Label = t.Label,
|
|
SIN = t.SIN,
|
|
Id = t.Value
|
|
}).ToArray();
|
|
return View(model);
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("/api-tokens")]
|
|
[Route("{storeId}/Tokens/Create")]
|
|
public async Task<IActionResult> CreateToken(string storeId, CreateTokenViewModel model)
|
|
{
|
|
if (!ModelState.IsValid)
|
|
{
|
|
return View(model);
|
|
}
|
|
model.Label = model.Label ?? String.Empty;
|
|
if (storeId == null) // Permissions are not checked by Policy if the storeId is not passed by url
|
|
{
|
|
storeId = model.StoreId;
|
|
var userId = GetUserId();
|
|
if (userId == null)
|
|
return Unauthorized();
|
|
var store = await _Repo.FindStore(storeId, userId);
|
|
if (store == null)
|
|
return Unauthorized();
|
|
}
|
|
|
|
var tokenRequest = new TokenRequest()
|
|
{
|
|
Facade = model.Facade,
|
|
Label = model.Label,
|
|
Id = model.PublicKey == null ? null : NBitpayClient.Extensions.BitIdExtensions.GetBitIDSIN(new PubKey(model.PublicKey))
|
|
};
|
|
|
|
string pairingCode = null;
|
|
if (model.PublicKey == null)
|
|
{
|
|
tokenRequest.PairingCode = await _TokenRepository.CreatePairingCodeAsync();
|
|
await _TokenRepository.UpdatePairingCode(new PairingCodeEntity()
|
|
{
|
|
Id = tokenRequest.PairingCode,
|
|
Facade = model.Facade,
|
|
Label = model.Label,
|
|
});
|
|
await _TokenRepository.PairWithStoreAsync(tokenRequest.PairingCode, storeId);
|
|
pairingCode = tokenRequest.PairingCode;
|
|
}
|
|
else
|
|
{
|
|
pairingCode = ((DataWrapper<List<PairingCodeResponse>>)await _TokenController.Tokens(tokenRequest)).Data[0].PairingCode;
|
|
}
|
|
|
|
return RedirectToAction(nameof(RequestPairing), new
|
|
{
|
|
pairingCode = pairingCode,
|
|
selectedStore = storeId
|
|
});
|
|
}
|
|
|
|
[HttpGet]
|
|
[Route("/api-tokens")]
|
|
[Route("{storeId}/Tokens/Create")]
|
|
public async Task<IActionResult> CreateToken(string storeId)
|
|
{
|
|
var userId = GetUserId();
|
|
if (string.IsNullOrWhiteSpace(userId))
|
|
return Unauthorized();
|
|
var model = new CreateTokenViewModel();
|
|
model.Facade = "merchant";
|
|
ViewBag.HidePublicKey = storeId == null;
|
|
ViewBag.ShowStores = storeId == null;
|
|
ViewBag.ShowMenu = storeId != null;
|
|
model.StoreId = storeId;
|
|
if (storeId == null)
|
|
{
|
|
model.Stores = new SelectList(await _Repo.GetStoresByUserId(userId), nameof(StoreData.Id), nameof(StoreData.StoreName), storeId);
|
|
}
|
|
|
|
return View(model);
|
|
}
|
|
|
|
|
|
[HttpPost]
|
|
[Route("{storeId}/Tokens/Delete")]
|
|
public async Task<IActionResult> DeleteToken(string storeId, string tokenId)
|
|
{
|
|
var token = await _TokenRepository.GetToken(tokenId);
|
|
if (token == null ||
|
|
token.StoreId != storeId ||
|
|
!await _TokenRepository.DeleteToken(tokenId))
|
|
StatusMessage = "Failure to revoke this token";
|
|
else
|
|
StatusMessage = "Token revoked";
|
|
return RedirectToAction(nameof(ListTokens));
|
|
}
|
|
|
|
|
|
[HttpGet]
|
|
[Route("/api-access-request")]
|
|
public async Task<IActionResult> RequestPairing(string pairingCode, string selectedStore = null)
|
|
{
|
|
var pairing = await _TokenRepository.GetPairingAsync(pairingCode);
|
|
if (pairing == null)
|
|
{
|
|
StatusMessage = "Unknown pairing code";
|
|
return RedirectToAction(nameof(ListStores));
|
|
}
|
|
else
|
|
{
|
|
var stores = await _Repo.GetStoresByUserId(GetUserId());
|
|
return View(new PairingModel()
|
|
{
|
|
Id = pairing.Id,
|
|
Facade = pairing.Facade,
|
|
Label = pairing.Label,
|
|
SIN = pairing.SIN ?? "Server-Initiated Pairing",
|
|
SelectedStore = selectedStore ?? stores.FirstOrDefault()?.Id,
|
|
Stores = stores.Select(s => new PairingModel.StoreViewModel()
|
|
{
|
|
Id = s.Id,
|
|
Name = string.IsNullOrEmpty(s.StoreName) ? s.Id : s.StoreName
|
|
}).ToArray()
|
|
});
|
|
}
|
|
}
|
|
|
|
[HttpPost]
|
|
[Route("api-access-request")]
|
|
public async Task<IActionResult> Pair(string pairingCode, string selectedStore)
|
|
{
|
|
if (pairingCode == null)
|
|
return NotFound();
|
|
var store = await _Repo.FindStore(selectedStore, GetUserId());
|
|
var pairing = await _TokenRepository.GetPairingAsync(pairingCode);
|
|
if (store == null || pairing == null)
|
|
return NotFound();
|
|
|
|
var pairingResult = await _TokenRepository.PairWithStoreAsync(pairingCode, store.Id);
|
|
if (pairingResult == PairingResult.Complete || pairingResult == PairingResult.Partial)
|
|
{
|
|
StatusMessage = "Pairing is successfull";
|
|
if (pairingResult == PairingResult.Partial)
|
|
StatusMessage = "Server initiated pairing code: " + pairingCode;
|
|
return RedirectToAction(nameof(ListTokens), new
|
|
{
|
|
storeId = store.Id
|
|
});
|
|
}
|
|
else
|
|
{
|
|
StatusMessage = $"Pairing failed ({pairingResult})";
|
|
return RedirectToAction(nameof(ListTokens), new
|
|
{
|
|
storeId = store.Id
|
|
});
|
|
}
|
|
}
|
|
|
|
private string GetUserId()
|
|
{
|
|
return _UserManager.GetUserId(User);
|
|
}
|
|
}
|
|
}
|