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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
|
package downloader
import (
"bvd/internal/api"
"encoding/json"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
)
type DownloadOptions struct {
OutputFile string // 单个视频下载时的输出文件名
OutputDir string // 下载目标文件夹
}
type Downloader struct{}
func NewDownloader() *Downloader {
return &Downloader{}
}
func requestSend(urls []string, video *api.VideoBaseInfo, opts *DownloadOptions) error {
// 确定下载目录
downloadDir := "./downloads/"
if opts != nil && opts.OutputDir != "" {
downloadDir = opts.OutputDir
}
// 创建下载目录
if err := os.MkdirAll(downloadDir, 0755); err != nil {
return fmt.Errorf("❌ 创建下载目录失败：%w", err)
}
for i := 0; i < len(urls); i++ {
fmt.Printf("✅ 开始下载第 %d 个文件\n", i+1)
// 创建支持重定向的 HTTP 客户端
client := &http.Client{
CheckRedirect: func(req *http.Request, via []*http.Request) error {
return nil
},
}
req, err := http.NewRequest("GET", urls[i], nil)
if err != nil {
return err
}
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
req.Header.Set("referer", "https://www.bilibili.com")
// 发送请求
resp, err := client.Do(req)
if err != nil {
return fmt.Errorf("❌ HTTP请求失败：%w", err)
}
defer resp.Body.Close()
// 检查响应状态
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("❌ 服务器返回错误状态码: %d", resp.StatusCode)
}
// 构建文件名
var filename string
if opts != nil && opts.OutputFile != "" {
// 使用用户指定的文件名
filename = filepath.Join(downloadDir, opts.OutputFile+".mp4")
} else {
// 使用视频分P名称作为文件名
filename = filepath.Join(downloadDir, video.Data[i].Part+".mp4")
}
fmt.Printf("✅ 保存到: %s\n", filename)
// 创建文件
out, err := os.Create(filename)
if err != nil {
return fmt.Errorf("❌ 创建文件失败：%w", err)
}
// 下载文件
_, err = io.Copy(out, resp.Body)
if err != nil {
out.Close()
return fmt.Errorf("❌ 下载失败：%w", err)
}
// 关闭文件
if err := out.Close(); err != nil {
return fmt.Errorf("❌ 关闭文件失败：%w", err)
}
fmt.Printf("✅ 文件下载成功: %s\n", filename)
}
return nil
}
func (d *Downloader) Start(bvid string, apiClient *api.BiliAPI, opts *DownloadOptions) error {
fmt.Println("✅ 获取到 BV 号:", bvid)
videoInfo, err := apiClient.GetVideoInfo(bvid)
if err != nil {
return err
}
// 2. 提取 提取所有 CID
cids := make([]int, len(videoInfo.Data))
for i, page := range videoInfo.Data {
cids[i] = page.Cid
}
fmt.Println("✅ 获取到 CID 列表", cids)
// 3. 为每个CID获取下载链接
downloadUrls := make([]string, len(cids))
for j, cid := range cids {
apiUrl := fmt.Sprintf("https://api.bilibili.com/x/player/playurl?&cid=%d&bvid=%s&qn=80", cid, bvid)
req, err := http.NewRequest("GET", apiUrl, nil)
if err != nil {
return fmt.Errorf("❌ CID %d 创建请求失败: %w", cid, err)
}
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
req.Header.Set("Referer", fmt.Sprintf("https://www.bilibili.com/video/%s", bvid))
resp, err := apiClient.Client.Do(req)
if err != nil {
return fmt.Errorf("❌ CID %d 请求失败: %w", cid, err)
}
body, err := io.ReadAll(resp.Body)
resp.Body.Close()
if err != nil {
return fmt.Errorf("❌ CID %d 读取响应失败: %w", cid, err)
}
var result struct {
Data struct {
Durl []struct {
Url string `json:"url"`
} `json:"durl"`
} `json:"data"`
}
if err := json.Unmarshal(body, &result); err != nil {
return fmt.Errorf("❌ CID %d JSON解析失败: %w", cid, err)
}
if len(result.Data.Durl) == 0 {
return fmt.Errorf("❌ CID %d 没有 没有获取到下载链接", cid)
}
downloadUrls[j] = result.Data.Durl[0].Url
}
fmt.Println("✅ 获取到所有下载链接")
requestSend(downloadUrls, videoInfo, opts)
return nil
}
|
