~❯ ssh company-
company-develop  ...

chmod 600 ~/.ssh/config

ssh chancel-pc

package main

import (
	"bufio"
	"flag"
	"fmt"
	"os"
	"os/exec"
	"strings"
)

// Function to parse SSH configuration file and return a slice of maps containing the configuration details
func parseSSHConfig(configPath string) ([]map[string]string, error) {
	// Open the SSH configuration file
	configFile, err := os.Open(configPath)
	if err != nil {
		fmt.Println(err)
		return nil, err
	}
	defer configFile.Close()

	// Initialize variables to store the configuration data
	configDicts := []map[string]string{} // Slice of maps to store multiple configurations
	currentDict := map[string]string{} // Map to store the current configuration

	// Create a scanner to read the configuration file line by line
	scanner := bufio.NewScanner(configFile)
	for scanner.Scan() {
		line := strings.TrimSpace(scanner.Text()) // Remove leading and trailing whitespaces

		// Skip empty lines and lines starting with "#"
		if line == "" || strings.HasPrefix(line, "#") {
			continue
		}

		// If line starts with "Host ", it indicates a new configuration block
		if strings.HasPrefix(line, "Host ") {
			// If there is any existing configuration, add it to the configDicts slice
			if len(currentDict) > 0 {
				configDicts = append(configDicts, currentDict)
			}

			// Create a new map for the current configuration block
			currentDict = map[string]string{}
			currentDict["Host"] = strings.TrimSpace(strings.TrimPrefix(line, "Host "))
		} else {
			// Split the line into key-value pairs
			parts := strings.SplitN(line, " ", 2)
			if len(parts) == 2 {
				key := strings.TrimSpace(parts[0])
				value := strings.TrimSpace(parts[1])

				// Store the key-value pair in the currentDict map
				currentDict[key] = value
			}
		}
	}

	// Add the last configuration block to the configDicts slice
	if len(currentDict) > 0 {
		configDicts = append(configDicts, currentDict)
	}

	// Return the slice of maps containing the configuration details and no error
	return configDicts, nil
}

// Function to print the SSH configuration
func printSSHConfig(configs []map[string]string) {
	// Printing header
	fmt.Println("************************ Hi, Welcome to use Go-SSH Tool *****************************")
	fmt.Println()
	fmt.Println("+-----+------------------------------+-------------------------+------------------------------------------+")
	fmt.Println("| id  | Host                         | username                | address                                  |")
	fmt.Println("+-----+------------------------------+-------------------------+------------------------------------------+")

	// Printing each SSH configuration
	for i, config := range configs {
		fmt.Printf("| %-3d | %-28s | %-23s | %-40s |\n", i+1, config["Host"], config["User"], config["HostName"])
	}

	fmt.Println("+-----+------------------------------+-------------------------+------------------------------------------+")
	fmt.Println()
	fmt.Println("Tips: Press a number between 1 and", len(configs)-1, "to select the host to connect, or \"q\" to quit.")
	fmt.Println()
}

func main() {
	// Parsing command line arguments
	configPath := flag.String("c", "$HOME/.ssh/config", "Path to SSH config file")
	flag.Parse()

	// Expanding environment variables in the config path
	expandedPath := os.ExpandEnv(*configPath)

	// Parsing the SSH config file
	configs, err := parseSSHConfig(expandedPath)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}

	// Printing the SSH configuration
	printSSHConfig(configs)

	// Creating a scanner to read user input
	scanner := bufio.NewScanner(os.Stdin)

	for {
		fmt.Print("# ")
		scanner.Scan()
		input := scanner.Text()

		// Exiting the program if user enters "q"
		if input == "q" {
			return
		}

		id := -1
		// Parsing the user input to get the selected SSH configuration ID
		fmt.Sscanf(input, "%d", &id)
		if id >= 1 && id <= len(configs) {
			// Getting the selected SSH configuration
			sshConfig := configs[id-1]

			// Creating arguments for the "ssh" command
			var args []string
			for key, value := range sshConfig {
				// Skipping "Host", "User", and "HostName" keys
				if key == "Host" || key == "User" || key == "HostName" {
					continue
				}
				arg := fmt.Sprintf("-o %s=%s", key, value)
				args = append(args, arg)
			}

			// Appending username and host name to the command arguments
			cmdArgs := append(args, fmt.Sprintf("%s@%s", sshConfig["User"], sshConfig["HostName"]))

			// Creating the "ssh" command
			sshCmd := exec.Command("ssh", cmdArgs...)
			sshCmd.Stdout = os.Stdout
			sshCmd.Stdin = os.Stdin
			sshCmd.Stderr = os.Stderr

			// Running the "ssh" command
			err := sshCmd.Run()
			if err != nil {
				fmt.Println("Error:", err)
				os.Exit(0)
			}
		} else {
			fmt.Println("Error: Invalid input")
		}

		// Printing the SSH configuration after each iteration
		printSSHConfig(configs)
	}
}

❯ ./go-ssh
************************ Hi, Welcome to use Go-SSH Tool *****************************

+-----+------------------------------+-------------------------+------------------------------------------+
| id  | Host                         | username                | address                                  |
+-----+------------------------------+-------------------------+------------------------------------------+
| 1   | 192.168.122.4                | chancel                 | 192.168.122.4                            |
| 2   | 192.168.4.15                 | chancel                 | 192.168.4.15                             |
| 3   | 192.168.11.2                 | chancel                 | 192.168.11.2                             |
| 4   | 192.168.11.3                 | chancel                 | 192.168.11.3                             |
| 5   | 192.168.11.12                | chancel                 | 192.168.11.12                            |
| 6   | 192.168.4.10                 | chancel                 | 192.168.4.10                             |
+-----+------------------------------+-------------------------+------------------------------------------+

Tips: Press a number between 1 and 5 to select the host to connect, or "q" to quit.

#