update qnotify for variable output

1 files changed, 10 insertions, 5 deletions

diff --git a/src-qt5/desktop-utils/lumina-notify/main.cpp b/src-qt5/desktop-utils/lumina-notify/main.cpp
index f9d76f28..c0e3bd05 100644
--- a/src-qt5/desktop-utils/lumina-notify/main.cpp
+++ b/src-qt5/desktop-utils/lumina-notify/main.cpp
@@ -1,19 +1,24 @@
 //-------------------------------------------------
-// Created by q5sys
+// Created by q5sys (JT)
 // Released under MIT License 2017-03-08
 // A Simple GUI Dialog Program
 //-------------------------------------------------
 
 #include <QApplication>
 #include <QMessageBox>
+#include <QDebug>
+#include <QTextStream>
 
 int main(int argc, char *argv[])
 {
     QApplication a(argc, argv);
+    int answer;
     QMessageBox *messageBox = new QMessageBox;
         messageBox->setText(argv[1]);
-        QPushButton *pushButtonOk = messageBox->addButton(argv[2], QMessageBox::YesRole);
-        messageBox->QDialog::setWindowTitle(argv[3]);
-    messageBox->show();
-    return a.exec();
+        QPushButton *pushButtonOk = messageBox->addButton(argv[2], QMessageBox::AcceptRole);
+        QPushButton *pushButtonNo = messageBox->addButton(argv[3], QMessageBox::RejectRole);
+        messageBox->QDialog::setWindowTitle(argv[4]);
+        messageBox->show();
+        if(messageBox->exec() == QMessageBox::AcceptRole){ answer = 0; QTextStream cout(stdout); cout << answer;}
+        else { answer = 1; QTextStream cout(stdout); cout << answer;}
 }