Import ABI for unknown contract

Author: dyng

internal/common/conv/argument.go

@@ -9,7 +9,7 @@ import (
 	"github.com/ethereum/go-ethereum/common"
 )
 
-func Unpack(t abi.Type, s string) (any, error) {
+func UnpackArgument(t abi.Type, s string) (any, error) {
 	switch t.T {
 	case abi.StringTy:
 		return s, nil

internal/service/contract.go

@@ -2,6 +2,7 @@ package service
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/ethereum/go-ethereum/accounts/abi"
 	"github.com/ethereum/go-ethereum/log"
@@ -13,6 +14,10 @@ type Contract struct {
 	source string
 }
 
+func (c *Contract) HasABI() bool {
+	return c.abi != nil
+}
+
 func (c *Contract) GetABI() *abi.ABI {
 	return c.abi
 }
@@ -34,3 +39,15 @@ func (c *Contract) Call(method string, args ...any) ([]any, error) {
 	log.Debug("Try to call contract", "method", method, "args", args)
 	return c.service.provider.CallContract(c.address, c.abi, method, args...)
 }
+
+func (c *Contract) ImportABI(abiJson string) error {
+	log.Debug("Try to parse abi json", "json", abiJson)
+	parsedAbi, err := abi.JSON(strings.NewReader(abiJson))
+	if err != nil {
+		return err
+	}
+
+	c.abi = &parsedAbi
+
+	return nil
+}

internal/service/service.go

@@ -168,7 +168,10 @@ func (s *Service) transactionsByTraverse(addr common.Address) (common.Transactio
 
 	txns := make([]common.Transaction, 0)
 	for _, t := range candidates {
-		if t.From().String() == addr.String() || t.To().String() == addr.String() {
+		if t.From().String() == addr.String() {
+			txns = append(txns, t)
+		}
+		if t.To() != nil && t.To().String() == addr.String() {
 			txns = append(txns, t)
 		}
 	}
@@ -241,7 +244,13 @@ func (s *Service) ToContract(account *Account) (*Contract, error) {
 		return nil, fmt.Errorf("Address %s is not a contract account", account.address.Hex())
 	}
 
-	// FIXME: support local chain
+	if s.GetNetwork().NetType() == TypeDevnet {
+		// in case of devnet, return a contract skeleton
+		return &Contract{
+			Account: account,
+		}, nil
+	}
+
 	source, abi, err := s.esclient.GetSourceCode(account.address)
 	if err != nil {
 		return nil, err

internal/view/account.go

@@ -18,7 +18,9 @@ type Account struct {
 	accountInfo     *AccountInfo
 	transactionList *TransactionList
 	methodCall      *MethodCallDialog
+	importABI       *ImportABIDialog
 	account         *serv.Account
+	contract        *serv.Contract
 }
 
 type AccountInfo struct {
@@ -50,6 +52,7 @@ func (a *Account) SetAccount(account *serv.Account) {
 	if account.IsContract() {
 		contract, err := account.AsContract()
 		if err == nil {
+			a.contract = contract
 			a.methodCall.SetContract(contract)
 		} else {
 			log.Error("Cannot upgrade account to contract", "account", account.GetAddress(), "error", err)
@@ -80,6 +83,10 @@ func (a *Account) initLayout() {
 	methodCall := NewMethodCallDialog(a.app)
 	a.methodCall = methodCall
 
+	// ImportABIDialog
+	importABI := NewImportABIDialog(a.app)
+	a.importABI = importABI
+
 	// Transactions
 	transactions := NewTransactionList(a.app, true)
 	transactions.SetTitleColor(s.SecondaryTitleColor)
@@ -99,16 +106,7 @@ func (a *Account) initLayout() {
 }
 
 func (a *Account) initKeymap() {
-	keymaps := a.KeyMaps()
-	a.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
-		handler, ok := keymaps.FindHandler(util.AsKey(event))
-		if ok {
-			handler(event)
-			return nil
-		} else {
-			return event
-		}
-	})
+	InitKeymap(a, a.app)
 }
 
 func (a *Account) KeyMaps() util.KeyMaps {
@@ -120,7 +118,14 @@ func (a *Account) KeyMaps() util.KeyMaps {
 		Shortcut:    "C",
 		Description: "Call Contract",
 		Handler: func(*tcell.EventKey) {
-			a.ShowMethodCallDialog()
+			// FIXME: don't show "Call Contract" for wallet account
+			if a.account.IsContract() {
+				if a.methodCall.contract.HasABI() {
+					a.ShowMethodCallDialog()
+				} else {
+					a.ShowImportABIDialog()
+				}
+			}
 		},
 	})
 
@@ -152,6 +157,27 @@ func (a *Account) HideMethodCallDialog() {
 	a.app.SetFocus(a)
 }
 
+func (a *Account) ShowImportABIDialog() {
+	log.Debug("Show importABI dialog")
+
+	if !a.importABI.IsDisplay() {
+		a.importABI.Clear()
+		a.importABI.Display(true)
+	}
+
+	a.app.SetFocus(a.importABI)
+}
+
+func (a *Account) HideImportABIDialog() {
+	log.Debug("Hide importABI dialog")
+
+	if a.importABI.IsDisplay() {
+		a.importABI.Display(false)
+	}
+
+	a.app.SetFocus(a)
+}
+
 func (a *Account) refresh() {
 	addr := a.account.GetAddress()
 	a.accountInfo.address.SetText(addr.Hex())
@@ -180,20 +206,29 @@ func (a *Account) HasFocus() bool {
 	if a.methodCall.HasFocus() {
 		return true
 	}
+	if a.importABI.HasFocus() {
+		return true
+	}
 	return a.Flex.HasFocus()
 }
 
 // InputHandler implements tview.Primitive
 func (a *Account) InputHandler() func(event *tcell.EventKey, setFocus func(p tview.Primitive)) {
 	return func(event *tcell.EventKey, setFocus func(p tview.Primitive)) {
-		if a.Flex.HasFocus() {
-			if handler := a.Flex.InputHandler(); handler != nil {
+		if a.methodCall.HasFocus() {
+			if handler := a.methodCall.InputHandler(); handler != nil {
 				handler(event, setFocus)
 				return
 			}
 		}
-		if a.methodCall.HasFocus() {
-			if handler := a.methodCall.InputHandler(); handler != nil {
+		if a.importABI.HasFocus() {
+			if handler := a.importABI.InputHandler(); handler != nil {
+				handler(event, setFocus)
+				return
+			}
+		}
+		if a.Flex.HasFocus() {
+			if handler := a.Flex.InputHandler(); handler != nil {
 				handler(event, setFocus)
 				return
 			}
@@ -205,10 +240,12 @@ func (a *Account) InputHandler() func(event *tcell.EventKey, setFocus func(p tvi
 func (a *Account) SetRect(x int, y int, width int, height int) {
 	a.Flex.SetRect(x, y, width, height)
 	a.methodCall.SetRect(a.GetInnerRect())
+	a.importABI.SetRect(a.GetInnerRect())
 }
 
 // Draw implements tview.Primitive
 func (a *Account) Draw(screen tcell.Screen) {
 	a.Flex.Draw(screen)
 	a.methodCall.Draw(screen)
+	a.importABI.Draw(screen)
 }

internal/view/contract.go

@@ -45,7 +45,7 @@ func (d *MethodCallDialog) initLayout() {
 	// method list
 	methods := tview.NewTable()
 	methods.SetBorder(true)
-	methods.SetBorderColor(s.MethNameBorderColor)
+	methods.SetBorderColor(s.DialogBorderColor)
 	methods.SetTitle(style.Padding("Method"))
 	methods.SetSelectable(true, false)
 	methods.SetSelectionChangedFunc(func(row, column int) {
@@ -73,7 +73,7 @@ func (d *MethodCallDialog) initLayout() {
 	// arguments form
 	args := tview.NewForm()
 	args.SetBorder(true)
-	args.SetBorderColor(s.MethArgsBorderColor)
+	args.SetBorderColor(s.DialogBorderColor)
 	args.SetTitle(style.Padding("Arguments"))
 	args.SetLabelColor(s.InputFieldLableColor)
 	args.SetFieldBackgroundColor(s.InputFieldBgColor)
@@ -133,9 +133,12 @@ func (d *MethodCallDialog) Clear() {
 }
 
 func (d *MethodCallDialog) refresh() {
+	d.methods.Clear()
 	d.args.Clear(true)
 	d.result.Clear()
-	d.showMethodList()
+	if d.contract.HasABI() {
+		d.showMethodList()
+	}
 }
 
 func (d *MethodCallDialog) showMethodList() {
@@ -199,7 +202,7 @@ func (d *MethodCallDialog) callMethod() {
 	for i := 0; i < d.args.GetFormItemCount(); i++ {
 		item := d.args.GetFormItem(i).(*tview.InputField)
 		arg := method.Inputs[i]
-		val, err := conv.Unpack(arg.Type, item.GetText())
+		val, err := conv.UnpackArgument(arg.Type, item.GetText())
 		if err == nil {
 			args = append(args, val)
 		} else {

internal/view/format/string.go

@@ -16,7 +16,7 @@ func TruncateText(text string, size int) string {
 
 func NormalizeReceiverAddress(receiver *common.Address) string {
 	if receiver == nil {
-		return "0x"
+		return "0x0"
 	} else {
 		return receiver.Hex()
 	}

internal/view/helper.go

@@ -5,8 +5,36 @@ import (
 
 	"github.com/dyng/ramen/internal/common"
 	serv "github.com/dyng/ramen/internal/service"
+	"github.com/dyng/ramen/internal/view/util"
+	"github.com/gdamore/tcell/v2"
+	"github.com/rivo/tview"
 )
 
+type KeymapPrimitive interface {
+	SetInputCapture(capture func(event *tcell.EventKey) *tcell.EventKey) *tview.Box
+
+	KeyMaps() util.KeyMaps
+}
+
+func InitKeymap(p KeymapPrimitive, app *App) {
+	keymaps := p.KeyMaps()
+	p.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		// do not capture characters for InputField and TextArea
+		switch app.GetFocus().(type) {
+		case *tview.InputField, *tview.TextArea:
+			return event
+		}
+
+		handler, ok := keymaps.FindHandler(util.AsKey(event))
+		if ok {
+			handler(event)
+			return nil
+		} else {
+			return event
+		}
+	})
+}
+
 func Inc(i *int) int {
 	t := *i
 	*i++