178 lines
3.3 KiB
Go
178 lines
3.3 KiB
Go
package file
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"mime/multipart"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/h2non/filetype"
|
|
gonanoid "github.com/matoous/go-nanoid/v2"
|
|
)
|
|
|
|
const (
|
|
MaxImageSize = 10 << 20 // 10 MB
|
|
MaxFileSize = 50 << 20 // 50 MB
|
|
)
|
|
|
|
var ErrUnsupported = errors.New("文件格式不支持")
|
|
|
|
type Type int
|
|
|
|
const (
|
|
ALL Type = 0
|
|
IMG Type = 1
|
|
)
|
|
|
|
//func UploadFilename(filepath string, t Type) (string, error) {
|
|
// fileOpen, err := os.Open(filepath)
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
// defer func(fileOpen *os.File) {
|
|
// _ = fileOpen.Close()
|
|
// }(fileOpen)
|
|
//
|
|
// fileBytes, err := io.ReadAll(fileOpen)
|
|
// if err != nil {
|
|
// return "", errors.New("failed to read file")
|
|
// }
|
|
//
|
|
// if t == IMG {
|
|
// // 判断是不是图片
|
|
// if !filetype.IsImage(fileBytes) {
|
|
// return "", ErrUnsupported
|
|
// }
|
|
// }
|
|
//
|
|
// kind, err := filetype.Match(fileBytes)
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
//
|
|
// if kind == filetype.Unknown {
|
|
// return "", ErrUnsupported
|
|
// }
|
|
//
|
|
// // 使用 filetype 判断类型后已经去读了一些bytes了
|
|
// // 要恢复文件读取位置
|
|
// _, err = fileOpen.Seek(0, io.SeekStart)
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
//
|
|
// dir := GetPath()
|
|
// exist, _ := Exists(dir)
|
|
// if !exist {
|
|
// if err := Mkdir(dir); err != nil {
|
|
// return "", err
|
|
// }
|
|
// }
|
|
//
|
|
// filename := GenFilename(kind.Extension)
|
|
// fullPath := path.Join(dir, filename)
|
|
// f, err := os.Create(fullPath)
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
// defer func(f *os.File) {
|
|
// _ = f.Close()
|
|
// }(f)
|
|
//
|
|
// _, err = io.Copy(f, fileOpen)
|
|
// if err != nil {
|
|
// return "", err
|
|
// }
|
|
//
|
|
// return "/" + fullPath, nil
|
|
//}
|
|
|
|
func UploadFile(file *multipart.FileHeader, t Type) (string, error) {
|
|
if t == IMG {
|
|
if file.Size > MaxImageSize {
|
|
return "", errors.New("failed to receive images too large")
|
|
}
|
|
} else {
|
|
if file.Size > MaxFileSize {
|
|
return "", errors.New("failed to receive file too large")
|
|
}
|
|
}
|
|
|
|
fileOpen, err := file.Open()
|
|
if err != nil {
|
|
return "", errors.New("failed to open file")
|
|
}
|
|
defer func(fileOpen multipart.File) {
|
|
_ = fileOpen.Close()
|
|
}(fileOpen)
|
|
|
|
fileBytes, err := io.ReadAll(fileOpen)
|
|
if err != nil {
|
|
return "", errors.New("failed to read file")
|
|
}
|
|
|
|
if t == IMG {
|
|
// 判断是不是图片
|
|
if !filetype.IsImage(fileBytes) {
|
|
return "", ErrUnsupported
|
|
}
|
|
}
|
|
|
|
kind, err := filetype.Match(fileBytes)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if kind == filetype.Unknown {
|
|
return "", ErrUnsupported
|
|
}
|
|
|
|
// 使用 filetype 判断类型后已经去读了一些bytes了
|
|
// 要恢复文件读取位置
|
|
_, err = fileOpen.Seek(0, io.SeekStart)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
dir := GetPath()
|
|
exist, _ := Exists(dir)
|
|
if !exist {
|
|
if err := Mkdir(dir); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
filename := GenFilename(kind.Extension)
|
|
fullPath := path.Join(dir, filename)
|
|
f, err := os.Create(fullPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer func(f *os.File) {
|
|
_ = f.Close()
|
|
}(f)
|
|
|
|
_, err = io.Copy(f, fileOpen)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return "/" + fullPath, nil
|
|
}
|
|
|
|
func GetPath() string {
|
|
return fmt.Sprintf("public/%s/%s/%s/", time.Now().Format("2006"), time.Now().Format("01"), time.Now().Format("02"))
|
|
}
|
|
|
|
func GenFilename(ext string) string {
|
|
id, err := gonanoid.New()
|
|
if err != nil {
|
|
return uuid.New().String()
|
|
}
|
|
return fmt.Sprintf("%s.%s", id, ext)
|
|
}
|