mirror of https://github.com/portainer/portainer
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"mime"
|
|
"net/http"
|
|
|
|
"github.com/portainer/portainer/api/archive"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
const OneMegabyte = 1024768
|
|
|
|
type postDockerfileRequest struct {
|
|
Content string
|
|
}
|
|
|
|
// buildOperation inspects the "Content-Type" header to determine if it needs to alter the request.
|
|
// If the value of the header is empty, it means that a Dockerfile is posted via upload, the function
|
|
// will extract the file content from the request body, tar it, and rewrite the body.
|
|
// If the value of the header contains "application/json", it means that the content of a Dockerfile is posted
|
|
// in the request payload as JSON, the function will create a new file called Dockerfile inside a tar archive and
|
|
// rewrite the body of the request.
|
|
// In any other case, it will leave the request unaltered.
|
|
func buildOperation(request *http.Request) error {
|
|
contentTypeHeader := request.Header.Get("Content-Type")
|
|
|
|
mediaType := ""
|
|
if contentTypeHeader != "" {
|
|
var err error
|
|
mediaType, _, err = mime.ParseMediaType(contentTypeHeader)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
var buffer []byte
|
|
switch mediaType {
|
|
case "":
|
|
body, err := io.ReadAll(request.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
buffer, err = archive.TarFileInBuffer(body, "Dockerfile", 0600)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
case "application/json":
|
|
var req postDockerfileRequest
|
|
err := json.NewDecoder(request.Body).Decode(&req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
buffer, err = archive.TarFileInBuffer([]byte(req.Content), "Dockerfile", 0600)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
case "multipart/form-data":
|
|
err := request.ParseMultipartForm(32 * OneMegabyte) // limit parser memory to 32MB
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if request.MultipartForm == nil || request.MultipartForm.File == nil {
|
|
return errors.New("uploaded files not found to build image")
|
|
}
|
|
|
|
tfb := archive.NewTarFileInBuffer()
|
|
defer tfb.Close()
|
|
|
|
for k := range request.MultipartForm.File {
|
|
f, hdr, err := request.FormFile(k)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
log.Info().Str("filename", hdr.Filename).Int64("size", hdr.Size).Msg("upload the file to build image")
|
|
|
|
content, err := io.ReadAll(f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
filename := hdr.Filename
|
|
if hdr.Filename == "blob" {
|
|
filename = "Dockerfile"
|
|
}
|
|
|
|
if err := tfb.Put(content, filename, 0600); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
buffer = tfb.Bytes()
|
|
request.Form = nil
|
|
request.PostForm = nil
|
|
request.MultipartForm = nil
|
|
|
|
default:
|
|
return nil
|
|
}
|
|
|
|
request.Body = io.NopCloser(bytes.NewReader(buffer))
|
|
request.ContentLength = int64(len(buffer))
|
|
request.Header.Set("Content-Type", "application/x-tar")
|
|
|
|
return nil
|
|
}
|