Implement beacon and vote display categories in UI transaction model

Author: jamescowens

src/qt/transactionrecord.cpp

@@ -166,11 +166,14 @@ QList<TransactionRecord> TransactionRecord::decomposeTransaction(const CWallet *
                             sub.credit = nCoinStakeReturnOutput - nDebit;
                         }
                     }
-                } else
+                }
+                else
+                {
                     sub.credit = wtx.vout[t].nValue;
+                }
 
                 parts.append(sub);
-            }
+            } // vout for loop
         }
     }
     else
@@ -234,6 +237,34 @@ QList<TransactionRecord> TransactionRecord::decomposeTransaction(const CWallet *
                 }
                 sub.debit = -nValue;
 
+                // Determine if the transaction is a beacon advertisement or a vote.
+                // For right now, there should only be one contract in a transaction.
+                // We will simply select the first and only one. Note that we are
+                // looping through the outputs one by one in the for loop above this,
+                // So if we get here, we are not a coinbase or coinstake, and we are on
+                // an ouput that isn't ours. The worst that can happen from this
+                // simple approach is to label more than one output with the
+                // first found contract type. For right now, this is sufficient, because
+                // the contracts that are sent right now only contain two outputs,
+                // the burn and the change. We will have to get more sophisticated
+                // when we allow more than one contract per transaction.
+
+                // Notice this doesn't mess with the value or debit, it simply
+                // overrides the TransactionRecord enum type.
+                if (!wtx.GetContracts().empty())
+                {
+                    const auto& contract = wtx.GetContracts().begin();
+
+                    if (contract->m_type == NN::ContractType::BEACON)
+                    {
+                        sub.type = TransactionRecord::BeaconAdvertisement;
+                    }
+                    else if(contract->m_type == NN::ContractType::VOTE)
+                    {
+                        sub.type = TransactionRecord::Vote;
+                    }
+                }
+
                 parts.append(sub);
             }
         }

src/qt/transactionrecord.h

@@ -70,7 +70,9 @@ class TransactionRecord
         SendToOther,
         RecvWithAddress,
         RecvFromOther,
-        SendToSelf
+        SendToSelf,
+        BeaconAdvertisement,
+        Vote
     };
 
     /** Number of confirmation recommended for accepting a transaction */

src/qt/transactiontablemodel.cpp

@@ -406,6 +406,10 @@ QVariant TransactionTableModel::txAddressDecoration(const TransactionRecord *wtx
     case TransactionRecord::SendToAddress:
     case TransactionRecord::SendToOther:
         return QIcon(":/icons/tx_output");
+    case TransactionRecord::BeaconAdvertisement:
+        return QIcon(":/icons/beacon_grey");
+    case TransactionRecord::Vote:
+        return QIcon(":/icons/voting_native");
     default:
         return QIcon(":/icons/tx_inout");
     }