feat(platform): support set ssh jump server in cls (#2001)

tkestack · Jun 27, 2022 · 7a6737b · 7a6737b
1 parent fd09358
commit 7a6737b
Show file tree

Hide file tree

Showing 17 changed files with 1,262 additions and 421 deletions.
diff --git a/api/openapi/zz_generated.openapi.go b/api/openapi/zz_generated.openapi.go
diff --git a/api/platform/cluster.go b/api/platform/cluster.go
@@ -40,6 +40,19 @@ func (in *ClusterMachine) SSH() (*ssh.SSH, error) {
 		DialTimeOut: time.Second,
 		Retry:       0,
 	}
+	switch in.Proxy.Type {
+	case SSHJumpServer:
+		proxy := ssh.JumpServer{}
+		proxy.Host = in.Proxy.IP
+		proxy.Port = int(in.Proxy.Port)
+		proxy.User = in.Proxy.Username
+		proxy.Password = string(in.Proxy.Password)
+		proxy.PrivateKey = in.Proxy.PrivateKey
+		proxy.PassPhrase = in.Proxy.PassPhrase
+		proxy.DialTimeOut = time.Second
+		proxy.Retry = 0
+		sshConfig.Proxy = proxy
+	}
 	return ssh.New(sshConfig)
 }
 

diff --git a/api/platform/types.go b/api/platform/types.go
@@ -77,8 +77,29 @@ type ClusterMachine struct {
 	PassPhrase []byte
 	Labels     map[string]string
 	Taints     []corev1.Taint
+	Proxy      ClusterMachineProxy
 }
 
+// ClusterMachine is the proxy definition of ClusterMachine.
+type ClusterMachineProxy struct {
+	Type       ProxyType
+	IP         string
+	Port       int32
+	Username   string
+	Password   []byte
+	PrivateKey []byte
+	PassPhrase []byte
+}
+
+// ProxyType describes diffirent type of proxy
+type ProxyType string
+
+const (
+	// SSH jumper server proxy
+	SSHJumpServer ProxyType = "SSHJumpServer"
+	// SOCKS5 ProxyType = "SOCKS5"
+)
+
 // KubeVendorType describe the kubernetes provider of the cluster
 // ref https://github.com/open-cluster-management/multicloud-operators-foundation/blob/e94b719de6d5f3541e948dd70ad8f1ff748aa452/pkg/apis/internal.open-cluster-management.io/v1beta1/clusterinfo_types.go#L137
 type KubeVendorType string

diff --git a/api/platform/v1/cluster.go b/api/platform/v1/cluster.go
@@ -43,6 +43,19 @@ func (in *ClusterMachine) SSH() (*ssh.SSH, error) {
 		DialTimeOut: time.Second,
 		Retry:       0,
 	}
+	switch in.Proxy.Type {
+	case SSHJumpServer:
+		proxy := ssh.JumpServer{}
+		proxy.Host = in.Proxy.IP
+		proxy.Port = int(in.Proxy.Port)
+		proxy.User = in.Proxy.Username
+		proxy.Password = string(in.Proxy.Password)
+		proxy.PrivateKey = in.Proxy.PrivateKey
+		proxy.PassPhrase = in.Proxy.PassPhrase
+		proxy.DialTimeOut = time.Second
+		proxy.Retry = 0
+		sshConfig.Proxy = proxy
+	}
 	return ssh.New(sshConfig)
 }