Typora 图片上传插件 golang

package main import ( "crypto/md5" "fmt" "io" "net/http" "os" "path/filepath" "github.com/gin-gonic/gin" ) type Config struct { MaxFileSize int64 `json:"max_file_size"` AllowedExts []string `json:"allowed_exts"` UploadPath string `json:"upload_path"` HttpUlr string `json:"http_url"` } func loadConfig() (*Config, error) { // 加载配置文件，定义各项参数 return &Config{ MaxFileSize: 2 * 1024 * 1024, // 文件最大2MB AllowedExts: []string{".jpg", ".jpeg", ".png", ".gif"}, // 允许的文件类型 UploadPath: "/www/wwwroot/website/upload/", // 图片上传存放目录 HttpUlr: "http://127.0.0.1:11259/upload/", // 图片访问URL }, nil } func isFileAllowed(fileName string, allowedExts []string) bool { ext := filepath.Ext(fileName) for _, allowedExt := range allowedExts { if ext == allowedExt { return true } } return false } func handleUpload(c *gin.Context) { // 添加header token验证 token := c.Request.Header.Get("Authorization") if token != "e583005c4b99107633bbe93a9fe59f83" { c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "Invalid token"}) return } // 获取上传文件和其他表单数据 file, header, err := c.Request.FormFile("upload_file") if err != nil { c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "Failed to fetch file"}) return } defer file.Close() // 检查文件大小是否超过限制 config, err := loadConfig() if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": "Failed to load config"}) return } if header.Size > config.MaxFileSize { c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "File too large"}) return } // 检查文件类型是否符合要求 if !isFileAllowed(header.Filename, config.AllowedExts) { c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": "Invalid file type"}) return } // 根据上传表单的名称生成存储路径和文件名 fileName := header.Filename // 保存文件时将文件名进行哈希，避免文件名重复 fileName = fmt.Sprintf("%x", md5.Sum([]byte(fileName))) + filepath.Ext(fileName) filePath := filepath.Join(config.UploadPath, fileName) // 创建一个新文件并将上传数据写入该文件中 newFile, err := os.Create(filePath) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": "Failed to create file"}) return } defer newFile.Close() _, err = io.Copy(newFile, file) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": "Failed to save file"}) return } // 响应 JSON 数据给客户端 c.JSON(http.StatusOK, gin.H{"success": true, "message": "", "url": config.HttpUlr + fileName}) } func main() { r := gin.New() r.Use(gin.Logger()) r.Use(gin.Recovery()) // 添加 Recovery 中间件 r.POST("/upload", handleUpload) r.Run(":11258") }