From 8eeff01939e3b9f09e5233f6cde60e1c10af0a63 Mon Sep 17 00:00:00 2001
From: benoitc <bchesneau@gmail.com>
Date: Fri, 5 Jul 2013 19:55:15 +0200
Subject: [PATCH] add websocket support to /container/<name>/attach/ws

This function add the possibility to attach containers streams to a
websocket. When a websocket is asked the request is upgraded to this
protocol..
---
 api.go | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/api.go b/api.go
index 1970feaec6..45261973a7 100644
--- a/api.go
+++ b/api.go
@@ -1,6 +1,7 @@
 package docker
 
 import (
+	"code.google.com/p/go.net/websocket"
 	"encoding/json"
 	"fmt"
 	"github.com/dotcloud/docker/auth"
@@ -693,6 +694,52 @@ func postContainersAttach(srv *Server, version float64, w http.ResponseWriter, r
 	return nil
 }
 
+func wsContainersAttach(srv *Server, ws *websocket.Conn, vars map[string]string) error {
+
+	r := ws.Request()
+
+	if err := parseForm(r); err != nil {
+		return err
+	}
+	logs, err := getBoolParam(r.Form.Get("logs"))
+	if err != nil {
+		return err
+	}
+	stream, err := getBoolParam(r.Form.Get("stream"))
+	if err != nil {
+		return err
+	}
+	stdin, err := getBoolParam(r.Form.Get("stdin"))
+	if err != nil {
+		return err
+	}
+	stdout, err := getBoolParam(r.Form.Get("stdout"))
+	if err != nil {
+		return err
+	}
+	stderr, err := getBoolParam(r.Form.Get("stderr"))
+	if err != nil {
+		return err
+	}
+
+	if vars == nil {
+		return fmt.Errorf("Missing parameter")
+	}
+	name := vars["name"]
+
+	if _, err := srv.ContainerInspect(name); err != nil {
+		return err
+	}
+
+	defer ws.Close()
+
+	if err := srv.ContainerAttach(name, logs, stream, stdin, stdout, stderr, ws, ws); err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func getContainersByName(srv *Server, version float64, w http.ResponseWriter, r *http.Request, vars map[string]string) error {
 	if vars == nil {
 		return fmt.Errorf("Missing parameter")
@@ -898,12 +945,14 @@ func createRouter(srv *Server, logging bool) (*mux.Router, error) {
 				if srv.enableCors {
 					writeCorsHeaders(w, r)
 				}
+
 				if version == 0 || version > APIVERSION {
 					w.WriteHeader(http.StatusNotFound)
 					return
 				}
 
 				if err := localFct(srv, version, w, r, mux.Vars(r)); err != nil {
+					utils.Debugf("Error: %s", err)
 					httpError(w, err)
 				}
 			}
@@ -914,8 +963,26 @@ func createRouter(srv *Server, logging bool) (*mux.Router, error) {
 				r.Path("/v{version:[0-9.]+}" + localRoute).Methods(localMethod).HandlerFunc(f)
 				r.Path(localRoute).Methods(localMethod).HandlerFunc(f)
 			}
+
 		}
 	}
+	attachHandler := websocket.Handler(func(ws *websocket.Conn) {
+		r := ws.Request()
+		utils.Debugf("Calling %s %s", r.Method, r.RequestURI)
+
+		if logging {
+			log.Println(r.Method, r.RequestURI)
+		}
+		if err := wsContainersAttach(srv, ws, mux.Vars(r)); err != nil {
+			utils.Debugf("Error: %s", err)
+			ws.Close()
+		}
+	})
+
+	attachRoute := "/containers/{name:.*}/attach/ws"
+	r.Path("/v{version:[0-9.]+}" + attachRoute).Methods("GET").Handler(attachHandler)
+	r.Path(attachRoute).Methods("GET").Handler(attachHandler)
+
 	return r, nil
 }