http.go

package main

import (
	"github.com/labstack/echo/v4"
	"github.com/labstack/echo/v4/middleware"
	"github.com/sirupsen/logrus"
	"path"
	"strconv"
	"strings"
)

func NewEcho() *echo.Echo {
	e := echo.New()
	e.Pre(middleware.RemoveTrailingSlash())

	g := e.Group("api")
	e.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
		Format: "${time_rfc3339} ${method} ${status} uri=${uri} latency=${latency}\n",
		Skipper: func(context echo.Context) bool {
			// health check log는 너무 verbose함.
			if context.Request().URL.RequestURI() == "/healthz" {
				return true
			}
			return false
		},
	}))
	e.GET("/healthz", func(c echo.Context) error { return c.String(200, "OK") })
	g.POST("/images", ImageUploadRequestHandler, ForceContentTypeMultipartFormDataMiddleware)

	return e
}

func ImageUploadRequestHandler(c echo.Context) error {
	// Source
	input, err := c.FormFile("image")
	if err != nil {
		logrus.Error(err)
		return err
	}
	inputFileName := input.Filename
	if err != nil {
		logrus.Error(err)
		return c.JSON(400, map[string]interface{}{
			"data":    nil,
			"message": ErrWrongImageFileName.Error(),
		})
	}
	var hashedFileName string
	if c.FormValue("hashing") == "false" {
		splited := strings.Split(inputFileName, ".")
		// .확장자의 형태가 아닌 경우
		if len(splited) == 1 {
			hashedFileName = inputFileName
		} else {
			hashedFileName = strings.Join(splited[:len(splited)-1], ".")
		}

		logrus.Println("Omit hashing. not hashed name:", hashedFileName)
	} else {
		hashedFileName = getHashedFileName(inputFileName)
		logrus.Println("Hashed", inputFileName, "into", hashedFileName)
	}
	src, err := input.Open()
	if err != nil {
		logrus.Error(err)
		return err
	}
	defer src.Close()
	//imageData, _, gifImageData, ext, err := DecodeImageFile(src)
	imageData, orientation, gifImageData, ext, err := DecodeImageFile(src)
	if orientation != 0 {
		imageData = RotateImage(imageData, orientation)
	}

	if err != nil {
		logrus.Error(err)
		return c.JSON(400, map[string]interface{}{
			"data":    nil,
			"message": ErrUnableToDecodeImage.Error(),
		})
	}

	DispatchMessages(&BaseImageTask{
		ImageData:        imageData,
		GIFImageData:     gifImageData,
		OriginalFileName: inputFileName,
		HashedFileName:   hashedFileName,
		Extension:        ext,
	})

	resp := GenerateSuccessfullyUploadedResponse(hashedFileName + "." + ext)
	logrus.Println(resp)
	return c.JSON(200, resp)
}

type BaseResponse struct {
	Data    interface{} `json:"data"`
	Message string      `json:"message"`
}

type SuccessfullyUploadedResponseData struct {
	RootEndpoint   string `json:"root_endpoint"`
	FileName       string `json:"file_name"`
	ThumbnailURL   string `json:"thumbnail_url"`
	Resized256URL  string `json:"resized_256_url"`
	Resized1024URL string `json:"resized_1024_url"`
}

// fileFullName은 파일 이름 자체와 ., 확장자명이 모두 연결된 문자.
// e.g. abcd123.png
func GenerateSuccessfullyUploadedResponse(fileFullName string) *BaseResponse {
	rootEndpoint := Config.Storage.Aws.Endpoint
	return &BaseResponse{
		Data: SuccessfullyUploadedResponseData{
			RootEndpoint:   rootEndpoint,
			FileName:       fileFullName,
			ThumbnailURL:   path.Join(rootEndpoint, "thumbnail", fileFullName),
			Resized256URL:  path.Join(rootEndpoint, "resized", strconv.Itoa(256), fileFullName),
			Resized1024URL: path.Join(rootEndpoint, "resized", strconv.Itoa(1024), fileFullName),
		},
	}

}

func ForceContentTypeMultipartFormDataMiddleware(handlerFunc echo.HandlerFunc) echo.HandlerFunc {
	return func(context echo.Context) error {

		if !strings.HasPrefix(context.Request().Header.Get("Content-Type"), "multipart/form-data") {
			logrus.Warn("Content-Type in Request", context.Request().Header)
			resp := BaseResponse{Message: "Unsupported Content-Type:" + context.Request().Header.Get("Content-Type") + ". Please use multipart/form-data."}
			logrus.Error(resp)
			return context.JSON(400, resp)
		}
		return handlerFunc(context)
	}
}