zhi
a6f2be44b7
Use Node.js built-in WebSocket to maintain a minimal Discord Gateway connection for the moderator bot, keeping it 'online' with a 'Watching Moderating' status. Handles heartbeat, reconnect, and resume. Also fix package-plugin.mjs to include moderator-presence.ts in dist.
822 lines
35 KiB
TypeScript
822 lines
35 KiB
TypeScript
import fs from "node:fs";
|
||
import path from "node:path";
|
||
import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
|
||
import { evaluateDecision, resolvePolicy, type ChannelPolicy, type Decision, type WhisperGateConfig } from "./rules.js";
|
||
import { checkTurn, advanceTurn, resetTurn, onNewMessage, onSpeakerDone, initTurnOrder, getTurnDebugInfo } from "./turn-manager.js";
|
||
import { startModeratorPresence, stopModeratorPresence } from "./moderator-presence.js";
|
||
|
||
type DiscordControlAction = "channel-private-create" | "channel-private-update" | "member-list";
|
||
|
||
type DecisionRecord = {
|
||
decision: Decision;
|
||
createdAt: number;
|
||
needsRestore?: boolean;
|
||
};
|
||
|
||
type PolicyState = {
|
||
filePath: string;
|
||
channelPolicies: Record<string, ChannelPolicy>;
|
||
};
|
||
|
||
type DebugConfig = {
|
||
enableDebugLogs?: boolean;
|
||
debugLogChannelIds?: string[];
|
||
};
|
||
|
||
const sessionDecision = new Map<string, DecisionRecord>();
|
||
const MAX_SESSION_DECISIONS = 2000;
|
||
const DECISION_TTL_MS = 5 * 60 * 1000;
|
||
function buildEndMarkerInstruction(endSymbols: string[], isGroupChat: boolean): string {
|
||
const symbols = endSymbols.length > 0 ? endSymbols.join("") : "🔚";
|
||
let instruction = `你的这次发言必须以${symbols}作为结尾。除非你的回复是 gateway 关键词（如 NO_REPLY、HEARTBEAT_OK），这些关键词不要加${symbols}。`;
|
||
if (isGroupChat) {
|
||
instruction += `\n\n群聊发言规则：如果这条消息与你无关、不需要你回应、或你没有有价值的补充，请主动回复 NO_REPLY。不要为了说话而说话。`;
|
||
}
|
||
return instruction;
|
||
}
|
||
|
||
const policyState: PolicyState = {
|
||
filePath: "",
|
||
channelPolicies: {},
|
||
};
|
||
|
||
function normalizeChannel(ctx: Record<string, unknown>): string {
|
||
const candidates = [ctx.commandSource, ctx.messageProvider, ctx.channelId, ctx.channel];
|
||
for (const c of candidates) {
|
||
if (typeof c === "string" && c.trim()) return c.trim().toLowerCase();
|
||
}
|
||
return "";
|
||
}
|
||
|
||
function normalizeSender(event: Record<string, unknown>, ctx: Record<string, unknown>): string | undefined {
|
||
const direct = [ctx.senderId, ctx.from, event.from];
|
||
for (const v of direct) {
|
||
if (typeof v === "string" && v.trim()) return v.trim();
|
||
}
|
||
|
||
const meta = (event.metadata || ctx.metadata) as Record<string, unknown> | undefined;
|
||
if (!meta) return undefined;
|
||
const metaCandidates = [meta.senderId, meta.sender_id, meta.userId, meta.user_id];
|
||
for (const v of metaCandidates) {
|
||
if (typeof v === "string" && v.trim()) return v.trim();
|
||
}
|
||
|
||
return undefined;
|
||
}
|
||
|
||
function extractUntrustedConversationInfo(text: string): Record<string, unknown> | undefined {
|
||
const marker = "Conversation info (untrusted metadata):";
|
||
const idx = text.indexOf(marker);
|
||
if (idx < 0) return undefined;
|
||
const tail = text.slice(idx + marker.length);
|
||
const m = tail.match(/```json\s*([\s\S]*?)\s*```/i);
|
||
if (!m) return undefined;
|
||
try {
|
||
const parsed = JSON.parse(m[1]);
|
||
return parsed && typeof parsed === "object" ? (parsed as Record<string, unknown>) : undefined;
|
||
} catch {
|
||
return undefined;
|
||
}
|
||
}
|
||
|
||
function deriveDecisionInputFromPrompt(
|
||
prompt: string,
|
||
messageProvider?: string,
|
||
): {
|
||
channel: string;
|
||
channelId?: string;
|
||
senderId?: string;
|
||
content: string;
|
||
conv: Record<string, unknown>;
|
||
} {
|
||
const conv = extractUntrustedConversationInfo(prompt) || {};
|
||
const channel = (messageProvider || "").toLowerCase();
|
||
const channelId =
|
||
(typeof conv.channel_id === "string" && conv.channel_id) ||
|
||
(typeof conv.chat_id === "string" && conv.chat_id.startsWith("channel:")
|
||
? conv.chat_id.slice("channel:".length)
|
||
: undefined);
|
||
const senderId =
|
||
(typeof conv.sender_id === "string" && conv.sender_id) ||
|
||
(typeof conv.sender === "string" && conv.sender) ||
|
||
undefined;
|
||
|
||
return { channel, channelId, senderId, content: prompt, conv };
|
||
}
|
||
|
||
function pruneDecisionMap(now = Date.now()) {
|
||
for (const [k, v] of sessionDecision.entries()) {
|
||
if (now - v.createdAt > DECISION_TTL_MS) sessionDecision.delete(k);
|
||
}
|
||
|
||
if (sessionDecision.size <= MAX_SESSION_DECISIONS) return;
|
||
const keys = sessionDecision.keys();
|
||
while (sessionDecision.size > MAX_SESSION_DECISIONS) {
|
||
const k = keys.next();
|
||
if (k.done) break;
|
||
sessionDecision.delete(k.value);
|
||
}
|
||
}
|
||
|
||
|
||
function getLivePluginConfig(api: OpenClawPluginApi, fallback: WhisperGateConfig): WhisperGateConfig {
|
||
const root = (api.config as Record<string, unknown>) || {};
|
||
const plugins = (root.plugins as Record<string, unknown>) || {};
|
||
const entries = (plugins.entries as Record<string, unknown>) || {};
|
||
const entry = (entries.whispergate as Record<string, unknown>) || {};
|
||
const cfg = (entry.config as Record<string, unknown>) || {};
|
||
if (Object.keys(cfg).length > 0) {
|
||
// Merge with defaults to ensure optional fields have values
|
||
return {
|
||
enableDiscordControlTool: true,
|
||
enableWhispergatePolicyTool: true,
|
||
discordControlApiBaseUrl: "http://127.0.0.1:8790",
|
||
enableDebugLogs: false,
|
||
debugLogChannelIds: [],
|
||
...cfg,
|
||
} as WhisperGateConfig;
|
||
}
|
||
return fallback;
|
||
}
|
||
|
||
function resolvePoliciesPath(api: OpenClawPluginApi, config: WhisperGateConfig): string {
|
||
return api.resolvePath(config.channelPoliciesFile || "~/.openclaw/whispergate-channel-policies.json");
|
||
}
|
||
|
||
function ensurePolicyStateLoaded(api: OpenClawPluginApi, config: WhisperGateConfig) {
|
||
if (policyState.filePath) return;
|
||
const filePath = resolvePoliciesPath(api, config);
|
||
policyState.filePath = filePath;
|
||
|
||
try {
|
||
if (!fs.existsSync(filePath)) {
|
||
fs.mkdirSync(path.dirname(filePath), { recursive: true });
|
||
fs.writeFileSync(filePath, "{}\n", "utf8");
|
||
policyState.channelPolicies = {};
|
||
return;
|
||
}
|
||
|
||
const raw = fs.readFileSync(filePath, "utf8");
|
||
const parsed = JSON.parse(raw) as Record<string, ChannelPolicy>;
|
||
policyState.channelPolicies = parsed && typeof parsed === "object" ? parsed : {};
|
||
} catch (err) {
|
||
api.logger.warn(`whispergate: failed init policy file ${filePath}: ${String(err)}`);
|
||
policyState.channelPolicies = {};
|
||
}
|
||
}
|
||
|
||
/** Resolve agentId → Discord accountId from config bindings */
|
||
function resolveAccountId(api: OpenClawPluginApi, agentId: string): string | undefined {
|
||
const root = (api.config as Record<string, unknown>) || {};
|
||
const bindings = root.bindings as Array<Record<string, unknown>> | undefined;
|
||
if (!Array.isArray(bindings)) return undefined;
|
||
for (const b of bindings) {
|
||
if (b.agentId === agentId) {
|
||
const match = b.match as Record<string, unknown> | undefined;
|
||
if (match?.channel === "discord" && typeof match.accountId === "string") {
|
||
return match.accountId;
|
||
}
|
||
}
|
||
}
|
||
return undefined;
|
||
}
|
||
|
||
/**
|
||
* Get all Discord bot accountIds from config bindings (excluding humanList-bound agents).
|
||
*/
|
||
function getAllBotAccountIds(api: OpenClawPluginApi): string[] {
|
||
const root = (api.config as Record<string, unknown>) || {};
|
||
const bindings = root.bindings as Array<Record<string, unknown>> | undefined;
|
||
if (!Array.isArray(bindings)) return [];
|
||
const ids: string[] = [];
|
||
for (const b of bindings) {
|
||
const match = b.match as Record<string, unknown> | undefined;
|
||
if (match?.channel === "discord" && typeof match.accountId === "string") {
|
||
ids.push(match.accountId);
|
||
}
|
||
}
|
||
return ids;
|
||
}
|
||
|
||
/**
|
||
* Ensure turn order is initialized for a channel.
|
||
* Uses all bot accounts from bindings as the turn order.
|
||
*/
|
||
function ensureTurnOrder(api: OpenClawPluginApi, channelId: string): void {
|
||
const botAccounts = getAllBotAccountIds(api);
|
||
if (botAccounts.length > 0) {
|
||
initTurnOrder(channelId, botAccounts);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* Build agent identity string for injection into group chat prompts.
|
||
*/
|
||
function buildAgentIdentity(api: OpenClawPluginApi, agentId: string): string | undefined {
|
||
const root = (api.config as Record<string, unknown>) || {};
|
||
const bindings = root.bindings as Array<Record<string, unknown>> | undefined;
|
||
const agents = ((root.agents as Record<string, unknown>)?.list as Array<Record<string, unknown>>) || [];
|
||
if (!Array.isArray(bindings)) return undefined;
|
||
|
||
// Find accountId for this agent
|
||
let accountId: string | undefined;
|
||
for (const b of bindings) {
|
||
if (b.agentId === agentId) {
|
||
const match = b.match as Record<string, unknown> | undefined;
|
||
if (match?.channel === "discord" && typeof match.accountId === "string") {
|
||
accountId = match.accountId;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
if (!accountId) return undefined;
|
||
|
||
// Find agent name
|
||
const agent = agents.find((a: Record<string, unknown>) => a.id === agentId);
|
||
const name = (agent?.name as string) || agentId;
|
||
|
||
// Find Discord bot user ID from account token (not available directly)
|
||
// We'll use accountId as the identifier
|
||
return `你是 ${name}（Discord 账号: ${accountId}）。`;
|
||
}
|
||
|
||
// --- Moderator bot helpers ---
|
||
|
||
/** Extract Discord user ID from a bot token (base64-encoded in first segment) */
|
||
function userIdFromToken(token: string): string | undefined {
|
||
try {
|
||
const segment = token.split(".")[0];
|
||
// Add padding
|
||
const padded = segment + "=".repeat((4 - (segment.length % 4)) % 4);
|
||
return Buffer.from(padded, "base64").toString("utf8");
|
||
} catch {
|
||
return undefined;
|
||
}
|
||
}
|
||
|
||
/** Resolve accountId → Discord user ID by reading the account's bot token from config */
|
||
function resolveDiscordUserId(api: OpenClawPluginApi, accountId: string): string | undefined {
|
||
const root = (api.config as Record<string, unknown>) || {};
|
||
const channels = (root.channels as Record<string, unknown>) || {};
|
||
const discord = (channels.discord as Record<string, unknown>) || {};
|
||
const accounts = (discord.accounts as Record<string, Record<string, unknown>>) || {};
|
||
const acct = accounts[accountId];
|
||
if (!acct?.token || typeof acct.token !== "string") return undefined;
|
||
return userIdFromToken(acct.token);
|
||
}
|
||
|
||
/** Get the moderator bot's Discord user ID from its token */
|
||
function getModeratorUserId(config: WhisperGateConfig): string | undefined {
|
||
if (!config.moderatorBotToken) return undefined;
|
||
return userIdFromToken(config.moderatorBotToken);
|
||
}
|
||
|
||
/** Send a message as the moderator bot via Discord REST API */
|
||
async function sendModeratorMessage(token: string, channelId: string, content: string, logger: { info: (msg: string) => void; warn: (msg: string) => void }): Promise<boolean> {
|
||
try {
|
||
const r = await fetch(`https://discord.com/api/v10/channels/${channelId}/messages`, {
|
||
method: "POST",
|
||
headers: {
|
||
"Authorization": `Bot ${token}`,
|
||
"Content-Type": "application/json",
|
||
},
|
||
body: JSON.stringify({ content }),
|
||
});
|
||
if (!r.ok) {
|
||
const text = await r.text();
|
||
logger.warn(`whispergate: moderator send failed (${r.status}): ${text}`);
|
||
return false;
|
||
}
|
||
logger.info(`whispergate: moderator message sent to channel=${channelId}`);
|
||
return true;
|
||
} catch (err) {
|
||
logger.warn(`whispergate: moderator send error: ${String(err)}`);
|
||
return false;
|
||
}
|
||
}
|
||
|
||
function persistPolicies(api: OpenClawPluginApi): void {
|
||
const filePath = policyState.filePath;
|
||
if (!filePath) throw new Error("policy file path not initialized");
|
||
const before = JSON.stringify(policyState.channelPolicies, null, 2) + "\n";
|
||
const tmp = `${filePath}.tmp`;
|
||
fs.mkdirSync(path.dirname(filePath), { recursive: true });
|
||
fs.writeFileSync(tmp, before, "utf8");
|
||
fs.renameSync(tmp, filePath);
|
||
api.logger.info(`whispergate: policy file persisted: ${filePath}`);
|
||
}
|
||
|
||
function pickDefined(input: Record<string, unknown>) {
|
||
const out: Record<string, unknown> = {};
|
||
for (const [k, v] of Object.entries(input)) {
|
||
if (v !== undefined) out[k] = v;
|
||
}
|
||
return out;
|
||
}
|
||
|
||
function shouldDebugLog(cfg: DebugConfig, channelId?: string): boolean {
|
||
if (!cfg.enableDebugLogs) return false;
|
||
const allow = Array.isArray(cfg.debugLogChannelIds) ? cfg.debugLogChannelIds : [];
|
||
if (allow.length === 0) return true;
|
||
if (!channelId) return true; // 允许打印，方便排查 channelId 为空的场景
|
||
return allow.includes(channelId);
|
||
}
|
||
|
||
function debugCtxSummary(ctx: Record<string, unknown>, event: Record<string, unknown>) {
|
||
const meta = ((ctx.metadata || event.metadata || {}) as Record<string, unknown>) || {};
|
||
return {
|
||
sessionKey: typeof ctx.sessionKey === "string" ? ctx.sessionKey : undefined,
|
||
commandSource: typeof ctx.commandSource === "string" ? ctx.commandSource : undefined,
|
||
messageProvider: typeof ctx.messageProvider === "string" ? ctx.messageProvider : undefined,
|
||
channel: typeof ctx.channel === "string" ? ctx.channel : undefined,
|
||
channelId: typeof ctx.channelId === "string" ? ctx.channelId : undefined,
|
||
senderId: typeof ctx.senderId === "string" ? ctx.senderId : undefined,
|
||
from: typeof ctx.from === "string" ? ctx.from : undefined,
|
||
metaSenderId:
|
||
typeof meta.senderId === "string"
|
||
? meta.senderId
|
||
: typeof meta.sender_id === "string"
|
||
? meta.sender_id
|
||
: undefined,
|
||
metaUserId:
|
||
typeof meta.userId === "string"
|
||
? meta.userId
|
||
: typeof meta.user_id === "string"
|
||
? meta.user_id
|
||
: undefined,
|
||
};
|
||
}
|
||
|
||
export default {
|
||
id: "whispergate",
|
||
name: "WhisperGate",
|
||
register(api: OpenClawPluginApi) {
|
||
// Merge pluginConfig with defaults (in case config is missing from openclaw.json)
|
||
const baseConfig = {
|
||
enableDiscordControlTool: true,
|
||
enableWhispergatePolicyTool: true,
|
||
discordControlApiBaseUrl: "http://127.0.0.1:8790",
|
||
...(api.pluginConfig || {}),
|
||
} as WhisperGateConfig & {
|
||
enableDiscordControlTool: boolean;
|
||
discordControlApiBaseUrl: string;
|
||
discordControlApiToken?: string;
|
||
discordControlCallerId?: string;
|
||
enableWhispergatePolicyTool: boolean;
|
||
};
|
||
|
||
const liveAtRegister = getLivePluginConfig(api, baseConfig as WhisperGateConfig);
|
||
ensurePolicyStateLoaded(api, liveAtRegister);
|
||
|
||
// Start moderator bot presence (keep it "online" on Discord)
|
||
if (liveAtRegister.moderatorBotToken) {
|
||
startModeratorPresence(liveAtRegister.moderatorBotToken, api.logger);
|
||
api.logger.info("whispergate: moderator bot presence starting");
|
||
}
|
||
|
||
api.registerTool(
|
||
{
|
||
name: "whispergate_tools",
|
||
description: "WhisperGate unified tool: Discord admin actions + in-memory policy management.",
|
||
parameters: {
|
||
type: "object",
|
||
additionalProperties: false,
|
||
properties: {
|
||
action: {
|
||
type: "string",
|
||
enum: ["channel-private-create", "channel-private-update", "member-list", "policy-get", "policy-set-channel", "policy-delete-channel", "turn-status", "turn-advance", "turn-reset"],
|
||
},
|
||
guildId: { type: "string" },
|
||
name: { type: "string" },
|
||
type: { type: "number" },
|
||
parentId: { type: "string" },
|
||
topic: { type: "string" },
|
||
position: { type: "number" },
|
||
nsfw: { type: "boolean" },
|
||
allowedUserIds: { type: "array", items: { type: "string" } },
|
||
allowedRoleIds: { type: "array", items: { type: "string" } },
|
||
allowMask: { type: "string" },
|
||
denyEveryoneMask: { type: "string" },
|
||
channelId: { type: "string" },
|
||
mode: { type: "string", enum: ["merge", "replace"] },
|
||
addUserIds: { type: "array", items: { type: "string" } },
|
||
addRoleIds: { type: "array", items: { type: "string" } },
|
||
removeTargetIds: { type: "array", items: { type: "string" } },
|
||
denyMask: { type: "string" },
|
||
limit: { type: "number" },
|
||
after: { type: "string" },
|
||
fields: { anyOf: [{ type: "string" }, { type: "array", items: { type: "string" } }] },
|
||
dryRun: { type: "boolean" },
|
||
listMode: { type: "string", enum: ["human-list", "agent-list"] },
|
||
humanList: { type: "array", items: { type: "string" } },
|
||
agentList: { type: "array", items: { type: "string" } },
|
||
endSymbols: { type: "array", items: { type: "string" } },
|
||
},
|
||
required: ["action"],
|
||
},
|
||
async execute(_id: string, params: Record<string, unknown>) {
|
||
const live = getLivePluginConfig(api, baseConfig as WhisperGateConfig) as WhisperGateConfig & {
|
||
discordControlApiBaseUrl?: string;
|
||
discordControlApiToken?: string;
|
||
discordControlCallerId?: string;
|
||
enableDiscordControlTool?: boolean;
|
||
enableWhispergatePolicyTool?: boolean;
|
||
};
|
||
ensurePolicyStateLoaded(api, live);
|
||
|
||
const action = String(params.action || "");
|
||
const discordActions = new Set(["channel-private-create", "channel-private-update", "member-list"]);
|
||
|
||
if (discordActions.has(action)) {
|
||
if (live.enableDiscordControlTool === false) {
|
||
return { content: [{ type: "text", text: "discord actions disabled by config" }], isError: true };
|
||
}
|
||
const baseUrl = (live.discordControlApiBaseUrl || "http://127.0.0.1:8790").replace(/\/$/, "");
|
||
const body = pickDefined({ ...params, action: action as DiscordControlAction });
|
||
const headers: Record<string, string> = { "Content-Type": "application/json" };
|
||
if (live.discordControlApiToken) headers.Authorization = `Bearer ${live.discordControlApiToken}`;
|
||
if (live.discordControlCallerId) headers["X-OpenClaw-Caller-Id"] = live.discordControlCallerId;
|
||
|
||
const r = await fetch(`${baseUrl}/v1/discord/action`, {
|
||
method: "POST",
|
||
headers,
|
||
body: JSON.stringify(body),
|
||
});
|
||
const text = await r.text();
|
||
if (!r.ok) {
|
||
return {
|
||
content: [{ type: "text", text: `whispergate_tools discord failed (${r.status}): ${text}` }],
|
||
isError: true,
|
||
};
|
||
}
|
||
return { content: [{ type: "text", text }] };
|
||
}
|
||
|
||
if (live.enableWhispergatePolicyTool === false) {
|
||
return { content: [{ type: "text", text: "policy actions disabled by config" }], isError: true };
|
||
}
|
||
|
||
if (action === "policy-get") {
|
||
return {
|
||
content: [{ type: "text", text: JSON.stringify({ file: policyState.filePath, policies: policyState.channelPolicies }, null, 2) }],
|
||
};
|
||
}
|
||
|
||
if (action === "policy-set-channel") {
|
||
const channelId = String(params.channelId || "").trim();
|
||
if (!channelId) return { content: [{ type: "text", text: "channelId is required" }], isError: true };
|
||
|
||
const prev = JSON.parse(JSON.stringify(policyState.channelPolicies));
|
||
try {
|
||
const next: ChannelPolicy = {
|
||
listMode: (params.listMode as "human-list" | "agent-list" | undefined) || undefined,
|
||
humanList: Array.isArray(params.humanList) ? (params.humanList as string[]) : undefined,
|
||
agentList: Array.isArray(params.agentList) ? (params.agentList as string[]) : undefined,
|
||
endSymbols: Array.isArray(params.endSymbols) ? (params.endSymbols as string[]) : undefined,
|
||
};
|
||
policyState.channelPolicies[channelId] = pickDefined(next as unknown as Record<string, unknown>) as ChannelPolicy;
|
||
persistPolicies(api);
|
||
return { content: [{ type: "text", text: JSON.stringify({ ok: true, channelId, policy: policyState.channelPolicies[channelId] }) }] };
|
||
} catch (err) {
|
||
policyState.channelPolicies = prev;
|
||
return { content: [{ type: "text", text: `persist failed: ${String(err)}` }], isError: true };
|
||
}
|
||
}
|
||
|
||
if (action === "policy-delete-channel") {
|
||
const channelId = String(params.channelId || "").trim();
|
||
if (!channelId) return { content: [{ type: "text", text: "channelId is required" }], isError: true };
|
||
const prev = JSON.parse(JSON.stringify(policyState.channelPolicies));
|
||
try {
|
||
delete policyState.channelPolicies[channelId];
|
||
persistPolicies(api);
|
||
return { content: [{ type: "text", text: JSON.stringify({ ok: true, channelId, deleted: true }) }] };
|
||
} catch (err) {
|
||
policyState.channelPolicies = prev;
|
||
return { content: [{ type: "text", text: `persist failed: ${String(err)}` }], isError: true };
|
||
}
|
||
}
|
||
|
||
if (action === "turn-status") {
|
||
const channelId = String(params.channelId || "").trim();
|
||
if (!channelId) return { content: [{ type: "text", text: "channelId is required" }], isError: true };
|
||
return { content: [{ type: "text", text: JSON.stringify(getTurnDebugInfo(channelId), null, 2) }] };
|
||
}
|
||
|
||
if (action === "turn-advance") {
|
||
const channelId = String(params.channelId || "").trim();
|
||
if (!channelId) return { content: [{ type: "text", text: "channelId is required" }], isError: true };
|
||
const next = advanceTurn(channelId);
|
||
return { content: [{ type: "text", text: JSON.stringify({ ok: true, channelId, nextSpeaker: next, ...getTurnDebugInfo(channelId) }) }] };
|
||
}
|
||
|
||
if (action === "turn-reset") {
|
||
const channelId = String(params.channelId || "").trim();
|
||
if (!channelId) return { content: [{ type: "text", text: "channelId is required" }], isError: true };
|
||
resetTurn(channelId);
|
||
return { content: [{ type: "text", text: JSON.stringify({ ok: true, channelId, ...getTurnDebugInfo(channelId) }) }] };
|
||
}
|
||
|
||
return { content: [{ type: "text", text: `unsupported action: ${action}` }], isError: true };
|
||
},
|
||
},
|
||
{ optional: false },
|
||
);
|
||
|
||
api.on("message_received", async (event, ctx) => {
|
||
try {
|
||
const c = (ctx || {}) as Record<string, unknown>;
|
||
const e = (event || {}) as Record<string, unknown>;
|
||
const preChannelId = typeof c.channelId === "string" ? c.channelId : undefined;
|
||
const livePre = getLivePluginConfig(api, baseConfig as WhisperGateConfig) as WhisperGateConfig & DebugConfig;
|
||
if (shouldDebugLog(livePre, preChannelId)) {
|
||
api.logger.info(`whispergate: debug message_received preflight ctx=${JSON.stringify(debugCtxSummary(c, e))}`);
|
||
}
|
||
|
||
// Turn management on message received
|
||
if (preChannelId) {
|
||
ensureTurnOrder(api, preChannelId);
|
||
const from = typeof (e as Record<string, unknown>).from === "string" ? (e as Record<string, unknown>).from as string : "";
|
||
|
||
// Ignore moderator bot messages — they don't affect turn state
|
||
const moderatorUserId = getModeratorUserId(livePre);
|
||
if (moderatorUserId && from === moderatorUserId) {
|
||
if (shouldDebugLog(livePre, preChannelId)) {
|
||
api.logger.info(`whispergate: ignoring moderator message in channel=${preChannelId}`);
|
||
}
|
||
// Don't call onNewMessage — moderator messages are transparent to turn logic
|
||
} else {
|
||
const humanList = livePre.humanList || livePre.bypassUserIds || [];
|
||
const isHuman = humanList.includes(from);
|
||
const senderAccountId = typeof c.accountId === "string" ? c.accountId : undefined;
|
||
onNewMessage(preChannelId, senderAccountId, isHuman);
|
||
if (shouldDebugLog(livePre, preChannelId)) {
|
||
api.logger.info(`whispergate: turn onNewMessage channel=${preChannelId} from=${from} isHuman=${isHuman} accountId=${senderAccountId ?? "unknown"}`);
|
||
}
|
||
}
|
||
}
|
||
} catch (err) {
|
||
api.logger.warn(`whispergate: message hook failed: ${String(err)}`);
|
||
}
|
||
});
|
||
|
||
api.on("before_model_resolve", async (event, ctx) => {
|
||
const key = ctx.sessionKey;
|
||
if (!key) return;
|
||
|
||
const live = getLivePluginConfig(api, baseConfig as WhisperGateConfig) as WhisperGateConfig & DebugConfig;
|
||
ensurePolicyStateLoaded(api, live);
|
||
|
||
const prompt = ((event as Record<string, unknown>).prompt as string) || "";
|
||
|
||
if (live.enableDebugLogs) {
|
||
api.logger.info(
|
||
`whispergate: DEBUG_BEFORE_MODEL_RESOLVE ctx=${JSON.stringify({ sessionKey: ctx.sessionKey, messageProvider: ctx.messageProvider, agentId: ctx.agentId })} ` +
|
||
`promptPreview=${prompt.slice(0, 300)}`,
|
||
);
|
||
}
|
||
|
||
const derived = deriveDecisionInputFromPrompt(prompt, ctx.messageProvider);
|
||
// Only proceed if: discord channel AND prompt contains untrusted metadata
|
||
const hasConvMarker = prompt.includes("Conversation info (untrusted metadata):");
|
||
if (live.discordOnly !== false && (!hasConvMarker || derived.channel !== "discord")) return;
|
||
|
||
let rec = sessionDecision.get(key);
|
||
if (!rec || Date.now() - rec.createdAt > DECISION_TTL_MS) {
|
||
if (rec) sessionDecision.delete(key);
|
||
const decision = evaluateDecision({
|
||
config: live,
|
||
channel: derived.channel,
|
||
channelId: derived.channelId,
|
||
channelPolicies: policyState.channelPolicies,
|
||
senderId: derived.senderId,
|
||
content: derived.content,
|
||
});
|
||
rec = { decision, createdAt: Date.now() };
|
||
sessionDecision.set(key, rec);
|
||
pruneDecisionMap();
|
||
if (shouldDebugLog(live, derived.channelId)) {
|
||
api.logger.info(
|
||
`whispergate: debug before_model_resolve recompute session=${key} ` +
|
||
`channel=${derived.channel} channelId=${derived.channelId ?? ""} senderId=${derived.senderId ?? ""} ` +
|
||
`convSenderId=${String((derived.conv as Record<string, unknown>).sender_id ?? "")} ` +
|
||
`convSender=${String((derived.conv as Record<string, unknown>).sender ?? "")} ` +
|
||
`convChannelId=${String((derived.conv as Record<string, unknown>).channel_id ?? "")} ` +
|
||
`decision=${decision.reason} shouldNoReply=${decision.shouldUseNoReply} shouldInject=${decision.shouldInjectEndMarkerPrompt}`,
|
||
);
|
||
}
|
||
}
|
||
|
||
// Turn-based check: if channel has turn order, only current speaker can respond
|
||
if (!rec.decision.shouldUseNoReply && derived.channelId) {
|
||
ensureTurnOrder(api, derived.channelId);
|
||
const accountId = resolveAccountId(api, ctx.agentId || "");
|
||
if (accountId) {
|
||
const turnCheck = checkTurn(derived.channelId, accountId);
|
||
if (!turnCheck.allowed) {
|
||
api.logger.info(
|
||
`whispergate: turn gate blocked session=${key} accountId=${accountId} currentSpeaker=${turnCheck.currentSpeaker} reason=${turnCheck.reason}`,
|
||
);
|
||
return {
|
||
providerOverride: live.noReplyProvider,
|
||
modelOverride: live.noReplyModel,
|
||
};
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!rec.decision.shouldUseNoReply) {
|
||
// 如果之前有 no-reply 执行过，现在不需要了，清除 override 恢复原模型
|
||
if (rec.needsRestore) {
|
||
sessionDecision.delete(key);
|
||
return {
|
||
providerOverride: undefined,
|
||
modelOverride: undefined,
|
||
};
|
||
}
|
||
return;
|
||
}
|
||
|
||
// 标记这次执行了 no-reply，下次需要恢复模型
|
||
rec.needsRestore = true;
|
||
sessionDecision.set(key, rec);
|
||
|
||
// 无论是否有缓存，只要 debug flag 开启就打印决策详情
|
||
if (live.enableDebugLogs) {
|
||
const prompt = ((event as Record<string, unknown>).prompt as string) || "";
|
||
const hasConvMarker = prompt.includes("Conversation info (untrusted metadata):");
|
||
api.logger.info(
|
||
`whispergate: DEBUG_NO_REPLY_TRIGGER session=${key} ` +
|
||
`channel=${derived.channel} channelId=${derived.channelId ?? ""} senderId=${derived.senderId ?? ""} ` +
|
||
`convSenderId=${String((derived.conv as Record<string, unknown>).sender_id ?? "")} ` +
|
||
`convSender=${String((derived.conv as Record<string, unknown>).sender ?? "")} ` +
|
||
`decision=${rec.decision.reason} ` +
|
||
`shouldNoReply=${rec.decision.shouldUseNoReply} shouldInject=${rec.decision.shouldInjectEndMarkerPrompt} ` +
|
||
`hasConvMarker=${hasConvMarker} promptLen=${prompt.length}`,
|
||
);
|
||
}
|
||
|
||
api.logger.info(
|
||
`whispergate: override model for session=${key}, provider=${live.noReplyProvider}, model=${live.noReplyModel}, reason=${rec.decision.reason}`,
|
||
);
|
||
|
||
return {
|
||
providerOverride: live.noReplyProvider,
|
||
modelOverride: live.noReplyModel,
|
||
};
|
||
});
|
||
|
||
api.on("before_prompt_build", async (event, ctx) => {
|
||
const key = ctx.sessionKey;
|
||
if (!key) return;
|
||
|
||
const live = getLivePluginConfig(api, baseConfig as WhisperGateConfig) as WhisperGateConfig & DebugConfig;
|
||
ensurePolicyStateLoaded(api, live);
|
||
|
||
let rec = sessionDecision.get(key);
|
||
if (!rec || Date.now() - rec.createdAt > DECISION_TTL_MS) {
|
||
if (rec) sessionDecision.delete(key);
|
||
|
||
const prompt = ((event as Record<string, unknown>).prompt as string) || "";
|
||
const derived = deriveDecisionInputFromPrompt(prompt, ctx.messageProvider);
|
||
|
||
const decision = evaluateDecision({
|
||
config: live,
|
||
channel: derived.channel,
|
||
channelId: derived.channelId,
|
||
channelPolicies: policyState.channelPolicies,
|
||
senderId: derived.senderId,
|
||
content: derived.content,
|
||
});
|
||
rec = { decision, createdAt: Date.now() };
|
||
if (shouldDebugLog(live, derived.channelId)) {
|
||
api.logger.info(
|
||
`whispergate: debug before_prompt_build recompute session=${key} ` +
|
||
`channel=${derived.channel} channelId=${derived.channelId ?? ""} senderId=${derived.senderId ?? ""} ` +
|
||
`convSenderId=${String((derived.conv as Record<string, unknown>).sender_id ?? "")} ` +
|
||
`convSender=${String((derived.conv as Record<string, unknown>).sender ?? "")} ` +
|
||
`convChannelId=${String((derived.conv as Record<string, unknown>).channel_id ?? "")} ` +
|
||
`decision=${decision.reason} shouldNoReply=${decision.shouldUseNoReply} shouldInject=${decision.shouldInjectEndMarkerPrompt}`,
|
||
);
|
||
}
|
||
}
|
||
|
||
sessionDecision.delete(key);
|
||
if (!rec.decision.shouldInjectEndMarkerPrompt) {
|
||
if (shouldDebugLog(live, undefined)) {
|
||
api.logger.info(
|
||
`whispergate: debug before_prompt_build session=${key} inject=false reason=${rec.decision.reason}`,
|
||
);
|
||
}
|
||
return;
|
||
}
|
||
|
||
// Resolve end symbols from config/policy for dynamic instruction
|
||
const prompt = ((event as Record<string, unknown>).prompt as string) || "";
|
||
const derived = deriveDecisionInputFromPrompt(prompt, ctx.messageProvider);
|
||
const policy = resolvePolicy(live, derived.channelId, policyState.channelPolicies);
|
||
const isGroupChat = derived.conv.is_group_chat === true || derived.conv.is_group_chat === "true";
|
||
const instruction = buildEndMarkerInstruction(policy.endSymbols, isGroupChat);
|
||
|
||
// Inject agent identity for group chats
|
||
let identity = "";
|
||
if (isGroupChat && ctx.agentId) {
|
||
const idStr = buildAgentIdentity(api, ctx.agentId);
|
||
if (idStr) identity = idStr + "\n\n";
|
||
}
|
||
|
||
api.logger.info(`whispergate: prepend end marker instruction for session=${key}, reason=${rec.decision.reason} isGroupChat=${isGroupChat}`);
|
||
return { prependContext: identity + instruction };
|
||
});
|
||
|
||
// Register slash commands for Discord
|
||
api.registerCommand({
|
||
name: "whispergate",
|
||
description: "WhisperGate 频道策略管理",
|
||
acceptsArgs: true,
|
||
handler: async (cmdCtx) => {
|
||
const args = cmdCtx.args || "";
|
||
const parts = args.trim().split(/\s+/);
|
||
const subCmd = parts[0] || "help";
|
||
|
||
if (subCmd === "help") {
|
||
return { text: `WhisperGate 命令:\n` +
|
||
`/whispergate status - 显示当前频道状态\n` +
|
||
`/whispergate turn-status - 显示轮流发言状态\n` +
|
||
`/whispergate turn-advance - 手动推进轮流\n` +
|
||
`/whispergate turn-reset - 重置轮流顺序` };
|
||
}
|
||
|
||
if (subCmd === "status") {
|
||
return { text: JSON.stringify({ policies: policyState.channelPolicies }, null, 2) };
|
||
}
|
||
|
||
if (subCmd === "turn-status") {
|
||
const channelId = cmdCtx.channelId;
|
||
if (!channelId) return { text: "无法获取频道ID", isError: true };
|
||
return { text: JSON.stringify(getTurnDebugInfo(channelId), null, 2) };
|
||
}
|
||
|
||
if (subCmd === "turn-advance") {
|
||
const channelId = cmdCtx.channelId;
|
||
if (!channelId) return { text: "无法获取频道ID", isError: true };
|
||
const next = advanceTurn(channelId);
|
||
return { text: JSON.stringify({ ok: true, nextSpeaker: next }) };
|
||
}
|
||
|
||
if (subCmd === "turn-reset") {
|
||
const channelId = cmdCtx.channelId;
|
||
if (!channelId) return { text: "无法获取频道ID", isError: true };
|
||
resetTurn(channelId);
|
||
return { text: JSON.stringify({ ok: true }) };
|
||
}
|
||
|
||
return { text: `未知子命令: ${subCmd}`, isError: true };
|
||
},
|
||
});
|
||
|
||
// Turn advance: when an agent sends a message, check if it signals end of turn
|
||
api.on("message_sent", async (event, ctx) => {
|
||
try {
|
||
const channelId = ctx.channelId;
|
||
const accountId = ctx.accountId;
|
||
const content = event.content || "";
|
||
|
||
if (!channelId || !accountId) return;
|
||
|
||
const live = getLivePluginConfig(api, baseConfig as WhisperGateConfig) as WhisperGateConfig & DebugConfig;
|
||
ensurePolicyStateLoaded(api, live);
|
||
const policy = resolvePolicy(live, channelId, policyState.channelPolicies);
|
||
|
||
const trimmed = content.trim();
|
||
const isEmpty = trimmed.length === 0;
|
||
const isNoReply = /^NO_REPLY$/i.test(trimmed) || /^HEARTBEAT_OK$/i.test(trimmed);
|
||
const lastChar = trimmed.length > 0 ? Array.from(trimmed).pop() || "" : "";
|
||
const hasEndSymbol = !!lastChar && policy.endSymbols.includes(lastChar);
|
||
const wasNoReply = isEmpty || isNoReply;
|
||
|
||
if (wasNoReply || hasEndSymbol) {
|
||
const nextSpeaker = onSpeakerDone(channelId, accountId, wasNoReply);
|
||
const trigger = wasNoReply ? (isEmpty ? "empty" : "no_reply_keyword") : "end_symbol";
|
||
api.logger.info(
|
||
`whispergate: turn onSpeakerDone channel=${channelId} from=${accountId} next=${nextSpeaker ?? "dormant"} trigger=${trigger}`,
|
||
);
|
||
// Moderator handoff: when current speaker NO_REPLY'd and there's a next speaker,
|
||
// send a handoff message via the moderator bot to trigger the next agent
|
||
if (wasNoReply && nextSpeaker && live.moderatorBotToken) {
|
||
const nextUserId = resolveDiscordUserId(api, nextSpeaker);
|
||
if (nextUserId) {
|
||
const handoffMsg = `轮到（<@${nextUserId}>）了，如果没有想说的请直接回复NO_REPLY`;
|
||
sendModeratorMessage(live.moderatorBotToken, channelId, handoffMsg, api.logger);
|
||
} else {
|
||
api.logger.warn(`whispergate: cannot resolve Discord userId for next speaker accountId=${nextSpeaker}`);
|
||
}
|
||
}
|
||
}
|
||
} catch (err) {
|
||
api.logger.warn(`whispergate: message_sent hook failed: ${String(err)}`);
|
||
}
|
||
});
|
||
},
|
||
};
|