Skip to content

Commit 1c8d895

Browse files
Code2qingCode2qing
authored
feat(terabox): add terabox driver (close #2825 close #2678 #2849)
1 parent fbf3fb8 commit 1c8d895

File tree

5 files changed

+524
-0
lines changed

5 files changed

+524
-0
lines changed

‎drivers/all.go

+1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ import (
2727
_ "github.com/alist-org/alist/v3/drivers/sftp"
2828
_ "github.com/alist-org/alist/v3/drivers/smb"
2929
_ "github.com/alist-org/alist/v3/drivers/teambition"
30+
_ "github.com/alist-org/alist/v3/drivers/terabox"
3031
_ "github.com/alist-org/alist/v3/drivers/thunder"
3132
_ "github.com/alist-org/alist/v3/drivers/uss"
3233
_ "github.com/alist-org/alist/v3/drivers/virtual"

‎drivers/terabox/driver.go

+214
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,214 @@
1+
package terbox
2+
3+
import (
4+
"bytes"
5+
"context"
6+
"crypto/md5"
7+
"encoding/hex"
8+
"fmt"
9+
"github.com/alist-org/alist/v3/drivers/base"
10+
"github.com/alist-org/alist/v3/pkg/utils"
11+
log "github.com/sirupsen/logrus"
12+
"io"
13+
"math"
14+
"net/http"
15+
"os"
16+
stdpath "path"
17+
"strconv"
18+
"strings"
19+
20+
"github.com/alist-org/alist/v3/internal/driver"
21+
"github.com/alist-org/alist/v3/internal/model"
22+
)
23+
24+
type Terabox struct {
25+
model.Storage
26+
Addition
27+
}
28+
29+
func (d *Terabox) Config() driver.Config {
30+
return config
31+
}
32+
33+
func (d *Terabox) GetAddition() driver.Additional {
34+
return &d.Addition
35+
}
36+
37+
func (d *Terabox) Init(ctx context.Context) error {
38+
_, err := d.request("https://www.terabox.com/api/check/#", http.MethodGet, nil, nil)
39+
return err
40+
}
41+
42+
func (d *Terabox) Drop(ctx context.Context) error {
43+
return nil
44+
}
45+
46+
func (d *Terabox) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
47+
files, err := d.getFiles(dir.GetPath())
48+
if err != nil {
49+
return nil, err
50+
}
51+
return utils.SliceConvert(files, func(src File) (model.Obj, error) {
52+
return fileToObj(src), nil
53+
})
54+
}
55+
56+
func (d *Terabox) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
57+
if d.DownloadAPI == "crack" {
58+
return d.linkCrack(file, args)
59+
}
60+
return d.linkOfficial(file, args)
61+
}
62+
63+
func (d *Terabox) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
64+
_, err := d.create(stdpath.Join(parentDir.GetPath(), dirName), 0, 1, "", "")
65+
return err
66+
}
67+
68+
func (d *Terabox) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
69+
data := []base.Json{
70+
{
71+
"path": srcObj.GetPath(),
72+
"dest": dstDir.GetPath(),
73+
"newname": srcObj.GetName(),
74+
},
75+
}
76+
_, err := d.manage("move", data)
77+
return err
78+
}
79+
80+
func (d *Terabox) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
81+
data := []base.Json{
82+
{
83+
"path": srcObj.GetPath(),
84+
"newname": newName,
85+
},
86+
}
87+
_, err := d.manage("rename", data)
88+
return err
89+
}
90+
91+
func (d *Terabox) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
92+
data := []base.Json{
93+
{
94+
"path": srcObj.GetPath(),
95+
"dest": dstDir.GetPath(),
96+
"newname": srcObj.GetName(),
97+
},
98+
}
99+
_, err := d.manage("copy", data)
100+
return err
101+
}
102+
103+
func (d *Terabox) Remove(ctx context.Context, obj model.Obj) error {
104+
data := []string{obj.GetPath()}
105+
_, err := d.manage("delete", data)
106+
return err
107+
}
108+
109+
func (d *Terabox) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
110+
tempFile, err := utils.CreateTempFile(stream.GetReadCloser())
111+
if err != nil {
112+
return err
113+
}
114+
defer func() {
115+
_ = tempFile.Close()
116+
_ = os.Remove(tempFile.Name())
117+
}()
118+
var Default int64 = 4 * 1024 * 1024
119+
defaultByteData := make([]byte, Default)
120+
count := int(math.Ceil(float64(stream.GetSize()) / float64(Default)))
121+
// cal md5
122+
h1 := md5.New()
123+
h2 := md5.New()
124+
block_list := make([]string, 0)
125+
left := stream.GetSize()
126+
for i := 0; i < count; i++ {
127+
byteSize := Default
128+
var byteData []byte
129+
if left < Default {
130+
byteSize = left
131+
byteData = make([]byte, byteSize)
132+
} else {
133+
byteData = defaultByteData
134+
}
135+
left -= byteSize
136+
_, err = io.ReadFull(tempFile, byteData)
137+
if err != nil {
138+
return err
139+
}
140+
h1.Write(byteData)
141+
h2.Write(byteData)
142+
block_list = append(block_list, fmt.Sprintf("\"%s\"", hex.EncodeToString(h2.Sum(nil))))
143+
h2.Reset()
144+
}
145+
146+
_, err = tempFile.Seek(0, io.SeekStart)
147+
if err != nil {
148+
return err
149+
}
150+
151+
rawPath := stdpath.Join(dstDir.GetPath(), stream.GetName())
152+
path := encodeURIComponent(rawPath)
153+
block_list_str := fmt.Sprintf("[%s]", strings.Join(block_list, ","))
154+
data := fmt.Sprintf("path=%s&size=%d&isdir=0&autoinit=1&block_list=%s",
155+
path, stream.GetSize(),
156+
block_list_str)
157+
params := map[string]string{}
158+
var precreateResp PrecreateResp
159+
_, err = d.post("/api/precreate", params, data, &precreateResp)
160+
if err != nil {
161+
return err
162+
}
163+
log.Debugf("%+v", precreateResp)
164+
if precreateResp.ReturnType == 2 {
165+
return nil
166+
}
167+
params = map[string]string{
168+
"method": "upload",
169+
"path": path,
170+
"uploadid": precreateResp.Uploadid,
171+
"app_id": "250528",
172+
"web": "1",
173+
"channel": "dubox",
174+
"clienttype": "0",
175+
}
176+
left = stream.GetSize()
177+
for i, partseq := range precreateResp.BlockList {
178+
if utils.IsCanceled(ctx) {
179+
return ctx.Err()
180+
}
181+
byteSize := Default
182+
var byteData []byte
183+
if left < Default {
184+
byteSize = left
185+
byteData = make([]byte, byteSize)
186+
} else {
187+
byteData = defaultByteData
188+
}
189+
left -= byteSize
190+
_, err = io.ReadFull(tempFile, byteData)
191+
if err != nil {
192+
return err
193+
}
194+
u := "https://c-jp.terabox.com/rest/2.0/pcs/superfile2"
195+
params["partseq"] = strconv.Itoa(partseq)
196+
res, err := base.RestyClient.R().
197+
SetContext(ctx).
198+
SetQueryParams(params).
199+
SetFileReader("file", stream.GetName(), bytes.NewReader(byteData)).
200+
SetHeader("Cookie", d.Cookie).
201+
Post(u)
202+
if err != nil {
203+
return err
204+
}
205+
log.Debugln(res.String())
206+
if len(precreateResp.BlockList) > 0 {
207+
up(i * 100 / len(precreateResp.BlockList))
208+
}
209+
}
210+
_, err = d.create(rawPath, stream.GetSize(), 0, precreateResp.Uploadid, block_list_str)
211+
return err
212+
}
213+
214+
var _ driver.Driver = (*Terabox)(nil)

‎drivers/terabox/meta.go

+25
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
package terbox
2+
3+
import (
4+
"github.com/alist-org/alist/v3/internal/driver"
5+
"github.com/alist-org/alist/v3/internal/op"
6+
)
7+
8+
type Addition struct {
9+
driver.RootPath
10+
Cookie string `json:"cookie" required:"true"`
11+
DownloadAPI string `json:"download_api" type:"select" options:"official,crack" default:"official"`
12+
OrderBy string `json:"order_by" type:"select" options:"name,time,size" default:"name"`
13+
OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
14+
}
15+
16+
var config = driver.Config{
17+
Name: "Terabox",
18+
DefaultRoot: "/",
19+
}
20+
21+
func init() {
22+
op.RegisterDriver(func() driver.Driver {
23+
return &Terabox{}
24+
})
25+
}

‎drivers/terabox/types.go

+92
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,92 @@
1+
package terbox
2+
3+
import (
4+
"github.com/alist-org/alist/v3/internal/model"
5+
"strconv"
6+
"time"
7+
)
8+
9+
type File struct {
10+
//TkbindId int `json:"tkbind_id"`
11+
//OwnerType int `json:"owner_type"`
12+
//Category int `json:"category"`
13+
//RealCategory string `json:"real_category"`
14+
FsId int64 `json:"fs_id"`
15+
ServerMtime int64 `json:"server_mtime"`
16+
//OperId int `json:"oper_id"`
17+
//ServerCtime int `json:"server_ctime"`
18+
Thumbs struct {
19+
//Icon string `json:"icon"`
20+
Url3 string `json:"url3"`
21+
//Url2 string `json:"url2"`
22+
//Url1 string `json:"url1"`
23+
} `json:"thumbs"`
24+
//Wpfile int `json:"wpfile"`
25+
//LocalMtime int `json:"local_mtime"`
26+
Size int64 `json:"size"`
27+
//ExtentTinyint7 int `json:"extent_tinyint7"`
28+
Path string `json:"path"`
29+
//Share int `json:"share"`
30+
//ServerAtime int `json:"server_atime"`
31+
//Pl int `json:"pl"`
32+
//LocalCtime int `json:"local_ctime"`
33+
ServerFilename string `json:"server_filename"`
34+
//Md5 string `json:"md5"`
35+
//OwnerId int `json:"owner_id"`
36+
//Unlist int `json:"unlist"`
37+
Isdir int `json:"isdir"`
38+
}
39+
40+
type ListResp struct {
41+
Errno int `json:"errno"`
42+
GuidInfo string `json:"guid_info"`
43+
List []File `json:"list"`
44+
RequestId int64 `json:"request_id"`
45+
Guid int `json:"guid"`
46+
}
47+
48+
func fileToObj(f File) *model.ObjThumb {
49+
return &model.ObjThumb{
50+
Object: model.Object{
51+
ID: strconv.FormatInt(f.FsId, 10),
52+
Name: f.ServerFilename,
53+
Size: f.Size,
54+
Modified: time.Unix(f.ServerMtime, 0),
55+
IsFolder: f.Isdir == 1,
56+
},
57+
Thumbnail: model.Thumbnail{Thumbnail: f.Thumbs.Url3},
58+
}
59+
}
60+
61+
type DownloadResp struct {
62+
Errno int `json:"errno"`
63+
Dlink []struct {
64+
Dlink string `json:"dlink"`
65+
} `json:"dlink"`
66+
}
67+
68+
type DownloadResp2 struct {
69+
Errno int `json:"errno"`
70+
Info []struct {
71+
Dlink string `json:"dlink"`
72+
} `json:"info"`
73+
RequestID int64 `json:"request_id"`
74+
}
75+
76+
type HomeInfoResp struct {
77+
Errno int `json:"errno"`
78+
Data struct {
79+
Sign1 string `json:"sign1"`
80+
Sign3 string `json:"sign3"`
81+
Timestamp int `json:"timestamp"`
82+
} `json:"data"`
83+
}
84+
85+
type PrecreateResp struct {
86+
Path string `json:"path"`
87+
Uploadid string `json:"uploadid"`
88+
ReturnType int `json:"return_type"`
89+
BlockList []int `json:"block_list"`
90+
Errno int `json:"errno"`
91+
RequestId int64 `json:"request_id"`
92+
}

0 commit comments

Comments
 (0)