forked from Angey40/BaiduPCS-Go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cp_mv_rename.go
105 lines (88 loc) · 2.25 KB
/
cp_mv_rename.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package baidupcs
import (
"encoding/json"
"fmt"
"github.com/bitly/go-simplejson"
"github.com/iikira/BaiduPCS-Go/requester"
"strings"
)
// CpMvJSON 源文件目录的地址和目标文件目录的地址
type CpMvJSON struct {
From string `json:"from"` // 源文件或目录
To string `json:"to"` // 目标文件或目录
}
// CpMvJSONList []CpMvJSON 对象数组
type CpMvJSONList struct {
List []CpMvJSON `json:"list"`
}
// Rename 重命名文件/目录
func (p PCSApi) Rename(from, to string) (err error) {
return p.cpmvOp("rename", CpMvJSON{
From: from,
To: to,
})
}
// Copy 批量拷贝文件/目录
func (p PCSApi) Copy(cpmvJSON ...CpMvJSON) (err error) {
return p.cpmvOp("copy", cpmvJSON...)
}
// Move 批量移动文件/目录
func (p PCSApi) Move(cpmvJSON ...CpMvJSON) (err error) {
return p.cpmvOp("move", cpmvJSON...)
}
func (p PCSApi) cpmvOp(op string, cpmvJSON ...CpMvJSON) (err error) {
ejs, err := cpmvJSONEncode(cpmvJSON...)
if err != nil {
return err
}
method := op
if method == "rename" {
method = "move"
}
p.addItem("file", method, map[string]string{
"param": ejs,
})
h := requester.NewHTTPClient()
body, err := h.Fetch("POST", p.url.String(), nil, map[string]string{
"Cookie": "BDUSS=" + p.bduss,
})
if err != nil {
return err
}
json, err := simplejson.NewJson(body)
if err != nil {
return
}
code, err := checkErr(json)
if err != nil {
switch op {
case "copy":
return fmt.Errorf("拷贝文件/目录 遇到错误, 错误代码: %d, 消息: %s", code, err)
case "move":
return fmt.Errorf("移动文件/目录 遇到错误, 错误代码: %d, 消息: %s", code, err)
case "rename":
return fmt.Errorf("重命名文件/目录 遇到错误, 错误代码: %d, 消息: %s", code, err)
default:
panic("Unknown op: " + op)
}
}
return nil
}
//cpmvJSONEncode 生成 json 串
func cpmvJSONEncode(cpmvJSON ...CpMvJSON) (string, error) {
pathsData := CpMvJSONList{
List: cpmvJSON,
}
ej, err := json.Marshal(&pathsData)
if err != nil {
return "", err
}
return string(ej[:]), nil
}
func (cl CpMvJSONList) String() string {
l := make([]string, len(cl.List))
for k := range cl.List {
l[k] = fmt.Sprintf("%d: %s -> %s", k+1, cl.List[k].From, cl.List[k].To)
}
return strings.Join(l, "\n")
}