Show transfer direction based on the transaction and operation

BunToy · Feb 17, 2018 · 688778d · 688778d
1 parent 72c6b61
commit 688778d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Trust/Transactions/ViewModels/TransactionViewModel.swift b/Trust/Transactions/ViewModels/TransactionViewModel.swift
@@ -25,12 +25,16 @@ struct TransactionViewModel {
         self.currentWallet = currentWallet
     }
 
+    var transactionTo: String {
+        guard let operation = transaction.operation else { return transaction.to }
+        return operation.to
+    }
+
     var direction: TransactionDirection {
-        if currentWallet.address.description == transaction.from ||
-            currentWallet.address.description.lowercased() == transaction.from.lowercased() {
-            return .outgoing
+        if currentWallet.address.description == transactionTo || currentWallet.address.description.lowercased() == transactionTo.lowercased() {
+            return .incoming
         }
-        return .incoming
+        return .outgoing
     }
 
     var confirmations: Int? {