mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
chore: Add workspace proxy enterprise cli commands (#7176)
* feat: Add workspace proxy enterprise cli commands * chore: Handle custom workspace proxy options. Remove excess * chore: Add endpoint to register workspace proxies
This commit is contained in:
@@ -8,6 +8,7 @@ import (
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/google/uuid"
|
||||
"golang.org/x/xerrors"
|
||||
|
||||
@@ -156,3 +157,53 @@ func ExtractWorkspaceProxy(opts ExtractWorkspaceProxyConfig) func(http.Handler)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
type workspaceProxyParamContextKey struct{}
|
||||
|
||||
// WorkspaceProxyParam returns the worksace proxy from the ExtractWorkspaceProxyParam handler.
|
||||
func WorkspaceProxyParam(r *http.Request) database.WorkspaceProxy {
|
||||
user, ok := r.Context().Value(workspaceProxyParamContextKey{}).(database.WorkspaceProxy)
|
||||
if !ok {
|
||||
panic("developer error: workspace proxy parameter middleware not provided")
|
||||
}
|
||||
return user
|
||||
}
|
||||
|
||||
// ExtractWorkspaceProxyParam extracts a workspace proxy from an ID/name in the {workspaceproxy} URL
|
||||
// parameter.
|
||||
//
|
||||
//nolint:revive
|
||||
func ExtractWorkspaceProxyParam(db database.Store) func(http.Handler) http.Handler {
|
||||
return func(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
|
||||
proxyQuery := chi.URLParam(r, "workspaceproxy")
|
||||
if proxyQuery == "" {
|
||||
httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
|
||||
Message: "\"workspaceproxy\" must be provided.",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
var proxy database.WorkspaceProxy
|
||||
var dbErr error
|
||||
if proxyID, err := uuid.Parse(proxyQuery); err == nil {
|
||||
proxy, dbErr = db.GetWorkspaceProxyByID(ctx, proxyID)
|
||||
} else {
|
||||
proxy, dbErr = db.GetWorkspaceProxyByName(ctx, proxyQuery)
|
||||
}
|
||||
if httpapi.Is404Error(dbErr) {
|
||||
httpapi.ResourceNotFound(rw)
|
||||
return
|
||||
}
|
||||
if dbErr != nil {
|
||||
httpapi.InternalServerError(rw, dbErr)
|
||||
return
|
||||
}
|
||||
|
||||
ctx = context.WithValue(ctx, workspaceProxyParamContextKey{}, proxy)
|
||||
next.ServeHTTP(rw, r.WithContext(ctx))
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user