From 0fd5de948958b9b2a4e3d797d4e0fd116aa618d7 Mon Sep 17 00:00:00 2001
From: tychovrahe <brunam@seznam.cz>
Date: Tue, 4 Apr 2023 16:16:02 +0200
Subject: [PATCH] TEST: poll ble events

---
 core/mocks/generated/trezorio/__init__.pyi     |  1 +
 core/src/trezor/ui/layouts/tt_v2/homescreen.py | 16 +++++++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/core/mocks/generated/trezorio/__init__.pyi b/core/mocks/generated/trezorio/__init__.pyi
index 6f042a9a0..2570c2ef3 100644
--- a/core/mocks/generated/trezorio/__init__.pyi
+++ b/core/mocks/generated/trezorio/__init__.pyi
@@ -202,4 +202,5 @@ BUTTON_RELEASED: int  # button up event
 BUTTON_LEFT: int  # button number of left button
 BUTTON_RIGHT: int  # button number of right button
 USB_CHECK: int # interface id for check of USB data connection
+BLE_CHECK: int # interface id for check of BLE data connection
 WireInterface = Union[HID, WebUSB, BleInterface]
diff --git a/core/src/trezor/ui/layouts/tt_v2/homescreen.py b/core/src/trezor/ui/layouts/tt_v2/homescreen.py
index 16478303b..6821fd017 100644
--- a/core/src/trezor/ui/layouts/tt_v2/homescreen.py
+++ b/core/src/trezor/ui/layouts/tt_v2/homescreen.py
@@ -75,8 +75,22 @@ class Homescreen(HomescreenBase):
             self.layout.paint()
             ui.refresh()
 
+    async def ble_checker_task(self) -> None:
+        from trezor import io, loop
+
+        blecheck = loop.wait(io.BLE_CHECK)
+        while True:
+            is_connected = await blecheck
+            self.layout.usb_event(is_connected)
+            self.layout.paint()
+            ui.refresh()
+
     def create_tasks(self) -> Tuple[loop.AwaitableTask, ...]:
-        return super().create_tasks() + (self.usb_checker_task(),)
+        return (
+            super().create_tasks()
+            + (self.usb_checker_task(),)
+            + (self.ble_checker_task(),)
+        )
 
 
 class Lockscreen(HomescreenBase):