feat(qqbot): 定时提醒技能与主动消息系统
**新增提醒技能** - 新增 skills/qqbot-cron/SKILL.md 定时提醒技能定义 - 支持一次性提醒（--at）和周期性提醒（--cron） - 支持设置、查询、取消提醒操作 **主动消息系统** - 新增 src/proactive.ts 主动消息发送模块 - 新增 src/known-users.ts 已知用户管理 - 新增 src/session-store.ts 会话存储 - 支持主动向用户/群组发送消息 **工具脚本** - 新增 scripts/proactive-api-server.ts 主动消息API服务
This commit is contained in:
rianli
358
src/known-users.ts
Normal file
358
src/known-users.ts
Normal file
@@ -0,0 +1,358 @@
|
||||
/**
|
||||
* 已知用户存储
|
||||
* 记录与机器人交互过的所有用户
|
||||
* 支持主动消息和批量通知功能
|
||||
*/
|
||||
|
||||
import fs from "node:fs";
|
||||
import path from "node:path";
|
||||
|
||||
// 已知用户信息接口
|
||||
export interface KnownUser {
|
||||
/** 用户 openid（唯一标识） */
|
||||
openid: string;
|
||||
/** 消息类型：私聊用户 / 群组 */
|
||||
type: "c2c" | "group";
|
||||
/** 用户昵称（如有） */
|
||||
nickname?: string;
|
||||
/** 群组 openid（如果是群消息） */
|
||||
groupOpenid?: string;
|
||||
/** 关联的机器人账户 ID */
|
||||
accountId: string;
|
||||
/** 首次交互时间戳 */
|
||||
firstSeenAt: number;
|
||||
/** 最后交互时间戳 */
|
||||
lastSeenAt: number;
|
||||
/** 交互次数 */
|
||||
interactionCount: number;
|
||||
}
|
||||
|
||||
// 存储文件路径
|
||||
const KNOWN_USERS_DIR = path.join(
|
||||
process.env.HOME || "/tmp",
|
||||
"clawd",
|
||||
"qqbot-data"
|
||||
);
|
||||
|
||||
const KNOWN_USERS_FILE = path.join(KNOWN_USERS_DIR, "known-users.json");
|
||||
|
||||
// 内存缓存
|
||||
let usersCache: Map<string, KnownUser> | null = null;
|
||||
|
||||
// 写入节流配置
|
||||
const SAVE_THROTTLE_MS = 5000; // 5秒写入一次
|
||||
let saveTimer: ReturnType<typeof setTimeout> | null = null;
|
||||
let isDirty = false;
|
||||
|
||||
/**
|
||||
* 确保目录存在
|
||||
*/
|
||||
function ensureDir(): void {
|
||||
if (!fs.existsSync(KNOWN_USERS_DIR)) {
|
||||
fs.mkdirSync(KNOWN_USERS_DIR, { recursive: true });
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件加载用户数据到缓存
|
||||
*/
|
||||
function loadUsersFromFile(): Map<string, KnownUser> {
|
||||
if (usersCache !== null) {
|
||||
return usersCache;
|
||||
}
|
||||
|
||||
usersCache = new Map();
|
||||
|
||||
try {
|
||||
if (fs.existsSync(KNOWN_USERS_FILE)) {
|
||||
const data = fs.readFileSync(KNOWN_USERS_FILE, "utf-8");
|
||||
const users = JSON.parse(data) as KnownUser[];
|
||||
|
||||
for (const user of users) {
|
||||
// 使用复合键：accountId + type + openid（群组还要加 groupOpenid）
|
||||
const key = makeUserKey(user);
|
||||
usersCache.set(key, user);
|
||||
}
|
||||
|
||||
console.log(`[known-users] Loaded ${usersCache.size} users from file`);
|
||||
}
|
||||
} catch (err) {
|
||||
console.error(`[known-users] Failed to load users: ${err}`);
|
||||
usersCache = new Map();
|
||||
}
|
||||
|
||||
return usersCache;
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存用户数据到文件（节流版本）
|
||||
*/
|
||||
function saveUsersToFile(): void {
|
||||
if (!isDirty) return;
|
||||
|
||||
if (saveTimer) {
|
||||
return; // 已有定时器在等待
|
||||
}
|
||||
|
||||
saveTimer = setTimeout(() => {
|
||||
saveTimer = null;
|
||||
doSaveUsersToFile();
|
||||
}, SAVE_THROTTLE_MS);
|
||||
}
|
||||
|
||||
/**
|
||||
* 实际执行保存
|
||||
*/
|
||||
function doSaveUsersToFile(): void {
|
||||
if (!usersCache || !isDirty) return;
|
||||
|
||||
try {
|
||||
ensureDir();
|
||||
const users = Array.from(usersCache.values());
|
||||
fs.writeFileSync(KNOWN_USERS_FILE, JSON.stringify(users, null, 2), "utf-8");
|
||||
isDirty = false;
|
||||
console.log(`[known-users] Saved ${users.length} users to file`);
|
||||
} catch (err) {
|
||||
console.error(`[known-users] Failed to save users: ${err}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 强制立即保存（用于进程退出前）
|
||||
*/
|
||||
export function flushKnownUsers(): void {
|
||||
if (saveTimer) {
|
||||
clearTimeout(saveTimer);
|
||||
saveTimer = null;
|
||||
}
|
||||
doSaveUsersToFile();
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成用户唯一键
|
||||
*/
|
||||
function makeUserKey(user: Partial<KnownUser>): string {
|
||||
const base = `${user.accountId}:${user.type}:${user.openid}`;
|
||||
if (user.type === "group" && user.groupOpenid) {
|
||||
return `${base}:${user.groupOpenid}`;
|
||||
}
|
||||
return base;
|
||||
}
|
||||
|
||||
/**
|
||||
* 记录已知用户（收到消息时调用）
|
||||
* @param user 用户信息（部分字段）
|
||||
*/
|
||||
export function recordKnownUser(user: {
|
||||
openid: string;
|
||||
type: "c2c" | "group";
|
||||
nickname?: string;
|
||||
groupOpenid?: string;
|
||||
accountId: string;
|
||||
}): void {
|
||||
const cache = loadUsersFromFile();
|
||||
const key = makeUserKey(user);
|
||||
const now = Date.now();
|
||||
|
||||
const existing = cache.get(key);
|
||||
|
||||
if (existing) {
|
||||
// 更新已存在的用户
|
||||
existing.lastSeenAt = now;
|
||||
existing.interactionCount++;
|
||||
if (user.nickname && user.nickname !== existing.nickname) {
|
||||
existing.nickname = user.nickname;
|
||||
}
|
||||
console.log(`[known-users] Updated user ${user.openid}, interactions: ${existing.interactionCount}`);
|
||||
} else {
|
||||
// 新用户
|
||||
const newUser: KnownUser = {
|
||||
openid: user.openid,
|
||||
type: user.type,
|
||||
nickname: user.nickname,
|
||||
groupOpenid: user.groupOpenid,
|
||||
accountId: user.accountId,
|
||||
firstSeenAt: now,
|
||||
lastSeenAt: now,
|
||||
interactionCount: 1,
|
||||
};
|
||||
cache.set(key, newUser);
|
||||
console.log(`[known-users] New user recorded: ${user.openid} (${user.type})`);
|
||||
}
|
||||
|
||||
isDirty = true;
|
||||
saveUsersToFile();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取单个用户信息
|
||||
* @param accountId 机器人账户 ID
|
||||
* @param openid 用户 openid
|
||||
* @param type 消息类型
|
||||
* @param groupOpenid 群组 openid（可选）
|
||||
*/
|
||||
export function getKnownUser(
|
||||
accountId: string,
|
||||
openid: string,
|
||||
type: "c2c" | "group" = "c2c",
|
||||
groupOpenid?: string
|
||||
): KnownUser | undefined {
|
||||
const cache = loadUsersFromFile();
|
||||
const key = makeUserKey({ accountId, openid, type, groupOpenid });
|
||||
return cache.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 列出所有已知用户
|
||||
* @param options 筛选选项
|
||||
*/
|
||||
export function listKnownUsers(options?: {
|
||||
/** 筛选特定机器人账户的用户 */
|
||||
accountId?: string;
|
||||
/** 筛选消息类型 */
|
||||
type?: "c2c" | "group";
|
||||
/** 最近活跃时间（毫秒，如 86400000 表示最近 24 小时） */
|
||||
activeWithin?: number;
|
||||
/** 返回数量限制 */
|
||||
limit?: number;
|
||||
/** 排序方式 */
|
||||
sortBy?: "lastSeenAt" | "firstSeenAt" | "interactionCount";
|
||||
/** 排序方向 */
|
||||
sortOrder?: "asc" | "desc";
|
||||
}): KnownUser[] {
|
||||
const cache = loadUsersFromFile();
|
||||
let users = Array.from(cache.values());
|
||||
|
||||
// 筛选
|
||||
if (options?.accountId) {
|
||||
users = users.filter(u => u.accountId === options.accountId);
|
||||
}
|
||||
if (options?.type) {
|
||||
users = users.filter(u => u.type === options.type);
|
||||
}
|
||||
if (options?.activeWithin) {
|
||||
const cutoff = Date.now() - options.activeWithin;
|
||||
users = users.filter(u => u.lastSeenAt >= cutoff);
|
||||
}
|
||||
|
||||
// 排序
|
||||
const sortBy = options?.sortBy ?? "lastSeenAt";
|
||||
const sortOrder = options?.sortOrder ?? "desc";
|
||||
users.sort((a, b) => {
|
||||
const aVal = a[sortBy] ?? 0;
|
||||
const bVal = b[sortBy] ?? 0;
|
||||
return sortOrder === "asc" ? aVal - bVal : bVal - aVal;
|
||||
});
|
||||
|
||||
// 限制数量
|
||||
if (options?.limit && options.limit > 0) {
|
||||
users = users.slice(0, options.limit);
|
||||
}
|
||||
|
||||
return users;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户统计信息
|
||||
* @param accountId 机器人账户 ID（可选，不传则返回所有账户的统计）
|
||||
*/
|
||||
export function getKnownUsersStats(accountId?: string): {
|
||||
totalUsers: number;
|
||||
c2cUsers: number;
|
||||
groupUsers: number;
|
||||
activeIn24h: number;
|
||||
activeIn7d: number;
|
||||
} {
|
||||
let users = listKnownUsers({ accountId });
|
||||
|
||||
const now = Date.now();
|
||||
const day = 24 * 60 * 60 * 1000;
|
||||
|
||||
return {
|
||||
totalUsers: users.length,
|
||||
c2cUsers: users.filter(u => u.type === "c2c").length,
|
||||
groupUsers: users.filter(u => u.type === "group").length,
|
||||
activeIn24h: users.filter(u => now - u.lastSeenAt < day).length,
|
||||
activeIn7d: users.filter(u => now - u.lastSeenAt < 7 * day).length,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除用户记录
|
||||
* @param accountId 机器人账户 ID
|
||||
* @param openid 用户 openid
|
||||
* @param type 消息类型
|
||||
* @param groupOpenid 群组 openid（可选）
|
||||
*/
|
||||
export function removeKnownUser(
|
||||
accountId: string,
|
||||
openid: string,
|
||||
type: "c2c" | "group" = "c2c",
|
||||
groupOpenid?: string
|
||||
): boolean {
|
||||
const cache = loadUsersFromFile();
|
||||
const key = makeUserKey({ accountId, openid, type, groupOpenid });
|
||||
|
||||
if (cache.has(key)) {
|
||||
cache.delete(key);
|
||||
isDirty = true;
|
||||
saveUsersToFile();
|
||||
console.log(`[known-users] Removed user ${openid}`);
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 清除所有用户记录
|
||||
* @param accountId 机器人账户 ID（可选，不传则清除所有）
|
||||
*/
|
||||
export function clearKnownUsers(accountId?: string): number {
|
||||
const cache = loadUsersFromFile();
|
||||
let count = 0;
|
||||
|
||||
if (accountId) {
|
||||
// 只清除指定账户的用户
|
||||
for (const [key, user] of cache.entries()) {
|
||||
if (user.accountId === accountId) {
|
||||
cache.delete(key);
|
||||
count++;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// 清除所有
|
||||
count = cache.size;
|
||||
cache.clear();
|
||||
}
|
||||
|
||||
if (count > 0) {
|
||||
isDirty = true;
|
||||
doSaveUsersToFile(); // 立即保存
|
||||
console.log(`[known-users] Cleared ${count} users`);
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户的所有群组（某用户在哪些群里交互过）
|
||||
* @param accountId 机器人账户 ID
|
||||
* @param openid 用户 openid
|
||||
*/
|
||||
export function getUserGroups(accountId: string, openid: string): string[] {
|
||||
const users = listKnownUsers({ accountId, type: "group" });
|
||||
return users
|
||||
.filter(u => u.openid === openid && u.groupOpenid)
|
||||
.map(u => u.groupOpenid!);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取群组的所有成员
|
||||
* @param accountId 机器人账户 ID
|
||||
* @param groupOpenid 群组 openid
|
||||
*/
|
||||
export function getGroupMembers(accountId: string, groupOpenid: string): KnownUser[] {
|
||||
return listKnownUsers({ accountId, type: "group" })
|
||||
.filter(u => u.groupOpenid === groupOpenid);
|
||||
}
|
||||
Reference in New Issue
Block a user