private repo ssh

Author: sukantoraymond

cmd/nodecmd/load_test_start.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/ava-labs/avalanche-cli/pkg/ansible"
 	"github.com/ava-labs/avalanche-cli/pkg/application"
@@ -29,13 +30,14 @@ import (
 )
 
 var (
-	loadTestRepoURL    string
-	loadTestBuildCmd   string
-	loadTestCmd        string
-	loadTestRepoCommit string
-	repoDirName        string
-	loadTestHostRegion string
-	loadTestBranch     string
+	loadTestRepoURL       string
+	loadTestBuildCmd      string
+	loadTestCmd           string
+	loadTestRepoCommit    string
+	repoDirName           string
+	loadTestHostRegion    string
+	loadTestBranch        string
+	loadTestDeployKeyPath string
 )
 
 type clusterInfo struct {
@@ -80,6 +82,7 @@ The command will then run the load test binary based on the provided load test r
 	cmd.Flags().StringVar(&loadTestCmd, "load-test-cmd", "", "command to run load test")
 	cmd.Flags().StringVar(&loadTestHostRegion, "region", "", "create load test node in a given region")
 	cmd.Flags().StringVar(&loadTestBranch, "load-test-branch", "", "load test branch or commit")
+	cmd.Flags().StringVar(&loadTestDeployKeyPath, "load-test-deploy-key", "", "path to SSH deploy key for private repository access")
 	return cmd
 }
 
@@ -357,6 +360,20 @@ func startLoadTest(_ *cobra.Command, args []string) error {
 	}
 
 	ux.Logger.GreenCheckmarkToUser("Load test environment is ready!")
+
+	// Setup deploy key for private repository access if provided
+	if loadTestDeployKeyPath != "" {
+		ux.Logger.PrintToUser("%s Setting up deploy key for private repository", logging.Green.Wrap(">"))
+		if err := ssh.RunSSHSetupDeployKey(currentLoadTestHost[0], loadTestDeployKeyPath); err != nil {
+			return fmt.Errorf("failed to setup deploy key for private repository: %w", err)
+		}
+
+		// Update repository URL to use SSH config if it's a GitHub SSH URL
+		if strings.HasPrefix(loadTestRepoURL, "git@github.com:") {
+			loadTestRepoURL = strings.Replace(loadTestRepoURL, "git@github.com:", "git@github.com-avalanche:", 1)
+		}
+	}
+
 	ux.Logger.PrintToUser("%s Building load test code", logging.Green.Wrap(">"))
 	if err := ssh.RunSSHBuildLoadTestCode(currentLoadTestHost[0], loadTestRepoURL, loadTestBuildCmd, loadTestRepoCommit, repoDirName, loadTestBranch, checkoutCommit); err != nil {
 		return err

pkg/ssh/shell/setupDeployKey.sh

@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+# Setup SSH deploy key for private repository access
+
+# Create .ssh directory if it doesn't exist
+mkdir -p ~/.ssh
+
+# Set proper permissions
+chmod 700 ~/.ssh
+
+# Upload the deploy key (this will be done by the Go code before calling this script)
+# The key should already be at ~/.ssh/avalanche-deploy-key
+
+# Set proper permissions for the deploy key
+chmod 600 ~/.ssh/avalanche-deploy-key
+
+# Configure SSH to use the deploy key
+cat >> ~/.ssh/config <<'EOF'
+Host github.com-avalanche
+  HostName github.com
+  User git
+  IdentityFile ~/.ssh/avalanche-deploy-key
+  IdentitiesOnly yes
+  StrictHostKeyChecking accept-new
+EOF
+
+# Test the SSH connection
+echo "Testing SSH connection to GitHub..."
+ssh -T github.com-avalanche || true
+
+echo "Deploy key setup completed successfully!"

pkg/ssh/ssh.go

@@ -820,6 +820,47 @@ func RunSSHBuildLoadTestCode(host *models.Host, loadTestRepo, loadTestPath, load
 	)
 }
 
+// RunSSHSetupDeployKey sets up SSH deploy key for private repository access
+func RunSSHSetupDeployKey(host *models.Host, deployKeyPath string) error {
+	// Expand tilde in the path
+	expandedPath, err := expandTilde(deployKeyPath)
+	if err != nil {
+		return fmt.Errorf("failed to expand deploy key path: %w", err)
+	}
+
+	// Check if the deploy key file exists
+	if _, err := os.Stat(expandedPath); os.IsNotExist(err) {
+		return fmt.Errorf("deploy key file not found: %s", expandedPath)
+	}
+
+	// Upload the deploy key first
+	remoteDeployKeyPath := "~/.ssh/avalanche-deploy-key"
+	if err := host.Upload(expandedPath, remoteDeployKeyPath, constants.SSHFileOpsTimeout); err != nil {
+		return fmt.Errorf("failed to upload deploy key: %w", err)
+	}
+
+	// Run the setup script
+	return RunOverSSH(
+		"Setup Deploy Key",
+		host,
+		constants.SSHFileOpsTimeout,
+		"shell/setupDeployKey.sh",
+		scriptInputs{},
+	)
+}
+
+// expandTilde expands ~ to the user's home directory
+func expandTilde(path string) (string, error) {
+	if strings.HasPrefix(path, "~/") {
+		homeDir, err := os.UserHomeDir()
+		if err != nil {
+			return "", err
+		}
+		return filepath.Join(homeDir, path[2:]), nil
+	}
+	return path, nil
+}
+
 func RunSSHBuildLoadTestDependencies(host *models.Host) error {
 	return RunOverSSH(
 		"Build Load Test",