4 files changed, 11 insertions, 45 deletions
diff --git a/src/ui/include/lvgl_task.hpp b/src/ui/include/lvgl_task.hpp
index 32e01b7f..0d6e4cd3 100644
--- a/src/ui/include/lvgl_task.hpp
+++ b/src/ui/include/lvgl_task.hpp
@@ -13,7 +13,6 @@
 namespace ui {
 
 auto StartLvgl(std::weak_ptr<drivers::TouchWheel> touch_wheel,
-               std::weak_ptr<drivers::Display> display,
-               std::atomic<bool>* quit) -> bool;
+               std::weak_ptr<drivers::Display> display) -> void;
 
 }  // namespace ui
diff --git a/src/ui/include/ui_fsm.hpp b/src/ui/include/ui_fsm.hpp
index 3c06b8a0..e902838e 100644
--- a/src/ui/include/ui_fsm.hpp
+++ b/src/ui/include/ui_fsm.hpp
@@ -35,7 +35,6 @@ class UiState : public tinyfsm::Fsm<UiState> {
   static std::weak_ptr<drivers::TouchWheel> sTouchWheel;
   static std::weak_ptr<drivers::Display> sDisplay;
   static std::weak_ptr<database::Database> sDatabase;
-  static std::atomic<bool> sTaskQuit;
 };
 
 namespace states {
diff --git a/src/ui/lvgl_task.cpp b/src/ui/lvgl_task.cpp
index 70df9f72..6664f754 100644
--- a/src/ui/lvgl_task.cpp
+++ b/src/ui/lvgl_task.cpp
@@ -24,6 +24,7 @@
 #include "misc/lv_style.h"
 #include "misc/lv_timer.h"
 #include "touchwheel.hpp"
+#include "tasks.hpp"
 #include "widgets/lv_label.h"
 
 #include "display.hpp"
@@ -37,23 +38,12 @@ auto tick_hook(TimerHandle_t xTimer) -> void {
   lv_tick_inc(1);
 }
 
-struct LvglArgs {
-  std::weak_ptr<drivers::TouchWheel> touch_wheel;
-  std::weak_ptr<drivers::Display> display;
-  std::atomic<bool>* quit;
-};
+void LvglMain(std::weak_ptr<drivers::TouchWheel> weak_touch_wheel, std::weak_ptr<drivers::Display> weak_display) {
+    ESP_LOGI(kTag, "init lvgl");
+    lv_init();
 
-void LvglMain(void* voidArgs) {
-  LvglArgs* args = reinterpret_cast<LvglArgs*>(voidArgs);
-  std::weak_ptr<drivers::TouchWheel> weak_touch_wheel = args->touch_wheel;
-  std::weak_ptr<drivers::Display> weak_display = args->display;
-
-  std::atomic<bool>* quit = args->quit;
-  delete args;
-
-  {
-    TimerHandle_t tick_timer =
-        xTimerCreate("lv_tick", pdMS_TO_TICKS(1), pdTRUE, NULL, &tick_hook);
+    // LVGL has been initialised, so we can now start reporting ticks to it.
+    xTimerCreate("lv_tick", pdMS_TO_TICKS(1), pdTRUE, NULL, &tick_hook);
 
     lv_style_t style;
     lv_style_init(&style);
@@ -68,38 +58,17 @@ void LvglMain(void* voidArgs) {
     lv_obj_center(label);
     lv_scr_load(label);
 
-    while (!quit->load()) {
+    while (1) {
       lv_timer_handler();
       // 30 FPS
       // TODO(jacqueline): make this dynamic
       vTaskDelay(pdMS_TO_TICKS(33));
     }
-
-    // TODO(robin? daniel?): De-init the UI stack here.
-    lv_obj_del(label);
-    lv_style_reset(&style);
-
-    xTimerDelete(tick_timer, portMAX_DELAY);
-  }
-
-  vTaskDelete(NULL);
 }
 
-static const size_t kLvglStackSize = 8 * 1024;
-static StaticTask_t sLvglTaskBuffer = {};
-static StackType_t sLvglStack[kLvglStackSize] = {0};
-
 auto StartLvgl(std::weak_ptr<drivers::TouchWheel> touch_wheel,
-               std::weak_ptr<drivers::Display> display,
-               std::atomic<bool>* quit) -> bool {
-  LvglArgs* args = new LvglArgs();
-  args->touch_wheel = touch_wheel;
-  args->display = display;
-  args->quit = quit;
-
-  return xTaskCreateStaticPinnedToCore(&LvglMain, "LVGL", kLvglStackSize,
-                                       reinterpret_cast<void*>(args), 1,
-                                       sLvglStack, &sLvglTaskBuffer, 1);
+               std::weak_ptr<drivers::Display> display) -> void {
+  tasks::StartPersistent<tasks::Type::kUi>([=]() { LvglMain(touch_wheel, display); });
 }
 
 }  // namespace ui
diff --git a/src/ui/ui_fsm.cpp b/src/ui/ui_fsm.cpp
index 8ca201e2..72d3f392 100644
--- a/src/ui/ui_fsm.cpp
+++ b/src/ui/ui_fsm.cpp
@@ -10,7 +10,6 @@ drivers::GpioExpander* UiState::sGpioExpander;
 std::weak_ptr<drivers::TouchWheel> UiState::sTouchWheel;
 std::weak_ptr<drivers::Display> UiState::sDisplay;
 std::weak_ptr<database::Database> UiState::sDatabase;
-std::atomic<bool> UiState::sTaskQuit;
 
 auto UiState::Init(drivers::GpioExpander* gpio_expander,
                    std::weak_ptr<drivers::TouchWheel> touchwheel,
@@ -25,7 +24,7 @@ auto UiState::Init(drivers::GpioExpander* gpio_expander,
 namespace states {
 
 void PreBoot::react(const system_fsm::DisplayReady& ev) {
-  transit<Splash>([&]() { StartLvgl(sTouchWheel, sDisplay, &sTaskQuit); });
+  transit<Splash>([&]() { StartLvgl(sTouchWheel, sDisplay); });
 }
 
 void Splash::react(const system_fsm::BootComplete& ev) {