From 6ec26db05377eb92dda499a6a375fe4183fea443 Mon Sep 17 00:00:00 2001
From: Francois Beutin <francois.beutin@ledger.fr>
Date: Fri, 30 Jun 2023 14:06:09 +0200
Subject: [PATCH] Add Spinner when starting in Swap mode on Stax

---
 src/handle_swap_sign_transaction.c | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/handle_swap_sign_transaction.c b/src/handle_swap_sign_transaction.c
index ddb32ec..16c8b7f 100644
--- a/src/handle_swap_sign_transaction.c
+++ b/src/handle_swap_sign_transaction.c
@@ -4,6 +4,9 @@
 #include "handle_swap_sign_transaction.h"
 #include "shared_context.h"
 #include "utils.h"
+#ifdef HAVE_NBGL
+#include "utils.h"
+#endif  // HAVE_NBGL
 
 bool copy_transaction_parameters(create_transaction_parameters_t* sign_transaction_params,
                                  chain_config_t* config) {
@@ -52,6 +55,11 @@ bool copy_transaction_parameters(create_transaction_parameters_t* sign_transacti
 }
 
 void handle_swap_sign_transaction(chain_config_t* config) {
+    UX_INIT();
+#ifdef HAVE_NBGL
+    nbgl_useCaseSpinner("Signing");
+#endif  // HAVE_NBGL
+
     chainConfig = config;
     reset_app_context();
     G_called_from_swap = true;
@@ -67,13 +75,6 @@ void handle_swap_sign_transaction(chain_config_t* config) {
         nvm_write((void*) &N_storage, (void*) &storage, sizeof(internalStorage_t));
     }
 
-#ifdef HAVE_BAGL
-    UX_INIT();
-#endif  // HAVE_BAGL
-#ifdef HAVE_NBGL
-    nbgl_objInit();
-#endif  // HAVE_NBGL
-
     USB_power(0);
     USB_power(1);
     // ui_idle();