37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
package upload
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/golang-jwt/jwt/v4"
|
|
|
|
"gitlab.com/gitlab-org/gitlab/workhorse/internal/api"
|
|
"gitlab.com/gitlab-org/gitlab/workhorse/internal/helper"
|
|
)
|
|
|
|
const RewrittenFieldsHeader = "Gitlab-Workhorse-Multipart-Fields"
|
|
|
|
type MultipartClaims struct {
|
|
RewrittenFields map[string]string `json:"rewritten_fields"`
|
|
jwt.StandardClaims
|
|
}
|
|
|
|
// Multipart is a request middleware. If the request has a MIME multipart
|
|
// request body, the middleware will iterate through the multipart parts.
|
|
// When it finds a file part (filename != ""), the middleware will save
|
|
// the file contents to a temporary location and replace the file part
|
|
// with a reference to the temporary location.
|
|
func Multipart(rails PreAuthorizer, h http.Handler, p Preparer) http.Handler {
|
|
return rails.PreAuthorizeHandler(func(w http.ResponseWriter, r *http.Request, a *api.Response) {
|
|
s := &SavedFileTracker{Request: r}
|
|
|
|
opts, _, err := p.Prepare(a)
|
|
if err != nil {
|
|
helper.Fail500(w, r, fmt.Errorf("Multipart: error preparing file storage options"))
|
|
return
|
|
}
|
|
|
|
InterceptMultipartFiles(w, r, h, a, s, opts)
|
|
}, "/authorize")
|
|
}
|