提几个建议

[woshichenghaibo]

一是没有favicon，强迫症受不了。 二是检查一下为何经常出现上传失败，而且毫无规律 三是建议加上图片压缩功能，用户可选

[woshichenghaibo]

因为对于vercel用户最大上传5mb，所以加入上传压缩功能

[woshichenghaibo]

package control

import ( "encoding/json" "html/template" "image" "image/jpeg" "io" "log" "net/http" "path/filepath" "strconv" "strings"

"golang.org/x/image/bmp"
"golang.org/x/image/tiff"

"csz.net/tgstate/assets"
"csz.net/tgstate/conf"
"csz.net/tgstate/utils"

)

// UploadImageAPI 上传图片api func UploadImageAPI(w http.ResponseWriter, r http.Request) { w.Header().Set("Access-Control-Allow-Origin", "") if r.Method == http.MethodPost { // 获取上传的文件 file, header, err := r.FormFile("image") if err != nil { errJsonMsg("Unable to get file", w) return } defer file.Close()

    // 检查文件大小
    fileInfo, err := file.Stat()
    if err != nil {
        errJsonMsg("Error getting file info", w)
        return
    }
    if fileInfo.Size() > 5*1024*1024 {
        errJsonMsg("File size exceeds 5MB limit", w)
        return
    }

    // 检查文件类型
    allowedExts := []string{".jpg", ".jpeg", ".png", ".bmp", ".tiff"}
    ext := strings.ToLower(filepath.Ext(header.Filename))
    valid := false
    for _, allowedExt := range allowedExts {
        if ext == allowedExt {
            valid = true
            break
        }
    }
    if !valid {
        errJsonMsg("Invalid file type. Only .jpg, .jpeg, .png, .bmp, .tiff are allowed.", w)
        return
    }

    // 读取文件内容
    imgData, err := io.ReadAll(file)
    if err != nil {
        errJsonMsg("Error reading file data", w)
        return
    }

    // 压缩图像
    img, _, err := image.Decode(bytes.NewReader(imgData))
    if err != nil {
        errJsonMsg("Error decoding image", w)
        return
    }

    // 将图像压缩到5MB以下
    buf := new(bytes.Buffer)
    if ext == ".jpg" || ext == ".jpeg" {
        err = jpeg.Encode(buf, img, &jpeg.Options{Quality: 75})
    } else if ext == ".png" {
        // PNG压缩较为复杂，这里简单示例
        err = png.Encode(buf, img)
    } else if ext == ".bmp" {
        err = bmp.Encode(buf, img)
    } else if ext == ".tiff" {
        err = tiff.Encode(buf, img, nil)
    }
    if err != nil {
        errJsonMsg("Error encoding image", w)
        return
    }

    // 检查压缩后的文件大小
    if buf.Len() > 5*1024*1024 {
        errJsonMsg("Compressed file size exceeds 5MB limit", w)
        return
    }

    // 保存压缩后的图像
    imgData = buf.Bytes()

    res := conf.UploadResponse{
        Code:    0,
        Message: "error",
    }
    img := conf.FileRoute + utils.UpDocument(utils.TgFileData(header.Filename, bytes.NewReader(imgData)))
    if img != conf.FileRoute {
        res = conf.UploadResponse{
            Code:    1,
            Message: img,
            ImgUrl:  strings.TrimSuffix(conf.BaseUrl, "/") + img,
        }
    }
    w.Header().Set("Content-Type", "application/json")
    w.WriteHeader(http.StatusOK)
    json.NewEncoder(w).Encode(res)
    return
}

// 如果不是POST请求，返回错误响应
http.Error(w, "Invalid request method", http.StatusMethodNotAllowed)

}

func errJsonMsg(msg string, w http.ResponseWriter) { // 这里示例直接返回JSON响应 response := conf.UploadResponse{ Code: 0, Message: msg, } w.Header().Set("Content-Type", "application/json") w.WriteHeader(http.StatusOK) json.NewEncoder(w).Encode(response) }