@zhin.js/plugin-group-suite 3.0.2 → 3.0.3

package/src/shared-runtime.ts

@@ -0,0 +1,44 @@
+export function todayStr(): string {
+  const d = new Date();
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}-${String(d.getDate()).padStart(2, '0')}`;
+}
+
+export function yesterdayStr(): string {
+  const d = new Date();
+  d.setDate(d.getDate() - 1);
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}-${String(d.getDate()).padStart(2, '0')}`;
+}
+
+export function ts(): string {
+  return new Date().toISOString();
+}
+
+export function randomInt(min: number, max: number): number {
+  return Math.floor(Math.random() * (max - min + 1)) + min;
+}
+
+/** Runtime Message → group/global context */
+export function resolveContextKey(input: {
+  target?: string;
+  metadata?: Readonly<Record<string, unknown>>;
+}): { type: string; id: string } {
+  const meta = input.metadata ?? {};
+  const channelType = String(meta.type ?? meta.channelType ?? '');
+  if (channelType === 'group' || channelType === 'guild') {
+    return { type: 'group', id: String(input.target ?? meta.channelId ?? '') };
+  }
+  if (input.target && channelType !== 'private') {
+    return { type: 'group', id: String(input.target) };
+  }
+  return { type: 'global', id: '' };
+}
+
+export function resolveSender(input: {
+  sender?: string;
+  metadata?: Readonly<Record<string, unknown>>;
+}): { id: string; name: string } {
+  const meta = input.metadata ?? {};
+  const id = String(input.sender ?? meta.senderId ?? '');
+  const name = String(meta.senderName ?? meta.name ?? '用户');
+  return { id, name };
+}

package/src/stats-lib.ts

@@ -0,0 +1,202 @@
+import type { GroupSuiteConfig } from './config.js';
+import { getStatsModel } from './db-store.js';
+import {
+  resolveSender,
+  todayStr,
+  ts,
+} from './shared-runtime.js';
+import {
+  buildStatsRankReportData,
+  formatMyStatsText,
+  formatRankText,
+  type MyStatsReportData,
+} from './stats-data.js';
+import type { GroupSuiteRuntime, PendingStatsIncrement } from './runtime-state.js';
+
+export function weekStartStr(): string {
+  const d = new Date();
+  const day = d.getDay() || 7;
+  d.setDate(d.getDate() - day + 1);
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}-${String(d.getDate()).padStart(2, '0')}`;
+}
+
+export function monthStartStr(): string {
+  const d = new Date();
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, '0')}-01`;
+}
+
+const buffer = new Map<string, PendingStatsIncrement>();
+
+function bufferKey(userId: string, groupId: string, date: string): string {
+  return `${userId}:${groupId}:${date}`;
+}
+
+export function resolveGroupId(input: {
+  target?: string;
+  metadata?: Readonly<Record<string, unknown>>;
+}): string {
+  const meta = input.metadata ?? {};
+  const channelType = String(meta.type ?? meta.channelType ?? '');
+  if (channelType === 'group' || channelType === 'guild') {
+    return String(input.target ?? meta.channelId ?? '');
+  }
+  if (input.target && channelType !== 'private') {
+    return String(input.target);
+  }
+  return '';
+}
+
+export function recordMessage(input: {
+  sender?: string;
+  target?: string;
+  metadata?: Readonly<Record<string, unknown>>;
+}, runtime?: GroupSuiteRuntime): void {
+  const targetBuffer = runtime?.statsBuffer ?? buffer;
+  const { id: userId, name: userName } = resolveSender(input);
+  if (!userId) return;
+  const groupId = resolveGroupId(input);
+  const key = bufferKey(userId, groupId, todayStr());
+  const pending = targetBuffer.get(key);
+  if (pending) {
+    pending.count++;
+    pending.user_name = userName;
+  } else {
+    targetBuffer.set(key, {
+      user_id: userId,
+      user_name: userName,
+      group_id: groupId,
+      date: todayStr(),
+      count: 1,
+    });
+  }
+}
+
+export async function flushStatsBuffer(runtime?: GroupSuiteRuntime): Promise<void> {
+  const targetBuffer = runtime?.statsBuffer ?? buffer;
+  const M = getStatsModel(runtime?.db);
+  if (!M || targetBuffer.size === 0) return;
+  const entries = [...targetBuffer.values()];
+  targetBuffer.clear();
+  for (const entry of entries) {
+    try {
+      const existing = (await M.select().where({
+        user_id: entry.user_id,
+        group_id: entry.group_id,
+        date: entry.date,
+      })) as Record<string, unknown>[];
+      if (existing.length > 0) {
+        await M.update({
+          count: Number(existing[0]!.count || 0) + entry.count,
+          user_name: entry.user_name,
+          updated_at: ts(),
+        }).where({
+          user_id: entry.user_id,
+          group_id: entry.group_id,
+          date: entry.date,
+        });
+      } else {
+        await M.insert({
+          user_id: entry.user_id,
+          user_name: entry.user_name,
+          group_id: entry.group_id,
+          date: entry.date,
+          count: entry.count,
+          updated_at: ts(),
+        });
+      }
+    } catch {
+      /* single row failure */
+    }
+  }
+}
+
+export async function queryStats(
+  groupId: string,
+  fromDate: string,
+  runtime?: GroupSuiteRuntime,
+): Promise<Map<string, { name: string; count: number }>> {
+  const M = getStatsModel(runtime?.db);
+  if (!M) return new Map();
+  const all = (groupId
+    ? await M.select().where({ group_id: groupId })
+    : await M.select()) as Record<string, unknown>[];
+  const result = new Map<string, { name: string; count: number }>();
+  for (const row of all) {
+    if (String(row.date) < fromDate) continue;
+    const userId = String(row.user_id);
+    const existing = result.get(userId);
+    if (existing) {
+      existing.count += Number(row.count || 0);
+      if (row.user_name) existing.name = String(row.user_name);
+    } else {
+      result.set(userId, {
+        name: String(row.user_name || row.user_id),
+        count: Number(row.count || 0),
+      });
+    }
+  }
+  return result;
+}
+
+type MessageInput = {
+  sender?: string;
+  target?: string;
+  metadata?: Readonly<Record<string, unknown>>;
+};
+
+export async function statsRankText(
+  input: MessageInput,
+  cfg: GroupSuiteConfig,
+  fromDate: string,
+  title: string,
+  runtime?: GroupSuiteRuntime,
+): Promise<string> {
+  await flushStatsBuffer(runtime);
+  const groupId = resolveGroupId(input);
+  const stats = await queryStats(groupId, fromDate, runtime);
+  const { id: userId } = resolveSender(input);
+  const data = buildStatsRankReportData(stats, title, cfg.rankSize, userId);
+  return formatRankText(data);
+}
+
+export async function myStatsText(input: MessageInput, runtime?: GroupSuiteRuntime): Promise<string> {
+  await flushStatsBuffer(runtime);
+  const M = getStatsModel(runtime?.db);
+  if (!M) return '统计数据库尚未就绪';
+  const { id: userId, name: userName } = resolveSender(input);
+  if (!userId) return '无法获取用户信息';
+  const groupId = resolveGroupId(input);
+  const rows = (groupId
+    ? await M.select().where({ user_id: userId, group_id: groupId })
+    : await M.select().where({ user_id: userId })) as Record<string, unknown>[];
+  if (rows.length === 0) return '暂无你的消息记录';
+  const today = todayStr();
+  const weekStart = weekStartStr();
+  const monthStart = monthStartStr();
+  let todayCount = 0;
+  let weekCount = 0;
+  let monthCount = 0;
+  let totalCount = 0;
+  for (const row of rows) {
+    const c = Number(row.count || 0);
+    totalCount += c;
+    if (String(row.date) >= monthStart) monthCount += c;
+    if (String(row.date) >= weekStart) weekCount += c;
+    if (row.date === today) todayCount += c;
+  }
+  const data: MyStatsReportData = {
+    userName: userName || '你',
+    scope: groupId ? '本群' : '全局',
+    todayCount,
+    weekCount,
+    monthCount,
+    totalCount,
+    activeDays: rows.length,
+  };
+  return formatMyStatsText(data);
+}
+
+/** Test helper */
+export function resetStatsBuffer(): void {
+  buffer.clear();
+}

package/src/tables.ts

@@ -0,0 +1,46 @@
+/**
+ * Group-suite table schemas for DatabaseHost / legacy DatabaseFeature.
+ */
+import {
+  CHECKIN_TABLE,
+  TEACH_TABLE,
+  STATS_TABLE,
+} from './memory-store.js';
+
+export function defineGroupSuiteTables(
+  db: { define: (name: string, schema: Record<string, unknown>) => void },
+): void {
+  db.define(CHECKIN_TABLE, {
+    user_id: { type: 'text', nullable: false },
+    user_name: { type: 'text', default: '' },
+    points: { type: 'integer', default: 0 },
+    total_checkins: { type: 'integer', default: 0 },
+    streak: { type: 'integer', default: 0 },
+    max_streak: { type: 'integer', default: 0 },
+    last_checkin: { type: 'text', default: '' },
+    context_type: { type: 'text', default: 'global' },
+    context_id: { type: 'text', default: '' },
+    created_at: { type: 'text', default: '' },
+    updated_at: { type: 'text', default: '' },
+  });
+  db.define(TEACH_TABLE, {
+    question: { type: 'text', nullable: false },
+    answer: { type: 'text', nullable: false },
+    is_regex: { type: 'integer', default: 0 },
+    context_type: { type: 'text', default: 'global' },
+    context_id: { type: 'text', default: '' },
+    creator_id: { type: 'text', default: '' },
+    creator_name: { type: 'text', default: '' },
+    hit_count: { type: 'integer', default: 0 },
+    created_at: { type: 'text', default: '' },
+    updated_at: { type: 'text', default: '' },
+  });
+  db.define(STATS_TABLE, {
+    user_id: { type: 'text', nullable: false },
+    user_name: { type: 'text', default: '' },
+    group_id: { type: 'text', default: '' },
+    date: { type: 'text', nullable: false },
+    count: { type: 'integer', default: 0 },
+    updated_at: { type: 'text', default: '' },
+  });
+}

package/src/teach-lib.ts

@@ -0,0 +1,301 @@
+import type { GroupSuiteConfig } from './config.js';
+import { getTeachModel } from './db-store.js';
+import {
+  resolveContextKey,
+  resolveSender,
+  ts,
+} from './shared-runtime.js';
+import type { GroupSuiteRuntime } from './runtime-state.js';
+
+const cooldownMap = new Map<string, number>();
+
+export const TEACH_USAGE_HINT =
+  '格式：teach 关键词 回答（回答可含空格），或 teach 问题|答案';
+
+function isSafeRegex(pattern: string): boolean {
+  if (pattern.length > 200) return false;
+  if (/(\+|\*|\{)\)?(\+|\*|\{)/.test(pattern)) return false;
+  if (/\([^)]*(\+|\*)[^)]*(\+|\*)[^)]*\)[+*]/.test(pattern)) return false;
+  const altGroups = pattern.match(/\([^)]*\|[^)]*\)/g) || [];
+  for (const g of altGroups) {
+    if ((g.match(/\|/g) || []).length > 10) return false;
+  }
+  return true;
+}
+
+export function parseTeachPair(raw: string): { question: string; answer: string } | null {
+  if (!raw) return null;
+  const pipe = raw.indexOf('|');
+  if (pipe >= 0) {
+    const question = raw.slice(0, pipe).trim();
+    const answer = raw.slice(pipe + 1).trim();
+    if (question && answer) return { question, answer };
+    return null;
+  }
+  const lastSpace = raw.lastIndexOf(' ');
+  if (lastSpace < 0) return null;
+  const question = raw.slice(0, lastSpace).trim();
+  const answer = raw.slice(lastSpace + 1).trim();
+  if (!question || !answer) return null;
+  return { question, answer };
+}
+
+function processAnswer(
+  answer: string,
+  sender: { id: string; name: string },
+  match?: RegExpMatchArray,
+): string {
+  let result = answer;
+  result = result.replace(/\{sender\}/g, sender.name || '你');
+  result = result.replace(/\{sender\.id\}/g, sender.id || '');
+  result = result.replace(/\{time\}/g, new Date().toLocaleTimeString('zh-CN'));
+  result = result.replace(/\{date\}/g, new Date().toLocaleDateString('zh-CN'));
+  if (match) {
+    for (let i = 1; i < match.length; i++) {
+      result = result.replace(new RegExp(`\\$${i}`, 'g'), match[i] || '');
+    }
+  }
+  return result;
+}
+
+function qaWhereKey(row: {
+  question: string;
+  context_type: string;
+  context_id: string;
+  is_regex?: number | boolean;
+}) {
+  return {
+    question: row.question,
+    context_type: row.context_type,
+    context_id: row.context_id,
+    is_regex: row.is_regex ? 1 : 0,
+  };
+}
+
+type MessageInput = {
+  sender?: string;
+  target?: string;
+  content?: unknown;
+  metadata?: Readonly<Record<string, unknown>>;
+};
+
+async function findMatch(
+  content: string,
+  ctxType: string,
+  ctxId: string,
+  cfg: GroupSuiteConfig,
+  runtime?: GroupSuiteRuntime,
+): Promise<Record<string, unknown> | null> {
+  const QA = getTeachModel(runtime?.db);
+  if (!QA) return null;
+  const allItems = (await QA.select()) as Record<string, unknown>[];
+  const candidates = allItems.filter(
+    (item) =>
+      item.context_type === 'global' ||
+      (item.context_type === ctxType && item.context_id === ctxId),
+  );
+  for (const item of candidates) {
+    if (!item.is_regex && item.question === content) return item;
+  }
+  if (cfg.teachAllowRegex) {
+    for (const item of candidates) {
+      if (!item.is_regex) continue;
+      try {
+        if (new RegExp(String(item.question), 'i').test(content)) return item;
+      } catch {
+        continue;
+      }
+    }
+  }
+  return null;
+}
+
+export async function teachAdd(
+  input: MessageInput,
+  cfg: GroupSuiteConfig,
+  payload: string,
+  regex = false,
+  runtime?: GroupSuiteRuntime,
+): Promise<string> {
+  const QA = getTeachModel(runtime?.db);
+  if (!QA) return '问答数据库尚未就绪，请稍后重试';
+  if (regex && !cfg.teachAllowRegex) return '管理员已禁用正则问答';
+
+  const parsed = parseTeachPair(payload.trim());
+  if (!parsed) {
+    return regex
+      ? '格式：teach-regex 正则 回答，或 teach-regex 正则|回答'
+      : TEACH_USAGE_HINT;
+  }
+  const { question, answer } = parsed;
+
+  if (regex) {
+    try {
+      new RegExp(question, 'i');
+    } catch (e) {
+      return `正则表达式不合法: ${(e as Error).message}`;
+    }
+    if (!isSafeRegex(question)) return '该正则可能导致性能问题，请简化';
+  }
+
+  const { type: ctxType, id: ctxId } = resolveContextKey(input);
+  const sender = resolveSender(input);
+
+  if (ctxType === 'group') {
+    const existing = (await QA.select().where({
+      context_type: ctxType,
+      context_id: ctxId,
+    })) as Record<string, unknown>[];
+    if (existing.length >= cfg.teachMaxPerGroup) {
+      return `该群问答数已达上限 (${cfg.teachMaxPerGroup})`;
+    }
+  }
+
+  const duplicate = (await QA.select().where({
+    question,
+    context_type: ctxType,
+    context_id: ctxId,
+  })) as Record<string, unknown>[];
+
+  if (duplicate.length > 0 && !regex) {
+    await QA.update({ answer, updated_at: ts() }).where(
+      qaWhereKey(duplicate[0] as {
+        question: string;
+        context_type: string;
+        context_id: string;
+        is_regex?: number | boolean;
+      }),
+    );
+    return `已更新问答：「${question}」→「${answer}」`;
+  }
+
+  await QA.insert({
+    question,
+    answer,
+    is_regex: regex ? 1 : 0,
+    context_type: ctxType,
+    context_id: ctxId,
+    creator_id: sender.id,
+    creator_name: sender.name,
+    hit_count: 0,
+    created_at: ts(),
+    updated_at: ts(),
+  });
+
+  return regex
+    ? `学会了！匹配 /${question}/ 的消息我会回复「${answer}」`
+    : `学会了！发送「${question}」我会回复「${answer}」`;
+}
+
+export async function teachList(
+  input: MessageInput,
+  cfg: GroupSuiteConfig,
+  page = 1,
+  runtime?: GroupSuiteRuntime,
+): Promise<string> {
+  const QA = getTeachModel(runtime?.db);
+  if (!QA) return '问答数据库尚未就绪';
+  const safePage = Math.max(1, page);
+  const { type: ctxType, id: ctxId } = resolveContextKey(input);
+  const allItems = (await QA.select().where(
+    ctxType === 'global'
+      ? { context_type: 'global' }
+      : { context_type: ctxType, context_id: ctxId },
+  )) as Record<string, unknown>[];
+  if (allItems.length === 0) return '还没有教过任何问答哦～';
+  const totalPages = Math.ceil(allItems.length / cfg.teachPageSize);
+  const pageNum = Math.min(safePage, totalPages);
+  const start = (pageNum - 1) * cfg.teachPageSize;
+  const pageItems = allItems.slice(start, start + cfg.teachPageSize);
+  const lines = pageItems.map((item, i) => {
+    const prefix = item.is_regex ? '[正则]' : '[精确]';
+    const q = item.is_regex ? `/${item.question}/` : item.question;
+    return `${start + i + 1}. ${prefix} ${q} → ${item.answer}`;
+  });
+  const header = ctxType === 'global' ? '全局问答列表' : '本群问答列表';
+  return `${header}\n${lines.join('\n')}\n第 ${pageNum}/${totalPages} 页 · 共 ${allItems.length} 条`;
+}
+
+export async function teachForget(
+  input: MessageInput,
+  questionRaw: string,
+  runtime?: GroupSuiteRuntime,
+): Promise<string> {
+  const QA = getTeachModel(runtime?.db);
+  if (!QA) return '问答数据库尚未就绪';
+  const question = questionRaw.trim();
+  if (!question) return '请提供要删除的问题';
+  const { type: ctxType, id: ctxId } = resolveContextKey(input);
+  const items = (await QA.select().where({
+    question,
+    context_type: ctxType,
+    context_id: ctxId,
+  })) as Record<string, unknown>[];
+  if (items.length === 0) return `没有找到问题「${question}」`;
+  await QA.delete().where(
+    qaWhereKey(items[0] as {
+      question: string;
+      context_type: string;
+      context_id: string;
+      is_regex?: number | boolean;
+    }),
+  );
+  return `已忘记「${question}」的回答`;
+}
+
+/** Try teach auto-reply; returns reply text or null. */
+export async function tryTeachReply(
+  input: MessageInput,
+  cfg: GroupSuiteConfig,
+  runtime?: GroupSuiteRuntime,
+): Promise<string | null> {
+  const content = typeof input.content === 'string' ? input.content.trim() : '';
+  if (!content) return null;
+  const { type: ctxType, id: ctxId } = resolveContextKey(input);
+  const matched = await findMatch(content, ctxType, ctxId, cfg, runtime);
+  if (!matched) return null;
+
+  const cooldownKey = `${matched.question}:${ctxType}:${ctxId}:${matched.is_regex ?? 0}`;
+  const cooldowns = runtime?.teachCooldowns ?? cooldownMap;
+  const last = cooldowns.get(cooldownKey);
+  if (last && Date.now() - last < cfg.teachCooldownMs) return null;
+  cooldowns.set(cooldownKey, Date.now());
+
+  const QA = getTeachModel(runtime?.db);
+  if (QA) {
+    try {
+      await QA.update({
+        hit_count: Number(matched.hit_count || 0) + 1,
+        updated_at: ts(),
+      }).where(
+        qaWhereKey(matched as {
+          question: string;
+          context_type: string;
+          context_id: string;
+          is_regex?: number | boolean;
+        }),
+      );
+    } catch {
+      /* ignore */
+    }
+  }
+
+  const sender = resolveSender(input);
+  let answer = String(matched.answer ?? '');
+  if (matched.is_regex) {
+    try {
+      const m = content.match(new RegExp(String(matched.question), 'i'));
+      answer = processAnswer(answer, sender, m || undefined);
+    } catch {
+      /* ignore */
+    }
+  } else {
+    answer = processAnswer(answer, sender);
+  }
+  return answer;
+}
+
+/** Test helper */
+export function resetTeachCooldown(): void {
+  cooldownMap.clear();
+}

package/tools/checkin-query/handler.ts

@@ -1,5 +1,5 @@
-import { getCheckinModel } from '../../src/checkin.js';
-import { todayStr } from '../../src/shared.js';
+import { getCheckinModel } from '../../src/db-store.js';
+import { todayStr } from '../../src/shared-runtime.js';
 
 export default async function (args: { user_id?: string }) {
   const M = getCheckinModel();

package/tools/checkin-rank/handler.ts

@@ -1,4 +1,4 @@
-import { getCheckinModel } from '../../src/checkin.js';
+import { getCheckinModel } from '../../src/db-store.js';
 
 export default async function (args: { limit?: number }) {
   const M = getCheckinModel();

package/tools/stats-query/handler.ts

@@ -3,8 +3,8 @@ import {
   queryStats,
   weekStartStr,
   monthStartStr,
-} from '../../src/stats.js';
-import { todayStr } from '../../src/shared.js';
+} from '../../src/stats-lib.js';
+import { todayStr } from '../../src/shared-runtime.js';
 
 export default async function (args: { group_id?: string; period?: string }) {
   await flushStatsBuffer();

package/tools/stats-user/handler.ts

@@ -1,5 +1,6 @@
-import { flushStatsBuffer, getStatsModel, weekStartStr } from '../../src/stats.js';
-import { todayStr } from '../../src/shared.js';
+import { flushStatsBuffer, weekStartStr } from '../../src/stats-lib.js';
+import { getStatsModel } from '../../src/db-store.js';
+import { todayStr } from '../../src/shared-runtime.js';
 
 export default async function (args: { user_id: string }) {
   await flushStatsBuffer();

package/lib/agent/tools/group_announce.js

@@ -1,13 +0,0 @@
-import { defineTool } from '@zhin.js/agent/tools';
-import { z } from 'zod';
-export default defineTool({
-    description: '向群聊发送公告/通知消息',
-    inputSchema: z.object({ message: z.string().min(1) }),
-    keywords: ['群公告', '通知', 'announce'],
-    async execute({ message }) {
-        if (!message?.trim())
-            return '公告内容不能为空';
-        return `📢 群公告：\n${message}`;
-    },
-});
-//# sourceMappingURL=group_announce.js.map

package/lib/agent/tools/group_announce.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"group_announce.js","sourceRoot":"","sources":["../../../agent/tools/group_announce.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAe,UAAU,CAAsB;IAC7C,WAAW,EAAE,cAAc;IAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IACrD,QAAQ,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,UAAU,CAAC;IACnC,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE;QACvB,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE;YAAE,OAAO,UAAU,CAAC;QACxC,OAAO,YAAY,OAAO,EAAE,CAAC;IAC/B,CAAC;CACF,CAAC,CAAC"}