Feat: add List method to Handler interface / implement local.List

This commit is contained in:
HFO4
2020-04-24 11:08:07 +08:00
parent 6006ff4d22
commit f7311f906b
11 changed files with 97 additions and 2 deletions

View File

@@ -46,6 +46,10 @@ type Driver struct {
HTTPClient request.Client
}
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
// CORS 创建跨域策略
func (handler Driver) CORS() error {
_, err := handler.Client.Bucket.PutCORS(context.Background(), &cossdk.BucketPutCORSOptions{

View File

@@ -23,6 +23,47 @@ type Driver struct {
Policy *model.Policy
}
// List 递归列取给定物理路径下所有文件
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
var res []response.Object
// 取得起始路径
root := util.RelativePath(filepath.FromSlash(path))
// 开始遍历路径下的文件、目录
err := filepath.Walk(root,
func(path string, info os.FileInfo, err error) error {
// 跳过根目录
if path == root {
return nil
}
if err != nil {
util.Log().Warning("无法遍历目录 %s, %s", path, err)
return filepath.SkipDir
}
// 将遍历对象的绝对路径转换为相对路径
rel, err := filepath.Rel(root, path)
if err != nil {
return err
}
res = append(res, response.Object{
Name: info.Name(),
RelativePath: filepath.ToSlash(rel),
Source: path,
Size: uint64(info.Size()),
IsDir: info.IsDir(),
LastModify: info.ModTime(),
})
return nil
})
return res, err
}
// Get 获取文件内容
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
// 打开文件

View File

@@ -2,6 +2,7 @@ package local
import (
"context"
"fmt"
model "github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/auth"
"github.com/HFO4/cloudreve/pkg/conf"
@@ -230,3 +231,12 @@ func TestHandler_Token(t *testing.T) {
_, err := handler.Token(ctx, 10, "123")
asserts.NoError(err)
}
func TestDriver_List(t *testing.T) {
//asserts := assert.New(t)
handler := Driver{}
ctx := context.Background()
res, err := handler.List(ctx, "KKV")
fmt.Println(res, err)
}

View File

@@ -22,6 +22,10 @@ type Driver struct {
HTTPClient request.Client
}
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
// Get 获取文件
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
// 获取文件源地址

View File

@@ -42,6 +42,10 @@ type Driver struct {
HTTPClient request.Client
}
func (handler *Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
type key int
const (

View File

@@ -22,6 +22,10 @@ type Driver struct {
Policy *model.Policy
}
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
// Get 获取文件
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
// 给文件名加上随机参数以强制拉取

View File

@@ -27,6 +27,10 @@ type Driver struct {
AuthInstance auth.Auth
}
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
// getAPIUrl 获取接口请求地址
func (handler Driver) getAPIUrl(scope string, routes ...string) string {
serverURL, err := url.Parse(handler.Policy.Server)

View File

@@ -40,6 +40,10 @@ type Driver struct {
Policy *model.Policy
}
func (handler Driver) List(ctx context.Context, path string) ([]response.Object, error) {
panic("implement me")
}
// Get 获取文件
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
// 给文件名加上随机参数以强制拉取