[GUI] Reset change address from within SendChangeAddressDialog

Fuzzbawls · Apr 23, 2020 · ca99eaa · ca99eaa
1 parent 15b56a2
commit ca99eaa
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/src/qt/pivx/sendchangeaddressdialog.cpp b/src/qt/pivx/sendchangeaddressdialog.cpp
@@ -4,7 +4,8 @@
 
 #include "qt/pivx/sendchangeaddressdialog.h"
 #include "qt/pivx/forms/ui_sendchangeaddressdialog.h"
-#include "walletmodel.h"
+
+#include "coincontrol.h"
 #include "qt/pivx/qtutils.h"
 
 SendChangeAddressDialog::SendChangeAddressDialog(QWidget *parent) :
@@ -32,16 +33,17 @@ SendChangeAddressDialog::SendChangeAddressDialog(QWidget *parent) :
     ui->btnEsc->setProperty("cssClass", "ic-close");
 
     ui->btnCancel->setProperty("cssClass", "btn-dialog-cancel");
-    ui->btnSave->setText("SAVE");
+    ui->btnSave->setText(tr("SAVE"));
     setCssBtnPrimary(ui->btnSave);
 
     connect(ui->btnEsc, &QPushButton::clicked, this, &SendChangeAddressDialog::close);
-    connect(ui->btnCancel, &QPushButton::clicked, this, &SendChangeAddressDialog::close);
+    connect(ui->btnCancel, &QPushButton::clicked, this, &SendChangeAddressDialog::reset);
     connect(ui->btnSave, &QPushButton::clicked, [this](){ selected = true; accept(); });
 }
 
 void SendChangeAddressDialog::setAddress(QString address){
     ui->lineEditAddress->setText(address);
+    ui->btnCancel->setText(tr("RESET"));
 }
 
 bool SendChangeAddressDialog::getAddress(WalletModel *model, QString *retAddress){
@@ -58,6 +60,16 @@ void SendChangeAddressDialog::showEvent(QShowEvent *event)
     if (ui->lineEditAddress) ui->lineEditAddress->setFocus();
 }
 
+void SendChangeAddressDialog::reset()
+{
+    if (!ui->lineEditAddress->text().isEmpty()) {
+        ui->lineEditAddress->clear();
+        ui->btnCancel->setText(tr("CANCEL"));
+        CoinControlDialog::coinControl->destChange = CNoDestination();
+    }
+    close();
+}
+
 SendChangeAddressDialog::~SendChangeAddressDialog(){
     delete ui;
 }
diff --git a/src/qt/pivx/sendchangeaddressdialog.h b/src/qt/pivx/sendchangeaddressdialog.h
@@ -26,8 +26,12 @@ class SendChangeAddressDialog : public QDialog
     bool selected = false;
 
     void showEvent(QShowEvent *event) override;
+
 private:
     Ui::SendChangeAddressDialog *ui;
+
+private Q_SLOTS:
+    void reset();
 };
 
 #endif // SENDCHANGEADDRESSDIALOG_H