mirror of https://github.com/coder/coder.git
71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package coderd
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
"nhooyr.io/websocket"
|
|
|
|
"github.com/coder/coder/v2/apiversion"
|
|
"github.com/coder/coder/v2/coderd/httpapi"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/tailnet/proto"
|
|
)
|
|
|
|
// @Summary Workspace Proxy Coordinate
|
|
// @ID workspace-proxy-coordinate
|
|
// @Security CoderSessionToken
|
|
// @Tags Enterprise
|
|
// @Success 101
|
|
// @Router /workspaceproxies/me/coordinate [get]
|
|
// @x-apidocgen {"skip": true}
|
|
func (api *API) workspaceProxyCoordinate(rw http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
|
|
version := "1.0"
|
|
msgType := websocket.MessageText
|
|
qv := r.URL.Query().Get("version")
|
|
if qv != "" {
|
|
version = qv
|
|
}
|
|
if err := proto.CurrentVersion.Validate(version); err != nil {
|
|
httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
|
|
Message: "Unknown or unsupported API version",
|
|
Validations: []codersdk.ValidationError{
|
|
{Field: "version", Detail: err.Error()},
|
|
},
|
|
})
|
|
return
|
|
}
|
|
maj, _, _ := apiversion.Parse(version)
|
|
if maj >= 2 {
|
|
// Versions 2+ use dRPC over a binary connection
|
|
msgType = websocket.MessageBinary
|
|
}
|
|
|
|
api.AGPL.WebsocketWaitMutex.Lock()
|
|
api.AGPL.WebsocketWaitGroup.Add(1)
|
|
api.AGPL.WebsocketWaitMutex.Unlock()
|
|
defer api.AGPL.WebsocketWaitGroup.Done()
|
|
|
|
conn, err := websocket.Accept(rw, r, nil)
|
|
if err != nil {
|
|
httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
|
|
Message: "Failed to accept websocket.",
|
|
Detail: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
ctx, nc := codersdk.WebsocketNetConn(ctx, conn, msgType)
|
|
defer nc.Close()
|
|
|
|
id := uuid.New()
|
|
err = api.tailnetService.ServeMultiAgentClient(ctx, version, nc, id)
|
|
if err != nil {
|
|
_ = conn.Close(websocket.StatusInternalError, err.Error())
|
|
} else {
|
|
_ = conn.Close(websocket.StatusGoingAway, "")
|
|
}
|
|
}
|