feat(controller-sdk-go): add pod exec

Author: duanhongyi

.gitignore

@@ -26,3 +26,4 @@ _testmain.go
 vendor
 coverage.txt
 .idea/*
+.vscode/

go.mod

@@ -1,8 +1,11 @@
 module github.com/drycc/controller-sdk-go
 
-go 1.17
+go 1.19
 
-require github.com/goware/urlx v0.3.1
+require (
+	github.com/gorilla/websocket v1.5.0
+	github.com/goware/urlx v0.3.1
+)
 
 require (
 	github.com/PuerkitoBio/purell v1.1.1 // indirect

go.sum

@@ -2,6 +2,8 @@ github.com/PuerkitoBio/purell v1.1.1 h1:WEQqlqaGbrPkxLJWfBwQmfEAE1Z7ONdDLqrN38tN
 github.com/PuerkitoBio/purell v1.1.1/go.mod h1:c11w/QuzBsJSee3cPx9rAFu61PvFxuPbtSwDGJws/X0=
 github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 h1:d+Bc7a5rLufV/sSk/8dngufqelfh6jnri85riMAaF/M=
 github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578/go.mod h1:uGdkoq3SwY9Y+13GIhn11/XLaGBb4BfwItxLd5jeuXE=
+github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
+github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
 github.com/goware/urlx v0.3.1 h1:BbvKl8oiXtJAzOzMqAQ0GfIhf96fKeNEZfm9ocNSUBI=
 github.com/goware/urlx v0.3.1/go.mod h1:h8uwbJy68o+tQXCGZNa9D73WN8n0r9OBae5bUnLcgjw=
 golang.org/x/net v0.0.0-20190213061140-3a22650c66bd h1:HuTn7WObtcDo9uEEU7rEqL0jYthdXAmZ6PP+meazmaU=

ps/ps.go

@@ -4,10 +4,13 @@ package ps
 import (
 	"encoding/json"
 	"fmt"
+	"net/http"
+	"net/url"
 	"sort"
 
 	drycc "github.com/drycc/controller-sdk-go"
 	"github.com/drycc/controller-sdk-go/api"
+	"github.com/gorilla/websocket"
 )
 
 // List lists an app's processes.
@@ -26,6 +29,36 @@ func List(c *drycc.Client, appID string, results int) (api.PodsList, int, error)
 	return procs, count, reqErr
 }
 
+// Exec a command in a container.
+func Exec(c *drycc.Client, app, pod string, tty, stdin bool, command []string) (*websocket.Conn, error) {
+	scheme := "ws"
+	if c.ControllerURL.Scheme == "https" {
+		scheme = "wss"
+	}
+	path := fmt.Sprintf("v2/apps/%s/pods/%s/exec/", app, pod)
+	endpoint := url.URL{Scheme: scheme, Host: c.ControllerURL.Host, Path: path}
+	conn, _, err := websocket.DefaultDialer.Dial(
+		endpoint.String(),
+		http.Header{
+			"User-Agent":           {c.UserAgent},
+			"Authorization":        {"token " + c.Token},
+			"X-Drycc-Builder-Auth": {c.HooksToken},
+		},
+	)
+	if err != nil {
+		return nil, err
+	}
+	data, err := json.Marshal(&command)
+	if err != nil {
+		return nil, err
+	}
+	conn.WriteMessage(
+		websocket.TextMessage,
+		[]byte(fmt.Sprintf(`{"tty": %t, "stdin": %t, "command": %s}`, tty, stdin, data)),
+	)
+	return conn, nil
+}
+
 // Scale increases or decreases an app's processes. The processes are specified in the target argument,
 // a key-value map, where the key is the process name and the value is the number of replicas
 func Scale(c *drycc.Client, appID string, targets map[string]int) error {

ps/ps_test.go

@@ -3,6 +3,7 @@ package ps
 import (
 	"fmt"
 	"io/ioutil"
+	"log"
 	"net/http"
 	"net/http/httptest"
 	"reflect"
@@ -11,6 +12,7 @@ import (
 	drycc "github.com/drycc/controller-sdk-go"
 	"github.com/drycc/controller-sdk-go/api"
 	"github.com/drycc/controller-sdk-go/pkg/time"
+	"github.com/gorilla/websocket"
 )
 
 const processesFixture string = `
@@ -64,6 +66,8 @@ const restartWebTwoFixture string = `[
 
 const scaleExpected string = `{"web":2}`
 
+var upgrader = websocket.Upgrader{} // use default options
+
 type fakeHTTPServer struct{}
 
 func (fakeHTTPServer) ServeHTTP(res http.ResponseWriter, req *http.Request) {
@@ -74,6 +78,29 @@ func (fakeHTTPServer) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 		return
 	}
 
+	if req.URL.Path == "/v2/apps/example-go/pods/example-go-web-111/exec/" {
+		c, err := upgrader.Upgrade(res, req, nil)
+		if err != nil {
+			log.Print("upgrade:", err)
+			return
+		}
+		defer c.Close()
+		for {
+			messageType, message, err := c.ReadMessage()
+			if err != nil {
+				log.Println("read:", err)
+				break
+			}
+
+			log.Printf("recv: %s", message)
+			err = c.WriteMessage(messageType, []byte("# "+"\n"))
+			if err != nil {
+				log.Println("write:", err)
+				break
+			}
+		}
+	}
+
 	if req.URL.Path == "/v2/apps/example-go/pods/restart/" && req.Method == "POST" {
 		res.Write([]byte(restartAllFixture))
 		return
@@ -155,6 +182,30 @@ func TestProcessesList(t *testing.T) {
 	}
 }
 
+func TestExec(t *testing.T) {
+	handler := fakeHTTPServer{}
+	server := httptest.NewServer(&handler)
+	defer server.Close()
+
+	drycc, err := drycc.New(false, server.URL, "abc")
+	if err != nil {
+		t.Fatal(err)
+	}
+	conn, err := Exec(drycc, "example-go", "example-go-web-111", true, true, []string{"/bin/sh"})
+	if err != nil {
+		t.Fatal(err)
+	}
+	expected := []byte("# " + "\n")
+	_, actual, err := conn.ReadMessage()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !reflect.DeepEqual(actual, expected) {
+		t.Errorf("Expected: %v, Got %v", expected, actual)
+	}
+}
+
 type testExpected struct {
 	Name     string
 	Type     string