diff --git a/packages/cli/internal/server/server.go b/packages/cli/internal/server/server.go index fff7b04..cdca3e6 100644 --- a/packages/cli/internal/server/server.go +++ b/packages/cli/internal/server/server.go @@ -1,11 +1,13 @@ package server import ( + "bufio" "context" "embed" "fmt" "io/fs" "log" + "net" "net/http" "sync" "time" @@ -238,6 +240,14 @@ func (lw *loggingResponseWriter) Write(b []byte) (int, error) { return n, err } +func (w *loggingResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { + hj, ok := w.ResponseWriter.(http.Hijacker) + if !ok { + return nil, nil, fmt.Errorf("http.Hijacker interface is not supported") + } + return hj.Hijack() +} + func loggingMiddleware(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { start := time.Now()