feat: support portforwarding over websocket

internal/dao/port_forwarder.go

@@ -18,6 +18,7 @@ import (
  "k8s.io/apimachinery/pkg/runtime"
  "k8s.io/apimachinery/pkg/runtime/schema"
  "k8s.io/apimachinery/pkg/runtime/serializer"
+ "k8s.io/apimachinery/pkg/util/httpstream"
  "k8s.io/cli-runtime/pkg/genericclioptions"
  "k8s.io/client-go/rest"
  "k8s.io/client-go/tools/portforward"
@@ -176,6 +177,16 @@ func (p *PortForwarder) forwardPorts(method string, url *url.URL, addr, portMap
  }
  dialer := spdy.NewDialer(upgrader, &http.Client{Transport: transport, Timeout: defaultTimeout}, method, url)
 
+ tunnelingDialer, err := portforward.NewSPDYOverWebsocketDialer(url, cfg)
+ if err != nil {
+ return nil, err
+ }
+
+ // First attempt tunneling (websocket) dialer, then fallback to spdy dialer.
+ dialer = portforward.NewFallbackDialer(tunnelingDialer, dialer, func(err error) bool {
+ return httpstream.IsUpgradeFailure(err) || httpstream.IsHTTPSProxyError(err)
+ })
+
  return portforward.NewOnAddresses(dialer, []string{addr}, []string{portMap}, p.stopChan, p.readyChan, p.Out, p.ErrOut)
 }