Fix display address usecase

LedgerHQ · Sep 27, 2024 · 61621c1 · 61621c1
1 parent 3e5ca69
commit 61621c1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/app_ui/address.rs b/src/app_ui/address.rs
@@ -21,7 +21,7 @@ use fmt_buffer::Buffer;
 #[cfg(any(target_os = "stax", target_os = "flex"))]
 use include_gif::include_gif;
 #[cfg(any(target_os = "stax", target_os = "flex"))]
-use ledger_device_sdk::nbgl::{NbglAddressReview, NbglGlyph};
+use ledger_device_sdk::nbgl::{NbglAddressReview, NbglGlyph, NbglReviewStatus, StatusType};
 #[cfg(not(any(target_os = "stax", target_os = "flex")))]
 use ledger_device_sdk::ui::bitmaps::{CROSSMARK, EYE, VALIDATE_14};
 #[cfg(not(any(target_os = "stax", target_os = "flex")))]
@@ -59,7 +59,18 @@ pub fn ui_display_pk_base58(public_key: &crypto::PublicKeyBe) -> Result<bool, Ap
             .glyph(&NEAR_LOGO)
             .verify_str("Confirm Public Key");
 
-        Ok(review.show(out_buf.as_str()))
+        let res = review.show(out_buf.as_str());
+        let mut status = NbglReviewStatus::new();
+        match res {
+            true => {
+                status.status_type(StatusType::Address).show(true);
+            }
+            false => {
+                status.status_type(StatusType::Address).show(false);
+            }
+        }
+
+        Ok(res)
     }
 }
 

diff --git a/tests/snapshots/stax/test_get_public_key_and_confirm_screen/00001.png b/tests/snapshots/stax/test_get_public_key_and_confirm_screen/00001.png