From 9e57b57ddf64820a83c3f587fbfcffa267db0ded Mon Sep 17 00:00:00 2001
From: Juan Gilsanz Polo <juangilsanz@gmail.com>
Date: Wed, 11 Oct 2023 22:14:32 +0200
Subject: [PATCH] Added loading indicator edit connection

---
 lib/widgets/add_server/add_server_modal.dart | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/lib/widgets/add_server/add_server_modal.dart b/lib/widgets/add_server/add_server_modal.dart
index 294cd7d..c587c7e 100644
--- a/lib/widgets/add_server/add_server_modal.dart
+++ b/lib/widgets/add_server/add_server_modal.dart
@@ -126,6 +126,8 @@ class _AddServerModalState extends State<AddServerModal> {
     }
 
     void connect() async {
+      setState(() => isConnecting = true);
+
       Server serverObj = Server(
         id: uuid.v4(),
         name: nameController.text, 
@@ -140,7 +142,6 @@ class _AddServerModalState extends State<AddServerModal> {
           : null,
         runningOnHa: homeAssistant
       );
-      setState(() => isConnecting = true);
 
       final result = homeAssistant == true 
         ? await loginHA(serverObj)
@@ -211,10 +212,11 @@ class _AddServerModalState extends State<AddServerModal> {
       else {
         Navigator.pop(context);
       }
-      return;
     }
 
     void edit() async {
+      setState(() => isConnecting = true);
+      
       final Server serverObj = Server(
         id: widget.server!.id,
         name: nameController.text, 
@@ -286,7 +288,6 @@ class _AddServerModalState extends State<AddServerModal> {
       else {
         Navigator.pop(context);
       }      
-      return; 
     }
 
     Widget actions() {