81 lines
2.7 KiB
TypeScript
81 lines
2.7 KiB
TypeScript
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 };
|