added lockwidget to replace all the other dialogs

rewrote ProcessRunner to have a bunch of setters and then a run()
fixed bad exit code when waiting on a process that's already completed
removed lock()/unlock() from main window, ProcessRunner is in charge of that now

1 files changed, 224 insertions, 0 deletions

diff --git a/src/lockwidget.cpp b/src/lockwidget.cpp
new file mode 100644
index 00000000..720cac36
--- /dev/null
+++ b/src/lockwidget.cpp
@@ -0,0 +1,224 @@
+#include "lockwidget.h"
+#include "mainwindow.h"
+#include <QGraphicsDropShadowEffect>
+#include <QMenuBar>
+#include <QStatusBar>
+
+QWidget* createTransparentWidget(QWidget* parent=nullptr)
+{
+  auto* w = new QWidget(parent);
+
+  w->setWindowOpacity(0);
+  w->setAttribute(Qt::WA_NoSystemBackground);
+  w->setAttribute(Qt::WA_TranslucentBackground);
+
+  return w;
+}
+
+
+LockWidget::LockWidget(QWidget* parent, Reasons reason) :
+  m_parent(parent), m_overlay(nullptr), m_info(nullptr), m_result(NoResult),
+  m_filter(nullptr)
+{
+  if (reason != NoReason) {
+    lock(reason);
+  }
+}
+
+LockWidget::~LockWidget()
+{
+  unlock();
+}
+
+void LockWidget::lock(Reasons reason)
+{
+  m_result = StillLocked;
+  createUi(reason);
+}
+
+void LockWidget::unlock()
+{
+  m_overlay.reset();
+
+  if (m_filter && m_parent) {
+    m_parent->removeEventFilter(m_filter.get());
+  }
+
+  enableAll();
+}
+
+void LockWidget::setInfo(DWORD pid, const QString& name)
+{
+  m_info->setText(QString("%1 (%2)").arg(name).arg(pid));
+}
+
+LockWidget::Results LockWidget::result() const
+{
+  return m_result;
+}
+
+void LockWidget::createUi(Reasons reason)
+{
+  if (m_parent) {
+    m_overlay.reset(createTransparentWidget(m_parent));
+    m_overlay->setWindowFlags(m_overlay->windowFlags() & Qt::FramelessWindowHint);
+    m_overlay->setGeometry(m_parent->rect());
+  } else {
+    m_overlay.reset(new QDialog);
+  }
+
+  auto* center = new QFrame;
+
+  if (m_parent) {
+    center->setFrameStyle(QFrame::StyledPanel);
+    center->setLineWidth(1);
+    center->setAutoFillBackground(true);
+
+    auto* shadow = new QGraphicsDropShadowEffect;
+    shadow->setBlurRadius(50);
+    shadow->setOffset(0);
+    shadow->setColor(QColor(0, 0, 0, 100));
+    center->setGraphicsEffect(shadow);
+  }
+
+  m_info = new QLabel(" ");
+  m_info->setAlignment(Qt::AlignCenter | Qt::AlignHCenter);
+
+  auto* ly = new QVBoxLayout(center);
+
+  if (!m_parent) {
+    ly->setContentsMargins(0, 0, 0, 0);
+  }
+
+  auto* message = new QLabel;
+  ly->addWidget(message);
+  ly->addWidget(m_info);
+
+  auto* buttons = new QWidget;
+  auto* buttonsLayout = new QHBoxLayout(buttons);
+  ly->addWidget(buttons);
+
+  switch (reason)
+  {
+    case LockUI:
+    {
+      message->setText(QObject::tr(
+        "Mod Organizer is locked while the executable is running."));
+
+      auto* unlockButton = new QPushButton(QObject::tr("Unlock"));
+      QObject::connect(unlockButton, &QPushButton::clicked, [&]{ onForceUnlock(); });
+      buttonsLayout->addWidget(unlockButton);
+
+      break;
+    }
+
+    case OutputRequired:
+    {
+      message->setText(QObject::tr(
+        "The executable must run to completion because a its output is "
+        "required."));
+
+      auto* unlockButton = new QPushButton(QObject::tr("Unlock"));
+      QObject::connect(unlockButton, &QPushButton::clicked, [&]{ onForceUnlock(); });
+      buttonsLayout->addWidget(unlockButton);
+
+      break;
+    }
+
+    case PreventExit:
+    {
+      message->setText(QObject::tr(
+        "Mod Organizer is waiting on processes to finish before exiting."));
+
+      auto* exit = new QPushButton(QObject::tr("Exit Now"));
+      QObject::connect(exit, &QPushButton::clicked, [&]{ onForceUnlock(); });
+      buttonsLayout->addWidget(exit);
+
+      auto* cancel = new QPushButton(QObject::tr("Cancel"));
+      QObject::connect(cancel, &QPushButton::clicked, [&]{ onCancel(); });
+      buttonsLayout->addWidget(cancel);
+
+      break;
+    }
+  }
+
+  auto* grid = new QGridLayout(m_overlay.get());
+  grid->addWidget(createTransparentWidget(), 0, 1);
+  grid->addWidget(createTransparentWidget(), 2, 1);
+  grid->addWidget(createTransparentWidget(), 1, 0);
+  grid->addWidget(createTransparentWidget(), 1, 2);
+  grid->addWidget(center, 1, 1);
+
+  if (!m_parent) {
+    grid->setContentsMargins(0, 0, 0, 0);
+  }
+
+  grid->setRowStretch(0, 1);
+  grid->setRowStretch(2, 1);
+  grid->setColumnStretch(0, 1);
+  grid->setColumnStretch(2, 1);
+
+  disableAll();
+
+  if (m_parent) {
+    m_filter.reset(new Filter);
+    m_filter->resized = [=]{ m_overlay->setGeometry(m_parent->rect()); };
+    m_parent->installEventFilter(m_filter.get());
+  }
+
+  m_overlay->setFocusPolicy(Qt::TabFocus);
+  m_overlay->setFocus();
+  m_overlay->show();
+  m_overlay->setEnabled(true);
+}
+
+void LockWidget::onForceUnlock()
+{
+  m_result = ForceUnlocked;
+  unlock();
+}
+
+void LockWidget::onCancel()
+{
+  m_result = Cancelled;
+  unlock();
+}
+
+void LockWidget::disableAll()
+{
+  if (!m_parent) {
+    // nothing to disable without a main window
+    return;
+  }
+
+  if (auto* mw=dynamic_cast<QMainWindow*>(m_parent)) {
+    disable(mw->centralWidget());
+    disable(mw->menuBar());
+    disable(mw->statusBar());
+  }
+
+  for (auto* tb : m_parent->findChildren<QToolBar*>()) {
+    disable(tb);
+  }
+
+  for (auto* d : m_parent->findChildren<QDockWidget*>()) {
+    disable(d);
+  }
+}
+
+void LockWidget::enableAll()
+{
+  for (auto* w : m_disabled) {
+    w->setEnabled(true);
+  }
+
+  m_disabled.clear();
+}
+
+void LockWidget::disable(QWidget* w)
+{
+  if (w->isEnabled()) {
+    w->setEnabled(false);
+    m_disabled.push_back(w);
+  }
+}