README.md

@@ -56,3 +56,9 @@ go build -o data-extractor
 ```
 go test tests/*.go -v
 ```
+
+### 2. Ergo extractors
+
+Ergo extractors are under development.
+At this moment, we developed erg-erg extractors to test gravity on our network.
+We add other extractors when all tests passed.

config/config.go

@@ -12,7 +12,7 @@ const (
 )
 
 const (
-	MountedVolume = "/etc/extractor"
+	MountedVolume = "."
 )
 
 type MainConfig struct {

extractors/susy/bridge/common.go

@@ -6,15 +6,16 @@ import (
 	"fmt"
 
 	"github.com/Gravity-Tech/gravity-node-data-extractor/v2/extractors"
+	"github.com/Gravity-Tech/gravity-node-data-extractor/v2/helpers"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/mr-tron/base58"
 	wavescrypto "github.com/wavesplatform/go-lib-crypto"
 
+	"math/big"
+
 	solclient "github.com/portto/solana-go-sdk/client"
 	solcommon "github.com/portto/solana-go-sdk/common"
 	soltoken "github.com/portto/solana-go-sdk/tokenprog"
-
-	"math/big"
 )
 
 /**
@@ -68,21 +69,20 @@ type Action int
 type RequestType int
 
 const (
-	NewStatus          Status = 1
-	CompletedStatus    Status = 2
+	NewStatus       Status = 1
+	CompletedStatus Status = 2
 
 	ApproveAction      Action = 1
 	UnlockAction       Action = 2
 	MintAction         Action = 1
 	ChangeStatusAction Action = 2
 
-	IssueType          RequestType = 1
-	BurnType           RequestType = 2
-	LockType           RequestType = 1
-	UnlockType         RequestType = 2
+	IssueType  RequestType = 1
+	BurnType   RequestType = 2
+	LockType   RequestType = 1
+	UnlockType RequestType = 2
 )
 
-
 func ValidateEthereumBasedAddress(address string) bool {
 	return common.IsHexAddress(address)
 }
@@ -124,7 +124,7 @@ func ValidateSolanaTokenAccountOwnershipByTokenProgram(client *solclient.Client,
 	if tokenAccountState.Mint.ToBase58() != tokenMint {
 		return false, fmt.Errorf("token mint mismatch - got: %v; expected: %v", tokenAccountState.Mint.ToBase58(), tokenMint)
 	}
-	
+
 	fmt.Printf("tokenAccount: %+v \n", tokenAccount)
 
 	return true, nil
@@ -136,4 +136,14 @@ func ValidateSolanaAddress(address string) bool {
 		return false
 	}
 	return true
-}
+}
+
+func ValidateErgoAddress(address string, proxyUrl string) bool {
+	isValid, err := helpers.ValidateErgoAddress(address, proxyUrl)
+	if err != nil {
+		fmt.Printf("error: %v", err)
+		return false
+	}
+	return isValid
+}
+

extractors/susy/bridge/ergo.go

@@ -0,0 +1,205 @@
+package bridge
+
+import (
+	"context"
+	"encoding/base64"
+	"encoding/binary"
+	"fmt"
+	"github.com/Gravity-Tech/gravity-node-data-extractor/v2/extractors"
+	"github.com/Gravity-Tech/gravity-node-data-extractor/v2/helpers"
+	"math/big"
+	"strconv"
+)
+
+// TODO: Implement general queue iterator (Waves & ETH)
+
+type ErgoRequestsState struct {
+	requests map[string]interface{}
+}
+
+const (
+	ErgoRequestStatusSuccess = true
+)
+
+type ErgoToErgoExtractionBridge struct {
+	config     ConfigureCommand
+	configured bool
+
+	ergClientSource *helpers.ErgClient
+	ergClientTarget *helpers.ErgClient
+}
+
+func (provider *ErgoToErgoExtractionBridge) Configure(config ConfigureCommand) error {
+	if provider.configured {
+		return fmt.Errorf("bridge is configured already")
+	}
+
+	provider.config = config
+
+	// Node clients instantiation
+	var err error
+	provider.ergClientSource, err = helpers.NewClient(helpers.ErgOptions{BaseUrl: config.SourceNodeUrl})
+	if err != nil {
+		return err
+	}
+	provider.ergClientTarget, err = helpers.NewClient(helpers.ErgOptions{BaseUrl: config.SourceNodeUrl})
+	if err != nil {
+		return err
+	}
+	provider.configured = true
+
+	return nil
+}
+
+func (provider *ErgoToErgoExtractionBridge) pickRequestFromLUPortQueue(luRequests []helpers.Request) (helpers.Request, error) {
+	var rq helpers.Request
+	allRequestIds, err := provider.ergClientTarget.GetAllRequestIds(provider.config.DestinationNodeUrl, provider.config.IBPortAddress)
+	if err != nil {
+		return rq, nil
+	}
+	fmt.Println(allRequestIds)
+	isExist := func(luRequest helpers.Request) bool {
+		flag := false
+		for _, requestId := range allRequestIds {
+			if luRequest.RequestId == requestId {
+				flag = true
+				break
+			}
+		}
+		return flag
+	}
+	for _, luRequest := range luRequests {
+		if isExist(luRequest) {
+			continue
+		}
+		isValid := ValidateErgoAddress(luRequest.Receiver, provider.config.SourceNodeUrl)
+		if !isValid {
+			continue
+		}
+		rq = luRequest
+		break
+	}
+
+	return rq, nil
+}
+
+func (provider *ErgoToErgoExtractionBridge) pickRequestFromIBPortQueue(ibRequests []helpers.Request) (helpers.Request, error) {
+	var rq helpers.Request
+	allRequestIds, err := provider.ergClientTarget.GetAllRequestIds(provider.config.SourceNodeUrl, provider.config.LUPortAddress)
+	if err != nil {
+		return rq, nil
+	}
+
+	isExist := func(ibRequest helpers.Request) bool {
+		flag := false
+		for _, requestId := range allRequestIds {
+			if ibRequest.RequestId == requestId {
+				flag = true
+				break
+			}
+		}
+		return flag
+	}
+	for _, ibRequest := range ibRequests {
+		if isExist(ibRequest) {
+			continue
+		}
+		isValid := ValidateErgoAddress(ibRequest.Receiver, provider.config.DestinationNodeUrl)
+		if !isValid {
+			continue
+		}
+		rq = ibRequest
+		break
+	}
+
+	return rq, nil
+}
+
+func (provider *ErgoToErgoExtractionBridge) ExtractDirectTransferRequest(ctx context.Context) (*extractors.Data, error) {
+	luRequests, err := provider.ergClientSource.GetRequestsList(provider.config.SourceNodeUrl, provider.config.LUPortAddress, ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	rq, _ := provider.pickRequestFromLUPortQueue(luRequests)
+
+	if rq.RequestId == "" {
+		return nil, extractors.NotFoundErr
+	}
+	rqId := new(big.Int)
+	rqId, _ = rqId.SetString(rq.RequestId, 10)
+	fmt.Println(rqId)
+	amount, err := strconv.ParseInt(rq.Amount, 10, 64)
+	bigAmount := big.NewInt(amount)
+	fmt.Println(bigAmount)
+
+	sourceDecimals := big.NewInt(10)
+	sourceDecimals.Exp(sourceDecimals, big.NewInt(provider.config.SourceDecimals), nil)
+	destinationDecimals := big.NewInt(10)
+	destinationDecimals.Exp(destinationDecimals, big.NewInt(provider.config.DestinationDecimals), nil)
+
+	newAmount := bigAmount.
+			Mul(bigAmount, destinationDecimals).
+			Div(bigAmount, sourceDecimals).Int64()
+	fmt.Println(newAmount)
+
+	result := []byte{'m'}
+	newAmountBytes := make([]byte, 32)
+	binary.BigEndian.PutUint64(newAmountBytes, uint64(newAmount))
+	var RequestIdBytes [32]byte
+	result = append(result, rqId.FillBytes(RequestIdBytes[:])...)
+	result = append(result, newAmountBytes[:]...)
+	receiver := []byte(rq.Receiver)
+	result = append(result, receiver[:]...)
+
+	println(base64.StdEncoding.EncodeToString(result))
+	return &extractors.Data{
+		Type:  extractors.Base64,
+		Value: base64.StdEncoding.EncodeToString(result),
+	}, err
+}
+
+func (provider *ErgoToErgoExtractionBridge) ExtractReverseTransferRequest(ctx context.Context) (*extractors.Data, error) {
+	ibRequests, err := provider.ergClientSource.GetRequestsList(provider.config.DestinationNodeUrl, provider.config.IBPortAddress, ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	rq, _ := provider.pickRequestFromIBPortQueue(ibRequests)
+
+	if rq.RequestId == "" {
+		return nil, extractors.NotFoundErr
+	}
+
+	rqId := new(big.Int)
+	rqId, _ = rqId.SetString(rq.RequestId, 10)
+
+	amount, err := strconv.ParseInt(rq.Amount, 10, 64)
+	bigAmount := big.NewInt(amount)
+
+	sourceDecimals := big.NewInt(10)
+	sourceDecimals.Exp(sourceDecimals, big.NewInt(provider.config.SourceDecimals), nil)
+	destinationDecimals := big.NewInt(10)
+	destinationDecimals.Exp(destinationDecimals, big.NewInt(provider.config.DestinationDecimals), nil)
+
+	newAmount := bigAmount.
+		Mul(bigAmount, destinationDecimals).
+		Div(bigAmount, sourceDecimals).Int64()
+	fmt.Println(newAmount)
+
+	result := []byte{'m'}
+	newAmountBytes := make([]byte, 32)
+	binary.BigEndian.PutUint64(newAmountBytes, uint64(newAmount))
+	var RequestIdBytes [32]byte
+	result = append(result, rqId.FillBytes(RequestIdBytes[:])...)
+	result = append(result, newAmountBytes[:]...)
+	receiver := []byte(rq.Receiver)
+	result = append(result, receiver[:]...)
+
+	fmt.Println(newAmount)
+	println(base64.StdEncoding.EncodeToString(result))
+	return &extractors.Data{
+		Type:  extractors.Base64,
+		Value: base64.StdEncoding.EncodeToString(result),
+	}, err
+}