[lvgl] Final stage (#7184)

Co-authored-by: Jesse Hills <3060199+jesserockz@users.noreply.github.com>

esphome/components/lvgl/select/__init__.py

@@ -0,0 +1,46 @@
+import esphome.codegen as cg
+from esphome.components import select
+import esphome.config_validation as cv
+from esphome.const import CONF_OPTIONS
+
+from ..defines import CONF_ANIMATED, CONF_LVGL_ID, CONF_WIDGET
+from ..lvcode import CUSTOM_EVENT, EVENT_ARG, LambdaContext, LvContext, lv, lv_add
+from ..schemas import LVGL_SCHEMA
+from ..types import LV_EVENT, LvSelect, lvgl_ns
+from ..widgets import get_widgets
+
+LVGLSelect = lvgl_ns.class_("LVGLSelect", select.Select)
+
+CONFIG_SCHEMA = (
+    select.select_schema(LVGLSelect)
+    .extend(LVGL_SCHEMA)
+    .extend(
+        {
+            cv.Required(CONF_WIDGET): cv.use_id(LvSelect),
+            cv.Optional(CONF_ANIMATED, default=False): cv.boolean,
+        }
+    )
+)
+
+
+async def to_code(config):
+    widget = await get_widgets(config, CONF_WIDGET)
+    widget = widget[0]
+    options = widget.config.get(CONF_OPTIONS, [])
+    selector = await select.new_select(config, options=options)
+    paren = await cg.get_variable(config[CONF_LVGL_ID])
+    async with LambdaContext(EVENT_ARG) as pub_ctx:
+        pub_ctx.add(selector.publish_index(widget.get_value()))
+    async with LambdaContext([(cg.uint16, "v")]) as control:
+        await widget.set_property("selected", "v", animated=config[CONF_ANIMATED])
+        lv.event_send(widget.obj, CUSTOM_EVENT, cg.nullptr)
+    async with LvContext(paren) as ctx:
+        lv_add(selector.set_control_lambda(await control.get_lambda()))
+        ctx.add(
+            paren.add_event_cb(
+                widget.obj,
+                await pub_ctx.get_lambda(),
+                LV_EVENT.VALUE_CHANGED,
+            )
+        )
+        lv_add(selector.publish_index(widget.get_value()))

esphome/components/lvgl/select/lvgl_select.h

@@ -0,0 +1,62 @@
+#pragma once
+
+#include "esphome/components/select/select.h"
+#include "esphome/core/automation.h"
+#include "esphome/core/component.h"
+#include "esphome/core/preferences.h"
+
+namespace esphome {
+namespace lvgl {
+
+static std::vector<std::string> split_string(const std::string &str) {
+  std::vector<std::string> strings;
+  auto delimiter = std::string("\n");
+
+  std::string::size_type pos;
+  std::string::size_type prev = 0;
+  while ((pos = str.find(delimiter, prev)) != std::string::npos) {
+    strings.push_back(str.substr(prev, pos - prev));
+    prev = pos + delimiter.size();
+  }
+
+  // To get the last substring (or only, if delimiter is not found)
+  strings.push_back(str.substr(prev));
+
+  return strings;
+}
+
+class LVGLSelect : public select::Select {
+ public:
+  void set_control_lambda(std::function<void(size_t)> lambda) {
+    this->control_lambda_ = lambda;
+    if (this->initial_state_.has_value()) {
+      this->control(this->initial_state_.value());
+      this->initial_state_.reset();
+    }
+  }
+
+  void publish_index(size_t index) {
+    auto value = this->at(index);
+    if (value)
+      this->publish_state(value.value());
+  }
+
+  void set_options(const char *str) { this->traits.set_options(split_string(str)); }
+
+ protected:
+  void control(const std::string &value) override {
+    if (this->control_lambda_ != nullptr) {
+      auto index = index_of(value);
+      if (index)
+        this->control_lambda_(index.value());
+    } else {
+      this->initial_state_ = value.c_str();
+    }
+  }
+
+  std::function<void(size_t)> control_lambda_{};
+  optional<const char *> initial_state_{};
+};
+
+}  // namespace lvgl
+}  // namespace esphome