From ba13a4b61cf9c8dba487fa9dad1116f07d7423a2 Mon Sep 17 00:00:00 2001
From: MihailRis <mexaruc@gmail.com>
Date: Sat, 28 Jun 2025 14:01:22 +0300
Subject: [PATCH] add 'options' selectbox property

---
 src/graphics/ui/elements/SelectBox.cpp  |  4 ++
 src/graphics/ui/elements/SelectBox.hpp  |  2 +
 src/logic/scripting/lua/libs/libgui.cpp | 49 +++++++++++++++++++++++++
 3 files changed, 55 insertions(+)

diff --git a/src/graphics/ui/elements/SelectBox.cpp b/src/graphics/ui/elements/SelectBox.cpp
index 98dfc826..102c346c 100644
--- a/src/graphics/ui/elements/SelectBox.cpp
+++ b/src/graphics/ui/elements/SelectBox.cpp
@@ -58,6 +58,10 @@ const std::vector<SelectBox::Option>& SelectBox::getOptions() const {
     return options;
 }
 
+void SelectBox::setOptions(std::vector<Option>&& options) {
+    this->options = std::move(options);
+}
+
 void SelectBox::drawBackground(const DrawContext& pctx, const Assets&) {
     glm::vec2 pos = calcPos();
     auto batch = pctx.getBatch2D();
diff --git a/src/graphics/ui/elements/SelectBox.hpp b/src/graphics/ui/elements/SelectBox.hpp
index 28795f22..1ee96839 100644
--- a/src/graphics/ui/elements/SelectBox.hpp
+++ b/src/graphics/ui/elements/SelectBox.hpp
@@ -32,6 +32,8 @@ namespace gui {
 
         const std::vector<Option>& getOptions() const;
 
+        void setOptions(std::vector<Option>&& options);
+
         void drawBackground(const DrawContext& pctx, const Assets&) override;
     };
 }
diff --git a/src/logic/scripting/lua/libs/libgui.cpp b/src/logic/scripting/lua/libs/libgui.cpp
index 66ac25c5..66a8f648 100644
--- a/src/logic/scripting/lua/libs/libgui.cpp
+++ b/src/logic/scripting/lua/libs/libgui.cpp
@@ -512,6 +512,28 @@ static int p_get_scroll(UINode* node, lua::State* L) {
     return 0;
 }
 
+static int p_get_options(UINode* node, lua::State* L) {
+    if (auto selectbox = dynamic_cast<SelectBox*>(node)) {
+        const auto& options = selectbox->getOptions();
+        size_t size = options.size();
+        lua::createtable(L, size, 0);
+        for (size_t i = 0; i < size; i++) {
+            const auto& option = options[i];
+            lua::createtable(L, 0, 2);
+
+            lua::pushstring(L, option.value);
+            lua::setfield(L, "value");
+
+            lua::pushwstring(L, option.text);
+            lua::setfield(L, "text");
+
+            lua::rawseti(L, i + 1);
+        }
+        return 1;
+    }
+    return 0;
+}
+
 static int l_gui_getattr(lua::State* L) {
     if (!lua::isstring(L, 1)) {
         throw std::runtime_error("document name is not a string");
@@ -598,6 +620,7 @@ static int l_gui_getattr(lua::State* L) {
             {"data", p_get_data},
             {"parent", p_get_parent},
             {"region", p_get_region},
+            {"options", p_get_options},
         };
     auto func = getters.find(attr);
     if (func != getters.end()) {
@@ -709,6 +732,31 @@ static void p_set_region(UINode* node, lua::State* L, int idx) {
         image->setRegion(UVRegion(vec.x, vec.y, vec.z, vec.w));
     }
 }
+static void p_set_options(UINode* node, lua::State* L, int idx) {
+    if (auto selectbox = dynamic_cast<SelectBox*>(node)) {
+        if (!lua::istable(L, idx)) {
+            throw std::runtime_error("options table expected");
+        }
+        std::vector<SelectBox::Option> options;
+        size_t size = lua::objlen(L, idx);
+        for (size_t i = 0; i < size; i++) {
+            lua::rawgeti(L, i + 1, idx);
+            
+            SelectBox::Option option;
+
+            lua::getfield(L, "value");
+            option.value = lua::require_string(L, -1);
+            lua::pop(L);
+
+            lua::getfield(L, "text");
+            option.text = lua::require_wstring(L, -1);
+            lua::pop(L, 2);
+            
+            options.push_back(std::move(option));
+        }
+        selectbox->setOptions(std::move(options));
+    }
+}
 static void p_set_value(UINode* node, lua::State* L, int idx) {
     if (auto bar = dynamic_cast<TrackBar*>(node)) {
         bar->setValue(lua::tonumber(L, idx));
@@ -856,6 +904,7 @@ static int l_gui_setattr(lua::State* L) {
             {"cursor", p_set_cursor},
             {"focused", p_set_focused},
             {"region", p_set_region},
+            {"options", p_set_options},
         };
     auto func = setters.find(attr);
     if (func != setters.end()) {