erigon-pulse/cmd/devnettest/services/utility.go

package services

import (
	"fmt"
	"github.com/ledgerwatch/erigon/cmd/devnettest/requests"
	"github.com/ledgerwatch/erigon/common"
	"github.com/ledgerwatch/erigon/crypto"
)

var devnetSignPrivateKey, _ = crypto.HexToECDSA("26e86e45f6fc45ec6e2ecd128cec80fa1d1505e5507dcd2ae58c3130a7a97b48")

func ValidateInputs(getBalance *bool, sendTx *bool, txpoolContent *bool, blockNum *string, value *uint64, to *string) {
	if !(*getBalance) && !(*sendTx) && !(*txpoolContent) {
		panic("At least one function flag (get-balance, send-tx, txpool-content) should be true")
	}

	seen := false
	for _, val := range []bool{*getBalance, *sendTx, *txpoolContent} {
		if val {
			if seen {
				panic("Only function flag (get-balance, send-tx, txpool-content) can be true at a time")
			}
			seen = true
		}
	}

	if *value <= 0 {
		panic("Value must be greater than zero")
	}

	if *getBalance {
		if *to == "" {
			panic("Cannot check balance of empty address")
		}
		if *blockNum != "pending" && *blockNum != "latest" && *blockNum != "earliest" {
			panic("Block number must be 'pending', 'latest' or 'earliest'")
		}
	}

	if *sendTx && *to == "" {
		panic("Cannot send to empty address")
	}

}

func ParseRequests(getBalance *bool, sendTx *bool, txpoolContent *bool, clearDev *bool, blockNum *string, value *uint64, to *string) {
	if *getBalance {
		toAddress := common.HexToAddress(*to)
		requests.GetBalance(toAddress, *blockNum)
	}

	if *sendTx {
		toAddress := common.HexToAddress(*to)
		requests.SendTx(devnetSignPrivateKey, toAddress, *value)
	}

	if *txpoolContent {
		requests.TxpoolContent()
	}

	if *clearDev {
		clearDevDB()
	}
}

func clearDevDB() {
	fmt.Printf("Clearing ~/dev\n")
}