主动消息推送，支持Rss订阅

Author: MorvenCat

.gitignore

@@ -1,6 +1,6 @@
 node_modules
 WechatEveryDay.memory-card.json
-.env
+src/.env
 test.js
 package-lock.json
 yarn.lock

package.json

@@ -38,6 +38,7 @@
     "p-timeout": "^6.0.0",
     "qrcode-terminal": "^0.12.0",
     "remark": "^14.0.2",
+    "rss-parser": "^3.13.0",
     "strip-markdown": "^5.0.0",
     "wechaty": "^1.20.2",
     "wechaty-puppet-wechat4u": "^1.14.14",

src/Rss/index.js

@@ -0,0 +1,36 @@
+import Parser from 'rss-parser'
+import dotenv from 'dotenv'
+import { broadcastMessage } from '../wechaty/sendMessage.js'
+
+export function getRssMsg() {}
+
+const env = dotenv.config().parsed // 环境参数
+const rssList = env.RSS_URLS ? env.RSS_URLS.split(',') : [] //
+const lastUpdate = new Map() //
+const latestcontent = new Map()
+/**
+ * 监听rss更新
+ */
+export async function startRssWatch(bot) {
+  const parser = new Parser() // 初始化 RSS 解析器
+
+  // 每60秒检查一次更新
+  setInterval(async () => {
+    for (const url of rssList) {
+      try {
+        const feed = await parser.parseURL(url) // 解析RSS源
+        const latestItem = feed.items[0] // 获取最新的文章
+
+        if (!lastUpdate.has(url) || new Date(latestItem.pubDate) > lastUpdate.get(url)) {
+          // 如果是第一次检查，或者发现有更新
+          lastUpdate.set(url, new Date(latestItem.pubDate)) // 更新lastUpdate
+          latestcontent.set(url, latestItem.content) // 保存最新的文章标题
+          console.log(`New update found in ${url}: ${latestItem.content}`)
+          await broadcastMessage(bot, 'Rss', `${latestItem.content}`)
+        }
+      } catch (error) {
+        console.error(`Failed to fetch RSS feed from ${url}:`, error)
+      }
+    }
+  }, 60000) // 60000ms = 60秒
+}

src/index.js

@@ -8,6 +8,7 @@ import fs from 'fs'
 import path, { dirname } from 'path'
 import { fileURLToPath } from 'url'
 import { defaultMessage } from './wechaty/sendMessage.js'
+import { startRssWatch } from './Rss/index.js'
 
 const __filename = fileURLToPath(import.meta.url)
 const __dirname = dirname(__filename)
@@ -50,7 +51,7 @@ async function onFriendShip(friendship) {
 }
 
 /**
- * 消息发送
+ * 回复
  * @param msg
  * @param isSharding
  * @returns {Promise<void>}
@@ -102,7 +103,10 @@ bot.on('error', (e) => {
 function botStart() {
   bot
     .start()
-    .then(() => console.log('Start to log in wechat...'))
+    .then(() => {
+      console.log('Start to log in wechat...')
+      startRssWatch(bot) //启动Rss模块
+    })
     .catch((e) => console.error('❌ botStart error: ', e))
 }
 
@@ -154,6 +158,12 @@ function handleStart(type) {
       }
       console.log('❌ 请先配置.env文件中的 DIFY_API_KEY')
       break
+    case 'Rss':
+      if (env.RSS_URLS) {
+        return botStart()
+      }
+      console.log('❌ 请先配置.env文件中的 RSS_URLS')
+      break
     default:
       console.log('❌ 服务类型错误, 目前支持： ChatGPT | Kimi | Xunfei')
   }
@@ -166,6 +176,7 @@ export const serveList = [
   { name: 'deepseek-free', value: 'deepseek-free' },
   { name: '302AI', value: '302AI' },
   { name: 'dify', value: 'dify' },
+  { name: 'Rss', value: 'Rss' },
   // ... 欢迎大家接入更多的服务
 ]
 const questions = [

src/wechaty/sendMessage.js

@@ -62,6 +62,21 @@ export async function defaultMessage(msg, bot, ServiceType = 'GPT') {
   }
 }
 
+/**
+ * 消息广播
+ */
+export async function broadcastMessage(bot, ServiceType = 'Rss', msg) {
+  for (let roomName of roomWhiteList) {
+    const room = await bot.Room.find({ topic: roomName })
+    if (room) {
+      // 检查 room 是否找到
+      await room.say(msg)
+    } else {
+      console.error(`Room not found: ${roomName}`) // 记录未找到的房间
+    }
+  }
+}
+
 /**
  * 分片消息发送
  * @param message

src/wechaty/serve.js

@@ -4,11 +4,12 @@ import { getXunfeiReply } from '../xunfei/index.js'
 import { getDeepSeekFreeReply } from '../deepseek-free/index.js'
 import { get302AiReply } from '../302ai/index.js'
 import { getDifyReply } from '../dify/index.js'
+import { getRssMsg } from '../Rss/index.js'
 
 /**
  * 获取ai服务
  * @param serviceType 服务类型 'GPT' | 'Kimi'
- * @returns {Promise<void>}
+ * @returns {string}
  */
 export function getServe(serviceType) {
   switch (serviceType) {
@@ -24,6 +25,8 @@ export function getServe(serviceType) {
       return get302AiReply
     case 'dify':
       return getDifyReply
+    case 'Rss':
+      return getRssMsg()
     default:
       return getGptReply
   }