chore(e2e): add allow validators command

e2e/app/admin/allowoperator.go

@@ -0,0 +1,81 @@
+package admin
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/omni-network/omni/contracts/bindings"
+	"github.com/omni-network/omni/e2e/app"
+	"github.com/omni-network/omni/e2e/app/eoa"
+	"github.com/omni-network/omni/halo/genutil/evm/predeploys"
+	"github.com/omni-network/omni/lib/errors"
+	"github.com/omni-network/omni/lib/log"
+	"github.com/omni-network/omni/lib/netconf"
+
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
+	"github.com/ethereum/go-ethereum/common"
+)
+
+var ensureOmegaOperators = []common.Address{
+	// TODO(corver): Add omega operators to ensure are added here.
+}
+
+// AllowOperators ensures that all operators hard-coded in this package is allowed as validators.
+// Note it only adds any of the operators that are missing, it doesn't remove any ever.
+func AllowOperators(ctx context.Context, def app.Definition, cfg PortalAdminConfig) error {
+	network := def.Testnet.Network
+	if err := cfg.Validate(); err != nil {
+		return err
+	} else if cfg.Chain != "omni_evm" {
+		return errors.New("allow operator only supported on omni_evm", "chain", cfg.Chain)
+	} else if network.Static().Network != netconf.Omega {
+		return errors.New("allow operator only supported on omega", "network", network.Static().Network.String())
+	}
+
+	backend, err := def.Backends().Backend(network.Static().OmniExecutionChainID)
+	if err != nil {
+		return err
+	}
+
+	contract, err := bindings.NewStaking(common.HexToAddress(predeploys.Staking), backend)
+	if err != nil {
+		return errors.Wrap(err, "new staking contract")
+	}
+
+	var toAllow []common.Address
+	for _, operator := range ensureOmegaOperators {
+		if ok, err := contract.IsAllowedValidator(&bind.CallOpts{}, operator); err != nil {
+			return errors.Wrap(err, "call is allowed validator")
+		} else if ok {
+			log.Info(ctx, "Operator already allowed as validator, skipping", "operator", operator)
+			continue
+		}
+
+		toAllow = append(toAllow, operator)
+		log.Info(ctx, "Operator not allowed yet, adding to transaction", "operator", operator)
+	}
+
+	if len(toAllow) == 0 {
+		log.Info(ctx, "All operators already allowed to be validators", "count", len(ensureOmegaOperators))
+		return nil
+	}
+
+	txOpts, err := backend.BindOpts(ctx, eoa.MustAddress(network, eoa.RoleAdmin))
+	if err != nil {
+		return errors.Wrap(err, "bind tx opts")
+	}
+
+	tx, err := contract.AllowValidators(txOpts, toAllow)
+	if err != nil {
+		return errors.Wrap(err, "allow validators")
+	}
+
+	if _, err := backend.WaitMined(ctx, tx); err != nil {
+		return errors.Wrap(err, "wait minded")
+	}
+
+	link := fmt.Sprintf("https://%s.omniscan.network/tx/%s", network, tx.Hash().Hex())
+	log.Info(ctx, "🎉 Successfully allowed operators as validators", "count", len(toAllow), "link", link)
+
+	return nil
+}

e2e/cmd/admin.go

@@ -17,11 +17,28 @@ func newAdminCmd(def *app.Definition) *cobra.Command {
 		newPausePortalCmd(def),
 		newUnpausePortalCmd(def),
 		newUpgradePortalCmd(def),
+		newAllowValidatorsCmd(def),
 	)
 
 	return cmd
 }
 
+func newAllowValidatorsCmd(def *app.Definition) *cobra.Command {
+	cfg := admin.DefaultPortalAdminConfig()
+
+	cmd := &cobra.Command{
+		Use:   "allow-operators",
+		Short: "Ensure all operators are allowed as validators",
+		RunE: func(cmd *cobra.Command, _ []string) error {
+			return admin.AllowOperators(cmd.Context(), *def, cfg)
+		},
+	}
+
+	bindPortalAdminFlags(cmd.Flags(), &cfg)
+
+	return cmd
+}
+
 func newPausePortalCmd(def *app.Definition) *cobra.Command {
 	cfg := admin.DefaultPortalAdminConfig()