File size: 2,732 Bytes
bf48b89 | 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 | import { load } from 'cheerio';
import iconv from 'iconv-lite';
import ofetch from '@/utils/ofetch';
import { parseDate as _parseDate } from '@/utils/parse-date';
import _timezone from '@/utils/timezone';
function transElemText($, prop) {
const regex = /\$\((.*)\)/g;
let result = prop;
// eslint-disable-next-line @typescript-eslint/no-unused-vars
const parseDate = _parseDate;
// eslint-disable-next-line @typescript-eslint/no-unused-vars
const timezone = _timezone;
if (regex.test(result)) {
// eslint-disable-next-line no-eval
result = eval(result);
}
return result;
}
function replaceParams(data, prop, $) {
const regex = /%(.*)%/g;
let result = prop;
let group = regex.exec(prop);
while (group) {
// FIXME Multi vars
result = result.replace(group[0], transElemText($, data.params[group[1]]));
group = regex.exec(prop);
}
return result;
}
function getProp(data, prop, $) {
let result = data;
if (Array.isArray(prop)) {
for (const e of prop) {
result = transElemText($, result[e]);
}
} else {
result = transElemText($, result[prop]);
}
return replaceParams(data, result, $);
}
async function buildData(data) {
const response = await ofetch.raw(data.url);
const contentType = response.headers.get('content-type') || '';
// 若没有指定编码，则默认utf-8
let charset = 'utf-8';
for (const attr of contentType.split(';')) {
if (attr.includes('charset=')) {
charset = (attr.split('=').pop() || 'utf-8').toLowerCase();
}
}
// @ts-expect-error custom property
const responseData = charset === 'utf-8' ? response._data : iconv.decode(await ofetch(data.url, { responseType: 'buffer' }), charset);
const $ = load(responseData);
const $item = $(data.item.item);
// 这里应该是可以通过参数注入一些代码的，不过应该无伤大雅
return {
link: data.link,
title: getProp(data, 'title', $),
description: getProp(data, 'description', $),
allowEmpty: data.allowEmpty || false,
item: $item.toArray().map((e) => {
const $elem = (selector) => $(e).find(selector);
return {
title: getProp(data, ['item', 'title'], $elem),
description: getProp(data, ['item', 'description'], $elem),
pubDate: getProp(data, ['item', 'pubDate'], $elem),
link: getProp(data, ['item', 'link'], $elem),
guid: getProp(data, ['item', 'guid'], $elem),
};
}),
};
}
export default buildData;
export { getProp, replaceParams, transElemText };
|