feat: vanilla opt-in programmatic script

tools/vanilla-stake/example.txt

@@ -0,0 +1,30 @@
+0xa1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6
+0xb2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1
+0xc3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2
+0xd4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3
+0xe5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4
+0xf6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5
+0xa2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7
+0xb3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2
+0xc4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3
+0xd5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4
+0xe6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5
+0xf7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6f7a2b3c4d5e6
+0xa3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8
+0xb4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3
+0xc5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4
+0xd6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5
+0xe7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6
+0xf8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7f8a3b4c5d6e7
+0xa4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9
+0xb5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4
+0xc6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5
+0xd7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6
+0xe8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7
+0xf9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8f9a4b5c6d7e8
+0xa5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0
+0xb6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5
+0xc7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6
+0xd8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7
+0xe9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8
+0xf0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9f0a5b6c7d8e9

tools/vanilla-stake/main.go

@@ -0,0 +1,222 @@
+package main
+
+import (
+	"bufio"
+	"context"
+	"fmt"
+	"math/big"
+	"os"
+	"time"
+
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/ethereum/go-ethereum/ethclient"
+	"github.com/urfave/cli/v2"
+
+	vanillaregistry "github.com/primev/mev-commit/contracts-abi/clients/VanillaRegistry"
+	"github.com/primev/mev-commit/x/keysigner"
+)
+
+var (
+	optionKeystorePath = &cli.StringFlag{
+		Name:     "keystore-dir",
+		Usage:    "directory where keystore file is stored",
+		EnvVars:  []string{"KEYSTORE_DIR"},
+		Required: true,
+	}
+	optionKeystorePassword = &cli.StringFlag{
+		Name:     "keystore-password",
+		Usage:    "use to access keystore",
+		EnvVars:  []string{"KEYSTORE_PASSWORD"},
+		Required: true,
+	}
+	optionL1RPCURL = &cli.StringFlag{
+		Name:     "l1-rpc-url",
+		Usage:    "URL of the L1 RPC server",
+		EnvVars:  []string{"L1_RPC_URL"},
+		Required: true,
+	}
+	optionPubkeyFilePath = &cli.StringFlag{
+		Name:     "pubkey-file-path",
+		Usage:    "path to the file containing the public keys",
+		EnvVars:  []string{"PUBKEY_FILE_PATH"},
+		Required: true,
+	}
+	optionVanillaRegistryAddress = &cli.StringFlag{
+		Name:    "vanilla-registry-address",
+		Usage:   "address of the vanilla registry contract",
+		EnvVars: []string{"VANILLA_REGISTRY_ADDRESS"},
+		Value:   "0x47afdcB2B089C16CEe354811EA1Bbe0DB7c335E9",
+	}
+)
+
+func main() {
+
+	flags := []cli.Flag{
+		optionKeystorePath,
+		optionKeystorePassword,
+		optionL1RPCURL,
+		optionPubkeyFilePath,
+		optionVanillaRegistryAddress,
+	}
+
+	app := &cli.App{
+		Name:  "vanilla-stake",
+		Usage: "Stake validators programmatically with the mev-commit vanilla registry",
+		Flags: flags,
+		Action: func(c *cli.Context) error {
+			return stakeVanilla(c)
+		},
+	}
+
+	if err := app.Run(os.Args); err != nil {
+		fmt.Fprintf(os.Stderr, "error: %v\n", err)
+		os.Exit(1)
+	}
+}
+
+func stakeVanilla(c *cli.Context) error {
+	keystoreDir := c.String(optionKeystorePath.Name)
+	keystorePassword := c.String(optionKeystorePassword.Name)
+	l1RpcUrl := c.String(optionL1RPCURL.Name)
+	pubkeyFilePath := c.String(optionPubkeyFilePath.Name)
+	vanillaRegistryAddress := c.String(optionVanillaRegistryAddress.Name)
+
+	signer, err := keysigner.NewKeystoreSigner(keystoreDir, keystorePassword)
+	if err != nil {
+		return fmt.Errorf("failed to create signer: %w", err)
+	}
+
+	client, err := ethclient.Dial(l1RpcUrl)
+	if err != nil {
+		return fmt.Errorf("failed to connect to the Ethereum client: %w", err)
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	chainID, err := client.ChainID(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to get chain ID: %w", err)
+	}
+
+	vrc, err := vanillaregistry.NewVanillaregistryCaller(common.HexToAddress(vanillaRegistryAddress), client)
+	if err != nil {
+		return fmt.Errorf("failed to create Vanilla Registry caller: %w", err)
+	}
+	ctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	minStake, err := vrc.MinStake(&bind.CallOpts{Context: ctx})
+	if err != nil {
+		return fmt.Errorf("failed to get min stake: %w", err)
+	}
+	fmt.Println("Min stake: ", minStake)
+
+	ctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	isWhitelisted, err := vrc.WhitelistedStakers(&bind.CallOpts{Context: ctx}, signer.GetAddress())
+	if err != nil {
+		return fmt.Errorf("failed to check if whitelisted: %w", err)
+	}
+	if !isWhitelisted {
+		return fmt.Errorf("caller %v is not whitelisted", signer.GetAddress())
+	}
+
+	vrt, err := vanillaregistry.NewVanillaregistryTransactor(common.HexToAddress(vanillaRegistryAddress), client)
+	if err != nil {
+		return fmt.Errorf("failed to create Vanilla Registry transactor: %w", err)
+	}
+
+	pksAsBytes, err := readBLSPublicKeysFromFile(pubkeyFilePath)
+	if err != nil {
+		return fmt.Errorf("failed to read public keys from file: %w", err)
+	}
+
+	batchSize := 20
+	type Batch struct {
+		pubKeys [][]byte
+	}
+	batches := make([]Batch, 0)
+	for i := 0; i < len(pksAsBytes); i += batchSize {
+		end := i + batchSize
+		if end > len(pksAsBytes) {
+			end = len(pksAsBytes)
+		}
+		batches = append(batches, Batch{pubKeys: pksAsBytes[i:end]})
+	}
+
+	for idx, batch := range batches {
+
+		opts, err := signer.GetAuth(chainID)
+		if err != nil {
+			return fmt.Errorf("failed to create transact opts: %w", err)
+		}
+
+		amountPerValidator := new(big.Int)
+		amountPerValidator.Set(minStake)
+		totalAmount := new(big.Int).Mul(amountPerValidator, big.NewInt(int64(len(batch.pubKeys))))
+		opts.Value = totalAmount
+
+		ctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
+		balance, err := client.BalanceAt(ctx, signer.GetAddress(), nil)
+		cancel()
+		if err != nil {
+			return fmt.Errorf("failed to get balance: %w", err)
+		}
+		if balance.Cmp(totalAmount) < 0 {
+			return fmt.Errorf("balance is less than total amount")
+		}
+
+		fmt.Println("Staking batch", idx+1, "with total amount", totalAmount, "and", len(batch.pubKeys), "validators")
+
+		tx, err := vrt.Stake(opts, batch.pubKeys)
+		if err != nil {
+			return fmt.Errorf("failed to stake: %w", err)
+		}
+
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
+		receipt, err := bind.WaitMined(ctx, client, tx)
+		cancel()
+		if err != nil {
+			return fmt.Errorf("failed to wait for stake tx to be mined: %w", err)
+		}
+
+		if receipt.Status != types.ReceiptStatusSuccessful {
+			return fmt.Errorf("stake tx included, but failed")
+		}
+		fmt.Println("Batch", idx+1, "completed")
+	}
+	fmt.Println("All staking batches completed successfully")
+	return nil
+}
+
+func readBLSPublicKeysFromFile(filePath string) ([][]byte, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		if err := file.Close(); err != nil {
+			fmt.Fprintf(os.Stderr, "error closing file: %v\n", err)
+		}
+	}()
+
+	var keys [][]byte
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		key := scanner.Text()
+		if key == "" {
+			continue // Empty lines skipped
+		}
+		if len(key) != 98 && len(key) != 96 {
+			return nil, fmt.Errorf("invalid BLS public key: %s - ensure there are no extra characters, commas, or spaces besides the pubkey and a trailing newline", key)
+		}
+		keyBytes := common.FromHex(key)
+		keys = append(keys, keyBytes)
+	}
+
+	if err := scanner.Err(); err != nil {
+		return nil, err
+	}
+	return keys, nil
+}