Add depot, attendant, jetfire, optimus, ota services with kustomize overlays

services/ota_update_go/handlers/updatemanifestfile_add.go

@@ -0,0 +1,162 @@
+package handlers
+
+import (
+	"fmt"
+	"io"
+	"mime"
+	"net/http"
+	"regexp"
+	"strings"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/pkg/errors"
+
+	"github.com/fiskerinc/cloud-services/pkg/common"
+	"github.com/fiskerinc/cloud-services/pkg/s3"
+	u "github.com/fiskerinc/cloud-services/pkg/utils"
+	"github.com/fiskerinc/cloud-services/pkg/validator"
+
+	"otaupdate/messages"
+	"otaupdate/services"
+	"otaupdate/utils"
+	"github.com/fiskerinc/cloud-services/pkg/loggerdataresp"
+)
+
+const fileFieldName string = "file"
+
+var reURLSafe *regexp.Regexp
+
+func init() {
+	reURLSafe, _ = regexp.Compile(`[^\w\d\-\_\.]`)
+}
+
+// HandleUpdateManifestFileAdd godoc
+// @Summary Add update manifest
+// @Description Upload update manifest
+// @Accept  json
+// @Produce  json
+// @Param Authorization header string false "Bearer <ID token>"
+// @Param Api-Key header string false "<API token>"
+// @Param manifest_ecu_id formData int true "Manifest ECU id"
+// @Param version formData string true "ECU file version"
+// @Param offset formData string true "ECU file offset"
+// @Param checksum formData string true "ECU file checksum"
+// @Param type formData string false "ECU file type"
+// @Param order formData string false "ECU file order"
+// @Param file formData file true "Update file"
+// @Success 200 {object} common.UpdateManifestFile
+// @Failure 400 {object} common.JSONError "Bad request"
+// @Failure 401 {object} common.JSONError "Unauthorized"
+// @Failure 503 {object} common.JSONError "Service unavailable"
+// @Router /manifestfile [post]
+func HandleUpdateManifestFileAdd(w http.ResponseWriter, r *http.Request) {
+	info, manifestfile, err := parseUpdateManifestAddRequest(r)
+	if loggerdataresp.BadDataErrorResp(w, err, http.StatusBadRequest) {
+		return
+	}
+
+	result, encryptor, err := uploadFile(info)
+	if loggerdataresp.BadDataErrorResp(w, err, http.StatusServiceUnavailable) {
+		return
+	}
+
+	err = saveUpdateManifestFile(manifestfile, result, info)
+	if loggerdataresp.BadDataErrorResp(w, err, http.StatusServiceUnavailable) {
+		return
+	}
+
+	err = encryptor.SaveFileKey()
+	if loggerdataresp.BadDataErrorResp(w, err, http.StatusServiceUnavailable) {
+		return
+	}
+
+	u.RespJSON(w, http.StatusOK, manifestfile)
+}
+
+func parseUpdateManifestAddRequest(r *http.Request) (*u.FileInfo, *common.UpdateManifestFile, error) {
+	mediaType, params, err := mime.ParseMediaType(r.Header.Get("Content-Type"))
+	if err != nil {
+		return nil, nil, errors.WithStack(err)
+	}
+	if !strings.HasPrefix(mediaType, "multipart/") {
+		return nil, nil, errors.New(messages.RequiresMultipart)
+	}
+	if len(params["boundary"]) > 70 {
+		return nil, nil, errors.New(messages.BoundaryTooLong)
+	}
+
+	parser, file := utils.GetUpdateManifestFileFormParser()
+	info, err := u.FindFilePart(r, params["boundary"], fileFieldName, parser)
+	if err != nil {
+		return nil, nil, err
+	}
+	if info.Part == nil {
+		return nil, nil, errors.New(messages.RequiresFile)
+	}
+
+	err = validator.ValidateStruct(file)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	return info, file, nil
+}
+
+func saveUpdateManifestFile(um *common.UpdateManifestFile, link *common.JSONLink, info *u.FileInfo) error {
+	um.FileID = info.FileID
+	um.Filename = info.Filename
+	um.URL = link.Link
+	um.FileSize = info.FileSize
+	um.WriteRegion.Length = info.OrigFileSize
+
+	err := validator.ValidateStruct(um)
+	if err != nil {
+		return err
+	}
+
+	_, err = services.GetDB().GetUpdateManifests().FileInsert(um)
+
+	return err
+}
+
+func uploadFile(info *u.FileInfo) (link *common.JSONLink, encryptor *utils.FileEncryptor, err error) {
+	encryptor, err = utils.NewEncryptor()
+	if err != nil {
+		return nil, nil, errors.WithStack(err)
+	}
+	defer encryptor.Close()
+
+	pReader, pWriter := io.Pipe()
+	defer pReader.Close()
+	defer info.Part.Close()
+
+	go u.ChunkFilePart(pWriter, info, encryptor.Encrypt)
+	url, err := putFile(pReader, urlSafeStr(info.Filename), "application/octet-stream")
+	if err != nil {
+		return
+	}
+
+	info.FileID = encryptor.FileID
+	p := u.LinkResult(url, time.Now().Unix())
+	link = &p
+	return
+}
+
+func putFile(reader io.Reader, filename string, contentType string) (string, error) {
+	uid, err := uuid.NewRandom()
+	if err != nil {
+		return "", errors.WithStack(err)
+	}
+	key := fmt.Sprintf("%s/%s", uid, filename)
+	url, err := s3.GetS3().PutBucket(key, reader, contentType)
+	if err != nil {
+		return "", errors.WithStack(err)
+	}
+
+	return url, nil
+}
+
+func urlSafeStr(value string) string {
+	return reURLSafe.ReplaceAllLiteralString(value, "")
+}