Craig Raw
3 years ago
11 changed files with 205 additions and 169 deletions
@ -0,0 +1,177 @@ |
|||||
|
package com.sparrowwallet.sparrow.whirlpool; |
||||
|
|
||||
|
import com.google.common.eventbus.Subscribe; |
||||
|
import com.google.common.net.HostAndPort; |
||||
|
import com.samourai.whirlpool.client.wallet.WhirlpoolEventService; |
||||
|
import com.sparrowwallet.drongo.Network; |
||||
|
import com.sparrowwallet.drongo.wallet.MixConfig; |
||||
|
import com.sparrowwallet.drongo.wallet.Wallet; |
||||
|
import com.sparrowwallet.sparrow.AppServices; |
||||
|
import com.sparrowwallet.sparrow.WalletTabData; |
||||
|
import com.sparrowwallet.sparrow.event.*; |
||||
|
import com.sparrowwallet.sparrow.io.Config; |
||||
|
import com.sparrowwallet.sparrow.io.Storage; |
||||
|
import com.sparrowwallet.sparrow.net.TorService; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
import java.util.NoSuchElementException; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
public class WhirlpoolServices { |
||||
|
private static final Logger log = LoggerFactory.getLogger(WhirlpoolServices.class); |
||||
|
|
||||
|
private final Map<String, Whirlpool> whirlpoolMap = new HashMap<>(); |
||||
|
|
||||
|
public Whirlpool getWhirlpool(Wallet wallet) { |
||||
|
Wallet masterWallet = wallet.isMasterWallet() ? wallet : wallet.getMasterWallet(); |
||||
|
for(Map.Entry<Wallet, Storage> entry : AppServices.get().getOpenWallets().entrySet()) { |
||||
|
if(entry.getKey() == masterWallet) { |
||||
|
return whirlpoolMap.get(entry.getValue().getWalletId(entry.getKey())); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
public Whirlpool getWhirlpool(String walletId) { |
||||
|
Whirlpool whirlpool = whirlpoolMap.get(walletId); |
||||
|
if(whirlpool == null) { |
||||
|
HostAndPort torProxy = AppServices.isTorRunning() ? HostAndPort.fromParts("localhost", TorService.PROXY_PORT) : (Config.get().getProxyServer() == null || Config.get().getProxyServer().isEmpty() || !Config.get().isUseProxy() ? null : HostAndPort.fromString(Config.get().getProxyServer())); |
||||
|
whirlpool = new Whirlpool(Network.get(), torProxy); |
||||
|
whirlpoolMap.put(walletId, whirlpool); |
||||
|
} |
||||
|
|
||||
|
return whirlpool; |
||||
|
} |
||||
|
|
||||
|
private void startAllWhirlpool() { |
||||
|
for(Map.Entry<String, Whirlpool> entry : whirlpoolMap.entrySet().stream().filter(entry -> entry.getValue().hasWallet() && !entry.getValue().isStarted()).collect(Collectors.toList())) { |
||||
|
Wallet wallet = AppServices.get().getWallet(entry.getKey()); |
||||
|
Whirlpool whirlpool = entry.getValue(); |
||||
|
startWhirlpool(wallet, whirlpool); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void startWhirlpool(Wallet wallet, Whirlpool whirlpool) { |
||||
|
if(wallet.getMasterMixConfig().getMixOnStartup() != Boolean.FALSE) { |
||||
|
try { |
||||
|
String mixToWalletId = getWhirlpoolMixToWalletId(wallet.getMasterMixConfig()); |
||||
|
whirlpool.setMixToWallet(mixToWalletId, wallet.getMasterMixConfig().getMinMixes()); |
||||
|
} catch(NoSuchElementException e) { |
||||
|
AppServices.showWarningDialog("Mix to wallet not open", wallet.getName() + " is configured to mix to " + wallet.getMasterMixConfig().getMixToWalletName() + ", but this wallet is not open. Mix to wallets are required to be open to avoid address reuse."); |
||||
|
} |
||||
|
|
||||
|
Whirlpool.StartupService startupService = new Whirlpool.StartupService(whirlpool); |
||||
|
startupService.setOnFailed(workerStateEvent -> { |
||||
|
log.error("Failed to start whirlpool", workerStateEvent.getSource().getException()); |
||||
|
}); |
||||
|
startupService.start(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void shutdownAllWhirlpool() { |
||||
|
for(Whirlpool whirlpool : whirlpoolMap.values().stream().filter(Whirlpool::isStarted).collect(Collectors.toList())) { |
||||
|
Whirlpool.ShutdownService shutdownService = new Whirlpool.ShutdownService(whirlpool); |
||||
|
shutdownService.setOnFailed(workerStateEvent -> { |
||||
|
log.error("Failed to shutdown whirlpool", workerStateEvent.getSource().getException()); |
||||
|
}); |
||||
|
shutdownService.start(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public String getWhirlpoolMixToWalletId(MixConfig mixConfig) { |
||||
|
if(mixConfig == null || mixConfig.getMixToWalletFile() == null || mixConfig.getMixToWalletName() == null) { |
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
return AppServices.get().getOpenWallets().entrySet().stream() |
||||
|
.filter(entry -> entry.getValue().getWalletFile().equals(mixConfig.getMixToWalletFile()) && entry.getKey().getName().equals(mixConfig.getMixToWalletName())) |
||||
|
.map(entry -> entry.getValue().getWalletId(entry.getKey())) |
||||
|
.findFirst().orElseThrow(); |
||||
|
} |
||||
|
|
||||
|
public Whirlpool getWhirlpoolForMixToWallet(String walletId) { |
||||
|
return whirlpoolMap.values().stream().filter(whirlpool -> walletId.equals(whirlpool.getMixToWalletId())).findFirst().orElse(null); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void newConnection(ConnectionEvent event) { |
||||
|
startAllWhirlpool(); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void disconnection(DisconnectionEvent event) { |
||||
|
shutdownAllWhirlpool(); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void walletOpened(WalletOpenedEvent event) { |
||||
|
String walletId = event.getStorage().getWalletId(event.getWallet()); |
||||
|
Whirlpool whirlpool = whirlpoolMap.get(walletId); |
||||
|
if(whirlpool != null && !whirlpool.isStarted() && AppServices.isConnected()) { |
||||
|
startWhirlpool(event.getWallet(), whirlpool); |
||||
|
} |
||||
|
|
||||
|
Whirlpool mixFromWhirlpool = whirlpoolMap.entrySet().stream() |
||||
|
.filter(entry -> event.getStorage().getWalletFile().equals(AppServices.get().getWallet(entry.getKey()).getMasterMixConfig().getMixToWalletFile())) |
||||
|
.map(Map.Entry::getValue).findFirst().orElse(null); |
||||
|
|
||||
|
if(mixFromWhirlpool != null) { |
||||
|
mixFromWhirlpool.setMixToWallet(walletId, AppServices.get().getWallet(mixFromWhirlpool.getWalletId()).getMasterMixConfig().getMinMixes()); |
||||
|
if(mixFromWhirlpool.isStarted()) { |
||||
|
Whirlpool.RestartService restartService = new Whirlpool.RestartService(mixFromWhirlpool); |
||||
|
restartService.setOnFailed(workerStateEvent -> { |
||||
|
log.error("Failed to restart whirlpool", workerStateEvent.getSource().getException()); |
||||
|
}); |
||||
|
restartService.start(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void walletTabsClosed(WalletTabsClosedEvent event) { |
||||
|
for(WalletTabData walletTabData : event.getClosedWalletTabData()) { |
||||
|
String walletId = walletTabData.getStorage().getWalletId(walletTabData.getWallet()); |
||||
|
Whirlpool whirlpool = whirlpoolMap.remove(walletId); |
||||
|
if(whirlpool != null) { |
||||
|
if(whirlpool.isStarted()) { |
||||
|
Whirlpool.ShutdownService shutdownService = new Whirlpool.ShutdownService(whirlpool); |
||||
|
shutdownService.setOnSucceeded(workerStateEvent -> { |
||||
|
WhirlpoolEventService.getInstance().unregister(whirlpool); |
||||
|
}); |
||||
|
shutdownService.setOnFailed(workerStateEvent -> { |
||||
|
log.error("Failed to shutdown whirlpool", workerStateEvent.getSource().getException()); |
||||
|
}); |
||||
|
shutdownService.start(); |
||||
|
} else { |
||||
|
//Ensure http clients are shutdown
|
||||
|
whirlpool.shutdown(); |
||||
|
WhirlpoolEventService.getInstance().unregister(whirlpool); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
Whirlpool mixToWhirlpool = getWhirlpoolForMixToWallet(walletId); |
||||
|
if(mixToWhirlpool != null && event.getClosedWalletTabData().stream().noneMatch(walletTabData1 -> walletTabData1.getWalletForm().getWalletId().equals(mixToWhirlpool.getWalletId()))) { |
||||
|
mixToWhirlpool.setMixToWallet(null, null); |
||||
|
if(mixToWhirlpool.isStarted()) { |
||||
|
Whirlpool.RestartService restartService = new Whirlpool.RestartService(mixToWhirlpool); |
||||
|
restartService.setOnFailed(workerStateEvent -> { |
||||
|
log.error("Failed to restart whirlpool", workerStateEvent.getSource().getException()); |
||||
|
}); |
||||
|
restartService.start(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void walletHistoryChanged(WalletHistoryChangedEvent event) { |
||||
|
Whirlpool whirlpool = getWhirlpool(event.getWallet()); |
||||
|
if(whirlpool != null) { |
||||
|
whirlpool.refreshUtxos(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue