update

Author: PWZER

config/ssh_config.go

@@ -12,6 +12,46 @@ import (
 	"github.com/kevinburke/ssh_config"
 )
 
+func hostFromSSHConfig(hostConfig *ssh_config.Host) (host *Host, err error) {
+	host = &Host{Patterns: make([]string, 0)}
+
+	// patterns
+	for _, pattern := range hostConfig.Patterns {
+		host.Patterns = append(host.Patterns, pattern.String())
+	}
+
+	// host info
+	for _, node := range hostConfig.Nodes {
+		switch node.(type) {
+		case *ssh_config.Empty:
+			continue
+		case *ssh_config.KV:
+			kv := node.(*ssh_config.KV)
+			switch kv.Key {
+			case "HostName":
+				host.HostName = kv.Value
+			case "User":
+				host.Username = kv.Value
+			case "Port":
+				port, err := strconv.Atoi(kv.Value)
+				if err != nil {
+					return nil, err
+				}
+				if port <= 0 || port >= 65536 {
+					return nil, fmt.Errorf("invalid port: %s", kv.Value)
+				}
+				host.Port = uint16(port)
+			case "ProxyJump":
+				host.ProxyJump = kv.Value
+			case "IdentityFile":
+				host.IdentityFiles = append(host.IdentityFiles, kv.Value)
+			}
+		}
+	}
+
+	return host, nil
+}
+
 func GetHostsFromSSHConfig() (hosts []*Host, err error) {
 	hosts = make([]*Host, 0)
 
@@ -31,40 +71,9 @@ func GetHostsFromSSHConfig() (hosts []*Host, err error) {
 	}
 
 	for _, hostConfig := range sshConfig.Hosts {
-		host := &Host{Patterns: make([]string, 0)}
-
-		// patterns
-		for _, pattern := range hostConfig.Patterns {
-			host.Patterns = append(host.Patterns, pattern.String())
-		}
-
-		// host info
-		for _, node := range hostConfig.Nodes {
-			switch node.(type) {
-			case *ssh_config.Empty:
-				continue
-			case *ssh_config.KV:
-				kv := node.(*ssh_config.KV)
-				switch kv.Key {
-				case "HostName":
-					host.HostName = kv.Value
-				case "User":
-					host.Username = kv.Value
-				case "Port":
-					port, err := strconv.Atoi(kv.Value)
-					if err != nil {
-						return hosts, err
-					}
-					if port <= 0 || port >= 65536 {
-						return hosts, fmt.Errorf("invalid port: %s", kv.Value)
-					}
-					host.Port = uint16(port)
-				case "ProxyJump":
-					host.ProxyJump = kv.Value
-				case "IdentityFile":
-					host.IdentityFiles = append(host.IdentityFiles, kv.Value)
-				}
-			}
+		host, err := hostFromSSHConfig(hostConfig)
+		if err != nil {
+			return hosts, err
 		}
 
 		// 没有 HostName 的都是正则类型的配置
@@ -90,7 +99,6 @@ func GetHostsFromSSHConfig() (hosts []*Host, err error) {
 		}
 
 		logger.Debugf("host: %+#v", host)
-
 		hosts = append(hosts, host)
 	}