Craig Raw
4 years ago
38 changed files with 583 additions and 529 deletions
@ -0,0 +1,423 @@ |
|||||
|
package com.sparrowwallet.sparrow; |
||||
|
|
||||
|
import com.google.common.eventbus.Subscribe; |
||||
|
import com.sparrowwallet.drongo.address.Address; |
||||
|
import com.sparrowwallet.drongo.protocol.Transaction; |
||||
|
import com.sparrowwallet.drongo.uri.BitcoinURI; |
||||
|
import com.sparrowwallet.drongo.wallet.KeystoreSource; |
||||
|
import com.sparrowwallet.drongo.wallet.Wallet; |
||||
|
import com.sparrowwallet.sparrow.event.*; |
||||
|
import com.sparrowwallet.sparrow.io.Config; |
||||
|
import com.sparrowwallet.sparrow.io.Device; |
||||
|
import com.sparrowwallet.sparrow.io.Hwi; |
||||
|
import com.sparrowwallet.sparrow.io.Storage; |
||||
|
import com.sparrowwallet.sparrow.net.ElectrumServer; |
||||
|
import com.sparrowwallet.sparrow.net.ExchangeSource; |
||||
|
import com.sparrowwallet.sparrow.net.MempoolRateSize; |
||||
|
import com.sparrowwallet.sparrow.net.VersionCheckService; |
||||
|
import javafx.application.Platform; |
||||
|
import javafx.beans.property.BooleanProperty; |
||||
|
import javafx.beans.property.SimpleBooleanProperty; |
||||
|
import javafx.beans.value.ChangeListener; |
||||
|
import javafx.beans.value.ObservableValue; |
||||
|
import javafx.concurrent.ScheduledService; |
||||
|
import javafx.concurrent.Worker; |
||||
|
import javafx.scene.control.Alert; |
||||
|
import javafx.scene.image.Image; |
||||
|
import javafx.scene.text.Font; |
||||
|
import javafx.stage.Stage; |
||||
|
import javafx.stage.Window; |
||||
|
import javafx.util.Duration; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
|
||||
|
import java.io.File; |
||||
|
import java.time.LocalDateTime; |
||||
|
import java.time.ZoneId; |
||||
|
import java.time.temporal.ChronoUnit; |
||||
|
import java.util.*; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
public class AppServices { |
||||
|
private static final Logger log = LoggerFactory.getLogger(AppServices.class); |
||||
|
|
||||
|
private static final int SERVER_PING_PERIOD = 1 * 60 * 1000; |
||||
|
private static final int ENUMERATE_HW_PERIOD = 30 * 1000; |
||||
|
private static final int RATES_PERIOD = 5 * 60 * 1000; |
||||
|
private static final int VERSION_CHECK_PERIOD_HOURS = 24; |
||||
|
private static final ExchangeSource DEFAULT_EXCHANGE_SOURCE = ExchangeSource.COINGECKO; |
||||
|
private static final Currency DEFAULT_FIAT_CURRENCY = Currency.getInstance("USD"); |
||||
|
|
||||
|
private static AppServices INSTANCE; |
||||
|
|
||||
|
private final MainApp application; |
||||
|
|
||||
|
private static final BooleanProperty onlineProperty = new SimpleBooleanProperty(false); |
||||
|
|
||||
|
private ExchangeSource.RatesService ratesService; |
||||
|
|
||||
|
private final ElectrumServer.ConnectionService connectionService; |
||||
|
|
||||
|
private Hwi.ScheduledEnumerateService deviceEnumerateService; |
||||
|
|
||||
|
private VersionCheckService versionCheckService; |
||||
|
|
||||
|
private static Integer currentBlockHeight; |
||||
|
|
||||
|
private static Map<Integer, Double> targetBlockFeeRates; |
||||
|
|
||||
|
private static final Map<Date, Set<MempoolRateSize>> mempoolHistogram = new TreeMap<>(); |
||||
|
|
||||
|
private static Double minimumRelayFeeRate; |
||||
|
|
||||
|
private static CurrencyRate fiatCurrencyExchangeRate; |
||||
|
|
||||
|
private static List<Device> devices; |
||||
|
|
||||
|
private static final Map<Address, BitcoinURI> payjoinURIs = new HashMap<>(); |
||||
|
|
||||
|
private final ChangeListener<Boolean> onlineServicesListener = new ChangeListener<>() { |
||||
|
@Override |
||||
|
public void changed(ObservableValue<? extends Boolean> observable, Boolean oldValue, Boolean online) { |
||||
|
Config.get().setMode(online ? Mode.ONLINE : Mode.OFFLINE); |
||||
|
if(online) { |
||||
|
restartService(connectionService); |
||||
|
|
||||
|
if(ratesService.getExchangeSource() != ExchangeSource.NONE) { |
||||
|
restartService(ratesService); |
||||
|
} |
||||
|
|
||||
|
if(Config.get().isCheckNewVersions()) { |
||||
|
restartService(versionCheckService); |
||||
|
} |
||||
|
} else { |
||||
|
connectionService.cancel(); |
||||
|
ratesService.cancel(); |
||||
|
versionCheckService.cancel(); |
||||
|
} |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
private void restartService(ScheduledService<?> service) { |
||||
|
if(service.getState() == Worker.State.CANCELLED) { |
||||
|
service.reset(); |
||||
|
} |
||||
|
|
||||
|
if(!service.isRunning()) { |
||||
|
service.start(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public AppServices(MainApp application) { |
||||
|
this.application = application; |
||||
|
|
||||
|
EventManager.get().register(this); |
||||
|
|
||||
|
Config config = Config.get(); |
||||
|
connectionService = createConnectionService(); |
||||
|
if(config.getMode() == Mode.ONLINE && config.getElectrumServer() != null && !config.getElectrumServer().isEmpty()) { |
||||
|
connectionService.start(); |
||||
|
} |
||||
|
|
||||
|
ExchangeSource source = config.getExchangeSource() != null ? config.getExchangeSource() : DEFAULT_EXCHANGE_SOURCE; |
||||
|
Currency currency = config.getFiatCurrency() != null ? config.getFiatCurrency() : DEFAULT_FIAT_CURRENCY; |
||||
|
ratesService = createRatesService(source, currency); |
||||
|
if(config.getMode() == Mode.ONLINE && source != ExchangeSource.NONE) { |
||||
|
ratesService.start(); |
||||
|
} |
||||
|
|
||||
|
versionCheckService = createVersionCheckService(); |
||||
|
if(config.getMode() == Mode.ONLINE && config.isCheckNewVersions()) { |
||||
|
versionCheckService.start(); |
||||
|
} |
||||
|
|
||||
|
onlineProperty.addListener(onlineServicesListener); |
||||
|
} |
||||
|
|
||||
|
private ElectrumServer.ConnectionService createConnectionService() { |
||||
|
ElectrumServer.ConnectionService connectionService = new ElectrumServer.ConnectionService(); |
||||
|
connectionService.setPeriod(new Duration(SERVER_PING_PERIOD)); |
||||
|
connectionService.setRestartOnFailure(true); |
||||
|
|
||||
|
EventManager.get().register(connectionService); |
||||
|
connectionService.statusProperty().addListener((observable, oldValue, newValue) -> { |
||||
|
if(connectionService.isRunning()) { |
||||
|
EventManager.get().post(new StatusEvent(newValue)); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
connectionService.setOnSucceeded(successEvent -> { |
||||
|
onlineProperty.removeListener(onlineServicesListener); |
||||
|
onlineProperty.setValue(true); |
||||
|
onlineProperty.addListener(onlineServicesListener); |
||||
|
|
||||
|
if(connectionService.getValue() != null) { |
||||
|
EventManager.get().post(connectionService.getValue()); |
||||
|
} |
||||
|
}); |
||||
|
connectionService.setOnFailed(failEvent -> { |
||||
|
//Close connection here to create a new transport next time we try
|
||||
|
connectionService.resetConnection(); |
||||
|
|
||||
|
onlineProperty.removeListener(onlineServicesListener); |
||||
|
onlineProperty.setValue(false); |
||||
|
onlineProperty.addListener(onlineServicesListener); |
||||
|
|
||||
|
log.debug("Connection failed", failEvent.getSource().getException()); |
||||
|
EventManager.get().post(new ConnectionFailedEvent(failEvent.getSource().getException())); |
||||
|
}); |
||||
|
|
||||
|
return connectionService; |
||||
|
} |
||||
|
|
||||
|
private ExchangeSource.RatesService createRatesService(ExchangeSource exchangeSource, Currency currency) { |
||||
|
ExchangeSource.RatesService ratesService = new ExchangeSource.RatesService(exchangeSource, currency); |
||||
|
ratesService.setPeriod(new Duration(RATES_PERIOD)); |
||||
|
ratesService.setRestartOnFailure(true); |
||||
|
|
||||
|
ratesService.setOnSucceeded(successEvent -> { |
||||
|
EventManager.get().post(ratesService.getValue()); |
||||
|
}); |
||||
|
|
||||
|
return ratesService; |
||||
|
} |
||||
|
|
||||
|
private VersionCheckService createVersionCheckService() { |
||||
|
VersionCheckService versionCheckService = new VersionCheckService(); |
||||
|
versionCheckService.setDelay(Duration.seconds(10)); |
||||
|
versionCheckService.setPeriod(Duration.hours(VERSION_CHECK_PERIOD_HOURS)); |
||||
|
versionCheckService.setRestartOnFailure(true); |
||||
|
|
||||
|
versionCheckService.setOnSucceeded(successEvent -> { |
||||
|
VersionUpdatedEvent event = versionCheckService.getValue(); |
||||
|
if(event != null) { |
||||
|
EventManager.get().post(event); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
return versionCheckService; |
||||
|
} |
||||
|
|
||||
|
private Hwi.ScheduledEnumerateService createDeviceEnumerateService() { |
||||
|
Hwi.ScheduledEnumerateService enumerateService = new Hwi.ScheduledEnumerateService(null); |
||||
|
enumerateService.setPeriod(new Duration(ENUMERATE_HW_PERIOD)); |
||||
|
enumerateService.setOnSucceeded(workerStateEvent -> { |
||||
|
List<Device> devices = enumerateService.getValue(); |
||||
|
|
||||
|
//Null devices are returned if the app is currently prompting for a pin. Otherwise, the enumerate clears the pin screen
|
||||
|
if(devices != null) { |
||||
|
//If another instance of HWI is currently accessing the usb interface, HWI returns empty device models. Ignore this run if that happens
|
||||
|
List<Device> validDevices = devices.stream().filter(device -> device.getModel() != null).collect(Collectors.toList()); |
||||
|
if(validDevices.size() == devices.size()) { |
||||
|
Platform.runLater(() -> EventManager.get().post(new UsbDeviceEvent(devices))); |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
return enumerateService; |
||||
|
} |
||||
|
|
||||
|
static void initialize(MainApp application) { |
||||
|
INSTANCE = new AppServices(application); |
||||
|
} |
||||
|
|
||||
|
public static AppServices get() { |
||||
|
return INSTANCE; |
||||
|
} |
||||
|
|
||||
|
public MainApp getApplication() { |
||||
|
return application; |
||||
|
} |
||||
|
|
||||
|
public static boolean isOnline() { |
||||
|
return onlineProperty.get(); |
||||
|
} |
||||
|
|
||||
|
public static BooleanProperty onlineProperty() { |
||||
|
return onlineProperty; |
||||
|
} |
||||
|
|
||||
|
public static Integer getCurrentBlockHeight() { |
||||
|
return currentBlockHeight; |
||||
|
} |
||||
|
|
||||
|
public static Map<Integer, Double> getTargetBlockFeeRates() { |
||||
|
return targetBlockFeeRates; |
||||
|
} |
||||
|
|
||||
|
public static Map<Date, Set<MempoolRateSize>> getMempoolHistogram() { |
||||
|
return mempoolHistogram; |
||||
|
} |
||||
|
|
||||
|
private void addMempoolRateSizes(Set<MempoolRateSize> rateSizes) { |
||||
|
LocalDateTime dateMinute = LocalDateTime.now().truncatedTo(ChronoUnit.MINUTES); |
||||
|
if(mempoolHistogram.isEmpty()) { |
||||
|
mempoolHistogram.put(Date.from(dateMinute.minusMinutes(1).atZone(ZoneId.systemDefault()).toInstant()), rateSizes); |
||||
|
} |
||||
|
|
||||
|
mempoolHistogram.put(Date.from(dateMinute.atZone(ZoneId.systemDefault()).toInstant()), rateSizes); |
||||
|
} |
||||
|
|
||||
|
public static Double getMinimumRelayFeeRate() { |
||||
|
return minimumRelayFeeRate == null ? Transaction.DEFAULT_MIN_RELAY_FEE : minimumRelayFeeRate; |
||||
|
} |
||||
|
|
||||
|
public static CurrencyRate getFiatCurrencyExchangeRate() { |
||||
|
return fiatCurrencyExchangeRate; |
||||
|
} |
||||
|
|
||||
|
public static List<Device> getDevices() { |
||||
|
return devices == null ? new ArrayList<>() : devices; |
||||
|
} |
||||
|
|
||||
|
public static BitcoinURI getPayjoinURI(Address address) { |
||||
|
return payjoinURIs.get(address); |
||||
|
} |
||||
|
|
||||
|
public static void addPayjoinURI(BitcoinURI bitcoinURI) { |
||||
|
if(bitcoinURI.getPayjoinUrl() == null) { |
||||
|
throw new IllegalArgumentException("Not a payjoin URI"); |
||||
|
} |
||||
|
payjoinURIs.put(bitcoinURI.getAddress(), bitcoinURI); |
||||
|
} |
||||
|
|
||||
|
public static void showErrorDialog(String title, String content) { |
||||
|
Alert alert = new Alert(Alert.AlertType.ERROR); |
||||
|
setStageIcon(alert.getDialogPane().getScene().getWindow()); |
||||
|
alert.getDialogPane().getScene().getStylesheets().add(AppServices.class.getResource("general.css").toExternalForm()); |
||||
|
alert.setTitle(title); |
||||
|
alert.setHeaderText(title); |
||||
|
alert.setContentText(content); |
||||
|
alert.showAndWait(); |
||||
|
} |
||||
|
|
||||
|
public static void setStageIcon(Window window) { |
||||
|
Stage stage = (Stage)window; |
||||
|
stage.getIcons().add(new Image(AppServices.class.getResourceAsStream("/image/sparrow.png"))); |
||||
|
|
||||
|
if(stage.getScene() != null && Config.get().getTheme() == Theme.DARK) { |
||||
|
stage.getScene().getStylesheets().add(AppServices.class.getResource("darktheme.css").toExternalForm()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public static Font getMonospaceFont() { |
||||
|
return Font.font("Roboto Mono", 13); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void newConnection(ConnectionEvent event) { |
||||
|
currentBlockHeight = event.getBlockHeight(); |
||||
|
targetBlockFeeRates = event.getTargetBlockFeeRates(); |
||||
|
addMempoolRateSizes(event.getMempoolRateSizes()); |
||||
|
minimumRelayFeeRate = event.getMinimumRelayFeeRate(); |
||||
|
String banner = event.getServerBanner(); |
||||
|
String status = "Connected to " + Config.get().getElectrumServer() + " at height " + event.getBlockHeight(); |
||||
|
EventManager.get().post(new StatusEvent(status)); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void connectionFailed(ConnectionFailedEvent event) { |
||||
|
String reason = event.getException().getCause() != null ? event.getException().getCause().getMessage() : event.getException().getMessage(); |
||||
|
String status = "Connection error: " + reason; |
||||
|
EventManager.get().post(new StatusEvent(status)); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void usbDevicesFound(UsbDeviceEvent event) { |
||||
|
devices = Collections.unmodifiableList(event.getDevices()); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void newBlock(NewBlockEvent event) { |
||||
|
currentBlockHeight = event.getHeight(); |
||||
|
String status = "Updating to new block height " + event.getHeight(); |
||||
|
EventManager.get().post(new StatusEvent(status)); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void feesUpdated(FeeRatesUpdatedEvent event) { |
||||
|
targetBlockFeeRates = event.getTargetBlockFeeRates(); |
||||
|
addMempoolRateSizes(event.getMempoolRateSizes()); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void feeRateSourceChanged(FeeRatesSourceChangedEvent event) { |
||||
|
ElectrumServer.FeeRatesService feeRatesService = new ElectrumServer.FeeRatesService(); |
||||
|
feeRatesService.setOnSucceeded(workerStateEvent -> { |
||||
|
EventManager.get().post(feeRatesService.getValue()); |
||||
|
}); |
||||
|
//Perform once-off fee rates retrieval to immediately change displayed rates
|
||||
|
feeRatesService.start(); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void fiatCurrencySelected(FiatCurrencySelectedEvent event) { |
||||
|
ratesService.cancel(); |
||||
|
|
||||
|
if(Config.get().getMode() != Mode.OFFLINE && event.getExchangeSource() != ExchangeSource.NONE) { |
||||
|
ratesService = createRatesService(event.getExchangeSource(), event.getCurrency()); |
||||
|
ratesService.start(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void exchangeRatesUpdated(ExchangeRatesUpdatedEvent event) { |
||||
|
fiatCurrencyExchangeRate = event.getCurrencyRate(); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void versionCheckStatus(VersionCheckStatusEvent event) { |
||||
|
versionCheckService.cancel(); |
||||
|
|
||||
|
if(Config.get().getMode() != Mode.OFFLINE && event.isEnabled()) { |
||||
|
versionCheckService = createVersionCheckService(); |
||||
|
versionCheckService.start(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void openWallets(OpenWalletsEvent event) { |
||||
|
List<File> walletFiles = event.getWalletsMap().values().stream().map(Storage::getWalletFile).collect(Collectors.toList()); |
||||
|
//TODO: Handle multiple windows
|
||||
|
Config.get().setRecentWalletFiles(walletFiles); |
||||
|
|
||||
|
boolean usbWallet = false; |
||||
|
for(Map.Entry<Wallet, Storage> entry : event.getWalletsMap().entrySet()) { |
||||
|
Wallet wallet = entry.getKey(); |
||||
|
Storage storage = entry.getValue(); |
||||
|
|
||||
|
if(!storage.getWalletFile().exists() || wallet.containsSource(KeystoreSource.HW_USB)) { |
||||
|
usbWallet = true; |
||||
|
|
||||
|
if(deviceEnumerateService == null) { |
||||
|
deviceEnumerateService = createDeviceEnumerateService(); |
||||
|
} |
||||
|
|
||||
|
if(deviceEnumerateService.getState() == Worker.State.CANCELLED) { |
||||
|
deviceEnumerateService.reset(); |
||||
|
} |
||||
|
|
||||
|
if(!deviceEnumerateService.isRunning()) { |
||||
|
deviceEnumerateService.start(); |
||||
|
} |
||||
|
|
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if(!usbWallet && deviceEnumerateService != null && deviceEnumerateService.isRunning()) { |
||||
|
deviceEnumerateService.cancel(); |
||||
|
EventManager.get().post(new UsbDeviceEvent(Collections.emptyList())); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void requestConnect(RequestConnectEvent event) { |
||||
|
onlineProperty.set(true); |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void requestDisconnect(RequestDisconnectEvent event) { |
||||
|
onlineProperty.set(false); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue