Update transaction history if new block received

Author: dyng

internal/view/account.go

@@ -1,7 +1,9 @@
 package view
 
 import (
+	"github.com/dyng/ramen/internal/common"
 	"github.com/dyng/ramen/internal/common/conv"
+	"github.com/dyng/ramen/internal/service"
 	serv "github.com/dyng/ramen/internal/service"
 	"github.com/dyng/ramen/internal/view/format"
 	"github.com/dyng/ramen/internal/view/style"
@@ -41,6 +43,9 @@ func NewAccount(app *App) *Account {
 	// setup keymap
 	account.initKeymap()
 
+	// subscribe to new blocks
+	app.eventBus.Subscribe(service.TopicNewBlock, account.onNewBlock)
+
 	return account
 }
 
@@ -178,6 +183,18 @@ func (a *Account) HideImportABIDialog() {
 	a.app.SetFocus(a)
 }
 
+func (a *Account) onNewBlock(block *common.Block) {
+	txns, err := a.app.service.GetTransactionsByBlock(block)
+	if err != nil {
+		log.Error("cannot extract transactions from block", "blockHash", block.Hash(), "error", err)
+		return
+	}
+
+	a.app.QueueUpdateDraw(func() {
+		a.transactionList.FilterAndPrependTransactions(txns)
+	})
+}
+
 func (a *Account) refresh() {
 	addr := a.account.GetAddress()
 	a.accountInfo.address.SetText(addr.Hex())

internal/view/helper.go

@@ -83,7 +83,11 @@ func StyledTxnDirection(base *common.Address, txn common.Transaction) string {
 
 	if txn.From().String() == base.String() {
 		return "[sandybrown]OUT[-]"
-	} else {
+	}
+
+	if txn.To() != nil && txn.To().String() == base.String() {
 		return "[lightgreen]IN[-]"
 	}
+
+	return ""
 }

internal/view/transactions.go

@@ -79,6 +79,24 @@ func (t *TransactionList) SetBaseAccount(account *common.Address) {
 	t.base = account
 }
 
+func (t *TransactionList) FilterAndPrependTransactions(txns common.Transactions) {
+	if t.base == nil {
+		t.PrependTransactions(txns)
+		return
+	}
+
+	toAdd := make(common.Transactions, 0)
+	for _, tx := range txns {
+		if tx.From().String() == t.base.String() {
+			toAdd = append(toAdd, tx)
+		}
+		if tx.To() != nil && tx.To().String() == t.base.String() {
+			toAdd = append(toAdd, tx)
+		}
+	}
+	t.PrependTransactions(toAdd)
+}
+
 func (t *TransactionList) PrependTransactions(txns common.Transactions) {
 	prepended := append(txns, t.txns...)
 	t.SetTransactions(prepended)