login flow

pedi-foods/Sources/PediFoods/Views/Main/ProfileView.swift

@@ -32,8 +32,8 @@ struct ProfileView: View {
 
             SecondaryButton(title: "Sair") {
                 tokenStore.clear()
-                appState.session.isAuthenticated = false
-                appState.session.jwt = nil
+                SessionStateStore.clearActiveUser()
+                appState = AppState()
                 root = .auth
             }
             .padding(.horizontal, 20)
@@ -81,6 +81,7 @@ struct ProfileView: View {
 struct AddressesView: View {
     let message: String?
     @Binding var appState: AppState
+    var selectionMode: Bool = false
     @Environment(\.dismiss) var dismiss
     @State var isLoading = false
     @State var errorMessage: String? = nil
@@ -127,7 +128,23 @@ struct AddressesView: View {
                                 .padding(.top, 24)
                         } else {
                             ForEach(Array(addresses.enumerated()), id: \.offset) { index, address in
-                                AddressCard(item: addressToListItem(address, isPrimary: index == 0))
+                                let isSelected: Bool = {
+                                    if let selectedId = appState.address.selectedId {
+                                        return address.id == selectedId
+                                    }
+                                    return index == 0
+                                }()
+
+                                if selectionMode {
+                                    Button {
+                                        selectAddress(address)
+                                    } label: {
+                                        AddressCard(item: addressToListItem(address, isPrimary: isSelected))
+                                    }
+                                    .buttonStyle(.plain)
+                                } else {
+                                    AddressCard(item: addressToListItem(address, isPrimary: isSelected))
+                                }
                             }
                         }
                     }
@@ -159,6 +176,7 @@ struct AddressesView: View {
                         appState.address.latitude = nil
                         appState.address.longitude = nil
                     }
+                    SessionStateStore.saveAddress(appState.address)
                     SnackbarCenter.shared.show(title: "Endereço adicionado com sucesso.", style: .success, icon: "checkmark.seal.fill", duration: 3.0)
                 }
             }
@@ -219,6 +237,26 @@ struct AddressesView: View {
         }
     }
 
+    private func selectAddress(_ address: CustomerAddress) {
+        appState.address.selectedId = address.id
+        let label = (address.label ?? "").trimmingCharacters(in: .whitespacesAndNewlines)
+        appState.address.display = label.isEmpty ? "Defina seu endereco" : label
+
+        if let lat = address.latLong?.first, let lng = address.latLong?.dropFirst().first {
+            appState.address.latitude = lat
+            appState.address.longitude = lng
+        } else {
+            appState.address.latitude = nil
+            appState.address.longitude = nil
+        }
+
+        SessionStateStore.saveAddress(appState.address)
+
+        if selectionMode {
+            dismiss()
+        }
+    }
+
     func addressToListItem(_ address: CustomerAddress, isPrimary: Bool) -> AddressListItem {
         let title = address.label?.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty == false ? (address.label ?? "Endereço") : "Endereço"
         let line1 = [address.address, address.number]
@@ -266,14 +304,26 @@ struct AddressesView: View {
                 return
             }
 
-            addresses = response.result?.addressBook ?? []
-            if let first = addresses.first {
-                appState.address.selectedId = first.id
-                appState.address.display = first.label ?? "Defina seu endereco"
-                if let lat = first.latLong?.first, let lng = first.latLong?.dropFirst().first {
+            if let customer = response.result {
+                appState.profile.id = customer.id
+                appState.profile.name = customer.name
+                appState.profile.email = customer.email
+                appState.profile.phone = customer.phoneNumber ?? appState.profile.phone
+                SessionStateStore.setActiveUserKey(
+                    SessionStateStore.makeUserKey(profileId: customer.id, email: customer.email)
+                )
+                addresses = customer.addressBook ?? []
+            } else {
+                addresses = []
+            }
+            if let selected = addresses.first(where: { $0.id == appState.address.selectedId }) ?? addresses.first {
+                appState.address.selectedId = selected.id
+                appState.address.display = selected.label ?? "Defina seu endereco"
+                if let lat = selected.latLong?.first, let lng = selected.latLong?.dropFirst().first {
                     appState.address.latitude = lat
                     appState.address.longitude = lng
                 }
+                SessionStateStore.saveAddress(appState.address)
             }
         } catch {
             errorMessage = error.localizedDescription