From fc64bcfe172d8f06c8f35b5fe95da8033adbbc06 Mon Sep 17 00:00:00 2001
From: Tr2e <peng.sun@plaso.me>
Date: Sat, 8 Aug 2026 15:54:39 +0800
Subject: [PATCH 1/2] feat: add Wi-Fi provisioning and status header

---
 main/apps/app_setup/app_setup.cpp             |  10 ++
 main/apps/app_setup/workers/wifi.cpp          | 118 +++++++++++++
 main/apps/app_setup/workers/workers.h         |  25 +++
 .../common/arc_top_clock/arc_top_clock.cpp    | 158 ++++++++++++++++--
 .../apps/common/arc_top_clock/arc_top_clock.h |   7 +
 main/main.cpp                                 |  73 ++++++++
 6 files changed, 378 insertions(+), 13 deletions(-)
 create mode 100644 main/apps/app_setup/workers/wifi.cpp

diff --git a/main/apps/app_setup/app_setup.cpp b/main/apps/app_setup/app_setup.cpp
index 28c98af5..29c3a794 100644
--- a/main/apps/app_setup/app_setup.cpp
+++ b/main/apps/app_setup/app_setup.cpp
@@ -55,6 +55,11 @@ void AppSetup::onOpen()
                      _destroy_menu = true;
                      _worker       = std::make_unique<ButtonWorker>();
                  }},
+                {"Wi-Fi",
+                 [&]() {
+                     _destroy_menu = true;
+                     _worker       = std::make_unique<WifiWorker>();
+                 }},
             },
         },
         {
@@ -114,7 +119,12 @@ void AppSetup::onRunning()
     if (_worker) {
         _worker->update();
         if (_worker->isDone()) {
+            const bool exit_app = _worker->shouldExitApp();
             _worker.reset();
+            if (exit_app) {
+                close();
+                return;
+            }
             _menu_page = std::make_unique<view::SelectMenuPage>(_menu_sections);
         }
     }
diff --git a/main/apps/app_setup/workers/wifi.cpp b/main/apps/app_setup/workers/wifi.cpp
new file mode 100644
index 00000000..5465d10d
--- /dev/null
+++ b/main/apps/app_setup/workers/wifi.cpp
@@ -0,0 +1,118 @@
+/*
+ * SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
+ *
+ * SPDX-License-Identifier: MIT
+ */
+#include "workers.h"
+#include <assets/assets.h>
+#include <mooncake_log.h>
+#include <ssid_manager.h>
+#include <wifi_manager.h>
+
+using namespace uitk;
+using namespace uitk::lvgl_cpp;
+using namespace setup_workers;
+
+class WifiWorker::WifiView {
+public:
+    WifiView()
+    {
+        _panel = std::make_unique<Container>(lv_screen_active());
+        _panel->setSize(466, 466);
+        _panel->setAlign(LV_ALIGN_CENTER);
+        _panel->setRadius(0);
+        _panel->setBorderWidth(0);
+        _panel->setPadding(42, 92, 42, 42);
+        _panel->setBgColor(lv_color_black());
+        _panel->removeFlag(LV_OBJ_FLAG_SCROLLABLE);
+
+        _title = std::make_unique<Label>(_panel->get());
+        _title->setTextFont(&MontserratSemiBold26);
+        _title->setTextColor(lv_color_hex(0xFFFFFF));
+        _title->align(LV_ALIGN_TOP_MID, 0, 0);
+        _title->setText("Wi-Fi");
+
+        _status = std::make_unique<Label>(_panel->get());
+        _status->setTextFont(&lv_font_montserrat_18);
+        _status->setTextColor(lv_color_hex(0xFFFFFF));
+        _status->align(LV_ALIGN_TOP_MID, 0, 72);
+        _status->setWidth(380);
+        _status->setTextAlign(LV_TEXT_ALIGN_CENTER);
+
+        _detail = std::make_unique<Label>(_panel->get());
+        _detail->setTextFont(&lv_font_montserrat_14);
+        _detail->setTextColor(lv_color_hex(0xBFBFBF));
+        _detail->align(LV_ALIGN_TOP_MID, 0, 150);
+        _detail->setWidth(380);
+        _detail->setTextAlign(LV_TEXT_ALIGN_CENTER);
+    }
+
+    void setText(const std::string& status, const std::string& detail)
+    {
+        _status->setText(status);
+        _detail->setText(detail);
+    }
+
+private:
+    std::unique_ptr<Container> _panel;
+    std::unique_ptr<Label> _title;
+    std::unique_ptr<Label> _status;
+    std::unique_ptr<Label> _detail;
+};
+
+WifiWorker::WifiWorker()
+{
+    auto& wifi        = WifiManager::GetInstance();
+    _restore_station  = !SsidManager::GetInstance().GetSsidList().empty();
+
+    if (_restore_station || !wifi.IsConfigMode()) {
+        wifi.StartConfigAp();
+    }
+
+    _view             = std::make_unique<WifiView>();
+    _next_update_tick = 0;
+}
+
+WifiWorker::~WifiWorker()
+{
+    auto& wifi = WifiManager::GetInstance();
+    if (wifi.IsConfigMode() && _restore_station) {
+        wifi.StopConfigAp();
+        wifi.StartStation();
+    }
+}
+
+void WifiWorker::update()
+{
+    const auto now = GetHAL().millis();
+    if (now < _next_update_tick || !_view) {
+        return;
+    }
+    _next_update_tick = now + 500;
+
+    auto& wifi = WifiManager::GetInstance();
+    const bool has_saved_wifi = !SsidManager::GetInstance().GetSsidList().empty();
+
+    // The portal exits only after the H5 form has been submitted successfully.
+    // Once the saved network is connected, return all the way to Launcher.
+    if (!_config_exit_seen && !wifi.IsConfigMode() && has_saved_wifi) {
+        _config_exit_seen = true;
+    }
+    if (_config_exit_seen && wifi.IsConnected()) {
+        _exit_to_launcher = true;
+        _is_done           = true;
+        return;
+    }
+
+    if (wifi.IsConfigMode()) {
+        _view->setText(
+            "Connect phone to device Wi-Fi",
+            fmt::format("{}\nOpen http://192.168.4.1\nThe phone may open this page automatically.", wifi.GetApSsid()));
+    } else if (wifi.IsConnected()) {
+        _view->setText("Wi-Fi connected", fmt::format("{}\nIP: {}", wifi.GetSsid(), wifi.GetIpAddress()));
+    } else if (has_saved_wifi) {
+        _view->setText("Connecting...", "Saved Wi-Fi is being retried.");
+    } else {
+        _view->setText("Wi-Fi setup", "Connect to the device Wi-Fi to configure it.");
+    }
+}
diff --git a/main/apps/app_setup/workers/workers.h b/main/apps/app_setup/workers/workers.h
index 0de6fe1b..a374d5a7 100644
--- a/main/apps/app_setup/workers/workers.h
+++ b/main/apps/app_setup/workers/workers.h
@@ -32,6 +32,11 @@ class WorkerBase {
         return _is_done;
     }
 
+    virtual bool shouldExitApp() const
+    {
+        return false;
+    }
+
 protected:
     bool _is_done = false;
 };
@@ -138,4 +143,24 @@ class AboutWorker : public WorkerBase {
     int _pending_burst_steps     = 0;
 };
 
+class WifiWorker : public WorkerBase {
+public:
+    WifiWorker();
+    ~WifiWorker();
+    void update() override;
+    bool shouldExitApp() const override
+    {
+        return _exit_to_launcher;
+    }
+
+private:
+    class WifiView;
+
+    std::unique_ptr<WifiView> _view;
+    uint32_t _next_update_tick = 0;
+    bool _restore_station       = false;
+    bool _config_exit_seen      = false;
+    bool _exit_to_launcher      = false;
+};
+
 }  // namespace setup_workers
diff --git a/main/apps/common/arc_top_clock/arc_top_clock.cpp b/main/apps/common/arc_top_clock/arc_top_clock.cpp
index 52361b62..83af6343 100644
--- a/main/apps/common/arc_top_clock/arc_top_clock.cpp
+++ b/main/apps/common/arc_top_clock/arc_top_clock.cpp
@@ -7,6 +7,7 @@
 #include <assets/assets.h>
 #include <mooncake_log.h>
 #include <hal/hal.h>
+#include <wifi_manager.h>
 #include <string>
 #include <ctime>
 #include <cmath>
@@ -15,28 +16,68 @@ using namespace uitk;
 using namespace uitk::lvgl_cpp;
 using namespace view;
 
-static const std::vector<int> _label_rotation_map = {-100, -46, 0, 32, 84};
-static const std::vector<int> _label_x_map        = {3, 3, 1, 4, 4};
-static const std::vector<int> _label_y_map        = {0, 0, -2, -1, -1};
+// All top indicators share the same circular geometry, based on the centre
+// of the 466 x 466 AMOLED. Angles start at 12 o'clock and increase clockwise.
+static constexpr float _screen_center       = 233.0f;
+static constexpr float _clock_radius        = 200.0f;
+// Balance the actual widths of MM/DD, HH:MM and the Wi-Fi glyph as one
+// header group. The time is intentionally offset a little right so the
+// complete three-part composition is centred at 12 o'clock.
+static constexpr float _date_start_angle    = -34.0f;
+static constexpr float _clock_start_angle   = -3.0f;
+static constexpr float _clock_angle_step    = 5.0f;
+static constexpr float _wifi_angle          = 31.5f;
+// The connection point is 10 px inward from the canvas pivot. Keeping the
+// pivot at 210 therefore puts that bottom point on the time's 200 px arc.
+static constexpr float _wifi_radius         = 210.0f;
+static constexpr int _wifi_icon_width       = 42;
+static constexpr int _wifi_icon_height      = 38;
+
+static float degrees_to_radians(float degrees)
+{
+    return degrees * static_cast<float>(M_PI) / 180.0f;
+}
 
 void ArcTopClock::init()
 {
-    setSize(114, 36);
+    // This is only a lifecycle owner. The clock glyphs themselves live on
+    // the top layer so they can be positioned directly on the circular edge.
+    setSize(1, 1);
     setBorderWidth(0);
     setOutlineWidth(0);
     setPaddingAll(0);
     setBgOpa(LV_OPA_TRANSP);
     removeFlag(LV_OBJ_FLAG_SCROLLABLE);
 
-    for (int i = 0; i < labels.size(); i++) {
-        labels[i] = std::make_unique<Label>(get());
-        labels[i]->setText(".");
-        labels[i]->setTextFont(&lv_font_maple_mono_medium_28);
-        labels[i]->setTextColor(lv_color_hex(color));
-        labels[i]->setTransformPivot(0, displayRadius);
-        labels[i]->align(LV_ALIGN_CENTER, _label_x_map[i], _label_y_map[i]);
-        labels[i]->setRotation(_label_rotation_map[i]);
-    }
+    auto init_labels = [this](auto& group) {
+        for (auto& label : group) {
+            label = std::make_unique<Label>(lv_layer_top());
+            label->setText(".");
+            label->setTextFont(&lv_font_maple_mono_medium_28);
+            label->setTextColor(lv_color_hex(color));
+            label->setBgOpa(LV_OPA_TRANSP);
+            label->setBorderWidth(0);
+        }
+    };
+    init_labels(date_labels);
+    init_labels(labels);
+
+    // Draw the complete indicator in one canvas. This avoids the unreliable
+    // layout of tiny nested LVGL objects on this display.
+    wifi_icon = std::make_unique<Canvas>(lv_layer_top());
+    wifi_icon->setSize(_wifi_icon_width, _wifi_icon_height);
+    wifi_icon->setBgOpa(LV_OPA_TRANSP);
+    wifi_icon->setBorderWidth(0);
+    wifi_icon->removeFlag(LV_OBJ_FLAG_SCROLLABLE);
+    wifi_icon->addFlag(LV_OBJ_FLAG_FLOATING);
+    wifi_icon->createBuffer(_wifi_icon_width, _wifi_icon_height, LV_COLOR_FORMAT_ARGB8888);
+
+    const float wifi_radians = degrees_to_radians(_wifi_angle);
+    const int wifi_center_x  = std::lround(_screen_center + _wifi_radius * std::sin(wifi_radians));
+    const int wifi_center_y  = std::lround(_screen_center - _wifi_radius * std::cos(wifi_radians));
+    wifi_icon->setPos(wifi_center_x - _wifi_icon_width / 2, wifi_center_y - _wifi_icon_height / 2);
+    wifi_icon->setTransformPivot(_wifi_icon_width / 2, _wifi_icon_height / 2);
+    wifi_icon->setRotation(std::lround(_wifi_angle * 10.0f));
 
     update(true);
 }
@@ -46,12 +87,56 @@ void ArcTopClock::update(bool force)
     if (force || GetHAL().millis() - update_time_count > updateInterval) {
         std::time_t now    = std::time(nullptr);
         std::tm* localTime = std::localtime(&now);
+        set_date_to(fmt::format("{:02d}/{:02d}", localTime->tm_mon + 1, localTime->tm_mday));
         set_clock_to(fmt::format("{:02d}:{:02d}", localTime->tm_hour, localTime->tm_min));
+        update_wifi_indicator();
         // set_clock_to("00:00");
         update_time_count = GetHAL().millis();
     }
 }
 
+void ArcTopClock::update_wifi_indicator()
+{
+    auto& wifi = WifiManager::GetInstance();
+
+    int level  = 0;
+    if (wifi.IsConnected()) {
+        const int rssi = wifi.GetRssi();
+        level          = rssi >= -55 ? 4 : rssi >= -67 ? 3 : rssi >= -75 ? 2 : rssi >= -85 ? 1 : 0;
+    }
+
+    draw_wifi_indicator(level);
+}
+
+void ArcTopClock::draw_wifi_indicator(int level)
+{
+    // Standard Wi-Fi glyph: a connection point and three nested arches.
+    // RSSI controls the number of white arches; inactive arches remain dim.
+    wifi_icon->fillBg(lv_color_black(), LV_OPA_TRANSP);
+    wifi_icon->startDrawing();
+
+    constexpr int center_x = _wifi_icon_width / 2;
+    constexpr int center_y = 29;
+    // Keep a clear gap between the connection dot and the inner arch while
+    // matching the thin stroke weight of the clock text.
+    constexpr int radii[]  = {10, 15, 20};
+    constexpr int widths[] = {3, 3, 3};
+
+    for (int i = 0; i < 3; ++i) {
+        const bool active      = level >= (i + 2);
+        const auto arc_color   = lv_color_hex(active ? 0xFFFFFF : 0x5D6875);
+        const auto arc_opacity = active ? LV_OPA_COVER : LV_OPA_50;
+        // 225° to 315° traces a top-facing arch in LVGL's clockwise system.
+        wifi_icon->drawArc(center_x, center_y, radii[i], 225, 315, arc_color, arc_opacity, widths[i], true);
+    }
+
+    const auto dot_color = lv_color_hex(level > 0 ? 0xFFFFFF : 0x5D6875);
+    wifi_icon->drawCircle(center_x, center_y, 3, dot_color, level > 0 ? LV_OPA_COVER : LV_OPA_50);
+
+    wifi_icon->finishDrawing();
+    lv_obj_invalidate(wifi_icon->get());
+}
+
 void ArcTopClock::set_clock_to(const std::string_view text)
 {
     const int count = std::min((int)text.size(), (int)labels.size());
@@ -63,4 +148,51 @@ void ArcTopClock::set_clock_to(const std::string_view text)
             labels[i]->setText(buf);
         }
     }
+    layout_clock();
+}
+
+void ArcTopClock::set_date_to(const std::string_view text)
+{
+    const int count = std::min((int)text.size(), (int)date_labels.size());
+    for (int i = 0; i < count; i++) {
+        if (text[i] == '0') {
+            date_labels[i]->setText("O");
+        } else {
+            char buf[2] = {text[i], '\0'};
+            date_labels[i]->setText(buf);
+        }
+    }
+    layout_date();
+}
+
+void ArcTopClock::layout_date()
+{
+    for (int i = 0; i < date_labels.size(); ++i) {
+        const float angle    = _date_start_angle + _clock_angle_step * i;
+        const float radians  = degrees_to_radians(angle);
+        const int center_x   = std::lround(_screen_center + _clock_radius * std::sin(radians));
+        const int center_y   = std::lround(_screen_center - _clock_radius * std::cos(radians));
+        const int label_w    = lv_obj_get_width(date_labels[i]->get());
+        const int label_h    = lv_obj_get_height(date_labels[i]->get());
+
+        date_labels[i]->setTransformPivot(label_w / 2, label_h / 2);
+        date_labels[i]->setRotation(std::lround(angle * 10.0f));
+        date_labels[i]->setPos(center_x - label_w / 2, center_y - label_h / 2);
+    }
+}
+
+void ArcTopClock::layout_clock()
+{
+    for (int i = 0; i < labels.size(); ++i) {
+        const float angle    = _clock_start_angle + _clock_angle_step * i;
+        const float radians  = degrees_to_radians(angle);
+        const int center_x   = std::lround(_screen_center + _clock_radius * std::sin(radians));
+        const int center_y   = std::lround(_screen_center - _clock_radius * std::cos(radians));
+        const int label_w    = lv_obj_get_width(labels[i]->get());
+        const int label_h    = lv_obj_get_height(labels[i]->get());
+
+        labels[i]->setTransformPivot(label_w / 2, label_h / 2);
+        labels[i]->setRotation(std::lround(angle * 10.0f));
+        labels[i]->setPos(center_x - label_w / 2, center_y - label_h / 2);
+    }
 }
diff --git a/main/apps/common/arc_top_clock/arc_top_clock.h b/main/apps/common/arc_top_clock/arc_top_clock.h
index a764c718..a9f0f858 100644
--- a/main/apps/common/arc_top_clock/arc_top_clock.h
+++ b/main/apps/common/arc_top_clock/arc_top_clock.h
@@ -28,10 +28,17 @@ class ArcTopClock : public uitk::lvgl_cpp::Widget<lv_obj_create> {
     void update(bool force = false);
 
 protected:
+    std::array<std::unique_ptr<uitk::lvgl_cpp::Label>, 5> date_labels;
     std::array<std::unique_ptr<uitk::lvgl_cpp::Label>, 5> labels;
+    std::unique_ptr<uitk::lvgl_cpp::Canvas> wifi_icon;
     uint32_t update_time_count = 0;
 
     void set_clock_to(const std::string_view text);
+    void set_date_to(const std::string_view text);
+    void layout_date();
+    void layout_clock();
+    void update_wifi_indicator();
+    void draw_wifi_indicator(int level);
 };
 
 }  // namespace view
diff --git a/main/main.cpp b/main/main.cpp
index ba08782b..642ed38f 100644
--- a/main/main.cpp
+++ b/main/main.cpp
@@ -11,6 +11,78 @@
 #include <hal/hal.h>
 #include <lv_demos.h>
 #include <apps/common/audio/audio.h>
+#include <freertos/FreeRTOS.h>
+#include <freertos/task.h>
+#include <esp_sntp.h>
+#include <ssid_manager.h>
+#include <sys/time.h>
+#include <hal/utils/settings/settings.h>
+#include <wifi_manager.h>
+
+static void sync_time_to_rtc(struct timeval*)
+{
+    GetHAL().syncSystemTimeToRtc();
+}
+
+static void start_network_time_sync()
+{
+    static bool initialized = false;
+    if (!initialized) {
+        esp_sntp_setoperatingmode(SNTP_OPMODE_POLL);
+        esp_sntp_setservername(0, "pool.ntp.org");
+        esp_sntp_setservername(1, "time.nist.gov");
+        esp_sntp_set_time_sync_notification_cb(sync_time_to_rtc);
+        esp_sntp_init();
+        initialized = true;
+    } else {
+        esp_sntp_restart();
+    }
+}
+
+static void init_wifi()
+{
+    auto& wifi = WifiManager::GetInstance();
+
+    WifiManagerConfig config;
+    config.ssid_prefix = "M5StopWatch";
+    config.language    = "zh-CN";
+    wifi.Initialize(config);
+
+    // Once the captive portal has saved credentials and exited, resume STA mode
+    // automatically. The small delay lets the HTTP response finish first.
+    wifi.SetEventCallback([](WifiEvent event, const std::string&) {
+        if (event == WifiEvent::Connected) {
+            start_network_time_sync();
+            return;
+        }
+        if (event != WifiEvent::ConfigModeExit) {
+            return;
+        }
+
+        xTaskCreate(
+            [](void*) {
+                vTaskDelay(pdMS_TO_TICKS(250));
+                auto& wifi = WifiManager::GetInstance();
+                if (!wifi.IsConfigMode() && !SsidManager::GetInstance().GetSsidList().empty()) {
+                    // Apply the timezone captured by the phone's browser before
+                    // the first SNTP result is rendered on the watch face.
+                    Settings settings("system", false);
+                    GetHAL().setTimezone(settings.GetString("tz", "GMT0"));
+                    wifi.StartStation();
+                }
+                vTaskDelete(nullptr);
+            },
+            "wifi_resume_sta", 4096, nullptr, 5, nullptr);
+    });
+
+    if (SsidManager::GetInstance().GetSsidList().empty()) {
+        // First boot / no saved credentials: start the phone-based captive portal.
+        wifi.StartConfigAp();
+    } else {
+        // Saved credentials are kept in NVS and retried with backoff by the component.
+        wifi.StartStation();
+    }
+}
 
 using namespace mooncake;
 using namespace smooth_ui_toolkit;
@@ -23,6 +95,7 @@ extern "C" void app_main(void)
 
     // HAL init
     GetHAL().init();
+    init_wifi();
 
     // Setup ui hal
     ui_hal::on_delay([](uint32_t ms) { GetHAL().delay(ms); });

From 2a1ac3bc3db05f05b565e7483a9989c581923eab Mon Sep 17 00:00:00 2001
From: Tr2e <tr2e@sina.com>
Date: Mon, 17 Aug 2026 00:17:15 +0800
Subject: [PATCH 2/2] fix: stabilize arc clock layout after app return

---
 main/apps/common/arc_top_clock/arc_top_clock.cpp | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/main/apps/common/arc_top_clock/arc_top_clock.cpp b/main/apps/common/arc_top_clock/arc_top_clock.cpp
index 83af6343..115f5131 100644
--- a/main/apps/common/arc_top_clock/arc_top_clock.cpp
+++ b/main/apps/common/arc_top_clock/arc_top_clock.cpp
@@ -89,6 +89,13 @@ void ArcTopClock::update(bool force)
         std::tm* localTime = std::localtime(&now);
         set_date_to(fmt::format("{:02d}/{:02d}", localTime->tm_mon + 1, localTime->tm_mday));
         set_clock_to(fmt::format("{:02d}:{:02d}", localTime->tm_hour, localTime->tm_min));
+        // Text and font changes invalidate LVGL's content-sized labels. Flush
+        // that deferred layout before reading their dimensions, otherwise the
+        // first frame after returning to Launcher uses the old placeholder
+        // size and visibly jumps on the next one-second update.
+        lv_obj_update_layout(lv_layer_top());
+        layout_date();
+        layout_clock();
         update_wifi_indicator();
         // set_clock_to("00:00");
         update_time_count = GetHAL().millis();
@@ -148,7 +155,6 @@ void ArcTopClock::set_clock_to(const std::string_view text)
             labels[i]->setText(buf);
         }
     }
-    layout_clock();
 }
 
 void ArcTopClock::set_date_to(const std::string_view text)
@@ -162,7 +168,6 @@ void ArcTopClock::set_date_to(const std::string_view text)
             date_labels[i]->setText(buf);
         }
     }
-    layout_date();
 }
 
 void ArcTopClock::layout_date()