fusenapi/server/data-transfer/internal/logic/uploadlogologic.go

package logic

import (
	"context"
	"fusenapi/utils/auth"
	"fusenapi/utils/basic"
	"fusenapi/utils/image"
	"io/ioutil"
	"net/http"
	"os"
	"strings"

	"fusenapi/server/data-transfer/internal/svc"
	"fusenapi/server/data-transfer/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type UploadLogoLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewUploadLogoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadLogoLogic {
	return &UploadLogoLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UploadLogoLogic) UploadLogo(req *types.UploadLogoReq, loginInfo *auth.UserInfo, r *http.Request) (resp *basic.Response) {
	req.LogoFile = strings.Trim(req.LogoFile, " ")
	//图片是否做缩放[php代码不是bool是字符串？？然后给python处理]
	onlyScale := "false"
	imageUrl := ""
	if req.LogoFile != "" {
		onlyScale = "true"
		imageUrl = req.LogoFile
	} else {
		file, fileInfo, err := r.FormFile("upfile")
		if err != nil {
			return resp.SetStatusWithMessage(basic.CodeApiErr, err.Error())
		}
		defer file.Close()
		if !image.CheckUploadImageFormat(fileInfo.Filename) {
			return resp.SetStatusWithMessage(basic.CodeApiErr, "format of the file is invalid ")
		}
		fileData, err := ioutil.ReadAll(file)
		if err != nil {
			return resp.SetStatusWithMessage(basic.CodeApiErr, err.Error())
		}
		_ = os.WriteFile("a.jpeg", fileData, 0666)
	}

	return resp.SetStatus(basic.CodeOK)
}