style: Update auth wallet ui

Author: TejasvOnly

apps/inheritance_app/inheritance_auth_wallet.c

@@ -19,11 +19,13 @@
 
 #include "bip39.h"
 #include "card_fetch_data.h"
+#include "constant_texts.h"
 #include "inheritance/core.pb.h"
 #include "inheritance_api.h"
 #include "inheritance_priv.h"
 #include "reconstruct_wallet_flow.h"
 #include "status_api.h"
+#include "ui_core_confirm.h"
 #include "ui_delay.h"
 
 /*****************************************************************************
@@ -130,6 +132,44 @@ static bool verify_auth_wallet_inputs() {
   return true;
 }
 
+static bool check_which_request(const inheritance_query_t *query,
+                                pb_size_t which_request) {
+  if (which_request != query->encrypt.which_request) {
+    inheritance_send_error(ERROR_COMMON_ERROR_CORRUPT_DATA_TAG,
+                           ERROR_DATA_FLOW_INVALID_REQUEST);
+    return false;
+  }
+
+  return true;
+}
+
+STATIC bool auth_wallet_handle_inititate_query(inheritance_query_t *query) {
+  char wallet_name[NAME_SIZE] = "";
+  char msg[100] = "";
+
+  if (!check_which_request(
+          query, INHERITANCE_ENCRYPT_DATA_WITH_PIN_REQUEST_INITIATE_TAG) ||
+      !get_wallet_name_by_id(query->encrypt.initiate.wallet_id,
+                             (uint8_t *)wallet_name,
+                             inheritance_send_error)) {
+    // TODO: update this for nominee
+    return false;
+  }
+
+  snprintf(msg,
+           sizeof(msg),
+           ui_text_inheritance_wallet_auth_flow_confirmation,
+           wallet_name);
+
+  if (!core_confirmation(msg, inheritance_send_error)) {
+    return false;
+  }
+
+  // TODO: update flow status here
+
+  return true;
+}
+
 static bool auth_wallet_get_entropy() {
   if (auth->do_seed_based) {
     uint8_t seed[SIZE_SEED] = {0};
@@ -298,7 +338,8 @@ void inheritance_auth_wallet(inheritance_query_t *query) {
   auth->do_wallet_based = query->auth_wallet.initiate.do_wallet_based;
 
   set_app_flow_status(INHERITANCE_AUTH_WALLET_STATUS_INIT);
-  if (verify_auth_wallet_inputs() && auth_wallet_get_entropy() &&
+  if (verify_auth_wallet_inputs() &&
+      auth_wallet_handle_inititate_query(query) && auth_wallet_get_entropy() &&
       auth_wallet_get_pairs() && auth_wallet_get_signature() && send_result()) {
     delay_scr_init(ui_text_inheritance_wallet_auth_success, DELAY_TIME);
   }

src/constant_texts.c

@@ -530,6 +530,8 @@ const char *ui_critical_card_health_migrate_data =
 // App specific
 
 // Inheritance
+const char *ui_text_inheritance_wallet_auth_flow_confirmation =
+    "Start %s wallet authentication?";
 const char *ui_text_inheritance_wallet_authenticating =
     "Wallet\nauthenticating...";
 const char *ui_text_inheritance_wallet_auth_success =

src/constant_texts.h

@@ -374,6 +374,7 @@ extern const char *ui_critical_card_health_migrate_data;
 // App specific
 
 // Inheritance
+extern const char *ui_text_inheritance_wallet_auth_flow_confirmation;
 extern const char *ui_text_inheritance_wallet_authenticating;
 extern const char *ui_text_inheritance_wallet_auth_success;
 extern const char *ui_text_inheritance_wallet_auth_fail;