Skip to content

Commit

Permalink
intents: use ethcoder abi parser for more robust abi string parsing
Browse files Browse the repository at this point in the history
  • Loading branch information
pkieltyka committed Sep 26, 2024
1 parent da5f6da commit 3d750f3
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 46 deletions.
58 changes: 14 additions & 44 deletions intents/intent_data_transaction_delayed_abi.go
Original file line number Diff line number Diff line change
Expand Up @@ -116,55 +116,25 @@ func getMethodFromAbi(abi string, method string) (string, []string, error) {
} `json:"inputs"`
}

//
// First attempt to parse `abi` string as a plain method abi
// ie. transferFrom(address,address,uint256)
//

// Handle the case for already encoded method abi
if strings.Contains(abi, "(") && strings.Contains(abi, ")") && strings.HasPrefix(abi, method) {
// We may or may not have name information
// transferFrom(address,address,uint256)
// vs
// transferFrom(address from,address to,uint256 val)

// Start by obtaning only the args
args := strings.Split(abi, "(")[1]
args = strings.Split(args, ")")[0]

// Split the args by comma, to get the individual types
argTypes := strings.Split(args, ",")

order := make([]string, len(argTypes))
types := make([]string, len(argTypes))

incompleteNaming := false

for i, arg := range argTypes {
// If starts with space, trim it
arg = strings.TrimLeft(arg, " ")

if strings.Contains(arg, " ") {
// We have name information, so we need to extract it
spl := strings.Split(arg, " ")

order[i] = spl[1]
types[i] = spl[0]

} else {
// We don't have name information, so we must
// mark this case as incomplete
incompleteNaming = true

// Assume that arg is the type
types[i] = arg
}
}

if incompleteNaming {
order = nil
eventDef, err := ethcoder.ParseEventDef(abi)
if err != nil {
return "", nil, err
}

// Re encode abi, now without name information
fnc := method + "(" + strings.Join(types, ",") + ")"
return fnc, order, nil
return eventDef.Sig, eventDef.ArgNames, nil
}

//
// If above didn't work, attempt to parse `abi` string as
// a JSON object of the full abi definition
//

// Handle array of function abis and single function abi
var abis []FunctionAbi
if strings.HasPrefix(abi, "[") {
Expand Down
4 changes: 2 additions & 2 deletions intents/intent_data_transaction_delayed_abi_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ func TestGetMethodFromABI(t *testing.T) {
assert.Nil(t, err)

assert.Equal(t, "transfer(address,uint256)", res)
assert.Nil(t, order)
assert.Equal(t, []string{"arg1", "arg2"}, order)

// From plain method, with named args
res, order, err = getMethodFromAbi(`transfer(address _to,uint256 _value, bytes _mas)`, "transfer")
Expand All @@ -60,7 +60,7 @@ func TestGetMethodFromABI(t *testing.T) {

assert.Nil(t, err)
assert.Equal(t, "transfer(address,uint256,bytes)", res)
assert.Nil(t, order)
assert.Equal(t, []string{"_to", "arg2", "_mas"}, order)
}

func TestEncodeDelayedABI(t *testing.T) {
Expand Down

0 comments on commit 3d750f3

Please sign in to comment.