59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/moby/buildkit/session"
|
|
"github.com/moby/buildkit/session/upload"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func newTransport(rt http.RoundTripper, sm *session.Manager, g session.Group) http.RoundTripper {
|
|
return &sessionHandler{rt: rt, sm: sm, g: g}
|
|
}
|
|
|
|
type sessionHandler struct {
|
|
sm *session.Manager
|
|
rt http.RoundTripper
|
|
g session.Group
|
|
}
|
|
|
|
func (h *sessionHandler) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
if req.URL.Host != "buildkit-session" {
|
|
return h.rt.RoundTrip(req)
|
|
}
|
|
|
|
if req.Method != "GET" {
|
|
return nil, errors.Errorf("invalid request")
|
|
}
|
|
|
|
var resp *http.Response
|
|
err := h.sm.Any(context.TODO(), h.g, func(ctx context.Context, _ string, caller session.Caller) error {
|
|
up, err := upload.New(context.TODO(), caller, req.URL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pr, pw := io.Pipe()
|
|
go func() {
|
|
_, err := up.WriteTo(pw)
|
|
pw.CloseWithError(err)
|
|
}()
|
|
|
|
resp = &http.Response{
|
|
Status: "200 OK",
|
|
StatusCode: 200,
|
|
Body: pr,
|
|
ContentLength: -1,
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp, nil
|
|
}
|