forked from qjfoidnh/BaiduPCS-Go
/
locatedonwload.go
82 lines (71 loc) · 1.92 KB
/
locatedonwload.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
package pcscommand
import (
"fmt"
"github.com/qjfoidnh/BaiduPCS-Go/internal/pcsconfig"
"github.com/qjfoidnh/BaiduPCS-Go/pcstable"
"github.com/qjfoidnh/BaiduPCS-Go/pcsutil/converter"
"os"
"strconv"
)
// RunLocateDownload 执行获取直链
func RunLocateDownload(pcspaths []string, opt *LocateDownloadOption) {
if opt == nil {
opt = &LocateDownloadOption{}
}
absPaths, err := matchPathByShellPattern(pcspaths...)
if err != nil {
fmt.Println(err)
return
}
pcs := GetBaiduPCS()
if opt.FromPan {
fds, err := pcs.FilesDirectoriesBatchMeta(absPaths...)
if err != nil {
fmt.Printf("%s\n", err)
return
}
fidList := make([]int64, 0, len(fds))
for i := range fds {
fidList = append(fidList, fds[i].FsID)
}
list, err := pcs.LocatePanAPIDownload(fidList...)
if err != nil {
fmt.Printf("%s\n", err)
return
}
tb := pcstable.NewTable(os.Stdout)
tb.SetHeader([]string{"#", "fs_id", "路径", "链接"})
var (
i int
fidStrList = converter.SliceInt64ToString(fidList)
)
for k := range fidStrList {
for i = range list {
if fidStrList[k] == list[i].FsID {
tb.Append([]string{strconv.Itoa(k), list[i].FsID, fds[k].Path, list[i].Dlink})
list = append(list[:i], list[i+1:]...)
break
}
}
}
tb.Render()
fmt.Printf("\n注意: 以上链接不能直接访问, 需要登录百度帐号才可以下载\n")
return
}
for i, pcspath := range absPaths {
info, err := pcs.LocateDownload(pcspath)
if err != nil {
fmt.Printf("[%d] %s, 路径: %s\n", i, err, pcspath)
continue
}
fmt.Printf("[%d] %s: \n", i, pcspath)
tb := pcstable.NewTable(os.Stdout)
tb.SetHeader([]string{"#", "链接"})
for k, u := range info.URLStrings(pcsconfig.Config.EnableHTTPS) {
tb.Append([]string{strconv.Itoa(k), u.String()})
}
tb.Render()
fmt.Println()
}
fmt.Printf("提示: 访问下载链接, 需将下载器的 User-Agent 设置为: %s\n", pcsconfig.Config.PanUA)
}