1 files changed, 77 insertions, 0 deletions
diff --git a/libs/plugin_python/src/mobase/wrappers/widgets.cpp b/libs/plugin_python/src/mobase/wrappers/widgets.cpp
new file mode 100644
index 0000000..e626540
--- /dev/null
+++ b/libs/plugin_python/src/mobase/wrappers/widgets.cpp
@@ -0,0 +1,77 @@
+#include "wrappers.h"
+
+#include "../pybind11_all.h"
+
+#include <uibase/report.h>
+
+namespace py = pybind11;
+using namespace MOBase;
+
+namespace mo2::python {
+
+    void add_widget_bindings(pybind11::module_ m)
+    {
+        // TaskDialog is also in Windows System.
+        using TaskDialog = MOBase::TaskDialog;
+
+        // TaskDialog
+        py::class_<TaskDialogButton>(m, "TaskDialogButton")
+            .def(py::init<QString, QString, QMessageBox::StandardButton>(),
+                 py::arg("text"), py::arg("description"), py::arg("button"))
+            .def(py::init<QString, QMessageBox::StandardButton>(), py::arg("text"),
+                 py::arg("button"))
+            .def_readwrite("text", &TaskDialogButton::text)
+            .def_readwrite("description", &TaskDialogButton::description)
+            .def_readwrite("button", &TaskDialogButton::button);
+
+        py::class_<TaskDialog>(m, "TaskDialog")
+            .def(py::init([](QWidget* parent, QString const& title, QString const& main,
+                             QString const& content, QString const& details,
+                             QMessageBox::Icon icon,
+                             std::vector<TaskDialogButton> const& buttons,
+                             std::variant<QString, std::tuple<QString, QString>> const&
+                                 remember) {
+                     auto* dialog = new TaskDialog(parent, title);
+                     dialog->main(main).content(content).details(details).icon(icon);
+
+                     for (auto& button : buttons) {
+                         dialog->button(button);
+                     }
+
+                     std::visit(
+                         [dialog](auto const& item) {
+                             QString action, file;
+                             if constexpr (std::is_same_v<std::decay_t<decltype(item)>,
+                                                          QString>) {
+                                 action = item;
+                             }
+                             else {
+                                 action = std::get<0>(item);
+                                 file   = std::get<1>(item);
+                             }
+                             dialog->remember(action, file);
+                         },
+                         remember);
+
+                     return dialog;
+                 }),
+                 py::return_value_policy::take_ownership,
+                 py::arg("parent") = static_cast<QWidget*>(nullptr),
+                 py::arg("title") = "", py::arg("main") = "", py::arg("content") = "",
+                 py::arg("details") = "", py::arg("icon") = QMessageBox::NoIcon,
+                 py::arg("buttons")  = std::vector<TaskDialogButton>{},
+                 py::arg("remember") = "")
+            .def("setTitle", &TaskDialog::title, py::arg("title"))
+            .def("setMain", &TaskDialog::main, py::arg("main"))
+            .def("setContent", &TaskDialog::content, py::arg("content"))
+            .def("setDetails", &TaskDialog::details, py::arg("details"))
+            .def("setIcon", &TaskDialog::icon, py::arg("icon"))
+            .def("addButton", &TaskDialog::button, py::arg("button"))
+            .def("setRemember", &TaskDialog::remember, py::arg("action"),
+                 py::arg("file") = "")
+            .def("setWidth", &TaskDialog::setWidth, py::arg("width"))
+            .def("addContent", &TaskDialog::addContent, py::arg("widget"))
+            .def("exec", &TaskDialog::exec);
+    }
+
+}  // namespace mo2::python