aiforge/routers/modelapp/desensitization.go

package modelapp

import (
	"bytes"
	"crypto/tls"
	"image"
	"image/png"
	"strconv"

	"code.gitea.io/gitea/modules/setting"

	"code.gitea.io/gitea/modules/base"

	"code.gitea.io/gitea/modules/context"
	"github.com/go-resty/resty/v2"
)

var restyClient *resty.Client
var tplExploreUpload base.TplName = "model/tuomin/upload"
var uploadUrl string = "/extension/tuomin/upload"
var allowedContentType = []string{"image/jpeg", "image/jpg", "image/png"}

func ProcessImageUI(ctx *context.Context) {
	ctx.HTML(200, tplExploreUpload)
}

func ProcessImage(ctx *context.Context) {

	file, header, err := ctx.GetFile("file")
	if err != nil {
		ctx.Flash.Error(ctx.Tr("model_app.get_file_fail"))
		ctx.Redirect(setting.AppSubURL + uploadUrl)
		return
	}
	defer file.Close()

	contentType := header.Header.Get("Content-Type")

	if !isInAllowedContentType(contentType) {
		ctx.Flash.Error(ctx.Tr("model_app.content_type_unsupported"))
		ctx.Redirect(setting.AppSubURL + uploadUrl)
		return
	}

	client := getRestyClient()
	res, err := client.R().SetMultipartField(
		"file", header.Filename, contentType, file).Post(setting.ModelApp.DesensitizationUrl + "?mode=" + strconv.Itoa(ctx.QueryInt("mode")))
	if err != nil {
		ctx.Flash.Error(ctx.Tr("model_app.process_image_fail"))
		ctx.Redirect(setting.AppSubURL  + uploadUrl)
		return
	}
	image, _, err := image.Decode(bytes.NewReader(res.Body()))
	if err != nil {
		ctx.Flash.Error(ctx.Tr("model_app.process_image_fail"))
		ctx.Redirect(setting.AppSubURL + uploadUrl)
		return
	}

	png.Encode(ctx.Resp, image)
	return

}

func isInAllowedContentType(contentType string) bool {
	for _, allowType := range allowedContentType {
		if allowType == contentType {
			return true
		}
	}
	return false
}

func getRestyClient() *resty.Client {
	if restyClient == nil {
		restyClient = resty.New()
		restyClient.SetTLSClientConfig(&tls.Config{InsecureSkipVerify: true})
	}
	return restyClient
}