8000 fix(lenovonas_share): stoken expire (#7727) · AlistGo/alist@040dc14 · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Commit 040dc14

Browse files
authored
fix(lenovonas_share): stoken expire (#7727)
1 parent 4dce53d commit 040dc14

File tree

1 file changed

+31
-13
lines changed

1 file changed

+31
-13
lines changed

drivers/lenovonas_share/driver.go

+31-13
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package LenovoNasShare
33
import (
44
"context"
55
"net/http"
6+
"time"
67

78
"github.com/go-resty/resty/v2"
89

@@ -15,7 +16,8 @@ import (
1516
type LenovoNasShare struct {
1617
model.Storage
1718
Addition
18-
stoken string
19+
stoken string
20+
expireAt int64
1921
}
2022

2123
func (d *LenovoNasShare) Config() driver.Config {
@@ -27,20 +29,9 @@ func (d *LenovoNasShare) GetAddition() driver.Additional {
2729
}
2830

2931
func (d *LenovoNasShare) Init(ctx context.Context) error {
30-
if d.Host == "" {
31-
d.Host = "https://siot-share.lenovo.com.cn"
32-
}
33-
query := map[string]string{
34-
"code": d.ShareId,
35-
"password": d.SharePwd,
36-
}
37-
resp, err := d.request(d.Host+"/oneproxy/api/share/v1/access", http.MethodGet, func(req *resty.Request) {
38-
req.SetQueryParams(query)
39-
}, nil)
40-
if err != nil {
32+
if err := d.getStoken(); err != nil {
4133
return err
4234
}
43-
d.stoken = utils.Json.Get(resp, "data", "stoken").ToString()
4435
return nil
4536
}
4637

@@ -49,6 +40,7 @@ func (d *LenovoNasShare) Drop(ctx context.Context) error {
4940
}
5041

5142
func (d *LenovoNasShare) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
43+
d.checkStoken() // 检查stoken是否过期
5244
files := make([]File, 0)
5345

5446
var resp Files
@@ -71,7 +63,33 @@ func (d *LenovoNasShare) List(ctx context.Context, dir model.Obj, args model.Lis
7163
})
7264
}
7365

66+
func (d *LenovoNasShare) checkStoken() { // 检查stoken是否过期
67+
if d.expireAt < time.Now().Unix() {
68+
d.getStoken()
69+
}
70+
}
71+
72+
func (d *LenovoNasShare) getStoken() error { // 获取stoken
73+
if d.Host == "" {
74+
d.Host = "https://siot-share.lenovo.com.cn"
75+
}
76+
query := map[string]string{
77+
"code": d.ShareId,
78+
"password": d.SharePwd,
79+
}
80+
resp, err := d.request(d.Host+"/oneproxy/api/share/v1/access", http.MethodGet, func(req *resty.Request) {
81+
req.SetQueryParams(query)
82+
}, nil)
83+
if err != nil {
84+
return err
85+
}
86+
d.stoken = utils.Json.Get(resp, "data", "stoken").ToString()
87+
d.expireAt = utils.Json.Get(resp, "data", "expires_in").ToInt64() + time.Now().Unix() - 60
88+
return nil
89+
}
90+
7491
func (d *LenovoNasShare) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
92+
d.checkStoken() // 检查stoken是否过期
7593
query := map[string]string{
7694
"code": d.ShareId,
7795
"stoken": d.stoken,

0 commit comments

Comments
 (0)
0