@zhin.js/plugin-rss 1.0.1 → 1.1.0

package/src/db-store.ts

@@ -0,0 +1,59 @@
+export {
+  createInMemoryRssDb,
+  RSS_SEEN_TABLE,
+  RSS_SUBS_TABLE,
+  type RssMemoryDb,
+  type RssModel,
+} from './memory-store.js';
+
+import { createGenerationStore, type Dispose, type GenerationStoreContext } from 'zhin.js';
+import {
+  createInMemoryRssDb,
+  RSS_SEEN_TABLE,
+  RSS_SUBS_TABLE,
+  type RssMemoryDb,
+  type RssModel,
+} from './memory-store.js';
+
+const rssDbStore = createGenerationStore<RssMemoryDb>('rss/db');
+
+let _db: RssMemoryDb | null = null;
+let _memoryDb: RssMemoryDb | null = null;
+
+/**
+ * Generation-owned database binding used by Plugin Runtime setup().
+ * Auto-unregisters when the generation lifecycle disposes, so no stale
+ * reference to the previous generation's host survives a reload.
+ */
+export function provideRssDb(context: GenerationStoreContext, db: RssMemoryDb): Dispose {
+  return rssDbStore.provide(context, db);
+}
+
+export function ensureRssMemoryDb(): RssMemoryDb {
+  const provided = rssDbStore.tryUse() ?? _db;
+  if (provided) return provided;
+  if (!_memoryDb) _memoryDb = createInMemoryRssDb();
+  return _memoryDb;
+}
+
+export function getRssDb(): RssMemoryDb | null {
+  return ensureRssMemoryDb();
+}
+
+export function setRssDb(db: RssMemoryDb | null): void {
+  _db = db;
+}
+
+export function resetRssDb(): void {
+  rssDbStore.clear();
+  _db = null;
+  _memoryDb = null;
+}
+
+export function getRssSubs(): RssModel | null {
+  return ensureRssMemoryDb().models.get(RSS_SUBS_TABLE) ?? null;
+}
+
+export function getRssSeen(): RssModel | null {
+  return ensureRssMemoryDb().models.get(RSS_SEEN_TABLE) ?? null;
+}

package/src/feed.ts

@@ -0,0 +1,106 @@
+/**
+ * RSS feed fetch/preview helpers (no Plugin host).
+ */
+import Parser from 'rss-parser';
+
+export interface FeedItem {
+  title: string;
+  link: string;
+  date: string;
+  summary: string;
+  guid: string;
+}
+
+export interface RssConfig {
+  readonly pollCron: string;
+  readonly maxPerGroup: number;
+  readonly maxItems: number;
+  readonly timeout: number;
+}
+
+export const DEFAULT_RSS_CONFIG: RssConfig = {
+  pollCron: '0 */5 * * * *',
+  maxPerGroup: 30,
+  maxItems: 5,
+  timeout: 15_000,
+};
+
+export function resolveRssConfig(raw: Partial<RssConfig> | undefined): RssConfig {
+  return { ...DEFAULT_RSS_CONFIG, ...raw };
+}
+
+/**
+ * 剔除 HTML 标签与 `&nbsp;` 实体并折叠空白。
+ * 线性单遍扫描实现（手写状态机），语义等价于原先的
+ * `/<[^>]*>/g` → `/&nbsp;/g` → `/\s+/g` 三步 replace，但避免：
+ * - js/polynomial-redos：大量 `<` 无闭合时正则逐位重扫的二次方回溯；
+ * - js/incomplete-multi-character-sanitization：多趟 replace 的残留问题。
+ */
+export function stripHtml(html: string): string {
+  let out = '';
+  let pendingSpace = false;
+  let mayHaveTagEnd = html.includes('>');
+  let i = 0;
+  while (i < html.length) {
+    const ch = html[i]!;
+    if (ch === '<' && mayHaveTagEnd) {
+      const end = html.indexOf('>', i + 1);
+      if (end >= 0) {
+        i = end + 1;
+        continue;
+      }
+      // 后续不再有 `>`：剩余的 `<` 都是字面字符，不再重复扫描。
+      mayHaveTagEnd = false;
+    }
+    if (ch === '&' && html.startsWith('&nbsp;', i)) {
+      pendingSpace = true;
+      i += '&nbsp;'.length;
+      continue;
+    }
+    if (/\s/u.test(ch)) {
+      pendingSpace = true;
+      i += 1;
+      continue;
+    }
+    if (pendingSpace && out.length > 0) out += ' ';
+    pendingSpace = false;
+    out += ch;
+    i += 1;
+  }
+  return out;
+}
+
+export async function fetchFeed(
+  url: string,
+  timeout = DEFAULT_RSS_CONFIG.timeout,
+): Promise<{ title: string; items: FeedItem[] }> {
+  const parser = new Parser({
+    timeout,
+    headers: {
+      'User-Agent': 'ZhinBot-RSS/1.0',
+      Accept: 'application/rss+xml, application/atom+xml, application/xml, text/xml',
+    },
+  });
+  const feed = await parser.parseURL(url);
+  const items: FeedItem[] = (feed.items || []).slice(0, 20).map((item) => ({
+    title: (item.title || '').trim(),
+    link: (item.link || '').trim(),
+    date: item.isoDate || item.pubDate || '',
+    summary: stripHtml(item.contentSnippet || item.content || '').slice(0, 200),
+    guid: item.guid || item.link || item.title || '',
+  }));
+  return { title: (feed.title || url).trim(), items };
+}
+
+export function formatFeedPreview(title: string, items: FeedItem[]): string {
+  if (items.length === 0) return `${title}: 暂无内容`;
+  const lines = [`${title} (共 ${items.length} 条)`];
+  for (const item of items.slice(0, 5)) {
+    lines.push('');
+    lines.push(`${item.title}`);
+    if (item.date) lines.push(`  ${new Date(item.date).toLocaleString('zh-CN')}`);
+    if (item.link) lines.push(`  ${item.link}`);
+  }
+  if (items.length > 5) lines.push(`\n...还有 ${items.length - 5} 条`);
+  return lines.join('\n');
+}