89 lines
2.0 KiB
Go
89 lines
2.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/loveuer/nf-disk/internal/manager"
|
|
"github.com/loveuer/nf-disk/internal/s3"
|
|
"github.com/loveuer/nf-disk/ndh"
|
|
"github.com/loveuer/nf/nft/log"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
func FileUpload(c *ndh.Ctx) error {
|
|
type Req struct {
|
|
ConnId uint64 `json:"conn_id"`
|
|
Bucket string `json:"bucket"`
|
|
Location string `json:"location"`
|
|
Name string `json:"name"`
|
|
PublicRead bool `json:"public_read"`
|
|
PublicReadWrite bool `json:"public_read_write"`
|
|
DetectContentType bool `json:"detect_content_type"`
|
|
}
|
|
|
|
var (
|
|
err error
|
|
req = new(Req)
|
|
client *s3.Client
|
|
reader *os.File
|
|
info os.FileInfo
|
|
)
|
|
|
|
if err = c.ReqParse(req); err != nil {
|
|
return c.Send400(c, err.Error())
|
|
}
|
|
|
|
if req.Location == "" {
|
|
return c.Send400(req, "缺少文件位置")
|
|
}
|
|
|
|
if req.Name == "" {
|
|
req.Name = filepath.Base(req.Location)
|
|
}
|
|
|
|
if _, client, err = manager.Manager.Use(req.ConnId); err != nil {
|
|
return c.Send500(err.Error())
|
|
}
|
|
|
|
if reader, err = os.Open(req.Location); err != nil {
|
|
return c.Send400(err.Error(), fmt.Sprintf("文件: %s 打开错误", req.Location))
|
|
}
|
|
|
|
if info, err = reader.Stat(); err != nil {
|
|
log.Error("FileUpload: stat file info err = %s", err.Error())
|
|
return c.Send500(err.Error())
|
|
}
|
|
|
|
obj := &s3.PutFilesObj{
|
|
Key: req.Name,
|
|
Reader: reader,
|
|
ContentLength: info.Size(),
|
|
ContentType: "",
|
|
ExpireAt: 0,
|
|
PublicRead: req.PublicRead,
|
|
PublicReadWrite: req.PublicReadWrite,
|
|
}
|
|
|
|
if req.DetectContentType {
|
|
bs := make([]byte, 128)
|
|
if _, err = reader.ReadAt(bs, 0); err != nil {
|
|
if !errors.Is(err, io.EOF) {
|
|
log.Error("FileUpload: read file to detect content_type err = %s", err.Error())
|
|
return c.Send500(err.Error())
|
|
}
|
|
}
|
|
|
|
obj.ContentType = http.DetectContentType(bs)
|
|
}
|
|
|
|
if err = client.PutFile(c.Context(), req.Bucket, obj); err != nil {
|
|
log.Error("FileUpload: client.PutFile err = %s", err.Error())
|
|
return c.Send500(err.Error())
|
|
}
|
|
|
|
return c.Send200(req)
|
|
}
|