feat: Add multi node (validator) testnet

ignite/cmd/model/testnet_multi_node.go

@@ -0,0 +1,162 @@
+package cmdmodel
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+	"path/filepath"
+	"strconv"
+	"syscall"
+
+	tea "github.com/charmbracelet/bubbletea"
+
+	"github.com/ignite/cli/v29/ignite/services/chain"
+)
+
+type NodeStatus int
+
+const (
+	Stopped NodeStatus = iota
+	Running
+)
+
+type Model struct {
+	appd string
+	args chain.MultiNodeArgs
+	ctx  context.Context
+
+	nodeStatuses []NodeStatus
+	pids         []int // Store the PIDs of the running processes
+	numNodes     int   // Number of nodes
+}
+
+type ToggleNodeMsg struct {
+	nodeIdx int
+}
+
+type UpdateStatusMsg struct {
+	nodeIdx int
+	status  NodeStatus
+}
+
+// Initialize the model
+func NewModel(chainname string, ctx context.Context, args chain.MultiNodeArgs) Model {
+	numNodes, err := strconv.Atoi(args.NumValidator)
+	if err != nil {
+		panic(err)
+	}
+	return Model{
+		appd:         chainname + "d",
+		args:         args,
+		ctx:          ctx,
+		nodeStatuses: make([]NodeStatus, numNodes), // initial states of nodes
+		pids:         make([]int, numNodes),
+		numNodes:     numNodes,
+	}
+}
+
+// Implement the Update function
+func (m Model) Init() tea.Cmd {
+	return nil
+}
+
+// ToggleNode toggles the state of a node
+func ToggleNode(nodeIdx int) tea.Cmd {
+	return func() tea.Msg {
+		return ToggleNodeMsg{nodeIdx: nodeIdx}
+	}
+}
+
+// Run or stop the node based on its status
+func RunNode(nodeIdx int, start bool, pid *int, args chain.MultiNodeArgs, appd string) tea.Cmd {
+	return func() tea.Msg {
+		if start {
+			nodeHome := filepath.Join(args.OutputDir, args.NodeDirPrefix+strconv.Itoa(nodeIdx))
+			// Create the command to run in background as a daemon
+			cmd := exec.Command(appd, "start", "--home", nodeHome)
+
+			// Start the process as a daemon
+			cmd.SysProcAttr = &syscall.SysProcAttr{
+				Setpgid: true, // Ensure it runs in a new process group
+			}
+
+			err := cmd.Start() // Start the node in the background
+			if err != nil {
+				fmt.Printf("Failed to start node %d: %v\n", nodeIdx+1, err)
+				return UpdateStatusMsg{nodeIdx: nodeIdx, status: Stopped}
+			}
+
+			*pid = cmd.Process.Pid // Store the PID
+			go cmd.Wait()          // Let the process run asynchronously
+			return UpdateStatusMsg{nodeIdx: nodeIdx, status: Running}
+		} else {
+			// Use kill to stop the node process by PID
+			if *pid != 0 {
+				err := syscall.Kill(-*pid, syscall.SIGTERM) // Stop the daemon process
+				if err != nil {
+					fmt.Printf("Failed to stop node %d: %v\n", nodeIdx+1, err)
+				} else {
+					*pid = 0 // Reset PID after stopping
+				}
+			}
+			return UpdateStatusMsg{nodeIdx: nodeIdx, status: Stopped}
+		}
+	}
+}
+
+// Stop all nodes
+func (m *Model) StopAllNodes() {
+	for i := 0; i < m.numNodes; i++ {
+		if m.nodeStatuses[i] == Running {
+			RunNode(i, false, &m.pids[i], m.args, m.appd)() // Stop node
+		}
+	}
+}
+
+// Update handles messages and updates the model
+func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.String() {
+		case "q":
+			m.StopAllNodes() // Stop all nodes before quitting
+			return m, tea.Quit
+		default:
+			// Check for numbers from 1 to numNodes
+			for i := 0; i < m.numNodes; i++ {
+				if msg.String() == fmt.Sprintf("%d", i+1) {
+					return m, ToggleNode(i)
+				}
+			}
+		}
+
+	case ToggleNodeMsg:
+		if m.nodeStatuses[msg.nodeIdx] == Running {
+			return m, RunNode(msg.nodeIdx, false, &m.pids[msg.nodeIdx], m.args, m.appd) // Stop node
+		}
+		return m, RunNode(msg.nodeIdx, true, &m.pids[msg.nodeIdx], m.args, m.appd) // Start node
+
+	case UpdateStatusMsg:
+		m.nodeStatuses[msg.nodeIdx] = msg.status
+		return m, nil
+	}
+
+	return m, nil
+}
+
+// View renders the interface
+func (m Model) View() string {
+	statusText := func(status NodeStatus) string {
+		if status == Running {
+			return "[Running]"
+		}
+		return "[Stopped]"
+	}
+
+	output := "Press keys 1,2,3.. to start and stop node 1,2,3.. respectively \nNode Control:\n"
+	for i := 0; i < m.numNodes; i++ {
+		output += fmt.Sprintf("%d. Node %d %s\n", i+1, i+1, statusText(m.nodeStatuses[i]))
+	}
+	output += "Press q to quit.\n"
+	return output
+}

ignite/cmd/testnet.go

@@ -18,6 +18,7 @@ func NewTestnet() *cobra.Command {
 
 	c.AddCommand(
 		NewTestnetInPlace(),
+		NewTestnetMultiNode(),
 	)
 
 	return c

ignite/cmd/testnet_multi_node.go

@@ -0,0 +1,175 @@
+package ignitecmd
+
+import (
+	"fmt"
+	"math/rand"
+	"strconv"
+	"time"
+
+	"cosmossdk.io/math"
+	"github.com/spf13/cobra"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	tea "github.com/charmbracelet/bubbletea"
+
+	cmdmodel "github.com/ignite/cli/v29/ignite/cmd/model"
+	"github.com/ignite/cli/v29/ignite/config/chain/base"
+	"github.com/ignite/cli/v29/ignite/pkg/cliui"
+	"github.com/ignite/cli/v29/ignite/services/chain"
+)
+
+func NewTestnetMultiNode() *cobra.Command {
+	c := &cobra.Command{
+		Use:   "multi-node",
+		Short: "Create a network test multi node",
+		Long: `Create a test network with the number of nodes from the config.yml file:
+			...
+			multi-node:
+				validators:
+					- name: validator1
+					stake: 100000000stake
+					- name: validator2
+					stake: 200000000stake
+					- name: validator3
+					stake: 200000000stake
+					- name: validator4
+					stake: 200000000stake
+				output-dir: ./.testchain-testnet/
+				chain-id: testchain-test-1
+				node-dir-prefix: validator
+
+	or random amount stake
+			....
+			multi-node:
+				random_validators:
+					count: 4
+					min_stake: 50000000stake
+					max_stake: 150000000stake
+				output-dir: ./.testchain-testnet/
+				chain-id: testchain-test-1
+				node-dir-prefix: validator
+
+
+		`,
+		Args: cobra.NoArgs,
+		RunE: testnetMultiNodeHandler,
+	}
+	flagSetPath(c)
+	flagSetClearCache(c)
+	c.Flags().AddFlagSet(flagSetHome())
+	c.Flags().AddFlagSet(flagSetCheckDependencies())
+	c.Flags().AddFlagSet(flagSetSkipProto())
+	c.Flags().AddFlagSet(flagSetVerbose())
+
+	c.Flags().Bool(flagQuitOnFail, false, "quit program if the app fails to start")
+	return c
+}
+
+func testnetMultiNodeHandler(cmd *cobra.Command, _ []string) error {
+	session := cliui.New(
+		cliui.WithVerbosity(getVerbosity(cmd)),
+	)
+	defer session.End()
+
+	return testnetMultiNode(cmd, session)
+}
+
+func testnetMultiNode(cmd *cobra.Command, session *cliui.Session) error {
+	chainOption := []chain.Option{
+		chain.WithOutputer(session),
+		chain.CollectEvents(session.EventBus()),
+		chain.CheckCosmosSDKVersion(),
+	}
+
+	if flagGetCheckDependencies(cmd) {
+		chainOption = append(chainOption, chain.CheckDependencies())
+	}
+
+	// check if custom config is defined
+	config, _ := cmd.Flags().GetString(flagConfig)
+	if config != "" {
+		chainOption = append(chainOption, chain.ConfigFile(config))
+	}
+
+	c, err := chain.NewWithHomeFlags(cmd, chainOption...)
+	if err != nil {
+		return err
+	}
+
+	cfg, err := c.Config()
+	if err != nil {
+		return err
+	}
+
+	numVal, amountDetails, err := getValidatorAmountStake(cfg.MultiNode)
+	if err != nil {
+		return err
+	}
+	args := chain.MultiNodeArgs{
+		ChainID:               cfg.MultiNode.ChainID,
+		ValidatorsStakeAmount: amountDetails,
+		OutputDir:             cfg.MultiNode.OutputDir,
+		NumValidator:          strconv.Itoa(numVal),
+		NodeDirPrefix:         cfg.MultiNode.NodeDirPrefix,
+	}
+
+	fmt.Printf("Creating %s nodes \n\n", args.NumValidator)
+	err = c.TestnetMultiNode(cmd.Context(), args)
+	if err != nil {
+		return err
+	}
+
+	time.Sleep(7 * time.Second)
+	fmt.Println()
+
+	m := cmdmodel.NewModel(c.Name(), cmd.Context(), args)
+	_, err = tea.NewProgram(m).Run()
+	return err
+}
+
+// getValidatorAmountStake returns the number of validators and the amountStakes arg from config.MultiNode
+func getValidatorAmountStake(cfg base.MultiNode) (int, string, error) {
+	var amounts string
+	count := 0
+
+	if len(cfg.Validators) == 0 {
+		numVal := cfg.RandomValidators.Count
+		minStake, err := sdk.ParseCoinNormalized(cfg.RandomValidators.MinStake)
+		if err != nil {
+			return count, amounts, err
+		}
+		maxStake, err := sdk.ParseCoinNormalized(cfg.RandomValidators.MaxStake)
+		if err != nil {
+			return count, amounts, err
+		}
+		minS := minStake.Amount.Uint64()
+		maxS := maxStake.Amount.Uint64()
+		for i := 0; i < numVal; i++ {
+			stakeAmount := minS + rand.Uint64()%(maxS-minS+1)
+			if amounts == "" {
+				amounts = math.NewIntFromUint64(stakeAmount).String()
+				count++
+			} else {
+				amounts = amounts + "," + math.NewIntFromUint64(stakeAmount).String()
+				count++
+			}
+		}
+	} else {
+		for _, v := range cfg.Validators {
+			stakeAmount, err := sdk.ParseCoinNormalized(v.Stake)
+			if err != nil {
+				return count, amounts, err
+			}
+			if amounts == "" {
+				amounts = stakeAmount.Amount.String()
+				count += 1
+			} else {
+				amounts = amounts + "," + stakeAmount.Amount.String()
+				count += 1
+			}
+		}
+	}
+
+	return count, amounts, nil
+}

ignite/config/chain/base/config.go

@@ -141,6 +141,29 @@ type Config struct {
 	Client     Client          `yaml:"client,omitempty" doc:"Configures client code generation."`
 	Genesis    xyaml.Map       `yaml:"genesis,omitempty" doc:"Custom genesis block modifications. Follow the nesting of the genesis file here to access all the parameters."`
 	Minimal    bool            `yaml:"minimal,omitempty" doc:"Indicates if the blockchain is minimal with the required Cosmos SDK modules."`
+	MultiNode  MultiNode       `yaml:"multi-node" doc:"Configuration for testnet multi node."`
+}
+
+// Validator defines the configuration for a single validator.
+type ValidatorDetails struct {
+	Name  string `yaml:"name" doc:"Name of the validator."`
+	Stake string `yaml:"stake" doc:"Amount of stake associated with the validator."`
+}
+
+// RandomValidator defines the configuration for random validators.
+type RandomValidatorDetails struct {
+	Count    int    `yaml:"count" doc:"Number of random validators to be generated."`
+	MinStake string `yaml:"min_stake" doc:"Minimum stake for each random validator."`
+	MaxStake string `yaml:"max_stake" doc:"Maximum stake for each random validator."`
+}
+
+// MultiNode holds the configuration related to multiple validators and random validators.
+type MultiNode struct {
+	Validators       []ValidatorDetails     `yaml:"validators" doc:"List of manually configured validators."`
+	RandomValidators RandomValidatorDetails `yaml:"random_validators" doc:"Configuration for randomly generated validators."`
+	OutputDir        string                 `yaml:"output-dir" doc:"Directory to store initialization data for the testnet"`
+	ChainID          string                 `yaml:"chain-id" doc:"Chain id for the testnet"`
+	NodeDirPrefix    string                 `yaml:"node-dir-prefix" doc:"Node directory prefix for the testnet"`
 }
 
 // GetVersion returns the config version.

ignite/pkg/chaincmd/chaincmd.go

@@ -28,6 +28,7 @@ const (
 	commandExport            = "export"
 	commandTendermint        = "tendermint"
 	commandTestnetInPlace    = "in-place-testnet"
+	commandTestnetMultiNode  = "multi-node"
 
 	optionHome                             = "--home"
 	optionNode                             = "--node"
@@ -59,6 +60,9 @@ const (
 	optionValidatorPrivateKey              = "--validator-privkey"
 	optionAccountToFund                    = "--accounts-to-fund"
 	optionSkipConfirmation                 = "--skip-confirmation"
+	optionAmountStakes                     = "--validators-stake-amount"
+	optionOutPutDir                        = "--output-dir"
+	optionNumValidator                     = "--v"
 
 	constTendermint = "tendermint"
 	constJSON       = "json"