-
Notifications
You must be signed in to change notification settings - Fork 1
/
app.js
516 lines (449 loc) · 14.8 KB
/
app.js
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
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
/**
* Instagram 爬虫
* 爬取某个用户发布的所有视频+图片
*/
const request = require('request')
const async = require('async')
const path = require('path')
const fs = require('fs')
const cheerio = require('cheerio')
const util = require('util')
// log
const log = require('./utils/log').getLogger('debug')
// 要抓取的用户/tag
let users = ['iqb_c', 'funny_videos']
users = []
//let tags = ['movies', 'funnyvideos']
// let tags = ['animals', 'pets', 'mashup', 'movies', 'gaming', 'cartoons', 'art', 'music', 'sports', 'sciences', 'celebrity', 'nature', 'travel', 'fashion', 'dance', 'car', 'nsfw']
let tags = ['arbicdance']
let target = [].concat(
users.map(item => {
return {
type: 'user',
name: item
}
}),
tags.map(item => {
return {
type: 'tag',
name: item
}
})
)
let baseUrl = 'https://www.instagram.com/'
let purePage = 50 // 每次最多返回50条数据
let headers = {
cookie: '',
'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.70 Safari/537.36'
}
fetchUserUrl = (id, offset) => encodeURI(`https://www.instagram.com/graphql/query/?query_hash=2c5d4d8b70cad329c4a6ebe3abb6eedd&variables={"id":"${id}","first":${purePage},"after":"${offset}"}`)
let userCookie = 'mcd=3; mid=XDRFuAAEAAFTzWmhvjlWuFH7w-oa; csrftoken=G240QxAtheabum0dWDGBxSNmAX8q5b1I; sessionid=14274675657%3Aj0j4H1djZqvlHI%3A18; ds_user_id=14274675657; rur=VLL; urlgen="{\"119.28.85.131\": 132203\054 \"104.199.164.217\": 15169\054 \"35.213.124.13\": 19527}:1iQROm:nBFWuWJvZRioW9pT9H2ELy23y9k"'
let fetchTagUrl = (tagName, offset) => encodeURI(`https://www.instagram.com/graphql/query/?query_hash=174a5243287c5f3a7de741089750ab3b&variables={"tag_name":"${tagName}","first":${purePage},"after":"${offset}"}`)
let tagCookie = 'mcd=3; mid=XDRFuAAEAAFTzWmhvjlWuFH7w-oa; csrftoken=G240QxAtheabum0dWDGBxSNmAX8q5b1I; sessionid=14274675657%3Aj0j4H1djZqvlHI%3A18; ds_user_id=14274675657; rur=VLL; urlgen="{\"119.28.85.131\": 132203\054 \"104.199.164.217\": 15169\054 \"35.213.124.13\": 19527}:1iQNQC:CbHNVkT7TjcOQ_DNb8CM-Lvh4_w"'
let getVideoUrl = 'https://www.instagram.com/p/%s/?__a=1'
// 存储所有的媒体资源
let media = []
let videoCount = 0
let imgCount = 0
let videoDl = 0
let imgDl = 0
let fetchPageCount = 0
// 获取页数限制,ins tag下的页数达十万多页,选择性抓取最新的一部分
let pageLimit = 100
// 存储路径
let videoDlPath = path.resolve(__dirname, `./downloads/video`);
let imgDlPath = path.resolve(__dirname, `./downloads/img`);
let videoJsonPath = path.resolve(__dirname, `./downloads/video_json`);
let imgJsonPath = path.resolve(__dirname, `./downloads/img_json`);
mkdirsSync(videoDlPath)
mkdirsSync(imgDlPath)
mkdirsSync(videoJsonPath)
mkdirsSync(imgJsonPath)
/**
* 同步递归创建目录
*/
function mkdirsSync(dirname) {
if (fs.existsSync(dirname)) {
return true;
} else {
if (mkdirsSync(path.dirname(dirname))) {
fs.mkdirSync(dirname);
return true;
}
}
}
/**
* 获取指定用户的主页
*/
const getHtml = item => {
let userName = item.name,
type = item.type
let url
if (item.type == 'user') {
url = `${baseUrl}${userName}/`
headers.cookie = userCookie
} else {
url = `${baseUrl}explore/tags/${userName}/`
headers.cookie = tagCookie
}
let options = {
method: 'GET',
url: url,
headers: headers
}
return new Promise((resolve, reject) => {
request(options, function (error, response, body) {
if (error) return reject(error);
const $ = cheerio.load(body)
let html = $.html()
// 获取uid/tag name
userId = item.type == 'user' ? html.match(/"profilePage_([0-9]+)"/)[1] : html.match(/"name":"([a-zA-Z_]+)",/)[1]
log.info(`${userName} id/name 获取成功 ${userId}`)
// 获取首页数据
data = html.match(/<script type="text\/javascript">window._sharedData = (.*?);<\/script>/)[1]
data = JSON.parse(data)
let edges, count, pageInfo, cursor, flag, totalPage
let firstPageDate
if (item.type == 'user') {
firstPageDate = data.entry_data.ProfilePage[0].graphql.user.edge_owner_to_timeline_media
} else {
firstPageDate = data.entry_data.TagPage[0].graphql.hashtag.edge_hashtag_to_media
}
edges = firstPageDate.edges
count = firstPageDate.count
pageInfo = firstPageDate.page_info
cursor = pageInfo.end_cursor
flag = pageInfo.has_next_page
totalPage = Math.ceil(count / purePage)
// 存储首页信息
// 优化: 等待所有视频下载地址获取完毕再进行下一步,避免丢失一部分未完成数据
let actions = []
edges.forEach(item => {
item.mode = type
actions.push(storeMedia(item))
})
Promise.all(actions)
.then(() => {
// 返回分页信息
return resolve({
totalPage: totalPage,
userId: userId,
cursor: cursor
})
})
});
})
}
/**
* 获取该用户的所有内容
*/
const getAllUrls = (item, totalPage, uid, cursor) => {
let userName = item.name
let actions = [async.constant(item, uid, cursor)]
let limit = totalPage > pageLimit ? pageLimit : totalPage
for (let i = 0; i < limit; i++) {
actions.push(fetchData)
}
log.info(`${userName} 数据共 ${totalPage} 页`)
return new Promise((resolve, reject) => {
async.waterfall(actions, (error, result) => {
log.info(`${userName} 的所有帖子数据获取成功,共${media.length}个帖子,视频${videoCount}个,图片${imgCount}个`, )
fetchPageCount = 0
//console.log(media)
return resolve(media)
})
})
}
/**
* 请求获取数据
*/
const fetchData = (item, uid, offset, next) => {
let userName = item.name,
type = item.type
let url
if (item.type == 'user') {
// url = util.format(fetchUserUrl, uid, offset)
url = fetchUserUrl(uid, offset)
headers.cookie = userCookie
} else {
// url = util.format(fetchTagUrl, uid, offset)
url = fetchTagUrl(uid, offset)
headers.cookie = tagCookie
}
// console.log(url)
let options = {
method: 'GET',
url: url,
headers: headers
};
request(options, function (error, response, body) {
if (error) {
log.error('fetch data error', error)
log.info('休息1min~')
return setTimeout(function () {
return next(null, item, uid, offset)
}, 1 * 60 * 1000)
}
let data
try {
data = JSON.parse(body)
} catch (error) {
log.error('json序列化失败', error)
return next(null, item, uid, offset)
}
if (data.status == 'fail') {
log.error('返回内容失败', data)
log.info('休息1min~')
//return next(data.message)
return setTimeout(function () {
return next(null, item, uid, offset)
}, 1 * 60 * 1000)
}
let listData
try {
if (item.type == 'user') {
listData = data.data.user.edge_owner_to_timeline_media
} else {
listData = data.data.hashtag.edge_hashtag_to_media
}
} catch (error) {
log.error('数据获取失败', error)
next(error)
}
let edges = listData.edges
let actions = []
edges.forEach(item => {
item.mode = type
// storeMedia(item)
actions.push(storeMedia(item))
})
let {
has_next_page,
end_cursor
} = listData.page_info
log.info(`page:${++fetchPageCount} ${userName} 数据获取成功,帖子 ${edges.length} 个, has_next_page: ${has_next_page} ,end_cursor: ${end_cursor}`)
Promise.all(actions)
.then(() => {
if (!has_next_page) {
return next('所有数据获取完毕,无下页')
}
setTimeout(function () {
return next(null, item, uid, end_cursor)
}, 2000)
})
});
}
/**
* 根据视频的shortcode获取视频的下载地址
*/
const fetchVideoUrl = (mode, shortcode) => {
let url = util.format(getVideoUrl, shortcode)
if (mode == 'user') {
headers.cookie = userCookie
} else {
headers.cookie = tagCookie
}
let options = {
method: 'GET',
url: url,
headers: headers
}
return new Promise((resolve, reject) => {
request(options, function (error, response, body) {
let videoUrl = ''
if (error) {
log.error(`获取 ${shortcode} 视频地址失败`, error)
return resolve(videoUrl)
}
try {
let data = JSON.parse(body)
videoUrl = data.graphql.shortcode_media.video_url
} catch (error) {
log.error(`获取 ${shortcode} videoUrl 为空`)
}
return resolve(videoUrl)
})
})
}
/**
* 根据不同的类型存储数据
*/
const storeMedia = async item => {
let result = {
id: item.node.id,
desc: item.node.edge_media_to_caption.edges[0] ? item.node.edge_media_to_caption.edges[0].node.text : ''
}
if (item.node.is_video) {
// video
// 如果有video_url直接获取
// 如果没有video_url,通过接口获取
let videoUrl = item.node.video_url
if (!videoUrl) videoUrl = await fetchVideoUrl(item.mode, item.node.shortcode)
if (videoUrl) {
result.type = 'video'
result.url = videoUrl
videoCount++
}
} else {
// img
let imgUrl = item.node.display_url
if (imgUrl) {
result.type = 'img'
result.url = imgUrl
imgCount++
}
}
media.push(result)
}
/**
* 下载视频/图片
*/
const download = (category, media, next) => {
let isExist = isFileExist(media.id)
if (isExist) return next(null)
let filePath
if (media.type == 'video') {
filePath = `${videoDlPath}/${media.id}.mp4`
} else if (media.type == 'img') {
filePath = `${imgDlPath}/${media.id}.jpg`
} else return next(null)
let st = new Date()
request(media.url)
.on('response', function (res) {
// create file write stream
let fws = fs.createWriteStream(filePath);
// setup piping
res.pipe(fws);
// finish
res.on('end', function (e) {
let et = new Date()
let ut = timeUsed((et - st) / 1000)
log.info(`${videoDl + imgDl} finish download ${category} ${filePath},用时${ut}`)
saveJsonData(media.type, {
id: media.id,
category: category,
desc: media.desc
})
if (media.type == 'video') videoDl++
else imgDl++
return next(null)
});
// error handler
res.on('error', err => {
log.error('download error', err)
return next(null)
})
})
.on('error', function (err) {
log.error('request source failed', media.url, err)
// 大约3分钟可恢复
log.info('超频啦!休息1分钟~')
setTimeout(function () {
return next(null)
}, 1 * 60 * 1000)
})
}
/**
* 视频是否已下载
*/
const isFileExist = id => {
let videoPath = `${videoDlPath}/${id}.mp4`
let imgPath = `${imgDlPath}/${id}.jpg`
if (fs.existsSync(videoPath)) {
log.info('video file exist', videoPath)
return true
} else if (fs.existsSync(imgPath)) {
log.info('img file exist', imgPath)
return true
} else return false
}
/**
* 视频下载成功后,实时更新json数据。防止程序中途奔溃后视频信息未保存
*/
const saveJsonData = (type, data) => {
try {
// 读取已有json信息
let jsonFile = type == 'video' ? videoJsonPath : imgJsonPath
jsonFile += `/data.json`
let jsonData = []
if (fs.existsSync(jsonFile)) {
fileData = fs.readFileSync(jsonFile, {
encoding: 'utf8'
})
if (fileData) {
jsonData = JSON.parse(fileData)
}
}
// 写入
jsonData.push(data)
fs.writeFileSync(jsonFile, JSON.stringify(jsonData));
} catch (error) {
log.error('写入json文件失败', data)
}
}
const clearData = () => {
media = []
videoCount = 0
imgCount = 0
videoDl = 0
imgDl = 0
}
/**
* 下载某用户/标签下获取的所有资源
*/
const downloadAll = (userName, data) => {
let dlActions = data.map(item => next => {
download(userName, item, next)
})
return new Promise((resolve, reject) => {
async.series(dlActions, (error, result) => {
return resolve(result)
})
})
}
/**
* 用时显示
*/
const timeUsed = t => {
// [1s, 1m)
if (t < 60) return `${Math.ceil(t)}s`
// [1m, 1h)
else if (t >= 60 && t < 60 * 60) return `${Math.floor(t/60)}m${Math.floor(t%60)}s`
// [1h, 1d)
else if (t >= 60 * 60 && t < 60 * 60 * 24) return `${Math.floor(t/(60*60))}h${Math.floor(t%(60*60)/60)}m`
// [1d, ~)
else return `${ Math.floor(t/(24*60*60)) }d ${ Math.floor( t%(24*60*60)/(60*60) ) }h`
}
/**
* 某个用户/标签的抓取任务
*/
const task = async (item, next) => {
let userName = item.name
let {
totalPage,
userId,
cursor
} = await getHtml(item).catch(error => {
log.error('fetch error', error)
return next(null)
})
let data = await getAllUrls(item, totalPage, userId, cursor)
clearData()
let st = new Date()
let download = await downloadAll(userName, data)
let et = new Date()
let ut = timeUsed((et - st) / 1000)
log.info(`${userName} 所有下载完成, video ${videoDl} 个,img ${imgDl} 个,共用时 ${ut}`)
clearData()
return next(null)
}
const main = () => {
let actions = target.map(item => next => {
task(item, next)
})
async.series(actions, (error, result) => {
log.info(`所有 ${result.length} 个任务完成`, error)
process.exit(0)
})
}
main()