Add system tray notification for new polls

This adds a GUI hook that displays a system notification when a node receives a new poll as well as an accompanying GUI option to disable those notifications.
gridcoin-community · Jun 9, 2021 · 74de3ed · 74de3ed
1 parent 0d548a5
commit 74de3ed
Show file tree

Hide file tree

Showing 6 changed files with 45 additions and 0 deletions.
diff --git a/src/qt/bitcoingui.cpp b/src/qt/bitcoingui.cpp
@@ -42,6 +42,7 @@
 #include "clicklabel.h"
 #include "univalue.h"
 #include "upgradeqt.h"
+#include "voting/votingmodel.h"
 
 #ifdef Q_OS_MAC
 #include "macdockiconhandler.h"
@@ -807,6 +808,12 @@ void BitcoinGUI::setResearcherModel(ResearcherModel *researcherModel)
 void BitcoinGUI::setVotingModel(VotingModel *votingModel)
 {
     votingPage->setVotingModel(votingModel);
+
+    if (!votingModel) {
+        return;
+    }
+
+    connect(votingModel, SIGNAL(newPollReceived()), this, SLOT(handleNewPoll()));
 }
 
 void BitcoinGUI::createTrayIcon()
@@ -1786,6 +1793,20 @@ void BitcoinGUI::updateBeaconIcon()
         .arg(researcherModel->formatBeaconStatus()));
 }
 
+void BitcoinGUI::handleNewPoll()
+{
+    if (!clientModel || !clientModel->getOptionsModel()) {
+        return;
+    }
+
+    if (!clientModel->getOptionsModel()->getDisablePollNotifications()) {
+        notificator->notify(
+            Notificator::Information,
+            tr("New Poll"),
+            tr("A new poll is available. Open Gridcoin to vote."));
+    }
+}
+
 // -----------------------------------------------------------------------------
 // Class: ToolbarButtonIconFilter
 // -----------------------------------------------------------------------------

diff --git a/src/qt/bitcoingui.h b/src/qt/bitcoingui.h
@@ -274,6 +274,7 @@ private slots:
     QString GetEstimatedStakingFrequency(unsigned int nEstimateTime);
 
     void updateGlobalStatus();
+    void handleNewPoll();
 };
 
 //!

diff --git a/src/qt/forms/optionsdialog.ui b/src/qt/forms/optionsdialog.ui
@@ -286,6 +286,13 @@
          </property>
         </widget>
        </item>
+       <item>
+        <widget class="QCheckBox" name="disablePollNotifications">
+         <property name="text">
+          <string>Disable Poll Notifications</string>
+         </property>
+        </widget>
+       </item>
        <item>
         <spacer name="verticalSpacer_Window">
          <property name="orientation">

diff --git a/src/qt/optionsdialog.cpp b/src/qt/optionsdialog.cpp
@@ -148,6 +148,7 @@ void OptionsDialog::setMapper()
 
     /* Window */
     mapper->addMapping(ui->disableTransactionNotifications, OptionsModel::DisableTrxNotifications);
+    mapper->addMapping(ui->disablePollNotifications, OptionsModel::DisablePollNotifications);
 #ifndef Q_OS_MAC
     mapper->addMapping(ui->minimizeToTray, OptionsModel::MinimizeToTray);
     mapper->addMapping(ui->minimizeOnClose, OptionsModel::MinimizeOnClose);

diff --git a/src/qt/optionsmodel.cpp b/src/qt/optionsmodel.cpp
@@ -46,6 +46,7 @@ void OptionsModel::Init()
     fStartMin = settings.value("fStartMin", true).toBool();
     fMinimizeToTray = settings.value("fMinimizeToTray", false).toBool();
     fDisableTrxNotifications = settings.value("fDisableTrxNotifications", false).toBool();
+    fDisablePollNotifications = settings.value("fDisablePollNotifications", false).toBool();
     bDisplayAddresses = settings.value("bDisplayAddresses", false).toBool();
     fMinimizeOnClose = settings.value("fMinimizeOnClose", false).toBool();
     fCoinControlFeatures = settings.value("fCoinControlFeatures", false).toBool();
@@ -97,6 +98,8 @@ QVariant OptionsModel::data(const QModelIndex & index, int role) const
             return QVariant(fMinimizeToTray);
         case DisableTrxNotifications:
             return QVariant(fDisableTrxNotifications);
+        case DisablePollNotifications:
+            return QVariant(fDisablePollNotifications);
         case MapPortUPnP:
             return settings.value("fUseUPnP", gArgs.GetBoolArg("-upnp", true));
         case MinimizeOnClose:
@@ -178,6 +181,10 @@ bool OptionsModel::setData(const QModelIndex & index, const QVariant & value, in
             fDisableTrxNotifications = value.toBool();
             settings.setValue("fDisableTrxNotifications", fDisableTrxNotifications);
             break;
+        case DisablePollNotifications:
+            fDisablePollNotifications = value.toBool();
+            settings.setValue("fDisablePollNotifications", fDisablePollNotifications);
+            break;
         case MapPortUPnP:
             fUseUPnP = value.toBool();
             settings.setValue("fUseUPnP", fUseUPnP);
@@ -338,6 +345,11 @@ bool OptionsModel::getDisableTrxNotifications()
     return fDisableTrxNotifications;
 }
 
+bool OptionsModel::getDisablePollNotifications()
+{
+    return fDisablePollNotifications;
+}
+
 bool OptionsModel::getMinimizeOnClose()
 {
     return fMinimizeOnClose;

diff --git a/src/qt/optionsmodel.h b/src/qt/optionsmodel.h
@@ -22,6 +22,7 @@ class OptionsModel : public QAbstractListModel
         MinimizeToTray,          // bool
         StartMin,                // bool
         DisableTrxNotifications, // bool
+        DisablePollNotifications,// bool
         MapPortUPnP,             // bool
         MinimizeOnClose,         // bool
         ProxyUse,                // bool
@@ -54,6 +55,7 @@ class OptionsModel : public QAbstractListModel
     bool getStartMin();
     bool getMinimizeToTray();
     bool getDisableTrxNotifications();
+    bool getDisablePollNotifications();
     bool getMinimizeOnClose();
     int getDisplayUnit();
 	bool getDisplayAddresses();
@@ -75,6 +77,7 @@ class OptionsModel : public QAbstractListModel
     bool fStartAtStartup;
     bool fStartMin;
     bool fDisableTrxNotifications;
+    bool fDisablePollNotifications;
 	bool bDisplayAddresses;
     bool fMinimizeOnClose;
     bool fCoinControlFeatures;