[touch] use useCaseHomeAndSettings for home and settings display

Instead of useCaseHome and nbgl_useCaseSettings
trilitech · Jul 25, 2024 · 7ef9244 · 7ef9244
1 parent 04c2c7d
commit 7ef9244
Show file tree

Hide file tree

Showing 117 changed files with 130 additions and 74 deletions.
diff --git a/src/ui_nbgl.c b/src/ui_nbgl.c
@@ -37,6 +37,10 @@
 
 #include "nbgl_use_case.h"
 
+//  -----------------------------------------------------------
+//  -------------------------- INFO ---------------------------
+//  -----------------------------------------------------------
+
 typedef enum {
     CHAIN_IDX = 0,
     PKH_IDX,
@@ -54,36 +58,20 @@ static const char* const infoTypes[INFO_NB] =
 static const char* infoContents[INFO_NB];
 static char infoContentsBridge[INFO_NB][MAX_LENGTH];
 
-enum {
-    HWM_ENABLED_TOKEN = FIRST_USER_TOKEN
-};
-enum {
-    HWM_ENABLED_TOKEN_ID = 0,
-    SETTINGS_SWITCHES_NB
-};
-
-static nbgl_layoutSwitch_t switches[SETTINGS_SWITCHES_NB] = {0};
+static const nbgl_contentInfoList_t infoList = {.nbInfos = INFO_NB,
+                                                .infoTypes = infoTypes,
+                                                .infoContents = infoContents};
 
 /**
- * @brief Callback to fill the settings page content
- *
- * @param page: page of the settings
- * @param content: content to fill
- * @return bool: if the page is not out of bounds
+ * @brief Initializes info values
  */
-static bool navigation_cb_baking(uint8_t page, nbgl_pageContent_t* content) {
-    if (page > 2u) {
-        return false;
-    }
-
+static void initInfo(void) {
     tz_exc exc = SW_OK;
 
     for (tz_infoIndex_t idx = 0; idx < INFO_NB; idx++) {
         infoContents[idx] = infoContentsBridge[idx];
     }
 
-    const bool hwm_disabled = g_hwm.hwm_disabled;
-
     TZ_ASSERT(chain_id_to_string_with_aliases(infoContentsBridge[CHAIN_IDX],
                                               MAX_LENGTH,
                                               &g_hwm.main_chain_id) >= 0,
@@ -110,72 +98,96 @@ static bool navigation_cb_baking(uint8_t page, nbgl_pageContent_t* content) {
     TZ_ASSERT(copy_string(infoContentsBridge[COPYRIGHT_IDX], MAX_LENGTH, "(c) 2023 Ledger") >= 0,
               EXC_WRONG_LENGTH);
 
-    switch (page) {
-        case 0:
-            switches[HWM_ENABLED_TOKEN_ID].initState = (nbgl_state_t) (!hwm_disabled);
-            switches[HWM_ENABLED_TOKEN_ID].text = "High Watermark";
-            switches[HWM_ENABLED_TOKEN_ID].subText = "Track high watermark\n in Ledger";
-            switches[HWM_ENABLED_TOKEN_ID].token = HWM_ENABLED_TOKEN;
-            switches[HWM_ENABLED_TOKEN_ID].tuneId = TUNE_TAP_CASUAL;
-            content->type = SWITCHES_LIST;
-            content->switchesList.nbSwitches = SETTINGS_SWITCHES_NB;
-            content->switchesList.switches = (nbgl_layoutSwitch_t*) switches;
-            break;
-        case 1:
-            content->type = INFOS_LIST;
-            content->infosList.nbInfos = 3;
-            content->infosList.infoTypes = infoTypes;
-            content->infosList.infoContents = (const char* const*) infoContents;
-            break;
-        case 2:
-            content->type = INFOS_LIST;
-            content->infosList.nbInfos = 3;
-            content->infosList.infoTypes = infoTypes + 3;
-            content->infosList.infoContents = (const char* const*) infoContents + 3;
-            break;
-        default:
-            return false;
-    }
-    return true;
 end:
     TZ_EXC_PRINT(exc);
-    return true;
 }
 
-static void controls_callback(int token, uint8_t index) {
+//  -----------------------------------------------------------
+//  ------------------------ SETTINGS -------------------------
+//  -----------------------------------------------------------
+
+typedef enum {
+    HWM_ENABLED_TOKEN = FIRST_USER_TOKEN,
+    NEXT_TOKEN
+} tz_settingsToken_t;
+
+//  -------------------- SWITCHES SETTINGS --------------------
+
+typedef enum {
+    HWM_ENABLED_IDX = 0,
+    SETTINGS_SWITCHES_NB
+} tz_settingsSwitchesIndex_t;
+
+static nbgl_contentSwitch_t settingsSwitches[SETTINGS_SWITCHES_NB] = {0};
+
+/**
+ * @brief Initializes switches settings values
+ */
+static void initSettingsSwitches(void) {
+    nbgl_contentSwitch_t* hwmSwitch = &settingsSwitches[HWM_ENABLED_IDX];
+    hwmSwitch->text = "High Watermark";
+    hwmSwitch->subText = "Track high watermark\n in Ledger";
+    hwmSwitch->initState = (nbgl_state_t) (!g_hwm.hwm_disabled);
+    hwmSwitch->token = HWM_ENABLED_TOKEN;
+}
+
+/**
+ * @brief Callback to controle switches impact
+ *
+ * @param token: token of the switch toggled
+ * @param index: settings index
+ * @param page: page index
+ */
+static void settingsSwitchesToggleCallback(tz_settingsToken_t token, uint8_t index, int page) {
     UNUSED(index);
+    UNUSED(page);
     if (token == HWM_ENABLED_TOKEN) {
         toggle_hwm();
     }
 }
 
-#define TOTAL_SETTINGS_PAGE  (3)
-#define INIT_SETTINGS_PAGE   (0)
-#define DISABLE_SUB_SETTINGS false
+#define SETTINGS_SWITCHES_CONTENTS                                                             \
+    {                                                                                          \
+        .type = SWITCHES_LIST,                                                                 \
+        .content.switchesList = {.switches = settingsSwitches,                                 \
+                                 .nbSwitches = SETTINGS_SWITCHES_NB},                          \
+        .contentActionCallback = (nbgl_contentActionCallback_t) settingsSwitchesToggleCallback \
+    }
+
+//  ---------------------- ALL SETTINGS -----------------------
+
+typedef enum {
+    SWITCHES_IDX = 0,
+    SETTINGS_PAGE_NB
+} tz_settingsIndex_t;
+
+static const nbgl_content_t settingsContentsList[SETTINGS_PAGE_NB] = {SETTINGS_SWITCHES_CONTENTS};
+
+static const nbgl_genericContents_t settingsContents = {.callbackCallNeeded = false,
+                                                        .contentsList = settingsContentsList,
+                                                        .nbContents = SETTINGS_PAGE_NB};
 
 /**
- * @brief Draws settings pages
- *
+ * @brief Initializes settings values
  */
-static void ui_menu_about_baking(void) {
-    nbgl_useCaseSettings("Tezos baking",
-                         INIT_SETTINGS_PAGE,
-                         TOTAL_SETTINGS_PAGE,
-                         DISABLE_SUB_SETTINGS,
-                         ui_initial_screen,
-                         navigation_cb_baking,
-                         controls_callback);
+static void initSettings(void) {
+    initSettingsSwitches();
 }
 
-#define SETTINGS_BUTTON_ENABLED (true)
+//  -----------------------------------------------------------
 
 void ui_initial_screen(void) {
-    nbgl_useCaseHome("Tezos Baking",
-                     &C_tezos,
-                     NULL,
-                     SETTINGS_BUTTON_ENABLED,
-                     ui_menu_about_baking,
-                     app_exit);
+    initSettings();
+    initInfo();
+
+    nbgl_useCaseHomeAndSettings("Tezos Baking",
+                                &C_tezos,
+                                NULL,
+                                INIT_HOME_PAGE,
+                                &settingsContents,
+                                &infoList,
+                                NULL,
+                                app_exit);
 }
 
 #endif  // HAVE_NBGL
diff --git a/test/snapshots/flex/description.png b/test/snapshots/flex/description.png
diff --git a/test/snapshots/flex/hwm_status_disabled.png b/test/snapshots/flex/hwm_status_disabled.png
diff --git a/test/snapshots/flex/hwm_status_enabled.png b/test/snapshots/flex/hwm_status_enabled.png
diff --git a/test/snapshots/flex/test_deauthorize/authorized_key_after_authorize.png b/test/snapshots/flex/test_deauthorize/authorized_key_after_authorize.png
diff --git a/test/snapshots/flex/test_deauthorize/authorized_key_before_authorize.png b/test/snapshots/flex/test_deauthorize/authorized_key_before_authorize.png
diff --git a/..._review_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context.png b/..._review_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context.png
diff --git a/...eview_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context_1.png b/...eview_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context_1.png
diff --git a/...eview_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context_2.png b/...eview_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context_2.png
diff --git a/...x/test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context.png b/...x/test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context.png
diff --git a/...test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context_1.png b/...test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context_1.png
diff --git a/...test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context_2.png b/...test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context_2.png
diff --git a/...x/test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context.png b/...x/test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context.png
diff --git a/...test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context_1.png b/...test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context_1.png
diff --git a/...test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context_2.png b/...test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context_2.png
diff --git a/...test_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context.png b/...test_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context.png
diff --git a/...st_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context_1.png b/...st_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context_1.png
diff --git a/...st_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context_2.png b/...st_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context_2.png
diff --git a/...test_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context.png b/...test_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context.png
diff --git a/...st_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context_1.png b/...st_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context_1.png
diff --git a/...st_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context_2.png b/...st_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context_2.png
diff --git a/test/snapshots/flex/test_review_home/app_context.png b/test/snapshots/flex/test_review_home/app_context.png
diff --git a/test/snapshots/flex/test_review_home/app_context_1.png b/test/snapshots/flex/test_review_home/app_context_1.png
diff --git a/test/snapshots/flex/test_review_home/app_context_2.png b/test/snapshots/flex/test_review_home/app_context_2.png
diff --git a/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...ign_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...ign_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...gn_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...gn_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...ign_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...ign_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...gn_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...gn_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...n_attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...n_attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/..._attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/..._attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...n_attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...n_attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/..._attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/..._attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...ation_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...ation_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...tion_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...tion_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...attestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...attestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...ttestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...ttestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...attestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...attestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...ttestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...ttestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...testation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...testation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...estation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...estation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...testation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...testation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...estation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...estation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...ign_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...ign_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...gn_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...gn_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...test_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...test_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...est_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...est_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...test_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...test_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...est_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...est_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...st_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...st_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...t_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...t_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...st_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...st_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...t_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...t_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/..._preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/..._preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/..._preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/..._preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...reattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...reattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...eattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...eattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...reattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...reattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...eattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...eattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/test/snapshots/stax/description.png b/test/snapshots/stax/description.png
diff --git a/test/snapshots/stax/hwm_status_disabled.png b/test/snapshots/stax/hwm_status_disabled.png
diff --git a/test/snapshots/stax/hwm_status_enabled.png b/test/snapshots/stax/hwm_status_enabled.png
diff --git a/test/snapshots/stax/test_deauthorize/authorized_key_after_authorize.png b/test/snapshots/stax/test_deauthorize/authorized_key_after_authorize.png
diff --git a/test/snapshots/stax/test_deauthorize/authorized_key_before_authorize.png b/test/snapshots/stax/test_deauthorize/authorized_key_before_authorize.png
diff --git a/..._review_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context.png b/..._review_home/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/app_context.png
diff --git a/...x/test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context.png b/...x/test_review_home/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/app_context.png
diff --git a/...x/test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context.png b/...x/test_review_home/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/app_context.png
diff --git a/...test_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context.png b/...test_review_home/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/app_context.png
diff --git a/...test_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context.png b/...test_review_home/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/app_context.png
diff --git a/test/snapshots/stax/test_review_home/app_context.png b/test/snapshots/stax/test_review_home/app_context.png
diff --git a/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...ign_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...ign_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...gn_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...gn_attestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...ign_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...ign_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...gn_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...gn_attestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...n_attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...n_attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/..._attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/..._attestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...n_attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...n_attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/..._attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/..._attestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...ation_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...ation_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...tion_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...tion_dal/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...attestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...attestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...ttestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...ttestation_dal/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...attestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...attestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...ttestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...ttestation_dal/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...testation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...testation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...estation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...estation_dal/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...testation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...testation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...estation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...estation_dal/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...ign_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...ign_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...gn_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...gn_block/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/...test_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/...test_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...est_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...est_sign_block/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/...test_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/...test_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...est_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...est_sign_block/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...st_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...st_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...t_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...t_sign_block/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...st_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...st_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...t_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...t_sign_block/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png b/...testation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_after_sign.png
diff --git a/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png b/...estation/BIP32_ED25519_tz1VKyZ3RFDwTkrz5LKcTc6fcYqZj6pvsyA7/hwm_before_sign.png
diff --git a/..._preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png b/..._preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_after_sign.png
diff --git a/...preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png b/...preattestation/ED25519_tz1dyX3B1CFYa2DfdFLyPtiJCfQRUgPVME6E/hwm_before_sign.png
diff --git a/..._preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png b/..._preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_after_sign.png
diff --git a/...preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png b/...preattestation/ED25519_tz1ez9eEyAxuDZACjHdCXym43UQDdMNa3LEL/hwm_before_sign.png
diff --git a/...reattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png b/...reattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_after_sign.png
diff --git a/...eattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png b/...eattestation/SECP256K1_tz2GB5YHqF4UzQ8GP5yUqdhY9oVWRXCY2hPU/hwm_before_sign.png
diff --git a/...reattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png b/...reattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_after_sign.png
diff --git a/...eattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png b/...eattestation/SECP256R1_tz3UMNyvQeMj6mQSftW2aV2XaWd3afTAM1d5/hwm_before_sign.png
diff --git a/test/test_instructions.py b/test/test_instructions.py
@@ -158,13 +158,31 @@ def test_review_home(account: Optional[Account],
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_HMW_ENABLED)
         tezos_navigator.settings.next()
         backend.wait_for_screen_change()
-        tezos_navigator.assert_screen("app_context", snap_path)
+        if tezos_navigator.firmware == Firmware.STAX:
+            # chain_id + pkh + hwm
+            tezos_navigator.assert_screen("app_context", snap_path)
+        elif tezos_navigator.firmware == Firmware.FLEX:
+            # chain_id + pkh
+            tezos_navigator.assert_screen("app_context_1", snap_path)
+            tezos_navigator.settings.next()
+            backend.wait_for_screen_change()
+            # hwm + version
+            tezos_navigator.assert_screen("app_context_2", snap_path)
         tezos_navigator.settings.next()
         backend.wait_for_screen_change()
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_DESCRIPTION)
         tezos_navigator.settings.previous()
         backend.wait_for_screen_change()
-        tezos_navigator.assert_screen("app_context", snap_path)
+        if tezos_navigator.firmware == Firmware.STAX:
+            # chain_id + pkh + hwm
+            tezos_navigator.assert_screen("app_context", snap_path)
+        elif tezos_navigator.firmware == Firmware.FLEX:
+            # hwm + version
+            tezos_navigator.assert_screen("app_context_2", snap_path)
+            tezos_navigator.settings.previous()
+            backend.wait_for_screen_change()
+            # chain_id + pkh
+            tezos_navigator.assert_screen("app_context_1", snap_path)
         tezos_navigator.settings.previous()
         backend.wait_for_screen_change()
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_HMW_ENABLED)
@@ -179,13 +197,31 @@ def test_review_home(account: Optional[Account],
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_HMW_DISABLED)
         tezos_navigator.settings.next()
         backend.wait_for_screen_change()
-        tezos_navigator.assert_screen("app_context", snap_path)
+        if tezos_navigator.firmware == Firmware.STAX:
+            # chain_id + pkh + hwm
+            tezos_navigator.assert_screen("app_context", snap_path)
+        elif tezos_navigator.firmware == Firmware.FLEX:
+            # chain_id + pkh
+            tezos_navigator.assert_screen("app_context_1", snap_path)
+            tezos_navigator.settings.next()
+            backend.wait_for_screen_change()
+            # hwm + version
+            tezos_navigator.assert_screen("app_context_2", snap_path)
         tezos_navigator.settings.next()
         backend.wait_for_screen_change()
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_DESCRIPTION)
         tezos_navigator.settings.previous()
         backend.wait_for_screen_change()
-        tezos_navigator.assert_screen("app_context", snap_path)
+        if tezos_navigator.firmware == Firmware.STAX:
+            # chain_id + pkh + hwm
+            tezos_navigator.assert_screen("app_context", snap_path)
+        elif tezos_navigator.firmware == Firmware.FLEX:
+            # hwm + version
+            tezos_navigator.assert_screen("app_context_2", snap_path)
+            tezos_navigator.settings.previous()
+            backend.wait_for_screen_change()
+            # chain_id + pkh
+            tezos_navigator.assert_screen("app_context_1", snap_path)
         tezos_navigator.settings.previous()
         backend.wait_for_screen_change()
         tezos_navigator.assert_screen(TouchFixedScreen.SETTINGS_HMW_DISABLED)

diff --git a/test/utils/navigator.py b/test/utils/navigator.py
@@ -348,6 +348,8 @@ def goto_home_public_key(self) -> Generator[None, None, None]:
             self.settings.next()
             self.backend.wait_for_screen_change()
 
+        # Stax: chain_id + pkh + hwm
+        # Flex: chain_id + pkh
         yield
 
         if self.firmware.is_nano:
@@ -386,7 +388,13 @@ def goto_home_hwm(self) -> Generator[None, None, None]:
             # hwm_status
             self.settings.next()
             self.backend.wait_for_screen_change()
+            if self.firmware == Firmware.FLEX:
+                # chain_id + pkh
+                self.settings.next()
+                self.backend.wait_for_screen_change()
 
+        # Stax: chain_id + pkh + hwm
+        # Flex: hwm + version
         yield
 
         if self.firmware.is_nano: