Craig Raw
2 years ago
12 changed files with 339 additions and 98 deletions
@ -0,0 +1,38 @@ |
|||
package com.sparrowwallet.sparrow.terminal; |
|||
|
|||
import com.googlecode.lanterna.TerminalSize; |
|||
import com.googlecode.lanterna.graphics.ThemeDefinition; |
|||
import com.googlecode.lanterna.gui2.ComponentRenderer; |
|||
import com.googlecode.lanterna.gui2.ProgressBar; |
|||
import com.googlecode.lanterna.gui2.TextGUIGraphics; |
|||
|
|||
public class BackgroundProgressBarRenderer implements ComponentRenderer<ProgressBar> { |
|||
@Override |
|||
public TerminalSize getPreferredSize(ProgressBar component) { |
|||
int preferredWidth = component.getPreferredWidth(); |
|||
if(preferredWidth > 0) { |
|||
return new TerminalSize(preferredWidth, 1); |
|||
} else { |
|||
return new TerminalSize(10, 1); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void drawComponent(TextGUIGraphics graphics, ProgressBar component) { |
|||
TerminalSize size = graphics.getSize(); |
|||
if(size.getRows() == 0 || size.getColumns() == 0) { |
|||
return; |
|||
} |
|||
|
|||
ThemeDefinition themeDefinition = component.getThemeDefinition(); |
|||
int columnOfProgress = (int)(component.getProgress() * size.getColumns()); |
|||
for(int row = 0; row < size.getRows(); row++) { |
|||
graphics.applyThemeStyle(themeDefinition.getActive()); |
|||
for(int column = 0; column < size.getColumns(); column++) { |
|||
if(column < columnOfProgress) { |
|||
graphics.setCharacter(column, row, themeDefinition.getCharacter("FILLER", ' ')); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,30 @@ |
|||
package com.sparrowwallet.sparrow.terminal; |
|||
|
|||
import com.googlecode.lanterna.TerminalSize; |
|||
import com.googlecode.lanterna.gui2.EmptySpace; |
|||
import com.googlecode.lanterna.gui2.Label; |
|||
import com.googlecode.lanterna.gui2.LinearLayout; |
|||
import com.googlecode.lanterna.gui2.Panel; |
|||
import com.googlecode.lanterna.gui2.dialogs.DialogWindow; |
|||
|
|||
import java.util.List; |
|||
|
|||
public final class ModalDialog extends DialogWindow { |
|||
public ModalDialog(String walletName, String description) { |
|||
super(walletName); |
|||
|
|||
setHints(List.of(Hint.CENTERED)); |
|||
setFixedSize(new TerminalSize(30, 5)); |
|||
|
|||
Panel mainPanel = new Panel(); |
|||
mainPanel.setLayoutManager(new LinearLayout()); |
|||
mainPanel.addComponent(new EmptySpace(), LinearLayout.createLayoutData(LinearLayout.Alignment.Beginning, LinearLayout.GrowPolicy.CanGrow)); |
|||
|
|||
Label label = new Label(description); |
|||
mainPanel.addComponent(label, LinearLayout.createLayoutData(LinearLayout.Alignment.Center)); |
|||
|
|||
mainPanel.addComponent(new EmptySpace(), LinearLayout.createLayoutData(LinearLayout.Alignment.Beginning, LinearLayout.GrowPolicy.CanGrow)); |
|||
|
|||
setComponent(mainPanel); |
|||
} |
|||
} |
|||
@ -0,0 +1,168 @@ |
|||
package com.sparrowwallet.sparrow.terminal.wallet; |
|||
|
|||
import com.googlecode.lanterna.TerminalPosition; |
|||
import com.googlecode.lanterna.TerminalSize; |
|||
import com.googlecode.lanterna.gui2.*; |
|||
import com.sparrowwallet.drongo.ExtendedKey; |
|||
import com.sparrowwallet.drongo.KeyDerivation; |
|||
import com.sparrowwallet.drongo.OutputDescriptor; |
|||
import com.sparrowwallet.drongo.policy.Policy; |
|||
import com.sparrowwallet.drongo.policy.PolicyType; |
|||
import com.sparrowwallet.drongo.protocol.ScriptType; |
|||
import com.sparrowwallet.drongo.wallet.Keystore; |
|||
import com.sparrowwallet.drongo.wallet.KeystoreSource; |
|||
import com.sparrowwallet.drongo.wallet.Wallet; |
|||
import com.sparrowwallet.drongo.wallet.WalletModel; |
|||
import com.sparrowwallet.sparrow.io.ImportException; |
|||
import com.sparrowwallet.sparrow.terminal.SparrowTerminal; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
|
|||
import java.util.*; |
|||
|
|||
import static com.sparrowwallet.sparrow.wallet.KeystoreController.DEFAULT_WATCH_ONLY_FINGERPRINT; |
|||
|
|||
public class WatchOnlyDialog extends NewWalletDialog { |
|||
private static final Logger log = LoggerFactory.getLogger(WatchOnlyDialog.class); |
|||
|
|||
private final TextBox descriptor; |
|||
private final Button importWallet; |
|||
|
|||
public WatchOnlyDialog(String walletName) { |
|||
super("Create Watch Only Wallet - " + walletName, walletName); |
|||
|
|||
setHints(List.of(Hint.CENTERED)); |
|||
|
|||
Panel mainPanel = new Panel(); |
|||
mainPanel.setLayoutManager(new GridLayout(2).setVerticalSpacing(0)); |
|||
|
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ZERO)); |
|||
|
|||
TerminalSize screenSize = SparrowTerminal.get().getScreen().getTerminalSize(); |
|||
int descriptorWidth = Math.min(Math.max(20, screenSize.getColumns() - 20), 120); |
|||
|
|||
mainPanel.addComponent(new Label("Output descriptor or xpub")); |
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ZERO)); |
|||
|
|||
descriptor = new TextBox(new TerminalSize(descriptorWidth, 10)); |
|||
mainPanel.addComponent(descriptor); |
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ZERO)); |
|||
|
|||
Panel buttonPanel = new Panel(); |
|||
buttonPanel.setLayoutManager(new GridLayout(2).setHorizontalSpacing(1)); |
|||
buttonPanel.addComponent(new Button("Cancel", this::onCancel)); |
|||
importWallet = new Button("Import Wallet", this::createWallet).setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.CENTER, GridLayout.Alignment.CENTER, true, false)); |
|||
importWallet.setEnabled(false); |
|||
buttonPanel.addComponent(importWallet); |
|||
|
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ZERO)); |
|||
|
|||
buttonPanel.setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.END, GridLayout.Alignment.CENTER,false,false)).addTo(mainPanel); |
|||
mainPanel.addComponent(new EmptySpace(TerminalSize.ZERO)); |
|||
|
|||
setComponent(mainPanel); |
|||
|
|||
descriptor.setTextChangeListener((newText, changedByUserInteraction) -> { |
|||
String line = newText.replaceAll("\\s+", ""); |
|||
try { |
|||
OutputDescriptor.getOutputDescriptor(line); |
|||
importWallet.setEnabled(true); |
|||
} catch(Exception e1) { |
|||
try { |
|||
ExtendedKey.fromDescriptor(line); |
|||
importWallet.setEnabled(true); |
|||
} catch(Exception e2) { |
|||
importWallet.setEnabled(false); |
|||
} |
|||
} |
|||
|
|||
if(changedByUserInteraction) { |
|||
List<String> lines = splitString(newText, descriptorWidth); |
|||
String splitText = lines.stream().reduce((s1, s2) -> s1 + "\n" + s2).get(); |
|||
if(!newText.equals(splitText)) { |
|||
descriptor.setText(splitText); |
|||
|
|||
TerminalPosition pos = descriptor.getCaretPosition(); |
|||
if(pos.getRow() == lines.size() - 2 && pos.getColumn() == lines.get(lines.size() - 2).length()) { |
|||
descriptor.setCaretPosition(lines.size() - 1, lines.get(lines.size() - 1).length()); |
|||
} |
|||
} |
|||
} |
|||
}); |
|||
} |
|||
|
|||
@Override |
|||
protected List<Wallet> getWallets() throws ImportException { |
|||
try { |
|||
return getWalletFromXpub(); |
|||
} catch(Exception e1) { |
|||
try { |
|||
return getWalletFromOutputDescriptor(); |
|||
} catch(Exception e2) { |
|||
log.error("Could not determine wallet from descriptor: " + descriptor.getText(), e2); |
|||
} |
|||
} |
|||
|
|||
return Collections.emptyList(); |
|||
} |
|||
|
|||
private List<Wallet> getWalletFromXpub() { |
|||
ExtendedKey xpub = ExtendedKey.fromDescriptor(descriptor.getText().replaceAll("\\s+", "")); |
|||
ExtendedKey.Header header = ExtendedKey.Header.fromExtendedKey(descriptor.getText()); |
|||
|
|||
Set<ScriptType> scriptTypes = new LinkedHashSet<>(); |
|||
scriptTypes.add(ScriptType.P2WPKH); |
|||
scriptTypes.add(header.getDefaultScriptType()); |
|||
scriptTypes.addAll(ScriptType.getAddressableScriptTypes(PolicyType.SINGLE)); |
|||
|
|||
List<Wallet> wallets = new ArrayList<>(); |
|||
for(ScriptType scriptType : scriptTypes) { |
|||
Wallet wallet = new Wallet(walletName); |
|||
wallet.setPolicyType(PolicyType.SINGLE); |
|||
wallet.setScriptType(scriptType); |
|||
|
|||
Keystore keystore = new Keystore(); |
|||
keystore.setSource(KeystoreSource.SW_WATCH); |
|||
keystore.setWalletModel(WalletModel.SPARROW); |
|||
keystore.setKeyDerivation(new KeyDerivation(DEFAULT_WATCH_ONLY_FINGERPRINT, scriptType.getDefaultDerivationPath())); |
|||
keystore.setExtendedPublicKey(xpub); |
|||
wallet.makeLabelsUnique(keystore); |
|||
wallet.getKeystores().add(keystore); |
|||
|
|||
wallet.setDefaultPolicy(Policy.getPolicy(wallet.getPolicyType(), wallet.getScriptType(), wallet.getKeystores(), 1)); |
|||
wallets.add(wallet); |
|||
} |
|||
|
|||
return wallets; |
|||
} |
|||
|
|||
private List<Wallet> getWalletFromOutputDescriptor() { |
|||
OutputDescriptor outputDescriptor = OutputDescriptor.getOutputDescriptor(descriptor.getText().replaceAll("\\s+", "")); |
|||
Wallet wallet = outputDescriptor.toWallet(); |
|||
wallet.setName(walletName); |
|||
return List.of(wallet); |
|||
} |
|||
|
|||
private List<String> splitString(String stringToSplit, int maxLength) { |
|||
String text = stringToSplit.replaceAll("\\s+", ""); |
|||
if(stringToSplit.endsWith("\n")) { |
|||
text += "\n"; |
|||
} |
|||
|
|||
List<String> lines = new ArrayList<>(); |
|||
while(text.length() >= maxLength) { |
|||
int breakAt = maxLength - 1; |
|||
lines.add(text.substring(0, breakAt)); |
|||
text = text.substring(breakAt); |
|||
} |
|||
|
|||
if(text.equals("\n")) { |
|||
text = ""; |
|||
} |
|||
|
|||
lines.add(text); |
|||
return lines; |
|||
} |
|||
} |
|||
Loading…
Reference in new issue