python ui: wifi manager (#34814)

* python wifi manager * fix ui * need auth callback * move to widgets * confirm forgot * add drag detection * improve keyboard & list * remove duplicate * typos * use gui_app render * refactor * cleanup * cleanup * shutdown * fix types * revert * scroll panel cleanup * reset is_dragging on mouse release * Revert "reset is_dragging on mouse release" This reverts commit ff5e51cf6f00848d93aa3ce0bab16602fea7a319. * unformat * cleanup * update state when connecting * forgotten callback * maybe fix? first tap didn't work * Revert "maybe fix? first tap didn't work" This reverts commit 739f0e3bd37323d10479b081a20d05c7fdff5495. * remove set_target_fps * Revert "remove set_target_fps" This reverts commit 96f74553ef6fa2ab2a4d1a07880175e7da628c9f. --------- Co-authored-by: Cameron Clough <cameronjclough@gmail.com>
2026-10-01 03:43:46 +08:00 · 2025-05-05 23:28:12 +08:00
parent 87fae0c6f2
commit c33e5b3209
4 changed files with 692 additions and 12 deletions
@@ -44,25 +44,28 @@ keyboard_layouts = {
 class Keyboard:
  def __init__(self, max_text_size: int = 255):
    self._layout = keyboard_layouts["lowercase"]
-    self._input_text = ""
    self._max_text_size = max_text_size
+    self._string_pointer = rl.ffi.new("char[]", max_text_size)
+    self._input_text = ""
+    self._clear()

  @property
-  def text(self) -> str:
-    return self._input_text
-
-  def clear(self):
-    self._input_text = ""
+  def text(self):
+    result = rl.ffi.string(self._string_pointer).decode("utf-8")
+    self._clear()
+    return result

  def render(self, rect, title, sub_title):
    gui_label(rl.Rectangle(rect.x, rect.y, rect.width, 95), title, 90)
    gui_label(rl.Rectangle(rect.x, rect.y + 95, rect.width, 60), sub_title, 55, rl.GRAY)
    if gui_button(rl.Rectangle(rect.x + rect.width - 300, rect.y, 300, 100), "Cancel"):
-      return -1
+      self._clear()
+      return 0

    # Text box for input
-    rl.gui_text_box(rl.Rectangle(rect.x, rect.y + 160, rect.width, 100), self._input_text, self._max_text_size, True)
-
+    self._sync_string_pointer()
+    rl.gui_text_box(rl.Rectangle(rect.x, rect.y + 160, rect.width, 100), self._string_pointer, self._max_text_size, True)
+    self._input_text = rl.ffi.string(self._string_pointer).decode("utf-8")
    h_space, v_space = 15, 15
    row_y_start = rect.y + 300  # Starting Y position for the first row
    key_height = (rect.height - 300 - 3 * v_space) / 4
@@ -87,7 +90,7 @@ class Keyboard:
          else:
            self.handle_key_press(key)

-    return 0
+    return -1

  def handle_key_press(self, key):
    if key in (SHIFT_DOWN_KEY, ABC_KEY):
@@ -102,3 +105,14 @@ class Keyboard:
      self._input_text = self._input_text[:-1]
    elif key != BACKSPACE_KEY and len(self._input_text) < self._max_text_size:
      self._input_text += key
+
+  def _clear(self):
+    self._input_text = ''
+    self._string_pointer[0] = b'\0'
+
+  def _sync_string_pointer(self):
+    """Sync the C-string pointer with the internal Python string."""
+    encoded = self._input_text.encode("utf-8")[:self._max_text_size - 1]  # Leave room for the null terminator
+    buffer = rl.ffi.buffer(self._string_pointer)
+    buffer[:len(encoded)] = encoded
+    self._string_pointer[len(encoded)] = b'\0'  # Null terminator
@@ -0,0 +1,172 @@
+from dataclasses import dataclass
+from typing import Literal
+
+import pyray as rl
+from openpilot.system.ui.lib.wifi_manager import NetworkInfo, WifiManagerCallbacks, WifiManagerWrapper
+from openpilot.system.ui.lib.application import gui_app
+from openpilot.system.ui.lib.button import gui_button
+from openpilot.system.ui.lib.label import gui_label
+from openpilot.system.ui.lib.scroll_panel import GuiScrollPanel
+from openpilot.system.ui.widgets.keyboard import Keyboard
+from openpilot.system.ui.widgets.confirm_dialog import confirm_dialog
+
+NM_DEVICE_STATE_NEED_AUTH = 60
+ITEM_HEIGHT = 160
+
+
+@dataclass
+class StateIdle:
+  action: Literal["idle"] = "idle"
+
+@dataclass
+class StateConnecting:
+  network: NetworkInfo
+  action: Literal["connecting"] = "connecting"
+
+@dataclass
+class StateNeedsAuth:
+  network: NetworkInfo
+  action: Literal["needs_auth"] = "needs_auth"
+
+@dataclass
+class StateShowForgetConfirm:
+  network: NetworkInfo
+  action: Literal["show_forget_confirm"] = "show_forget_confirm"
+
+@dataclass
+class StateForgetting:
+  network: NetworkInfo
+  action: Literal["forgetting"] = "forgetting"
+
+UIState = StateIdle | StateConnecting | StateNeedsAuth | StateShowForgetConfirm | StateForgetting
+
+
+class WifiManagerUI:
+  def __init__(self, wifi_manager: WifiManagerWrapper):
+    self.state: UIState = StateIdle()
+    self.btn_width = 200
+    self.scroll_panel = GuiScrollPanel()
+    self.keyboard = Keyboard()
+
+    self.wifi_manager = wifi_manager
+    self.wifi_manager.set_callbacks(WifiManagerCallbacks(self._on_need_auth, self._on_activated, self._on_forgotten))
+    self.wifi_manager.start()
+    self.wifi_manager.connect()
+
+  def render(self, rect: rl.Rectangle):
+    if not self.wifi_manager.networks:
+      gui_label(rect, "Scanning Wi-Fi networks...", 72, alignment=rl.GuiTextAlignment.TEXT_ALIGN_CENTER)
+      return
+
+    match self.state:
+      case StateNeedsAuth(network):
+        result = self.keyboard.render(rect, "Enter password", f"for {network.ssid}")
+        if result == 1:
+          self.connect_to_network(network, self.keyboard.text)
+        elif result == 0:
+          self.state = StateIdle()
+
+      case StateShowForgetConfirm(network):
+        result = confirm_dialog(rect, f'Forget Wi-Fi Network "{network.ssid}"?', "Forget")
+        if result == 1:
+          self.forget_network(network)
+        elif result == 0:
+          self.state = StateIdle()
+
+      case _:
+        self._draw_network_list(rect)
+
+  def _draw_network_list(self, rect: rl.Rectangle):
+    content_rect = rl.Rectangle(rect.x, rect.y, rect.width, len(self.wifi_manager.networks) * ITEM_HEIGHT)
+    offset = self.scroll_panel.handle_scroll(rect, content_rect)
+    clicked = self.scroll_panel.is_click_valid()
+
+    rl.begin_scissor_mode(int(rect.x), int(rect.y), int(rect.width), int(rect.height))
+    for i, network in enumerate(self.wifi_manager.networks):
+      y_offset = rect.y + i * ITEM_HEIGHT + offset.y
+      item_rect = rl.Rectangle(rect.x, y_offset, rect.width, ITEM_HEIGHT)
+      if not rl.check_collision_recs(item_rect, rect):
+        continue
+
+      self._draw_network_item(item_rect, network, clicked)
+      if i < len(self.wifi_manager.networks) - 1:
+        line_y = int(item_rect.y + item_rect.height - 1)
+        rl.draw_line(int(item_rect.x), int(line_y), int(item_rect.x + item_rect.width), line_y, rl.LIGHTGRAY)
+
+    rl.end_scissor_mode()
+
+  def _draw_network_item(self, rect, network: NetworkInfo, clicked: bool):
+    label_rect = rl.Rectangle(rect.x, rect.y, rect.width - self.btn_width * 2, ITEM_HEIGHT)
+    state_rect = rl.Rectangle(rect.x + rect.width - self.btn_width * 2 - 150, rect.y, 300, ITEM_HEIGHT)
+
+    gui_label(label_rect, network.ssid, 55)
+
+    status_text = ""
+    if network.is_connected:
+      status_text = "Connected"
+    match self.state:
+      case StateConnecting(network=connecting):
+        if connecting.ssid == network.ssid:
+          status_text = "CONNECTING..."
+      case StateForgetting(network=forgetting):
+        if forgetting.ssid == network.ssid:
+          status_text = "FORGETTING..."
+    if status_text:
+      rl.gui_label(state_rect, status_text)
+
+    # If the network is saved, show the "Forget" button
+    if self.wifi_manager.is_saved(network.ssid):
+      forget_btn_rect = rl.Rectangle(
+        rect.x + rect.width - self.btn_width,
+        rect.y + (ITEM_HEIGHT - 80) / 2,
+        self.btn_width,
+        80,
+      )
+      if isinstance(self.state, StateIdle) and gui_button(forget_btn_rect, "Forget") and clicked:
+        self.state = StateShowForgetConfirm(network)
+
+    if isinstance(self.state, StateIdle) and rl.check_collision_point_rec(rl.get_mouse_position(), label_rect) and clicked:
+      if not self.wifi_manager.is_saved(network.ssid):
+        self.state = StateNeedsAuth(network)
+      else:
+        self.connect_to_network(network)
+
+  def connect_to_network(self, network: NetworkInfo, password=''):
+    self.state = StateConnecting(network)
+    if self.wifi_manager.is_saved(network.ssid) and not password:
+      self.wifi_manager.activate_connection(network.ssid)
+    else:
+      self.wifi_manager.connect_to_network(network.ssid, password)
+
+  def forget_network(self, network: NetworkInfo):
+    self.state = StateForgetting(network)
+    self.wifi_manager.forget_connection(network.ssid)
+
+  def _on_need_auth(self):
+    match self.state:
+      case StateConnecting(network):
+        self.state = StateNeedsAuth(network)
+
+  def _on_activated(self):
+    if isinstance(self.state, StateConnecting):
+      self.state = StateIdle()
+
+  def _on_forgotten(self):
+    if isinstance(self.state, StateForgetting):
+      self.state = StateIdle()
+
+
+def main():
+  gui_app.init_window("Wi-Fi Manager")
+  wifi_manager = WifiManagerWrapper()
+  wifi_ui = WifiManagerUI(wifi_manager)
+
+  for _ in gui_app.render():
+    wifi_ui.render(rl.Rectangle(50, 50, gui_app.width - 100, gui_app.height - 100))
+
+  wifi_manager.shutdown()
+  gui_app.close()
+
+
+if __name__ == "__main__":
+  main()