Function UpdateSSHConfig

utils/sshconfig.go:11–109

UpdateSSHConfig updates the ~/.ssh/config file with the instance connection details

(instanceID, ip string, port int, uuid string, tunnelPorts []int, templatePorts []int)

// UpdateSSHConfig updates the ~/.ssh/config file with the instance connection details
func UpdateSSHConfig(instanceID, ip string, port int, uuid string, tunnelPorts []int, templatePorts []int) error {
	homeDir, err := os.UserHomeDir()
	if err != nil {
		return fmt.Errorf("failed to get home directory: %w", err)
	}

	sshDir := filepath.Join(homeDir, ".ssh")
	if err := os.MkdirAll(sshDir, 0700); err != nil {
		return fmt.Errorf("failed to create .ssh directory: %w", err)
	}

	configPath := filepath.Join(sshDir, "config")

	// Read existing config
	var existingLines []string
	if data, err := os.ReadFile(configPath); err == nil {
		existingLines = strings.Split(string(data), "\n")
	}

	// Check if entry exists
	hostName := fmt.Sprintf("tnr-%s", instanceID)
	existingIndex := -1
	inBlock := false
	blockStart := -1
	blockEnd := -1

	for i, line := range existingLines {
		trimmed := strings.TrimSpace(line)
		if strings.HasPrefix(trimmed, "Host ") {
			if inBlock {
				blockEnd = i - 1
				break
			}
			if strings.Contains(trimmed, hostName) {
				existingIndex = i
				blockStart = i
				inBlock = true
			}
		}
	}

	if inBlock && blockEnd == -1 {
		blockEnd = len(existingLines) - 1
	}

	// Build new SSH config entry
	keyFile := GetKeyFile(uuid)
	var configLines []string
	configLines = append(configLines, fmt.Sprintf("Host %s", hostName))
	configLines = append(configLines, fmt.Sprintf("    HostName %s", ip))
	configLines = append(configLines, "    User ubuntu")
	configLines = append(configLines, fmt.Sprintf("    IdentityFile \"%s\"", keyFile))
	configLines = append(configLines, "    IdentitiesOnly yes")
	configLines = append(configLines, "    IdentityAgent none")
	configLines = append(configLines, "    StrictHostKeyChecking no")
	configLines = append(configLines, fmt.Sprintf("    Port %d", port))

	// Add port forwarding