App: handle DialogExecScript closing properly

fougue · Jun 28, 2024 · 90d6034 · 90d6034
1 parent 073bea6
commit 90d6034
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 24 deletions.
diff --git a/src/app/commands_tools.cpp b/src/app/commands_tools.cpp
@@ -117,7 +117,7 @@ void CommandExecScript::execute()
     auto strFilePath = QFileDialog::getOpenFileName(
         this->widgetMain(),
         Command::tr("Choose JavaScript file"),
-        QString(/*dir*/),
+        QString{}/*dir*/,
         Command::tr("Script files(*.js)")
     );
     if (!strFilePath.isEmpty())

diff --git a/src/app/dialog_exec_script.cpp b/src/app/dialog_exec_script.cpp
@@ -9,13 +9,13 @@
 #include "qtwidgets_utils.h"
 #include "../qtcommon/filepath_conv.h"
 #include "../qtcommon/log_message_handler.h"
+#include "../qtcommon/qtcore_utils.h"
 #include "../qtscripting/script_global.h"
 #include "ui_dialog_exec_script.h"
 
 #include <QtCore/QDir>
 #include <QtCore/QFile>
-#include <QtCore/QFileinfo>
-#include <QtCore/QTimer>
+#include <QtCore/QFileInfo>
 #include <QtQml/QJSEngine>
 #include <gsl/util>
 
@@ -43,10 +43,12 @@ DialogExecScript::DialogExecScript(QWidget* parent)
     m_taskMgr.signalStarted.connectSlot(&DialogExecScript::onTaskStarted, this);
     m_taskMgr.signalEnded.connectSlot(&DialogExecScript::onTaskEnded, this);
     QObject::connect(
-        m_ui->btn_restartStop, &QAbstractButton::clicked, this, &DialogExecScript::restartOrStopScriptExec
+        m_ui->btn_restartStop, &QAbstractButton::clicked,
+        this, &DialogExecScript::restartOrStopScriptExec
     );
     QObject::connect(
-        m_ui->buttonBox->button(QDialogButtonBox::Close), &QAbstractButton::clicked, this, &QDialog::accept
+        m_ui->buttonBox->button(QDialogButtonBox::Close), &QAbstractButton::clicked,
+        this, &DialogExecScript::tryCloseDialog
     );
 }
 
@@ -79,7 +81,7 @@ void DialogExecScript::startScript()
         });
         auto fnAddConsoleOutput = [=](QtMsgType type, const QString& text, const QString& file, int line) {
             const Message msg = { type, text, file, line };
-            QTimer::singleShot(0, this, [=]{ this->addConsoleOutput(msg); });
+            QtCoreUtils::runJobOnMainThread([=]{ this->addConsoleOutput(msg); });
         };
         logMsgHandler.setJsConsoleOutputHandler(
             [=](QtMsgType type, const QMessageLogContext& context, const QString& text) {
@@ -105,16 +107,6 @@ void DialogExecScript::startScript()
     m_taskMgr.run(m_scriptExecTaskId);
 }
 
-void DialogExecScript::done(int resultCode)
-{
-    if (m_scriptExecIsRunning) {
-        // TODO Ask if script execution should be stopped
-    }
-    else {
-        QDialog::done(resultCode);
-    }
-}
-
 void DialogExecScript::onTaskStarted(TaskId taskId)
 {
     if (m_scriptExecTaskId != taskId)
@@ -149,15 +141,18 @@ void DialogExecScript::onTaskEnded(TaskId taskId)
         m_ui->treeWidget_Output->resizeColumnToContents(col);
 }
 
+void DialogExecScript::interruptScriptExec()
+{
+    m_wasScriptExecInterrupted = true;
+    m_jsEngine->setInterrupted(true);
+}
+
 void DialogExecScript::restartOrStopScriptExec()
 {
-    if (m_scriptExecIsRunning) {
-        m_wasScriptExecInterrupted = true;
-        m_jsEngine->setInterrupted(true);
-    }
-    else {
+    if (m_scriptExecIsRunning)
+        this->interruptScriptExec();
+    else
         this->startScript();
-    }
 }
 
 void DialogExecScript::recreateScriptEngine()
@@ -187,4 +182,26 @@ void DialogExecScript::addConsoleOutput(const Message& msg)
     m_ui->treeWidget_Output->addTopLevelItem(item);
 }
 
+void DialogExecScript::tryCloseDialog()
+{
+    if (m_scriptExecIsRunning) {
+        auto msgBox = QtWidgetsUtils::asyncMsgBoxWarning(
+            this,
+            tr("Warning"),
+            tr("Script execution isn't finished\n\nInterrupt and exit?"),
+            QMessageBox::Yes | QMessageBox::No
+        );
+        QObject::connect(msgBox, &QMessageBox::buttonClicked, this, [=](QAbstractButton* btn) {
+            if (btn == msgBox->button(QMessageBox::Yes)) {
+                this->interruptScriptExec();
+                m_taskMgr.waitForDone(m_scriptExecTaskId, 10000/*ms*/);
+                QtCoreUtils::runJobOnMainThread([=]{ this->tryCloseDialog(); });
+            }
+        });
+    }
+    else {
+        this->reject();
+    }
+}
+
 } // namespace Mayo
diff --git a/src/app/dialog_exec_script.h b/src/app/dialog_exec_script.h
@@ -28,12 +28,11 @@ class DialogExecScript : public QDialog {
 
     void startScript();
 
-    void done(int resultCode) override;
-
 private:
     void onTaskStarted(TaskId taskId);
     void onTaskEnded(TaskId taskId);
 
+    void interruptScriptExec();
     void restartOrStopScriptExec();
     void recreateScriptEngine();
 
@@ -46,6 +45,8 @@ class DialogExecScript : public QDialog {
 
     void addConsoleOutput(const Message& msg);
 
+    void tryCloseDialog();
+
     class Ui_DialogExecScript* m_ui = nullptr;
     ScriptEngineCreator m_fnScriptEngineCreator;
     QJSEngine* m_jsEngine = nullptr;