mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
22e781eced
* chore: add /v2 to import module path go mod requires semantic versioning with versions greater than 1.x This was a mechanical update by running: ``` go install github.com/marwan-at-work/mod/cmd/mod@latest mod upgrade ``` Migrate generated files to import /v2 * Fix gen
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package coderd
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
"nhooyr.io/websocket"
|
|
|
|
"github.com/coder/coder/v2/coderd/httpapi"
|
|
"github.com/coder/coder/v2/coderd/httpmw"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/enterprise/tailnet"
|
|
"github.com/coder/coder/v2/enterprise/wsproxy/wsproxysdk"
|
|
)
|
|
|
|
// @Summary Agent is legacy
|
|
// @ID agent-is-legacy
|
|
// @Security CoderSessionToken
|
|
// @Produce json
|
|
// @Tags Enterprise
|
|
// @Param workspaceagent path string true "Workspace Agent ID" format(uuid)
|
|
// @Success 200 {object} wsproxysdk.AgentIsLegacyResponse
|
|
// @Router /workspaceagents/{workspaceagent}/legacy [get]
|
|
// @x-apidocgen {"skip": true}
|
|
func (api *API) agentIsLegacy(rw http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
|
|
agentID, ok := httpmw.ParseUUIDParam(rw, r, "workspaceagent")
|
|
if !ok {
|
|
httpapi.Write(r.Context(), rw, http.StatusBadRequest, codersdk.Response{
|
|
Message: "Missing UUID in URL.",
|
|
})
|
|
return
|
|
}
|
|
|
|
node := (*api.AGPL.TailnetCoordinator.Load()).Node(agentID)
|
|
httpapi.Write(ctx, rw, http.StatusOK, wsproxysdk.AgentIsLegacyResponse{
|
|
Found: node != nil,
|
|
Legacy: node != nil &&
|
|
len(node.Addresses) > 0 &&
|
|
node.Addresses[0].Addr() == codersdk.WorkspaceAgentIP,
|
|
})
|
|
}
|
|
|
|
// @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()
|
|
|
|
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
|
|
}
|
|
|
|
id := uuid.New()
|
|
sub := (*api.AGPL.TailnetCoordinator.Load()).ServeMultiAgent(id)
|
|
nc := websocket.NetConn(ctx, conn, websocket.MessageText)
|
|
defer nc.Close()
|
|
|
|
err = tailnet.ServeWorkspaceProxy(ctx, nc, sub)
|
|
if err != nil {
|
|
_ = conn.Close(websocket.StatusInternalError, err.Error())
|
|
}
|
|
}
|