mirror of
https://github.com/Jerryplusy/crystelf-plugin.git
synced 2025-12-05 15:41:56 +00:00
fix RSSpush and add listFeeds functionality
详见注释
This commit is contained in:
devil233
GitHub
parent
21d8546486
commit
b86743f885
224
apps/rssPush.js
224
apps/rssPush.js
@ -6,7 +6,7 @@ import fs from 'fs';
|
||||
import rssCache from '../lib/rss/rssCache.js';
|
||||
import schedule from 'node-schedule';
|
||||
import tools from '../components/tool.js';
|
||||
import ConfigControl from '../lib/config/configControl.js';
|
||||
// 移除了重复的 import ConfigControl
|
||||
|
||||
export default class RssPlugin extends plugin {
|
||||
constructor() {
|
||||
@ -31,6 +31,12 @@ export default class RssPlugin extends plugin {
|
||||
permission: 'master',
|
||||
priority: 100,
|
||||
},
|
||||
// 【新增】查看列表功能
|
||||
{
|
||||
reg: '^#rss列表$',
|
||||
fnc: 'listFeeds',
|
||||
permission: 'master',
|
||||
},
|
||||
{
|
||||
reg: /(https?:\/\/\S+(?:\.atom|\/feed))/i,
|
||||
fnc: 'autoAddFeed',
|
||||
@ -39,21 +45,24 @@ export default class RssPlugin extends plugin {
|
||||
},
|
||||
],
|
||||
});
|
||||
|
||||
// 【修复】定时任务启动逻辑
|
||||
// 只要插件加载，且未注册过任务，就启动定时器
|
||||
if (!global.__rss_job_scheduled) {
|
||||
if (ConfigControl.get()?.rss) {
|
||||
schedule.scheduleJob('*/10 * * * *', () => this.pushFeeds());
|
||||
global.__rss_job_scheduled = true;
|
||||
}
|
||||
// 默认每10分钟执行一次
|
||||
schedule.scheduleJob('*/10 * * * *', () => this.pushFeeds());
|
||||
global.__rss_job_scheduled = true;
|
||||
logger.mark('[RSS] 定时检测任务已启动');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加rss
|
||||
* @param e
|
||||
* @returns {Promise<*>}
|
||||
*/
|
||||
async addFeed(e) {
|
||||
const url = e.msg.replace(/^#rss添加/, '').trim();
|
||||
if (!url) return e.reply('请输入有效的RSS链接', true);
|
||||
|
||||
const feeds = configControl.get('feeds') || [];
|
||||
const groupId = e.group_id;
|
||||
|
||||
@ -74,12 +83,9 @@ export default class RssPlugin extends plugin {
|
||||
|
||||
/**
|
||||
* 自动添加
|
||||
* @param e
|
||||
* @returns {Promise<*|boolean>}
|
||||
*/
|
||||
async autoAddFeed(e) {
|
||||
//if (/^#rss/i.test(e.msg.trim())) return false;
|
||||
if (!ConfigControl.get()?.config?.rss) {
|
||||
if (!configControl.get()?.config?.rss) {
|
||||
return;
|
||||
}
|
||||
const url = e.msg.match(/(https?:\/\/\S+(?:\.atom|\/feed))/i)?.[1];
|
||||
@ -89,86 +95,180 @@ export default class RssPlugin extends plugin {
|
||||
}
|
||||
|
||||
/**
|
||||
* 移除rss
|
||||
* @param e
|
||||
* @returns {Promise<*>}
|
||||
* 【新增】查看当前群组订阅列表
|
||||
*/
|
||||
async removeFeed(e) {
|
||||
const index = parseInt(e.msg.replace(/^#rss移除/, '').trim(), 10);
|
||||
async listFeeds(e) {
|
||||
const feeds = configControl.get('feeds') || [];
|
||||
const groupId = e.group_id;
|
||||
|
||||
if (index < 0 || index >= feeds.length) return e.reply('索引无效..', true);
|
||||
// 筛选当前群组的订阅，保留全局索引以便移除
|
||||
const currentGroupFeeds = feeds
|
||||
.map((feed, index) => ({ index, ...feed }))
|
||||
.filter((feed) => feed.targetGroups.includes(groupId));
|
||||
|
||||
feeds[index].targetGroups = feeds[index].targetGroups.filter((id) => id !== groupId);
|
||||
await configControl.set('feeds', feeds);
|
||||
return await e.reply('群已移除该订阅');
|
||||
if (currentGroupFeeds.length === 0) {
|
||||
return e.reply('当前群组暂无任何RSS订阅。', true);
|
||||
}
|
||||
|
||||
const msg = [
|
||||
`≡ 当前群组订阅列表 (${currentGroupFeeds.length}) ≡`,
|
||||
...currentGroupFeeds.map((f) => `[${f.index}] ${f.url}`),
|
||||
'----------------',
|
||||
'提示: 使用 #rss移除+索引号 取消订阅'
|
||||
].join('\n');
|
||||
|
||||
return e.reply(msg);
|
||||
}
|
||||
|
||||
/**
|
||||
* 手动拉取
|
||||
* @param e
|
||||
* @returns {Promise<*>}
|
||||
* 【修复】移除rss (增强健壮性)
|
||||
*/
|
||||
async removeFeed(e) {
|
||||
// 提取数字
|
||||
const match = e.msg.match(/#rss移除\s*(\d+)/);
|
||||
if (!match || !match[1]) {
|
||||
return e.reply('请指定要移除的订阅索引，例如：#rss移除0', true);
|
||||
}
|
||||
|
||||
const index = parseInt(match[1], 10);
|
||||
const feeds = configControl.get('feeds') || [];
|
||||
const groupId = e.group_id;
|
||||
|
||||
// 越界与有效性检查
|
||||
if (isNaN(index) || index < 0 || index >= feeds.length) {
|
||||
return e.reply('索引无效，请发送 #rss列表 查看正确索引。', true);
|
||||
}
|
||||
|
||||
const targetFeed = feeds[index];
|
||||
if (!targetFeed) return e.reply('未找到该配置。', true);
|
||||
|
||||
// 兼容性处理
|
||||
if (!Array.isArray(targetFeed.targetGroups)) {
|
||||
targetFeed.targetGroups = [];
|
||||
}
|
||||
|
||||
// 检查本群是否已订阅
|
||||
if (!targetFeed.targetGroups.includes(groupId)) {
|
||||
return e.reply('当前群组未订阅此源，无需移除。', true);
|
||||
}
|
||||
|
||||
// 执行移除
|
||||
targetFeed.targetGroups = targetFeed.targetGroups.filter((id) => id !== groupId);
|
||||
await configControl.set('feeds', feeds);
|
||||
|
||||
return await e.reply(`已取消订阅：${targetFeed.title || targetFeed.url}`);
|
||||
}
|
||||
|
||||
/**
|
||||
* 【修复】手动拉取 (增加错误处理)
|
||||
*/
|
||||
async pullFeedNow(e) {
|
||||
const url = e.msg.replace(/^#rss拉取/, '').trim();
|
||||
const latest = await rssTools.fetchFeed(url);
|
||||
//logger.info(latest);
|
||||
if (!url) return e.reply('请提供RSS链接', true);
|
||||
|
||||
let latest;
|
||||
try {
|
||||
latest = await rssTools.fetchFeed(url);
|
||||
} catch (err) {
|
||||
logger.error(`[RSS] 手动拉取失败: ${err.message}`);
|
||||
return await e.reply(`拉取失败: ${err.message}`, true);
|
||||
}
|
||||
|
||||
if (!latest || !latest.length) {
|
||||
return await e.reply('拉取失败或无内容..', true);
|
||||
return await e.reply('拉取成功但无内容..', true);
|
||||
}
|
||||
|
||||
const post = latest[0];
|
||||
//console.log(post);
|
||||
const tempPath = path.join(process.cwd(), 'data', `rss-test-${Date.now()}.png`);
|
||||
await screenshot.generateScreenshot(post, tempPath);
|
||||
await e.reply([segment.image(tempPath)]);
|
||||
fs.unlinkSync(tempPath);
|
||||
|
||||
try {
|
||||
await e.reply(`最新文章：${post.title}\n正在生成预览...`);
|
||||
await screenshot.generateScreenshot(post, tempPath);
|
||||
await e.reply([segment.image(tempPath)]);
|
||||
} catch (err) {
|
||||
logger.error(`[RSS] 截图失败: ${err}`);
|
||||
await e.reply('生成预览图失败..', true);
|
||||
} finally {
|
||||
if (fs.existsSync(tempPath)) fs.unlinkSync(tempPath);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查rss更新
|
||||
* @returns {Promise<void>}
|
||||
* 【修复】核心逻辑：检查rss更新
|
||||
* 修复了时区问题、逻辑死循环问题和报错导致插件停止的问题
|
||||
*/
|
||||
async pushFeeds() {
|
||||
const feeds = configControl.get('feeds') || [];
|
||||
logger.mark(`正在检查rss流更新..`);
|
||||
// logger.mark(`正在检查rss流更新..`); // 调试时可解开
|
||||
|
||||
for (const feed of feeds) {
|
||||
const latest = await rssTools.fetchFeed(feed.url);
|
||||
let latest;
|
||||
try {
|
||||
latest = await rssTools.fetchFeed(feed.url);
|
||||
} catch (error) {
|
||||
logger.warn(`[RSS] 自动检查 ${feed.url} 失败: ${error.message}，跳过`);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!latest || !latest.length) continue;
|
||||
const todayStr = new Date().toISOString().split('T')[0];
|
||||
|
||||
const newItems = [];
|
||||
for (const item of latest) {
|
||||
const pubDate = item.date;
|
||||
if (!pubDate) continue;
|
||||
const itemDate = new Date(pubDate).toISOString().split('T')[0];
|
||||
if (itemDate !== todayStr) continue;
|
||||
if (!(await rssCache.has(feed.url, item.link))) {
|
||||
newItems.push(item);
|
||||
// 限制只检查前3条，避免刷屏
|
||||
const checkLimit = Math.min(latest.length, 3);
|
||||
|
||||
for (let i = 0; i < checkLimit; i++) {
|
||||
const item = latest[i];
|
||||
if (!item.link) continue;
|
||||
|
||||
// 【核心修复】不再强制对比 todayStr (解决时区问题)
|
||||
// 而是依赖 rssCache 判断是否推送过
|
||||
const isCached = await rssCache.has(feed.url, item.link);
|
||||
|
||||
if (!isCached) {
|
||||
// 增加一个宽松的时间检查 (48小时内)，防止第一次添加把几年前的文章全发出来
|
||||
const pubDate = item.date ? new Date(item.date).getTime() : Date.now();
|
||||
// 如果文章没有时间，默认允许推送；如果有时间，则必须在48小时内
|
||||
if (!item.date || (Date.now() - pubDate < 172800000)) {
|
||||
newItems.push(item);
|
||||
}
|
||||
}
|
||||
}
|
||||
if (newItems.length) {
|
||||
await rssCache.set(feed.url, newItems[0].link);
|
||||
for (const groupId of feed.targetGroups) {
|
||||
const post = newItems[0];
|
||||
const tempPath = path.join(process.cwd(), 'data', `rss-${Date.now()}.png`);
|
||||
if (feed.screenshot) {
|
||||
await Bot.pickGroup(groupId)?.sendMsg(
|
||||
`${configControl.get('profile')?.nickName}发现了一条新的rss推送!`
|
||||
);
|
||||
await tools.sleep(1000);
|
||||
//await Bot.pickGroup(groupId)?.sendMsg(`让${configControl.get('nickName')}看看内容是什么..`);
|
||||
// TODO 通过人工智能查看内容
|
||||
await Bot.pickGroup(groupId)?.sendMsg(
|
||||
`[标题] ${post.title}\n[作者] ${post.author}\n[来源] ${post.feedTitle}\n正在努力截图..`
|
||||
);
|
||||
await screenshot.generateScreenshot(post, tempPath);
|
||||
await Bot.pickGroup(groupId)?.sendMsg([segment.image(tempPath)]);
|
||||
fs.unlinkSync(tempPath);
|
||||
} else {
|
||||
await Bot.pickGroup(groupId)?.sendMsg(`[RSS推送]\n${post.title}\n${post.link}`);
|
||||
|
||||
if (newItems.length > 0) {
|
||||
// 【优化】倒序排列，先发旧的，后发新的
|
||||
newItems.reverse();
|
||||
|
||||
for (const post of newItems) {
|
||||
// 写入缓存
|
||||
await rssCache.set(feed.url, post.link);
|
||||
|
||||
for (const groupId of feed.targetGroups) {
|
||||
// 稍微延迟，防止风控
|
||||
await tools.sleep(2000);
|
||||
|
||||
const tempPath = path.join(process.cwd(), 'data', `rss-${Date.now()}.png`);
|
||||
try {
|
||||
if (feed.screenshot) {
|
||||
logger.info(`[RSS] 推送更新: ${post.title} -> 群 ${groupId}`);
|
||||
// 先发个文字提示
|
||||
await Bot.pickGroup(groupId)?.sendMsg(
|
||||
`[RSS] ${post.feedTitle || '订阅更新'}\n${post.title}`
|
||||
);
|
||||
|
||||
// 再发图片
|
||||
await screenshot.generateScreenshot(post, tempPath);
|
||||
await Bot.pickGroup(groupId)?.sendMsg([segment.image(tempPath)]);
|
||||
} else {
|
||||
await Bot.pickGroup(groupId)?.sendMsg(`[RSS推送]\n${post.title}\n${post.link}`);
|
||||
}
|
||||
} catch (err) {
|
||||
logger.error(`[RSS] 推送消息异常: ${err.message}`);
|
||||
} finally {
|
||||
// 确保删除临时文件
|
||||
if (fs.existsSync(tempPath)) {
|
||||
fs.unlinkSync(tempPath);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user