Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
alist-org
GitHub Repository: alist-org/alist
 Path: blob/main/drivers/115_share/driver.go
1986 views
1
package _115_share

2


3
import (

4
	"context"

5


6
	driver115 "github.com/SheltonZhu/115driver/pkg/driver"

7
	"github.com/alist-org/alist/v3/internal/driver"

8
	"github.com/alist-org/alist/v3/internal/errs"

9
	"github.com/alist-org/alist/v3/internal/model"

10
	"github.com/alist-org/alist/v3/pkg/utils"

11
	"golang.org/x/time/rate"

12
)

13


14
type Pan115Share struct {

15
	model.Storage

16
	Addition

17
	client  *driver115.Pan115Client

18
	limiter *rate.Limiter

19
}

20


21
func (d *Pan115Share) Config() driver.Config {

22
	return config

23
}

24


25
func (d *Pan115Share) GetAddition() driver.Additional {

26
	return &d.Addition

27
}

28


29
func (d *Pan115Share) Init(ctx context.Context) error {

30
	if d.LimitRate > 0 {

31
		d.limiter = rate.NewLimiter(rate.Limit(d.LimitRate), 1)

32
	}

33


34
	return d.login()

35
}

36


37
func (d *Pan115Share) WaitLimit(ctx context.Context) error {

38
	if d.limiter != nil {

39
		return d.limiter.Wait(ctx)

40
	}

41
	return nil

42
}

43


44
func (d *Pan115Share) Drop(ctx context.Context) error {

45
	return nil

46
}

47


48
func (d *Pan115Share) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {

49
	if err := d.WaitLimit(ctx); err != nil {

50
		return nil, err

51
	}

52


53
	files := make([]driver115.ShareFile, 0)

54
	fileResp, err := d.client.GetShareSnap(d.ShareCode, d.ReceiveCode, dir.GetID(), driver115.QueryLimit(int(d.PageSize)))

55
	if err != nil {

56
		return nil, err

57
	}

58
	files = append(files, fileResp.Data.List...)

59
	total := fileResp.Data.Count

60
	count := len(fileResp.Data.List)

61
	for total > count {

62
		fileResp, err := d.client.GetShareSnap(

63
			d.ShareCode, d.ReceiveCode, dir.GetID(),

64
			driver115.QueryLimit(int(d.PageSize)), driver115.QueryOffset(count),

65
		)

66
		if err != nil {

67
			return nil, err

68
		}

69
		files = append(files, fileResp.Data.List...)

70
		count += len(fileResp.Data.List)

71
	}

72


73
	return utils.SliceConvert(files, transFunc)

74
}

75


76
func (d *Pan115Share) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {

77
	if err := d.WaitLimit(ctx); err != nil {

78
		return nil, err

79
	}

80
	downloadInfo, err := d.client.DownloadByShareCode(d.ShareCode, d.ReceiveCode, file.GetID())

81
	if err != nil {

82
		return nil, err

83
	}

84


85
	return &model.Link{URL: downloadInfo.URL.URL}, nil

86
}

87


88
func (d *Pan115Share) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {

89
	return errs.NotSupport

90
}

91


92
func (d *Pan115Share) Move(ctx context.Context, srcObj, dstDir model.Obj) error {

93
	return errs.NotSupport

94
}

95


96
func (d *Pan115Share) Rename(ctx context.Context, srcObj model.Obj, newName string) error {

97
	return errs.NotSupport

98
}

99


100
func (d *Pan115Share) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {

101
	return errs.NotSupport

102
}

103


104
func (d *Pan115Share) Remove(ctx context.Context, obj model.Obj) error {

105
	return errs.NotSupport

106
}

107


108
func (d *Pan115Share) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {

109
	return errs.NotSupport

110
}

111


112
var _ driver.Driver = (*Pan115Share)(nil)

113


114