feat: add divvi submit referral tag

Author: kamikazechaser

README.md

@@ -16,7 +16,8 @@ Features:
 * Grassroots Economics [address](https://software.grassecon.org/addresses) loaders
 * Dump reverted tx reason
 * Batch balances scanner
-* Index iterator
+* Index scanner
+* Divvi support
 
 ## Installation
 

divvi.go

@@ -1,27 +1,112 @@
 package ethutils
 
 import (
+	"bytes"
+	"context"
 	"encoding/hex"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
 	"math/big"
+	"net/http"
 
 	"github.com/ethereum/go-ethereum/common"
 )
 
+type SubmitBody struct {
+	TxHash  string `json:"txHash"`
+	ChainID int64  `json:"chainId"`
+}
+
 const (
 	divviMagicPrefix               = "6decb85d"
 	knownEmptyAddressArrayEncoding = "00000000000000000000000000000000000000000000000000000000000000600000000000000000000000000000000000000000000000000000000000000000"
+
+	divviAPIEndpoint = "https://api.divvi.xyz/submitReferral"
+	userAgent        = "ethutils/v1.x.x"
+	contentType      = "application/json"
 )
 
-var referalTagFormat1Byte = []byte{0x01}
+var (
+	referalTagFormat1Byte = []byte{0x01}
+
+	ErrDivviClientNotInitialized = errors.New("divvi client not initialized")
+)
 
 func (p *Provider) GetReferalTag(user common.Address) []byte {
-	encodedBytes := concatBytes(encodeAddress(user), encodeAddress(p.DivviConsumerAddress), encodeAddressArrayEmpty())
+	encodedBytes := ConcatBytes(encodeAddress(user), encodeAddress(p.DivviConsumerAddress), encodeAddressArrayEmpty())
 	payloadLen := []byte{byte(len(encodedBytes) >> 8), byte(len(encodedBytes))}
 	magicPrefix, _ := hex.DecodeString(divviMagicPrefix)
-	payload := concatBytes(magicPrefix, referalTagFormat1Byte, payloadLen, encodedBytes)
+	payload := ConcatBytes(magicPrefix, referalTagFormat1Byte, payloadLen, encodedBytes)
 	return payload
 }
 
+func (p *Provider) SubmitReferral(ctx context.Context, txHash common.Hash) error {
+	if p.DivviClient == nil {
+		return ErrDivviClientNotInitialized
+	}
+	body := SubmitBody{
+		TxHash:  txHash.Hex(),
+		ChainID: CeloMainnet,
+	}
+
+	b, err := json.Marshal(&body)
+	if err != nil {
+		return err
+	}
+
+	resp, err := p.requestWithCtx(ctx, http.MethodPost, divviAPIEndpoint, bytes.NewBuffer(b))
+	if err != nil {
+		return err
+	}
+
+	if err := parseResponse(resp); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (p *Provider) requestWithCtx(ctx context.Context, method string, url string, body io.Reader) (*http.Response, error) {
+	req, err := http.NewRequestWithContext(ctx, method, url, body)
+	if err != nil {
+		return nil, err
+	}
+	return p.do(req)
+}
+
+func (p *Provider) do(req *http.Request) (*http.Response, error) {
+	builtRequest, err := setHeaders(req)
+	if err != nil {
+		return nil, err
+	}
+
+	return p.DivviClient.Do(builtRequest)
+}
+
+func setHeaders(req *http.Request) (*http.Request, error) {
+	req.Header.Set("User-Agent", userAgent)
+	req.Header.Set("Accept", contentType)
+	req.Header.Set("Content-Type", contentType)
+
+	return req, nil
+}
+
+func parseResponse(resp *http.Response) error {
+	defer resp.Body.Close()
+
+	if resp.StatusCode >= http.StatusBadRequest {
+		b, err := io.ReadAll(resp.Body)
+		if err != nil {
+			return fmt.Errorf("divvi api error: status=%s", resp.Status)
+		}
+		return fmt.Errorf("divvi api error: status=%s body=%s", resp.Status, string(b))
+	}
+
+	return nil
+}
+
 func encodeAddress(address common.Address) []byte {
 	return common.LeftPadBytes(address.Bytes(), 32)
 }
@@ -51,7 +136,7 @@ func encodeAddressArrayEmpty() []byte {
 	return emptyBytes
 }
 
-func concatBytes(slices ...[]byte) []byte {
+func ConcatBytes(slices ...[]byte) []byte {
 	var result []byte
 	for _, slice := range slices {
 		result = append(result, slice...)

provider.go

@@ -2,6 +2,8 @@ package ethutils
 
 import (
 	"math/big"
+	"net/http"
+	"time"
 
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
@@ -16,6 +18,7 @@ type (
 		Signer                        types.Signer
 		BalanceScannerContractAddress string
 		DivviConsumerAddress          common.Address
+		DivviClient                   *http.Client
 	}
 )
 
@@ -40,6 +43,9 @@ func WithBalanceScannerAddress(address string) Option {
 func WithDivviConsumerAddress(address string) Option {
 	return func(p *Provider) {
 		p.DivviConsumerAddress = w3.A(address)
+		p.DivviClient = &http.Client{
+			Timeout: time.Second * 10,
+		}
 	}
 }
 

transaction.go

@@ -20,6 +20,7 @@ type (
 
 	GasTransferTxOpts struct {
 		To        common.Address
+		InputData []byte
 		Value     *big.Int
 		GasFeeCap *big.Int
 		GasTipCap *big.Int
@@ -54,9 +55,10 @@ func (p *Provider) SignContractExecutionTx(privateKey *ecdsa.PrivateKey, txData
 func (p *Provider) SignGasTransferTx(privateKey *ecdsa.PrivateKey, txData GasTransferTxOpts) (*types.Transaction, error) {
 	tx, err := types.SignNewTx(privateKey, p.Signer, &types.DynamicFeeTx{
 		Value:     txData.Value,
+		Data:      txData.InputData,
 		To:        &txData.To,
 		Nonce:     txData.Nonce,
-		Gas:       21000,
+		Gas:       25000,
 		GasFeeCap: txData.GasFeeCap,
 		GasTipCap: txData.GasTipCap,
 	})