From 5501814e00c7e8584f42bd6ff69827ac7d8705fd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mika=C3=ABl=20Capelle?= <capelle.mikael@gmail.com>
Date: Tue, 26 Sep 2023 21:41:27 +0200
Subject: [PATCH] Add onNextRefresh() and 2nd onAboutToRun bindings.

---
 src/mobase/wrappers/basic_classes.cpp | 12 +++++++++++-
 tests/mocks/MockOrganizer.h           |  2 ++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/mobase/wrappers/basic_classes.cpp b/src/mobase/wrappers/basic_classes.cpp
index 8219fb3..26e1ddb 100644
--- a/src/mobase/wrappers/basic_classes.cpp
+++ b/src/mobase/wrappers/basic_classes.cpp
@@ -553,10 +553,20 @@ namespace mo2::python {
             .def("managedGame", &IOrganizer::managedGame,
                  py::return_value_policy::reference)
 
-            .def("onAboutToRun", &IOrganizer::onAboutToRun, "callback"_a)
+            .def("onAboutToRun",
+                 py::overload_cast<std::function<bool(QString const&)> const&>(
+                     &IOrganizer::onAboutToRun),
+                 "callback"_a)
+            .def("onAboutToRun",
+                 py::overload_cast<std::function<bool(QString const&, const QDir&,
+                                                      const QString&)> const&>(
+                     &IOrganizer::onAboutToRun),
+                 "callback"_a)
             .def("onFinishedRun", &IOrganizer::onFinishedRun, "callback"_a)
             .def("onUserInterfaceInitialized", &IOrganizer::onUserInterfaceInitialized,
                  "callback"_a)
+            .def("onNextRefresh", &IOrganizer::onNextRefresh, "callback"_a,
+                 "immediate_if_possible"_a = true)
             .def("onProfileCreated", &IOrganizer::onProfileCreated, "callback"_a)
             .def("onProfileRenamed", &IOrganizer::onProfileRenamed, "callback"_a)
             .def("onProfileRemoved", &IOrganizer::onProfileRemoved, "callback"_a)
diff --git a/tests/mocks/MockOrganizer.h b/tests/mocks/MockOrganizer.h
index 316ba9e..1e393ea 100644
--- a/tests/mocks/MockOrganizer.h
+++ b/tests/mocks/MockOrganizer.h
@@ -39,10 +39,12 @@ public:
 	MOCK_METHOD(HANDLE, startApplication, (const QString &executable, const QStringList &args, const QString &cwd, const QString &profile, const QString &forcedCustomOverwrite, bool ignoreCustomOverwrite), (override));
 	MOCK_METHOD(bool, waitForApplication, (HANDLE handle, bool refresh, LPDWORD exitCode), (const, override));
 	MOCK_METHOD(bool, onAboutToRun, (const std::function<bool(const QString&)> &func), (override));
+	MOCK_METHOD(bool, onAboutToRun, (const std::function<bool(const QString&, const QDir&, const QString&)> &func), (override));
 	MOCK_METHOD(bool, onFinishedRun, (const std::function<void(const QString&, unsigned int)> &func), (override));
 	MOCK_METHOD(void, refresh, (bool saveChanges), (override));
 	MOCK_METHOD(MOBase::IPluginGame const *, managedGame, (), (const, override));
 	MOCK_METHOD(bool, onUserInterfaceInitialized, (const std::function<void (QMainWindow *)> &), (override));
+	MOCK_METHOD(bool, onNextRefresh, (const std::function<void()>&, bool), (override));
 	MOCK_METHOD(bool, onProfileCreated, (const std::function<void(MOBase::IProfile*)>&), (override));
 	MOCK_METHOD(bool, onProfileRemoved, (const std::function<void(const QString&)>&), (override));
 	MOCK_METHOD(bool, onProfileRenamed, (const std::function<void(MOBase::IProfile*, QString const&, QString const&)>&), (override));