|
|
@ -11,6 +11,7 @@ import (
|
|
|
|
model "github.com/HFO4/cloudreve/models"
|
|
|
|
model "github.com/HFO4/cloudreve/models"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/fsctx"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/response"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/filesystem/response"
|
|
|
|
|
|
|
|
"github.com/HFO4/cloudreve/pkg/request"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
|
|
"github.com/HFO4/cloudreve/pkg/serializer"
|
|
|
|
"github.com/google/go-querystring/query"
|
|
|
|
"github.com/google/go-querystring/query"
|
|
|
|
cossdk "github.com/tencentyun/cos-go-sdk-v5"
|
|
|
|
cossdk "github.com/tencentyun/cos-go-sdk-v5"
|
|
|
@ -42,16 +43,50 @@ type urlOption struct {
|
|
|
|
type Driver struct {
|
|
|
|
type Driver struct {
|
|
|
|
Policy *model.Policy
|
|
|
|
Policy *model.Policy
|
|
|
|
Client *cossdk.Client
|
|
|
|
Client *cossdk.Client
|
|
|
|
|
|
|
|
HTTPClient request.Client
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Get 获取文件
|
|
|
|
// Get 获取文件
|
|
|
|
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
|
|
|
|
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
|
|
|
|
return nil, errors.New("未实现")
|
|
|
|
// 获取文件源地址
|
|
|
|
|
|
|
|
downloadURL, err := handler.Source(
|
|
|
|
|
|
|
|
ctx,
|
|
|
|
|
|
|
|
path,
|
|
|
|
|
|
|
|
url.URL{},
|
|
|
|
|
|
|
|
int64(model.GetIntSetting("preview_timeout", 60)),
|
|
|
|
|
|
|
|
false,
|
|
|
|
|
|
|
|
0,
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
return nil, err
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 获取文件数据流
|
|
|
|
|
|
|
|
resp, err := handler.HTTPClient.Request(
|
|
|
|
|
|
|
|
"GET",
|
|
|
|
|
|
|
|
downloadURL,
|
|
|
|
|
|
|
|
nil,
|
|
|
|
|
|
|
|
request.WithContext(ctx),
|
|
|
|
|
|
|
|
).CheckHTTPResponse(200).GetRSCloser()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
return nil, err
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
resp.SetFirstFakeChunk()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 尝试自主获取文件大小
|
|
|
|
|
|
|
|
if file, ok := ctx.Value(fsctx.FileModelCtx).(model.File); ok {
|
|
|
|
|
|
|
|
resp.SetContentLength(int64(file.Size))
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return resp, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Put 将文件流保存到指定目录
|
|
|
|
// Put 将文件流保存到指定目录
|
|
|
|
func (handler Driver) Put(ctx context.Context, file io.ReadCloser, dst string, size uint64) error {
|
|
|
|
func (handler Driver) Put(ctx context.Context, file io.ReadCloser, dst string, size uint64) error {
|
|
|
|
return errors.New("未实现")
|
|
|
|
opt := &cossdk.ObjectPutOptions{}
|
|
|
|
|
|
|
|
_, err := handler.Client.Object.Put(ctx, dst, file, opt)
|
|
|
|
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Delete 删除一个或多个文件,
|
|
|
|
// Delete 删除一个或多个文件,
|
|
|
|