56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package upload
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"mime/multipart"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"gitlab.com/gitlab-org/gitlab-workhorse/internal/filestore"
|
||
|
|
"gitlab.com/gitlab-org/gitlab-workhorse/internal/secret"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SavedFileTracker struct {
|
||
|
|
Request *http.Request
|
||
|
|
rewrittenFields map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) Track(fieldName string, localPath string) {
|
||
|
|
if s.rewrittenFields == nil {
|
||
|
|
s.rewrittenFields = make(map[string]string)
|
||
|
|
}
|
||
|
|
s.rewrittenFields[fieldName] = localPath
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) Count() int {
|
||
|
|
return len(s.rewrittenFields)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) ProcessFile(_ context.Context, fieldName string, file *filestore.FileHandler, _ *multipart.Writer) error {
|
||
|
|
s.Track(fieldName, file.LocalPath)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) ProcessField(_ context.Context, _ string, _ *multipart.Writer) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) Finalize(_ context.Context) error {
|
||
|
|
if s.rewrittenFields == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
claims := MultipartClaims{RewrittenFields: s.rewrittenFields, StandardClaims: secret.DefaultClaims}
|
||
|
|
tokenString, err := secret.JWTTokenString(claims)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("savedFileTracker.Finalize: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
s.Request.Header.Set(RewrittenFieldsHeader, tokenString)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SavedFileTracker) Name() string {
|
||
|
|
return "accelerate"
|
||
|
|
}
|