RSSHub/lib/v2/zhihu/collection.js

78 lines
2.9 KiB
JavaScript

const got = require('@/utils/got');
const cheerio = require('cheerio');
const utils = require('./utils');
const { generateData } = require('./pin/utils');
const { parseDate } = require('@/utils/parse-date');
module.exports = async (ctx) => {
const id = ctx.params.id;
const getAll = ctx.params.getAll;
const response = await got({
method: 'get',
url: `https://www.zhihu.com/api/v4/collections/${id}/items?offset=0&limit=20`,
headers: {
...utils.header,
Referer: `https://www.zhihu.com/collection/${id}`,
},
});
const list = response.data.data;
if (getAll) {
const totals = response.data.paging.totals;
const offsetList = [...new Array(Math.round(totals / 20)).keys()].map((item) => item * 20).slice(1);
const otherList = await Promise.all(
offsetList.map((offset) =>
ctx.cache.tryGet(`https://www.zhihu.com/api/v4/collections/${id}/items?offset=${offset}&limit=20`, async () => {
const response = await got({
method: 'get',
url: `https://www.zhihu.com/api/v4/collections/${id}/items?offset=${offset}&limit=20`,
headers: {
...utils.header,
Referer: `https://www.zhihu.com/collection/${id}`,
},
});
return response.data.data;
})
)
).then((item) => item.flat());
list.push(...otherList);
}
const response2 = await got({
method: 'get',
url: `https://www.zhihu.com/collection/${id}`,
headers: {
...utils.header,
Referer: `https://www.zhihu.com/collection/${id}`,
},
});
const meta = response2.data;
const $ = cheerio.load(meta);
const collection_title = $('.CollectionDetailPageHeader-title').text() + ' - 知乎收藏夹';
const collection_description = $('.CollectionDetailPageHeader-description').text();
const generateDataPin = (item) => generateData([item.content])[0];
ctx.state.data = {
title: collection_title,
link: `https://www.zhihu.com/collection/${id}`,
description: collection_description,
item:
list &&
list.map((item) =>
item.content.type === 'pin'
? generateDataPin(item)
: {
title: item.content.type === 'article' || item.content.type === 'zvideo' ? item.content.title : item.content.question.title,
link: item.content.url,
description: item.content.type === 'zvideo' ? `<img src=${item.content.video.url}/>` : item.content.content,
pubDate: parseDate((item.content.type === 'article' ? item.content.updated : item.content.updated_time) * 1000),
}
),
};
};