From ea169f5112fedda309afd1d97cd6c7744846876c Mon Sep 17 00:00:00 2001
From: tychovrahe <brunam@seznam.cz>
Date: Thu, 6 Apr 2023 12:47:04 +0200
Subject: [PATCH] TEST: Homescreen notification

---
 .../ui/model_tt/component/homescreen/mod.rs   | 21 +++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/core/embed/rust/src/ui/model_tt/component/homescreen/mod.rs b/core/embed/rust/src/ui/model_tt/component/homescreen/mod.rs
index 1dbabb5b9..9228dd0e3 100644
--- a/core/embed/rust/src/ui/model_tt/component/homescreen/mod.rs
+++ b/core/embed/rust/src/ui/model_tt/component/homescreen/mod.rs
@@ -2,7 +2,7 @@ mod render;
 
 use crate::{
     time::{Duration, Instant},
-    trezorhal::usb::usb_configured,
+    trezorhal::{ble::ble_connected, ble::start_advertising, usb::usb_configured},
     ui::{
         component::{Component, Event, EventCtx, Pad, TimerToken},
         display::{self, tjpgd::jpeg_info, toif::Icon, Color, Font},
@@ -18,8 +18,9 @@ use crate::{
     ui::{
         constant::HEIGHT,
         display::{tjpgd::BufferInput, toif::Toif},
-        model_tt::component::homescreen::render::{
-            HomescreenJpeg, HomescreenToif, HOMESCREEN_TOIF_SIZE,
+        model_tt::{
+            component::homescreen::render::{HomescreenJpeg, HomescreenToif, HOMESCREEN_TOIF_SIZE},
+            theme::{BLUE, ICON_MAGIC},
         },
     },
 };
@@ -81,13 +82,19 @@ where
     }
 
     fn get_notification(&self) -> Option<HomescreenNotification> {
-        if !usb_configured() {
+        if !usb_configured() && !ble_connected() {
             let (color, icon) = Self::level_to_style(0);
             Some(HomescreenNotification {
-                text: "NO USB CONNECTION",
+                text: "NO CONNECTION",
                 icon,
                 color,
             })
+        } else if ble_connected() {
+            Some(HomescreenNotification {
+                text: "BLE CONNECTED",
+                icon: ICON_MAGIC,
+                color: BLUE,
+            })
         } else if let Some((notification, level)) = &self.notification {
             let (color, icon) = Self::level_to_style(*level);
             Some(HomescreenNotification {
@@ -188,7 +195,9 @@ where
         }
 
         if let Event::Button(ButtonEvent::ButtonPressed(PhysicalButton::Power)) = event {
-            Some(HomescreenMsg::Dismissed)
+            start_advertising(false);
+            None
+            //Some(HomescreenMsg::Dismissed)
         } else {
             None
         }