89 lines
3.4 KiB
TypeScript
89 lines
3.4 KiB
TypeScript
import { load } from 'cheerio';
|
|
|
|
import { config } from '@/config';
|
|
import ConfigNotFoundError from '@/errors/types/config-not-found';
|
|
import got from '@/utils/got';
|
|
|
|
const security_key = 'pianyuan-security_session_verify';
|
|
const PHPSESSID_key = 'pianyuan-PHPSESSID';
|
|
const loginauth_key = 'pianyuan-py_loginauth';
|
|
|
|
const ProcessFeed = async (list, cache) => {
|
|
const link_base = 'https://pianyuan.org';
|
|
|
|
const items = await Promise.all(
|
|
list.map(async (e) => {
|
|
const link = new URL(e, link_base).href;
|
|
const single = await cache.tryGet(link, async () => {
|
|
const response = await request(link, cache);
|
|
const content = load(response.data);
|
|
const magnet = content('.btn-primary').attr('href');
|
|
const torrent_name = content('body > div.jumbotron.masthead > div > div > div.col-sm-10.col-md-10.col-lg-10.text-left > h1').text();
|
|
const name = content('body > div.jumbotron.masthead > div > div > div.col-sm-10.col-md-10.col-lg-10.text-left > h2 > a').text();
|
|
const size = content('#main-container > div > div.col-sm-10.col-md-8.col-lg-8 > div > ul > li:nth-child(2)').text();
|
|
let length;
|
|
if (size.includes('MB')) {
|
|
length = size.replace('MB', '') * 1024;
|
|
} else if (size.includes('GB')) {
|
|
length = size.replace('GB', '') * 1024 * 1024;
|
|
}
|
|
// const description ='';
|
|
|
|
return {
|
|
title: `${torrent_name} [${name}] [${size}]`,
|
|
size: length,
|
|
enclosure_url: magnet,
|
|
enclosure_type: 'application/x-bittorrent',
|
|
enclosure_length: length,
|
|
};
|
|
});
|
|
return single;
|
|
})
|
|
);
|
|
|
|
return items;
|
|
};
|
|
|
|
async function getCookie(cache) {
|
|
const security_session_verify = await cache.get(security_key);
|
|
const PHPSESSID = await cache.get(PHPSESSID_key);
|
|
let py_loginauth = await cache.get(loginauth_key);
|
|
if (!py_loginauth) {
|
|
if (!config.pianyuan || !config.pianyuan.cookie) {
|
|
throw new ConfigNotFoundError('pianyuan is disabled due to the lack of <a href="https://docs.rsshub.app/deploy/config#route-specific-configurations">relevant config</a>');
|
|
}
|
|
py_loginauth = config.pianyuan.cookie;
|
|
}
|
|
return `${py_loginauth};${security_session_verify};${PHPSESSID};`;
|
|
}
|
|
|
|
async function request(link, cache) {
|
|
const cookie = await getCookie(cache);
|
|
const response = await got({
|
|
method: 'get',
|
|
url: link,
|
|
headers: {
|
|
Cookie: cookie,
|
|
},
|
|
});
|
|
// set cookie
|
|
const set_cookie = response.headers['set-cookie'];
|
|
if (set_cookie) {
|
|
for (const e of set_cookie) {
|
|
if (e.includes('security_session_verify')) {
|
|
cache.set(security_key, e.split(';')[0]);
|
|
} else if (e.includes('PHPSESSID')) {
|
|
cache.set(PHPSESSID_key, e.split(';')[0]);
|
|
} else if (e.includes('py_loginauth')) {
|
|
cache.set(loginauth_key, e.split(';')[0]);
|
|
}
|
|
}
|
|
}
|
|
if (response.data.includes('会员登录后才能访问')) {
|
|
throw new ConfigNotFoundError('pianyuan Cookie已失效');
|
|
}
|
|
return response;
|
|
}
|
|
|
|
export default { ProcessFeed, request };
|