Support executing command in containers

This feature is only valid on single node since container API is not Swarmable

controller/container.go

@@ -1,14 +1,18 @@
 package controller
 
 import (
+	"io"
 	"strings"
 
 	"github.com/cuigh/auxo/data"
+	"github.com/cuigh/auxo/log"
 	"github.com/cuigh/auxo/net/web"
 	"github.com/cuigh/auxo/util/cast"
 	"github.com/cuigh/swirl/biz/docker"
 	"github.com/cuigh/swirl/misc"
 	"github.com/cuigh/swirl/model"
+	"github.com/gobwas/ws"
+	"github.com/gobwas/ws/wsutil"
 )
 
 // ContainerController is a controller of docker container
@@ -19,6 +23,8 @@ type ContainerController struct {
 	Logs      web.HandlerFunc `path:"/:id/logs" name:"container.logs" authorize:"!" desc:"container logs page"`
 	FetchLogs web.HandlerFunc `path:"/:id/fetch_logs" name:"container.fetch_logs" authorize:"?" desc:"fetch container logs"`
 	Delete    web.HandlerFunc `path:"/delete" method:"post" name:"container.delete" authorize:"!" desc:"delete container"`
+	Exec      web.HandlerFunc `path:"/:id/exec" name:"container.exec" authorize:"!" desc:"run a command in a running container"`
+	Connect   web.HandlerFunc `path:"/:id/connect" name:"container.connect" authorize:"!" desc:"connect to a running container"`
 }
 
 // Container creates an instance of ContainerController
@@ -28,8 +34,10 @@ func Container() (c *ContainerController) {
 		Detail:    containerDetail,
 		Raw:       containerRaw,
 		Logs:      containerLogs,
-		Delete:    containerDelete,
 		FetchLogs: containerFetchLogs,
+		Delete:    containerDelete,
+		Exec:      containerExec,
+		Connect:   containerConnect,
 	}
 }
 
@@ -118,3 +126,103 @@ func containerDelete(ctx web.Context) error {
 	}
 	return ajaxSuccess(ctx, nil)
 }
+
+func containerExec(ctx web.Context) error {
+	id := ctx.P("id")
+	container, _, err := docker.ContainerInspectRaw(id)
+	if err != nil {
+		return err
+	}
+
+	m := newModel(ctx).Set("Container", container)
+	return ctx.Render("container/exec", m)
+}
+
+func containerConnect(ctx web.Context) error {
+	id := ctx.P("id")
+	_, _, err := docker.ContainerInspectRaw(id)
+	if err != nil {
+		return err
+	}
+
+	conn, _, _, err := ws.UpgradeHTTP(ctx.Request(), ctx.Response(), nil)
+	if err != nil {
+		return err
+	}
+
+	cmd := ctx.Q("cmd")
+	idResp, err := docker.ContainerExecCreate(id, cmd)
+	if err != nil {
+		return err
+	}
+
+	resp, err := docker.ContainerExecAttach(idResp.ID)
+	if err != nil {
+		return err
+	}
+
+	err = docker.ContainerExecStart(idResp.ID)
+	if err != nil {
+		return err
+	}
+
+	var (
+		closed   = false
+		logger   = log.Get("exec")
+		disposer = func() {
+			if !closed {
+				closed = true
+				conn.Close()
+				resp.Close()
+			}
+		}
+	)
+
+	// input
+	go func() {
+		defer disposer()
+
+		for {
+			msg, op, err := wsutil.ReadClientData(conn)
+			if err != nil {
+				if !closed {
+					logger.Error("Failed to read data from client: ", err)
+				}
+				break
+			}
+
+			if op == ws.OpClose {
+				break
+			}
+
+			_, err = resp.Conn.Write(msg)
+			if err != nil {
+				logger.Error("Failed to write data to container: ", err)
+				break
+			}
+		}
+	}()
+
+	// output
+	go func() {
+		defer disposer()
+
+		buf := make([]byte, 1024)
+		for {
+			n, err := resp.Reader.Read(buf)
+			if err == io.EOF {
+				break
+			} else if err != nil {
+				logger.Error("Failed to read data from container: ", err)
+				break
+			}
+
+			err = wsutil.WriteServerMessage(conn, ws.OpText, buf[:n])
+			if err != nil {
+				logger.Error("Failed to write data to client: ", err)
+				break
+			}
+		}
+	}()
+	return nil
+}