Feat/juno on sedge

README.md

@@ -132,12 +132,12 @@ Users acknowledge that no warranty is being made of a successful installation. S
 
 ### Mainnet
 
-| Execution  | Consensus  | Validator  |
-| ---------- | ---------- | ---------- |
-| Geth       | Lighthouse | Lighthouse |
-| Nethermind | Lodestar   | Lodestar   |
-| Erigon     | Prysm      | Prysm      |
-| Besu       | Teku       | Teku       |
+| Execution  | Consensus  | Validator  | Starknet  |
+| ---------- | ---------- | ---------- |---------- |
+| Geth       | Lighthouse | Lighthouse | Juno      |
+| Nethermind | Lodestar   | Lodestar   |           |
+| Erigon     | Prysm      | Prysm      |           |
+| Besu       | Teku       | Teku       |           |
 
 ### Sepolia
 
@@ -150,12 +150,12 @@ Users acknowledge that no warranty is being made of a successful installation. S
 
 ### Goerli
 
-| Execution  | Consensus  | Validator  |
-| ---------- | ---------- | ---------- |
-| Geth       | Lighthouse | Lighthouse |
-| Nethermind | Lodestar   | Lodestar   |
-| Erigon     | Prysm      | Prysm      |
-| Besu       | Teku       | Teku       |
+| Execution  | Consensus  | Validator  | Starknet   |
+| ---------- | ---------- | ---------- | ---------- |
+| Geth       | Lighthouse | Lighthouse | Juno       |
+| Nethermind | Lodestar   | Lodestar   |            |
+| Erigon     | Prysm      | Prysm      |            |
+| Besu       | Teku       | Teku       |            |
 
 ### Holesky
 

cli/actions/getContainers.go

@@ -54,6 +54,9 @@ func (actions *sedgeActions) GetContainersData(options GetContainersDataOptions)
 		if composeData.Services.Validator != nil && composeData.Services.Validator.ContainerName != "" {
 			containersNames = append(containersNames, composeData.Services.Validator.ContainerName)
 		}
+		if composeData.Services.Starknet != nil && composeData.Services.Starknet.ContainerName != "" {
+			containersNames = append(containersNames, composeData.Services.Starknet.ContainerName)
+		}
 
 		outputData := ContainersData{}
 		outputData.Containers = make([]ContainerData, 0, len(containersNames))

cli/actions/getContainers_test.go

@@ -41,6 +41,8 @@ const (
 	consensusNotFoundContainerName = "sedge-execution-client-nf"
 	validatorContainerName         = "sedge-validator-client"
 	validatorNotFoundContainerName = "sedge-validator-client-nf"
+	starknetContainerName          = "sedge-starknet-client"
+	starknetNotFoundContainerName  = "sedge-starknet-client-nf"
 	mevBoostContainerName          = "sedge-mev-boost"
 
 	executionContainerIp    = "192.168.1.1"
@@ -51,6 +53,8 @@ const (
 	validatorContainerImage = "consensys/teku"
 	mevBoostContainerIp     = "192.168.1.4"
 	mevBoostContainerImage  = "flashbots/mev-boost"
+	starknetContainerIp     = "192.168.1.5"
+	starknetContainerImage  = "nethermind/juno"
 
 	executionNotFoundErrorMsg = "execution container not found"
 	consensusNotFoundErrorMsg = "consensus container not found"

cli/actions/setup_test.go

@@ -38,11 +38,11 @@ func TestSetupContainers(t *testing.T) {
 			name: "with services",
 			options: actions.SetupContainersOptions{
 				GenerationPath: filepath.Join("a", "b", "c", "d"),
-				Services:       []string{"validator", "consensus", "execution"},
+				Services:       []string{"validator", "consensus", "execution", "starknet"},
 			},
-			expectedBuildCmd:  fmt.Sprintf("docker compose -f %s build validator consensus execution", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
-			expectedPullCmd:   fmt.Sprintf("docker compose -f %s pull validator consensus execution", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
-			expectedCreateCmd: fmt.Sprintf("docker compose -f %s create validator consensus execution", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
+			expectedBuildCmd:  fmt.Sprintf("docker compose -f %s build validator consensus execution starknet", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
+			expectedPullCmd:   fmt.Sprintf("docker compose -f %s pull validator consensus execution starknet", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
+			expectedCreateCmd: fmt.Sprintf("docker compose -f %s create validator consensus execution starknet", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
 		},
 		{
 			name: "without services",
@@ -58,12 +58,12 @@ func TestSetupContainers(t *testing.T) {
 			name: "skip-pull",
 			options: actions.SetupContainersOptions{
 				GenerationPath: filepath.Join("a", "b", "c", "d"),
-				Services:       []string{"execution", "consensus"},
+				Services:       []string{"execution", "consensus", "starknet"},
 				SkipPull:       true,
 			},
-			expectedBuildCmd:  fmt.Sprintf("docker compose -f %s build execution consensus", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
+			expectedBuildCmd:  fmt.Sprintf("docker compose -f %s build execution consensus starknet", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
 			expectedPullCmd:   "",
-			expectedCreateCmd: fmt.Sprintf("docker compose -f %s create execution consensus", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
+			expectedCreateCmd: fmt.Sprintf("docker compose -f %s create execution consensus starknet", filepath.Join("a", "b", "c", "d", "docker-compose.yml")),
 		},
 	}
 	for _, tc := range tests {

cli/cli.go

@@ -51,6 +51,7 @@ const (
 	NodeTypeExecution = "execution"
 	NodeTypeConsensus = "consensus"
 	NodeTypeValidator = "validator"
+	NodeTypeStarknet  = "starknet"
 
 	Randomize = "randomize"
 
@@ -67,6 +68,7 @@ type CliCmdOptions struct {
 	generationPath           string
 	nodeType                 string
 	withValidator            bool
+	withFullL1Node           bool
 	withMevBoost             bool
 	importSlashingProtection bool
 	slashingProtectionFrom   string
@@ -97,6 +99,7 @@ func CliCmd(p ui.Prompter, actions actions.SedgeActions, depsMgr dependencies.De
 - Execution Node
 - Consensus Node
 - Validator Node
+- Starknet  Node
 
 Follow the prompts to select the options you want for your node. At the end of the process, you will
 be asked to run the generated setup or not. If you chose to run the setup, it will be executed for you
@@ -120,6 +123,8 @@ using docker compose command behind the scenes.
 				return setupConsensusNode(p, o, actions, depsMgr)
 			case NodeTypeValidator:
 				return setupValidatorNode(p, o, actions, depsMgr)
+			case NodeTypeStarknet:
+				return setupStarknetNode(p, o, actions, depsMgr)
 			}
 			return nil
 		},
@@ -313,6 +318,61 @@ func setupValidatorNode(p ui.Prompter, o *CliCmdOptions, a actions.SedgeActions,
 	return postGenerate(p, o, a, depsManager)
 }
 
+// setup a starknet node
+func setupStarknetNode(p ui.Prompter, o *CliCmdOptions, a actions.SedgeActions, depsManager dependencies.DependenciesManager) (err error) {
+	o.genData.Services = []string{"starknet"}
+	if err := selectStarknetClient(p, o); err != nil {
+		return err
+	}
+	if err := confirmWithFullStarknetNode(p, o); err != nil {
+		return err
+	}
+	if o.genData.Network == NetworkCustom {
+		if err := runPromptActions(p, o,
+			inputCustomNetworkConfig,
+			inputCustomChainSpec,
+			inputCustomGenesis,
+			inputCustomDeployBlock,
+		); err != nil {
+			return err
+		}
+	}
+	if o.withFullL1Node {
+		o.genData.Services = append(o.genData.Services, "execution", "consensus")
+
+		if err := runPromptActions(p, o,
+			selectExecutionClient,
+			selectConsensusClient,
+			inputStarknetGracePeriod,
+			inputCheckpointSyncURL,
+			inputFeeRecipientNoValidator,
+		); err != nil {
+			return err
+		}
+	} else {
+		if err := runPromptActions(p, o,
+			inputEthNode,
+		); err != nil {
+			return err
+		}
+	}
+	if err := confirmExposeAllPorts(p, o); err != nil {
+		return err
+	}
+	if err := setupJWT(p, o, false); err != nil {
+		return err
+	}
+	// Call generate action
+	o.genData, err = a.Generate(actions.GenerateOptions{
+		GenerationData: o.genData,
+		GenerationPath: o.generationPath,
+	})
+	if err != nil {
+		return err
+	}
+	return postGenerate(p, o, a, depsManager)
+}
+
 func setupJWT(p ui.Prompter, o *CliCmdOptions, skip bool) error {
 	if skip {
 		if err := selectJWTSourceOrSkip(p, o); err != nil {
@@ -361,6 +421,8 @@ func postGenerate(p ui.Prompter, o *CliCmdOptions, a actions.SedgeActions, depsM
 		services = []string{"consensus"}
 	case NodeTypeValidator:
 		services = []string{"validator"}
+	case NodeTypeStarknet:
+		services = []string{"starknet"}
 	}
 	run, err := p.Confirm("Run services now?", false)
 	if err != nil {
@@ -615,6 +677,11 @@ func runPromptActions(p ui.Prompter, o *CliCmdOptions, actions ...promptAction)
 	return nil
 }
 
+// check if network is mainnet or sepolia for starknet
+func networkNotMainnetOrSepolia(o *CliCmdOptions) bool {
+	return o.genData.Network != NetworkMainnet && o.genData.Network != NetworkSepolia
+}
+
 func selectNetwork(p ui.Prompter, o *CliCmdOptions) error {
 	options := []string{NetworkMainnet, NetworkGoerli, NetworkSepolia, NetworkGnosis, NetworkChiado, NetworkHolesky}
 	index, err := p.Select("Select network", "", options)
@@ -625,8 +692,22 @@ func selectNetwork(p ui.Prompter, o *CliCmdOptions) error {
 	return nil
 }
 
+func removeFromNodeType(options []string, optionToRemove string) []string {
+	newSlice := make([]string, 0)
+	for _, item := range options {
+		if item != optionToRemove {
+			newSlice = append(newSlice, item)
+		}
+	}
+	return newSlice
+}
+
 func selectNodeType(p ui.Prompter, o *CliCmdOptions) error {
-	options := []string{NodeTypeFullNode, NodeTypeExecution, NodeTypeConsensus, NodeTypeValidator}
+	options := []string{NodeTypeFullNode, NodeTypeExecution, NodeTypeConsensus, NodeTypeValidator, NodeTypeStarknet}
+	// check if mainnet or goerli was selected, if not omit starknet
+	if networkNotMainnetOrSepolia(o) {
+		options = removeFromNodeType(options, NodeTypeStarknet)
+	}
 	index, err := p.Select("Select node type", "", options)
 	if err != nil {
 		return err
@@ -722,6 +803,35 @@ func selectValidatorClient(p ui.Prompter, o *CliCmdOptions) (err error) {
 	return nil
 }
 
+func selectStarknetClient(p ui.Prompter, o *CliCmdOptions) (err error) {
+	c := clients.ClientInfo{Network: o.genData.Network}
+	supportedClients, err := c.SupportedClients(starknet)
+	if err != nil {
+		return err
+	}
+	options := append(supportedClients, Randomize)
+	index, err := p.Select("Select starknet client", "", options)
+	if err != nil {
+		return err
+	}
+	selectedStarknetClient := options[index]
+	// In case random is selected, select a random client
+	if selectedStarknetClient == Randomize {
+		randomName, err := clients.RandomClientName(supportedClients)
+		if err != nil {
+			return err
+		}
+		selectedStarknetClient = randomName
+		log.Info("Random starknet client selected: ", selectedStarknetClient)
+	}
+	o.genData.StarknetClient = &clients.Client{
+		Name: selectedStarknetClient,
+		Type: "starknet",
+	}
+	o.genData.StarknetClient.SetImageOrDefault("")
+	return nil
+}
+
 func selectJWTSource(p ui.Prompter, o *CliCmdOptions) error {
 	options := []string{SourceTypeCreate, SourceTypeExisting}
 	index, err := p.Select("Select JWT source", "", options)
@@ -777,6 +887,11 @@ func confirmWithValidator(p ui.Prompter, o *CliCmdOptions) (err error) {
 	return
 }
 
+func confirmWithFullStarknetNode(p ui.Prompter, o *CliCmdOptions) (err error) {
+	o.withFullL1Node, err = p.Confirm("Do you want to set up a full starknet node (with execution and consensus)?", true)
+	return
+}
+
 func confirmExposeAllPorts(p ui.Prompter, o *CliCmdOptions) (err error) {
 	o.genData.MapAllPorts, err = p.Confirm("Do you want to expose all ports?", false)
 	return
@@ -889,6 +1004,15 @@ func inputValidatorGracePeriod(p ui.Prompter, o *CliCmdOptions) (err error) {
 	return nil
 }
 
+func inputStarknetGracePeriod(p ui.Prompter, o *CliCmdOptions) (err error) {
+	epochs, err := p.InputInt64("Starknet grace period. This is the number of epochs the starknet client will wait before starting", 1)
+	if err != nil {
+		return err
+	}
+	o.genData.SLStartGracePeriod = uint((configs.NetworkEpochTime(o.genData.Network) * time.Duration(epochs)).Seconds())
+	return nil
+}
+
 func inputGenerationPath(p ui.Prompter, o *CliCmdOptions) (err error) {
 	o.generationPath, err = p.Input("Generation path", configs.DefaultAbsSedgeDataPath, false, nil)
 	if err != nil {
@@ -980,6 +1104,11 @@ func inputContainerTag(p ui.Prompter, o *CliCmdOptions) (err error) {
 	return
 }
 
+func inputEthNode(p ui.Prompter, o *CliCmdOptions) (err error) {
+	o.genData.ExecutionApiUrl, err = p.InputURL("Input Eth 1 Endpoint", "", true)
+	return
+}
+
 func absPathInPlace(path *string) error {
 	absPath, err := filepath.Abs(*path)
 	if err != nil {