Balance view

Author: dr-orlovsky

Shared/Components/Controls.swift

@@ -54,6 +54,31 @@ struct AmountField: View {
     }
 }
 
+struct Copyable<Label>: View where Label: View {
+    let text: String
+    var useSpacer: Bool
+    let content: Label
+
+    public init(text: String, useSpacer: Bool = true, @ViewBuilder content: () -> Label) {
+        self.text = text
+        self.useSpacer = useSpacer
+        self.content = content()
+    }
+
+    var body: some View {
+        Button(action: { My_Citadel.clipboardCopy(text: text) }) {
+            HStack {
+                content
+                if useSpacer {
+                    Spacer()
+                }
+                Image(systemName: "doc.on.doc")
+            }
+        }
+        .foregroundColor(.primary)
+    }
+}
+
 struct CopyableText: View {
     let text: String
     var copyable: Bool = false
@@ -62,18 +87,11 @@ struct CopyableText: View {
 
     var body: some View {
         if copyable {
-            Button(action: { My_Citadel.clipboardCopy(text: text) }) {
-                HStack {
-                    Text(text)
-                        .font(.body)
-                        .multilineTextAlignment(multilineTextAlignment)
-                    if useSpacer {
-                        Spacer()
-                    }
-                    Image(systemName: "doc.on.doc")
-                }
-            }
-            .foregroundColor(.primary)
+            Copyable(text: text, useSpacer: useSpacer, content: {
+                Text(text)
+                    .font(.body)
+                    .multilineTextAlignment(multilineTextAlignment)
+            })
         } else {
             Text(text)
                 .font(.body)

Shared/Wallet/TransactionView.swift

@@ -53,6 +53,16 @@ struct TransactionCell: View {
 struct TransactionView: View {
     var wallet: WalletContract
     var assetId: String = CitadelVault.embedded.nativeAsset.id
+    var asset: Asset {
+        CitadelVault.embedded.assets[assetId]!
+    }
+
+    private struct AddressBalance: Identifiable {
+        public var id: String { address }
+        public let address: String
+        public var amount: Double
+        public var utxo: [(OutPoint, Double)]
+    }
 
     private var assetIdModified: String? {
         assetId == CitadelVault.embedded.nativeAsset.id ? nil : assetId
@@ -65,35 +75,86 @@ struct TransactionView: View {
         }
         return wallet.operations.filter { $0.assetId == assetIdModified }
     }
+    private var addressBalances: [AddressBalance] {
+        Array((wallet.balance(of: assetId)?.unspentAllocations.reduce(into: [String: AddressBalance]()) { ( balances: inout [String: AddressBalance], allocation) in
+            guard let address = allocation.address else { return }
+            var balance = balances[address] ?? AddressBalance(address: address, amount: 0, utxo: [])
+            balance.amount += allocation.amount
+            balance.utxo.append((allocation.outpoint, allocation.amount))
+            balances[address] = balance
+        } ?? [:]).values)
+    }
 
+    enum SelectedTab: Hashable {
+        case history, balance
+        
+        var title: String {
+            switch self {
+            case .balance: return "Balance"
+            case .history: return "History"
+            }
+        }
+    }
+    
+    @State private var selectedTab: SelectedTab = .history
     @State private var scannedInvoice: Invoice? = nil
     @State private var scannedString: String = ""
     @State var presentedSheet: PresentedSheet?
-    private let placement: ToolbarItemPlacement = {
-        #if os(iOS)
-        return ToolbarItemPlacement.navigationBarLeading
-        #else
-        return ToolbarItemPlacement.primaryAction
-        #endif
-    }()
 
     var body: some View {
-        List(operations) { transaction in
-            TransactionCell(transaction: transaction)
+        List{
+            if selectedTab == .history {
+                ForEach(operations) { transaction in
+                    TransactionCell(transaction: transaction)
+                }
+            } else {
+                ForEach(addressBalances) { balance in
+                    Section(header: Text("Address")) {
+                        Copyable(text: balance.address) {
+                            BechBrief(text: balance.address)
+                        }
+                        DetailsCell(title: "Balance:", details: "\(balance.amount) \(asset.ticker)", clipboardCopy: true)
+                        ForEach(balance.utxo, id: \.0) { (outpoint, amount) in
+                            HStack(alignment: .center) {
+                                VStack(alignment: .leading) {
+                                    Text(outpoint.txid)
+                                        .font(.subheadline)
+                                        .truncationMode(.middle)
+                                        .lineLimit(1)
+                                    HStack(alignment: .lastTextBaseline) {
+                                        Text("Output number:").font(.caption).foregroundColor(.secondary)
+                                        Text("\(outpoint.vout)").font(.subheadline)
+                                    }
+                                }
+                                Spacer()
+                                Text("\(amount) \(asset.ticker)").font(.subheadline)
+                            }.padding(.leading)
+                        }
+                    }
+                }
+            }
         }
-        .navigationTitle("History")
+        .listStyle(GroupedListStyle())
+        .navigationTitle(selectedTab.title)
         .toolbar {
-            ToolbarItemGroup(placement: placement) {
+            ToolbarItem(placement: .confirmationAction) {
                 Button("Receive") { presentedSheet = .invoice(wallet, assetId) }
                     .background(Color.accentColor)
                     .foregroundColor(.white)
                     .cornerRadius(13)
-                Spacer()
+            }
+            ToolbarItem(placement: .cancellationAction) {
                 Button("Send") { presentedSheet = .scan("invoice", .invoice) }
                     .background(Color.accentColor)
                     .foregroundColor(.white)
                     .cornerRadius(13)
             }
+            ToolbarItem(placement: .principal) {
+                Picker(selection: $selectedTab, label: EmptyView()) {
+                    Text("History").tag(SelectedTab.history)
+                    Text("Balance").tag(SelectedTab.balance)
+                }.pickerStyle(SegmentedPickerStyle())
+            }
         }
         .sheet(item: $presentedSheet) { item in
             switch item {