Craig Raw
2 years ago
6 changed files with 204 additions and 69 deletions
@ -0,0 +1,75 @@ |
|||
package com.sparrowwallet.sparrow.io; |
|||
|
|||
import com.google.zxing.BarcodeFormat; |
|||
import com.google.zxing.WriterException; |
|||
import com.google.zxing.client.j2se.MatrixToImageConfig; |
|||
import com.google.zxing.client.j2se.MatrixToImageWriter; |
|||
import com.google.zxing.common.BitMatrix; |
|||
import com.google.zxing.qrcode.QRCodeWriter; |
|||
import com.lowagie.text.*; |
|||
import com.lowagie.text.Font; |
|||
import com.lowagie.text.Image; |
|||
import com.lowagie.text.pdf.PdfWriter; |
|||
import com.sparrowwallet.hummingbird.UR; |
|||
import com.sparrowwallet.hummingbird.UREncoder; |
|||
import com.sparrowwallet.sparrow.AppServices; |
|||
import javafx.embed.swing.SwingFXUtils; |
|||
import javafx.stage.FileChooser; |
|||
import javafx.stage.Stage; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
|
|||
import java.awt.*; |
|||
import java.io.*; |
|||
|
|||
public class PdfUtils { |
|||
private static final Logger log = LoggerFactory.getLogger(PdfUtils.class); |
|||
|
|||
private static final int QR_WIDTH = 480; |
|||
private static final int QR_HEIGHT = 480; |
|||
|
|||
public static void saveOutputDescriptor(String walletName, String outputDescriptor, UR ur) { |
|||
Stage window = new Stage(); |
|||
FileChooser fileChooser = new FileChooser(); |
|||
fileChooser.setTitle("Save PDF"); |
|||
fileChooser.setInitialFileName(walletName + ".pdf"); |
|||
AppServices.moveToActiveWindowScreen(window, 800, 450); |
|||
File file = fileChooser.showSaveDialog(window); |
|||
if(file != null) { |
|||
try(Document document = new Document()) { |
|||
document.setMargins(36, 36, 48, 36); |
|||
PdfWriter.getInstance(document, new FileOutputStream(file)); |
|||
document.open(); |
|||
|
|||
Font titleFont = FontFactory.getFont(FontFactory.HELVETICA_BOLD, 16, Color.BLACK); |
|||
Chunk title = new Chunk("Output descriptor for " + walletName, titleFont); |
|||
document.add(title); |
|||
|
|||
UREncoder urEncoder = new UREncoder(ur, 2000, 10, 0); |
|||
String fragment = urEncoder.nextPart(); |
|||
if(urEncoder.isSinglePart()) { |
|||
Image image = Image.getInstance(SwingFXUtils.fromFXImage(getQrCode(fragment), null), Color.WHITE); |
|||
document.add(image); |
|||
} |
|||
|
|||
Font descriptorFont = FontFactory.getFont(FontFactory.COURIER, 14, Color.BLACK); |
|||
Paragraph descriptor = new Paragraph(outputDescriptor, descriptorFont); |
|||
document.add(descriptor); |
|||
} catch(Exception e) { |
|||
log.error("Error creating descriptor PDF", e); |
|||
AppServices.showErrorDialog("Error creating PDF", e.getMessage()); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private static javafx.scene.image.Image getQrCode(String fragment) throws IOException, WriterException { |
|||
QRCodeWriter qrCodeWriter = new QRCodeWriter(); |
|||
BitMatrix qrMatrix = qrCodeWriter.encode(fragment, BarcodeFormat.QR_CODE, QR_WIDTH, QR_HEIGHT); |
|||
|
|||
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
|||
MatrixToImageWriter.writeToStream(qrMatrix, "PNG", baos, new MatrixToImageConfig()); |
|||
|
|||
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray()); |
|||
return new javafx.scene.image.Image(bais); |
|||
} |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package com.sparrowwallet.sparrow.wallet; |
|||
|
|||
import com.sparrowwallet.drongo.wallet.Wallet; |
|||
import com.sparrowwallet.hummingbird.UR; |
|||
import com.sparrowwallet.sparrow.AppServices; |
|||
import com.sparrowwallet.sparrow.glyphfont.FontAwesome5; |
|||
import com.sparrowwallet.sparrow.io.PdfUtils; |
|||
import javafx.scene.Node; |
|||
import javafx.scene.control.*; |
|||
import javafx.scene.layout.HBox; |
|||
import javafx.scene.layout.Priority; |
|||
import org.controlsfx.glyphfont.Glyph; |
|||
|
|||
public class MultisigBackupDialog extends Dialog<String> { |
|||
private final Wallet wallet; |
|||
private final String descriptor; |
|||
private final UR ur; |
|||
|
|||
private final TextArea textArea; |
|||
|
|||
public MultisigBackupDialog(Wallet wallet, String descriptor, UR ur) { |
|||
this.wallet = wallet; |
|||
this.descriptor = descriptor; |
|||
this.ur = ur; |
|||
|
|||
setTitle("Backup Multisig Wallet?"); |
|||
|
|||
DialogPane dialogPane = new MultisigBackupDialogPane(); |
|||
dialogPane.setHeaderText("To restore this multisig wallet, you need at least " + wallet.getDefaultPolicy().getNumSignaturesRequired() + " seeds and ALL of the xpubs!\n" + |
|||
"It is recommended to backup either this wallet file, or the wallet output descriptor.\n\n" + |
|||
"The wallet output descriptor contains all the xpubs and is shown below.\n" + |
|||
"Alternatively, use the Export button below to export the Sparrow wallet file."); |
|||
setDialogPane(dialogPane); |
|||
|
|||
dialogPane.getStyleClass().addAll("alert", "warning"); |
|||
|
|||
HBox hbox = new HBox(); |
|||
this.textArea = new TextArea(descriptor); |
|||
this.textArea.setMaxWidth(Double.MAX_VALUE); |
|||
this.textArea.setWrapText(true); |
|||
this.textArea.getStyleClass().add("fixed-width"); |
|||
this.textArea.setEditable(false); |
|||
hbox.getChildren().add(textArea); |
|||
HBox.setHgrow(this.textArea, Priority.ALWAYS); |
|||
|
|||
dialogPane.setContent(hbox); |
|||
dialogPane.getStylesheets().add(AppServices.class.getResource("general.css").toExternalForm()); |
|||
AppServices.setStageIcon(dialogPane.getScene().getWindow()); |
|||
|
|||
dialogPane.getStyleClass().add("text-input-dialog"); |
|||
dialogPane.getButtonTypes().add(ButtonType.OK); |
|||
|
|||
final ButtonType qrButtonType = new javafx.scene.control.ButtonType("Save PDF...", ButtonBar.ButtonData.LEFT); |
|||
dialogPane.getButtonTypes().add(qrButtonType); |
|||
|
|||
dialogPane.setPrefWidth(700); |
|||
dialogPane.setPrefHeight(350); |
|||
AppServices.moveToActiveWindowScreen(this); |
|||
} |
|||
|
|||
private class MultisigBackupDialogPane extends DialogPane { |
|||
@Override |
|||
protected Node createButton(ButtonType buttonType) { |
|||
Node button; |
|||
if(buttonType.getButtonData() == ButtonBar.ButtonData.LEFT) { |
|||
Button pdfButton = new Button(buttonType.getText()); |
|||
pdfButton.setGraphicTextGap(5); |
|||
pdfButton.setGraphic(getGlyph(FontAwesome5.Glyph.FILE_PDF)); |
|||
|
|||
final ButtonBar.ButtonData buttonData = buttonType.getButtonData(); |
|||
ButtonBar.setButtonData(pdfButton, buttonData); |
|||
pdfButton.setOnAction(event -> { |
|||
PdfUtils.saveOutputDescriptor(wallet.getFullDisplayName(), descriptor, ur); |
|||
}); |
|||
|
|||
button = pdfButton; |
|||
} else { |
|||
button = super.createButton(buttonType); |
|||
} |
|||
|
|||
return button; |
|||
} |
|||
|
|||
private Glyph getGlyph(FontAwesome5.Glyph glyphName) { |
|||
Glyph glyph = new Glyph(FontAwesome5.FONT_NAME, glyphName); |
|||
glyph.setFontSize(11); |
|||
return glyph; |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue