GUI: SendCoinsEntry: Propagate asset types available to amount field

ElementsProject · Apr 9, 2019 · 2ebae1a · 2ebae1a
1 parent a163be0
commit 2ebae1a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/qt/sendcoinsentry.cpp b/src/qt/sendcoinsentry.cpp
@@ -81,8 +81,10 @@ void SendCoinsEntry::setModel(WalletModel *_model)
 {
     this->model = _model;
 
-    if (_model && _model->getOptionsModel())
+    if (_model && _model->getOptionsModel()) {
         connect(_model->getOptionsModel(), &OptionsModel::displayUnitChanged, this, &SendCoinsEntry::updateDisplayUnit);
+        connect(_model, SIGNAL(assetTypesChanged()), this, SLOT(updateAssetTypes()));
+    }
 
     clear();
 }
@@ -107,6 +109,7 @@ void SendCoinsEntry::clear()
     ui->payAmount_s->clear();
 
     // update the display unit, to not use the default ("BTC")
+    updateAssetTypes();
     updateDisplayUnit();
 }
 
@@ -251,6 +254,13 @@ void SendCoinsEntry::setFocus()
     ui->payTo->setFocus();
 }
 
+void SendCoinsEntry::updateAssetTypes()
+{
+    if (model) {
+        ui->payAmount->setAllowedAssets(model->getAssetTypes());
+    }
+}
+
 void SendCoinsEntry::updateDisplayUnit()
 {
     if(model && model->getOptionsModel())

diff --git a/src/qt/sendcoinsentry.h b/src/qt/sendcoinsentry.h
@@ -64,6 +64,7 @@ private Q_SLOTS:
     void on_addressBookButton_clicked();
     void on_pasteButton_clicked();
     void updateDisplayUnit();
+    void updateAssetTypes();
 
 private:
     SendCoinsRecipient recipient;