custom vm path for devnet

Author: sukantoraymond

cmd/blockchaincmd/create.go

@@ -159,6 +159,7 @@ func CallCreate(
 	customVMRepoURLParam string,
 	customVMBranchParam string,
 	customVMBuildScriptParam string,
+	customVMPath string,
 ) error {
 	forceCreate = forceCreateParam
 	genesisPath = genesisPathParam
@@ -174,6 +175,7 @@ func CallCreate(
 	customVMRepoURL = customVMRepoURLParam
 	customVMBranch = customVMBranchParam
 	customVMBuildScript = customVMBuildScriptParam
+	vmFile = customVMPath
 	return createBlockchainConfig(cmd, []string{blockchainName})
 }
 

cmd/nodecmd/wiz.go

@@ -63,6 +63,7 @@ var (
 	customVMRepoURL                 string
 	customVMBranch                  string
 	customVMBuildScript             string
+	customVMPath                    string
 	nodeConf                        string
 	subnetConf                      string
 	chainConf                       string
@@ -122,6 +123,7 @@ The node wiz command creates a devnet and deploys, sync and validate a subnet in
 	cmd.Flags().StringVar(&customVMRepoURL, "custom-vm-repo-url", "", "custom vm repository url")
 	cmd.Flags().StringVar(&customVMBranch, "custom-vm-branch", "", "custom vm branch or commit")
 	cmd.Flags().StringVar(&customVMBuildScript, "custom-vm-build-script", "", "custom vm build-script")
+	cmd.Flags().StringVar(&customVMPath, "custom-vm-branch", "", "custom vm branch or commit")
 	cmd.Flags().StringVar(&customGrafanaDashboardPath, "add-grafana-dashboard", "", "path to additional grafana dashboard json file")
 	cmd.Flags().StringVar(&nodeConf, "node-config", "", "path to avalanchego node configuration for subnet")
 	cmd.Flags().StringVar(&subnetConf, "subnet-config", "", "path to the subnet configuration for subnet")
@@ -210,6 +212,7 @@ func wiz(cmd *cobra.Command, args []string) error {
 			customVMRepoURL,
 			customVMBranch,
 			customVMBuildScript,
+			customVMPath,
 		); err != nil {
 			return err
 		}

pkg/models/sidecar.go

@@ -42,6 +42,7 @@ type Sidecar struct {
 	CustomVMRepoURL     string
 	CustomVMBranch      string
 	CustomVMBuildScript string
+	CustomVMPath        string
 	// ICM related
 	TeleporterReady   bool
 	TeleporterKey     string

pkg/node/sync.go

@@ -64,14 +64,26 @@ func prepareSubnetPlugin(app *application.Avalanche, hosts []*models.Host, block
 	}
 	wg := sync.WaitGroup{}
 	wgResults := models.NodeResults{}
-	for _, host := range hosts {
-		wg.Add(1)
-		go func(nodeResults *models.NodeResults, host *models.Host) {
-			defer wg.Done()
-			if err := ssh.RunSSHCreatePlugin(host, sc); err != nil {
-				nodeResults.AddResult(host.NodeID, nil, err)
-			}
-		}(&wgResults, host)
+	if sc.CustomVMRepoURL == "" && sc.CustomVMPath != "" {
+		for _, host := range hosts {
+			wg.Add(1)
+			go func(nodeResults *models.NodeResults, host *models.Host) {
+				defer wg.Done()
+				if err := ssh.RunSSHCopyBinaryFile(host, sc); err != nil {
+					nodeResults.AddResult(host.NodeID, nil, err)
+				}
+			}(&wgResults, host)
+		}
+	} else {
+		for _, host := range hosts {
+			wg.Add(1)
+			go func(nodeResults *models.NodeResults, host *models.Host) {
+				defer wg.Done()
+				if err := ssh.RunSSHCreatePlugin(host, sc); err != nil {
+					nodeResults.AddResult(host.NodeID, nil, err)
+				}
+			}(&wgResults, host)
+		}
 	}
 	wg.Wait()
 	if wgResults.HasErrors() {

pkg/ssh/ssh.go

@@ -609,6 +609,23 @@ func RunSSHRenderAvalancheNodeConfig(
 	return host.UploadBytes(nodeConf, remoteconfig.GetRemoteAvalancheNodeConfig(), constants.SSHFileOpsTimeout)
 }
 
+func RunSSHCopyBinaryFile(host *models.Host, sc models.Sidecar) error {
+	vmID, err := sc.GetVMID()
+	if err != nil {
+		return err
+	}
+	subnetVMBinaryPath := fmt.Sprintf(constants.CloudNodeSubnetEvmBinaryPath, vmID)
+	ux.Logger.Info("Building Custom VM for %s to %s", host.NodeID, subnetVMBinaryPath)
+	if err := host.Upload(
+		sc.CustomVMPath,
+		subnetVMBinaryPath,
+		constants.SSHFileOpsTimeout,
+	); err != nil {
+		return err
+	}
+	return nil
+}
+
 // RunSSHCreatePlugin runs script to create plugin
 func RunSSHCreatePlugin(host *models.Host, sc models.Sidecar) error {
 	vmID, err := sc.GetVMID()

pkg/vm/create_custom.go

@@ -90,6 +90,9 @@ func CreateCustomSidecar(
 
 	sc.RPCVersion = rpcVersion
 	sc.Sovereign = sovereign
+	if vmPath != "" {
+		sc.CustomVMPath = vmPath
+	}
 	return sc, nil
 }