Merge pull request #276 from multiversx/feat/relayedv3

Feat/RelayedV3
multiversx · Jul 19, 2024 · 953ec5c · 953ec5c
2 parents a77b241 + 626328c
commit 953ec5c
Show file tree

Hide file tree

Showing 13 changed files with 398 additions and 117 deletions.
diff --git a/data/interface.go b/data/interface.go
@@ -272,6 +272,7 @@ type TransactionHandler interface {
 	GetSndAddr() []byte
 	GetGasLimit() uint64
 	GetGasPrice() uint64
+	GetUserTransactions() []TransactionHandler
 
 	SetValue(*big.Int)
 	SetData([]byte)
@@ -398,6 +399,7 @@ type TransactionWithFeeHandler interface {
 	GetData() []byte
 	GetRcvAddr() []byte
 	GetValue() *big.Int
+	GetUserTransactions() []TransactionHandler
 }
 
 // UserAccountHandler models a user account

diff --git a/data/receipt/receipt.go b/data/receipt/receipt.go
@@ -58,6 +58,11 @@ func (_ *Receipt) GetRcvUserName() []byte {
 	return nil
 }
 
+// GetUserTransactions returns nil as Receipt does not have user transactions
+func (rpt *Receipt) GetUserTransactions() []data.TransactionHandler {
+	return nil
+}
+
 // CheckIntegrity checks for not nil fields and negative value
 func (rpt *Receipt) CheckIntegrity() error {
 	return nil

diff --git a/data/receipt/receipt_test.go b/data/receipt/receipt_test.go
@@ -29,6 +29,7 @@ func TestReceipt_SettersAndGetters(t *testing.T) {
 	assert.Equal(t, uint64(0), r.GetNonce())
 	assert.Equal(t, uint64(0), r.GetGasPrice())
 	assert.Equal(t, uint64(0), r.GetGasLimit())
+	assert.Nil(t, r.GetUserTransactions())
 }
 
 func TestReceipt_CheckIntegrityReturnsNil(t *testing.T) {

diff --git a/data/rewardTx/rewardTx.go b/data/rewardTx/rewardTx.go
@@ -62,6 +62,11 @@ func (rtx *RewardTx) GetRcvUserName() []byte {
 	return nil
 }
 
+// GetUserTransactions returns nil as RewardTx does not have user transactions
+func (rtx *RewardTx) GetUserTransactions() []data.TransactionHandler {
+	return nil
+}
+
 // CheckIntegrity checks for not nil fields and negative value
 func (rtx *RewardTx) CheckIntegrity() error {
 	if len(rtx.RcvAddr) == 0 {

diff --git a/data/rewardTx/rewardTx_test.go b/data/rewardTx/rewardTx_test.go
@@ -27,6 +27,7 @@ func TestRewardTx_GettersAndSetters(t *testing.T) {
 	assert.Equal(t, uint64(0), rwdTx.GetGasLimit())
 	assert.Equal(t, uint64(0), rwdTx.GetGasPrice())
 	assert.Equal(t, uint64(0), rwdTx.GetNonce())
+	assert.Nil(t, rwdTx.GetUserTransactions())
 }
 
 func TestRewardTx_CheckIntegrityShouldWork(t *testing.T) {

diff --git a/data/smartContractResult/smartContractResult.go b/data/smartContractResult/smartContractResult.go
@@ -39,6 +39,11 @@ func (_ *SmartContractResult) GetRcvUserName() []byte {
 	return nil
 }
 
+// GetUserTransactions returns nil as SmartContractResult does not have user transactions
+func (scr *SmartContractResult) GetUserTransactions() []data.TransactionHandler {
+	return nil
+}
+
 // TrimSlicePtr creates a copy of the provided slice without the excess capacity
 func TrimSlicePtr(in []*SmartContractResult) []*SmartContractResult {
 	if len(in) == 0 {

diff --git a/data/smartContractResult/smartContractResult_test.go b/data/smartContractResult/smartContractResult_test.go
@@ -38,6 +38,7 @@ func TestSmartContractResult_SettersAndGetters(t *testing.T) {
 	assert.Equal(t, gasLimit, scr.GetGasLimit())
 	assert.Equal(t, gasPrice, scr.GetGasPrice())
 	assert.Equal(t, nonce, scr.GetNonce())
+	assert.Nil(t, scr.GetUserTransactions())
 }
 
 func TestTrimSlicePtr(t *testing.T) {

diff --git a/data/transaction/apiTransactionResult.go b/data/transaction/apiTransactionResult.go
@@ -69,6 +69,7 @@ type ApiTransactionResult struct {
 	Options                           uint32                    `json:"options"`
 	GuardianAddr                      string                    `json:"guardian,omitempty"`
 	GuardianSignature                 string                    `json:"guardianSignature,omitempty"`
+	InnerTransactions                 []*FrontendTransaction    `json:"innerTransactions,omitempty"`
 }
 
 // ApiSmartContractResult represents a smart contract result with changed fields' types in order to make it friendly for API's json

diff --git a/data/transaction/frontendTransaction.go b/data/transaction/frontendTransaction.go
@@ -2,19 +2,21 @@ package transaction
 
 // FrontendTransaction represents the DTO used in transaction signing/validation.
 type FrontendTransaction struct {
-	Nonce             uint64 `json:"nonce"`
-	Value             string `json:"value"`
-	Receiver          string `json:"receiver"`
-	Sender            string `json:"sender"`
-	SenderUsername    []byte `json:"senderUsername,omitempty"`
-	ReceiverUsername  []byte `json:"receiverUsername,omitempty"`
-	GasPrice          uint64 `json:"gasPrice"`
-	GasLimit          uint64 `json:"gasLimit"`
-	Data              []byte `json:"data,omitempty"`
-	Signature         string `json:"signature,omitempty"`
-	ChainID           string `json:"chainID"`
-	Version           uint32 `json:"version"`
-	Options           uint32 `json:"options,omitempty"`
-	GuardianAddr      string `json:"guardian,omitempty"`
-	GuardianSignature string `json:"guardianSignature,omitempty"`
+	Nonce             uint64                 `json:"nonce"`
+	Value             string                 `json:"value"`
+	Receiver          string                 `json:"receiver"`
+	Sender            string                 `json:"sender"`
+	SenderUsername    []byte                 `json:"senderUsername,omitempty"`
+	ReceiverUsername  []byte                 `json:"receiverUsername,omitempty"`
+	GasPrice          uint64                 `json:"gasPrice"`
+	GasLimit          uint64                 `json:"gasLimit"`
+	Data              []byte                 `json:"data,omitempty"`
+	Signature         string                 `json:"signature,omitempty"`
+	ChainID           string                 `json:"chainID"`
+	Version           uint32                 `json:"version"`
+	Options           uint32                 `json:"options,omitempty"`
+	GuardianAddr      string                 `json:"guardian,omitempty"`
+	GuardianSignature string                 `json:"guardianSignature,omitempty"`
+	Relayer           string                 `json:"relayer,omitempty"`
+	InnerTransactions []*FrontendTransaction `json:"innerTransactions,omitempty"`
 }
diff --git a/data/transaction/transaction.go b/data/transaction/transaction.go
@@ -2,6 +2,7 @@
 package transaction
 
 import (
+	"encoding/hex"
 	"math/big"
 
 	"github.com/multiversx/mx-chain-core-go/core"
@@ -36,6 +37,17 @@ func (tx *Transaction) SetSndAddr(addr []byte) {
 	tx.SndAddr = addr
 }
 
+// GetUserTransactions returns the inner transactions
+func (tx *Transaction) GetUserTransactions() []data.TransactionHandler {
+	innerTxs := tx.GetInnerTransactions()
+	txsHandlers := make([]data.TransactionHandler, len(innerTxs))
+	for i := 0; i < len(innerTxs); i++ {
+		txsHandlers[i] = innerTxs[i]
+	}
+
+	return txsHandlers
+}
+
 // TrimSlicePtr creates a copy of the provided slice without the excess capacity
 func TrimSlicePtr(in []*Transaction) []*Transaction {
 	if len(in) == 0 {
@@ -68,38 +80,25 @@ func (tx *Transaction) GetDataForSigning(encoder data.Encoder, marshaller data.M
 		return nil, ErrNilHasher
 	}
 
-	receiverAddr, err := encoder.Encode(tx.RcvAddr)
+	ftx, err := tx.prepareTx(encoder)
 	if err != nil {
 		return nil, err
 	}
 
-	senderAddr, err := encoder.Encode(tx.SndAddr)
-	if err != nil {
-		return nil, err
-	}
+	numInnerTxs := len(tx.InnerTransactions)
+	if numInnerTxs > 0 {
+		ftx.InnerTransactions = make([]*FrontendTransaction, numInnerTxs)
+		var errPrepare error
+		for i := 0; i < numInnerTxs; i++ {
+			ftx.InnerTransactions[i], errPrepare = tx.InnerTransactions[i].prepareTx(encoder)
+			if errPrepare != nil {
+				return nil, errPrepare
+			}
 
-	ftx := &FrontendTransaction{
-		Nonce:            tx.Nonce,
-		Value:            tx.Value.String(),
-		Receiver:         receiverAddr,
-		Sender:           senderAddr,
-		GasPrice:         tx.GasPrice,
-		GasLimit:         tx.GasLimit,
-		SenderUsername:   tx.SndUserName,
-		ReceiverUsername: tx.RcvUserName,
-		Data:             tx.Data,
-		ChainID:          string(tx.ChainID),
-		Version:          tx.Version,
-		Options:          tx.Options,
-	}
-
-	if len(tx.GuardianAddr) > 0 {
-		guardianAddr, errGuardian := encoder.Encode(tx.GuardianAddr)
-		if errGuardian != nil {
-			return nil, errGuardian
+			ftx.InnerTransactions[i].Signature = hex.EncodeToString(tx.InnerTransactions[i].Signature)
+			ftx.InnerTransactions[i].GuardianSignature = hex.EncodeToString(tx.InnerTransactions[i].GuardianSignature)
 		}
 
-		ftx.GuardianAddr = guardianAddr
 	}
 
 	ftxBytes, err := marshaller.Marshal(ftx)
@@ -147,3 +146,50 @@ func (tx *Transaction) CheckIntegrity() error {
 
 	return nil
 }
+
+func (tx *Transaction) prepareTx(encoder data.Encoder) (*FrontendTransaction, error) {
+	receiverAddr, err := encoder.Encode(tx.RcvAddr)
+	if err != nil {
+		return nil, err
+	}
+
+	senderAddr, err := encoder.Encode(tx.SndAddr)
+	if err != nil {
+		return nil, err
+	}
+
+	ftx := &FrontendTransaction{
+		Nonce:            tx.Nonce,
+		Value:            tx.Value.String(),
+		Receiver:         receiverAddr,
+		Sender:           senderAddr,
+		GasPrice:         tx.GasPrice,
+		GasLimit:         tx.GasLimit,
+		SenderUsername:   tx.SndUserName,
+		ReceiverUsername: tx.RcvUserName,
+		Data:             tx.Data,
+		ChainID:          string(tx.ChainID),
+		Version:          tx.Version,
+		Options:          tx.Options,
+	}
+
+	if len(tx.RelayerAddr) > 0 {
+		relayerAddr, errRelayer := encoder.Encode(tx.RelayerAddr)
+		if errRelayer != nil {
+			return nil, errRelayer
+		}
+
+		ftx.Relayer = relayerAddr
+	}
+
+	if len(tx.GuardianAddr) > 0 {
+		guardianAddr, errGuardian := encoder.Encode(tx.GuardianAddr)
+		if errGuardian != nil {
+			return nil, errGuardian
+		}
+
+		ftx.GuardianAddr = guardianAddr
+	}
+
+	return ftx, nil
+}