Cloudreve/service/explorer/file.go

54 lines
1.4 KiB
Go
Raw Normal View History

package explorer
import (
"context"
"github.com/HFO4/cloudreve/pkg/filesystem"
2019-12-10 17:10:34 +08:00
"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
"github.com/HFO4/cloudreve/pkg/serializer"
"github.com/gin-gonic/gin"
"net/http"
)
// FileDownloadService 文件下载服务path为文件完整路径
type FileDownloadService struct {
2019-11-29 15:24:23 +08:00
Path string `uri:"path" binding:"required,min=1,max=65535"`
}
2019-12-10 20:17:21 +08:00
type FileAnonymousGetService struct {
ID uint `uri:"id" binding:"required,min=1"`
Name string `uri:"name" binding:"required"`
}
// Download 签名的匿名文件下载
func (service *FileAnonymousGetService) Download(ctx context.Context, c *gin.Context) serializer.Response {
return serializer.Response{
Code: 0,
}
}
// Download 文件下载
func (service *FileDownloadService) Download(ctx context.Context, c *gin.Context) serializer.Response {
// 创建文件系统
fs, err := filesystem.NewFileSystemFromContext(c)
if err != nil {
return serializer.Err(serializer.CodePolicyNotAllowed, err.Error(), err)
}
2019-11-26 20:59:57 +08:00
// 开始处理下载
2019-12-10 17:10:34 +08:00
ctx = context.WithValue(ctx, fsctx.GinCtx, c)
2019-11-27 13:10:19 +08:00
rs, err := fs.GetDownloadContent(ctx, service.Path)
if err != nil {
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
}
// 设置文件名
2019-11-30 15:09:56 +08:00
c.Header("Content-Disposition", "attachment; filename=\""+fs.FileTarget[0].Name+"\"")
// 发送文件
http.ServeContent(c.Writer, c.Request, "", fs.FileTarget[0].UpdatedAt, rs)
return serializer.Response{
Code: 0,
}
}