369 lines
12 KiB
TypeScript
369 lines
12 KiB
TypeScript
import {
|
|
listDiscordAccountIds,
|
|
type ResolvedDiscordAccount,
|
|
resolveDefaultDiscordAccountId,
|
|
resolveDiscordAccount,
|
|
} from "../../discord/accounts.js";
|
|
import {
|
|
auditDiscordChannelPermissions,
|
|
collectDiscordAuditChannelIds,
|
|
} from "../../discord/audit.js";
|
|
import { probeDiscord } from "../../discord/probe.js";
|
|
import { sendMessageDiscord, sendPollDiscord } from "../../discord/send.js";
|
|
import { shouldLogVerbose } from "../../globals.js";
|
|
import {
|
|
DEFAULT_ACCOUNT_ID,
|
|
normalizeAccountId,
|
|
} from "../../routing/session-key.js";
|
|
import { getChatProviderMeta } from "../registry.js";
|
|
import { discordMessageActions } from "./actions/discord.js";
|
|
import {
|
|
deleteAccountFromConfigSection,
|
|
setAccountEnabledInConfigSection,
|
|
} from "./config-helpers.js";
|
|
import { resolveDiscordGroupRequireMention } from "./group-mentions.js";
|
|
import { formatPairingApproveHint } from "./helpers.js";
|
|
import { normalizeDiscordMessagingTarget } from "./normalize-target.js";
|
|
import { discordOnboardingAdapter } from "./onboarding/discord.js";
|
|
import { PAIRING_APPROVED_MESSAGE } from "./pairing-message.js";
|
|
import {
|
|
applyAccountNameToProviderSection,
|
|
migrateBaseNameToDefaultAccount,
|
|
} from "./setup-helpers.js";
|
|
import { collectDiscordStatusIssues } from "./status-issues/discord.js";
|
|
import type { ProviderPlugin } from "./types.js";
|
|
|
|
const meta = getChatProviderMeta("discord");
|
|
|
|
export const discordPlugin: ProviderPlugin<ResolvedDiscordAccount> = {
|
|
id: "discord",
|
|
meta: {
|
|
...meta,
|
|
},
|
|
onboarding: discordOnboardingAdapter,
|
|
pairing: {
|
|
idLabel: "discordUserId",
|
|
normalizeAllowEntry: (entry) => entry.replace(/^(discord|user):/i, ""),
|
|
notifyApproval: async ({ id }) => {
|
|
await sendMessageDiscord(`user:${id}`, PAIRING_APPROVED_MESSAGE);
|
|
},
|
|
},
|
|
capabilities: {
|
|
chatTypes: ["direct", "channel", "thread"],
|
|
polls: true,
|
|
reactions: true,
|
|
threads: true,
|
|
media: true,
|
|
nativeCommands: true,
|
|
},
|
|
streaming: {
|
|
blockStreamingCoalesceDefaults: { minChars: 1500, idleMs: 1000 },
|
|
},
|
|
reload: { configPrefixes: ["discord"] },
|
|
config: {
|
|
listAccountIds: (cfg) => listDiscordAccountIds(cfg),
|
|
resolveAccount: (cfg, accountId) =>
|
|
resolveDiscordAccount({ cfg, accountId }),
|
|
defaultAccountId: (cfg) => resolveDefaultDiscordAccountId(cfg),
|
|
setAccountEnabled: ({ cfg, accountId, enabled }) =>
|
|
setAccountEnabledInConfigSection({
|
|
cfg,
|
|
sectionKey: "discord",
|
|
accountId,
|
|
enabled,
|
|
allowTopLevel: true,
|
|
}),
|
|
deleteAccount: ({ cfg, accountId }) =>
|
|
deleteAccountFromConfigSection({
|
|
cfg,
|
|
sectionKey: "discord",
|
|
accountId,
|
|
clearBaseFields: ["token", "name"],
|
|
}),
|
|
isConfigured: (account) => Boolean(account.token?.trim()),
|
|
describeAccount: (account) => ({
|
|
accountId: account.accountId,
|
|
name: account.name,
|
|
enabled: account.enabled,
|
|
configured: Boolean(account.token?.trim()),
|
|
tokenSource: account.tokenSource,
|
|
}),
|
|
resolveAllowFrom: ({ cfg, accountId }) =>
|
|
(
|
|
resolveDiscordAccount({ cfg, accountId }).config.dm?.allowFrom ?? []
|
|
).map((entry) => String(entry)),
|
|
formatAllowFrom: ({ allowFrom }) =>
|
|
allowFrom
|
|
.map((entry) => String(entry).trim())
|
|
.filter(Boolean)
|
|
.map((entry) => entry.toLowerCase()),
|
|
},
|
|
security: {
|
|
resolveDmPolicy: ({ cfg, accountId, account }) => {
|
|
const resolvedAccountId =
|
|
accountId ?? account.accountId ?? DEFAULT_ACCOUNT_ID;
|
|
const useAccountPath = Boolean(
|
|
cfg.discord?.accounts?.[resolvedAccountId],
|
|
);
|
|
const allowFromPath = useAccountPath
|
|
? `discord.accounts.${resolvedAccountId}.dm.`
|
|
: "discord.dm.";
|
|
return {
|
|
policy: account.config.dm?.policy ?? "pairing",
|
|
allowFrom: account.config.dm?.allowFrom ?? [],
|
|
allowFromPath,
|
|
approveHint: formatPairingApproveHint("discord"),
|
|
normalizeEntry: (raw) =>
|
|
raw.replace(/^(discord|user):/i, "").replace(/^<@!?(\d+)>$/, "$1"),
|
|
};
|
|
},
|
|
collectWarnings: ({ account }) => {
|
|
const groupPolicy = account.config.groupPolicy ?? "allowlist";
|
|
if (groupPolicy !== "open") return [];
|
|
const channelAllowlistConfigured =
|
|
Boolean(account.config.guilds) &&
|
|
Object.keys(account.config.guilds ?? {}).length > 0;
|
|
if (channelAllowlistConfigured) {
|
|
return [
|
|
`- Discord guilds: groupPolicy="open" allows any channel not explicitly denied to trigger (mention-gated). Set discord.groupPolicy="allowlist" and configure discord.guilds.<id>.channels.`,
|
|
];
|
|
}
|
|
return [
|
|
`- Discord guilds: groupPolicy="open" with no guild/channel allowlist; any channel can trigger (mention-gated). Set discord.groupPolicy="allowlist" and configure discord.guilds.<id>.channels.`,
|
|
];
|
|
},
|
|
},
|
|
groups: {
|
|
resolveRequireMention: resolveDiscordGroupRequireMention,
|
|
},
|
|
mentions: {
|
|
stripPatterns: () => ["<@!?\\d+>"],
|
|
},
|
|
threading: {
|
|
resolveReplyToMode: ({ cfg }) => cfg.discord?.replyToMode ?? "off",
|
|
},
|
|
messaging: {
|
|
normalizeTarget: normalizeDiscordMessagingTarget,
|
|
},
|
|
actions: discordMessageActions,
|
|
setup: {
|
|
resolveAccountId: ({ accountId }) => normalizeAccountId(accountId),
|
|
applyAccountName: ({ cfg, accountId, name }) =>
|
|
applyAccountNameToProviderSection({
|
|
cfg,
|
|
providerKey: "discord",
|
|
accountId,
|
|
name,
|
|
}),
|
|
validateInput: ({ accountId, input }) => {
|
|
if (input.useEnv && accountId !== DEFAULT_ACCOUNT_ID) {
|
|
return "DISCORD_BOT_TOKEN can only be used for the default account.";
|
|
}
|
|
if (!input.useEnv && !input.token) {
|
|
return "Discord requires --token (or --use-env).";
|
|
}
|
|
return null;
|
|
},
|
|
applyAccountConfig: ({ cfg, accountId, input }) => {
|
|
const namedConfig = applyAccountNameToProviderSection({
|
|
cfg,
|
|
providerKey: "discord",
|
|
accountId,
|
|
name: input.name,
|
|
});
|
|
const next =
|
|
accountId !== DEFAULT_ACCOUNT_ID
|
|
? migrateBaseNameToDefaultAccount({
|
|
cfg: namedConfig,
|
|
providerKey: "discord",
|
|
})
|
|
: namedConfig;
|
|
if (accountId === DEFAULT_ACCOUNT_ID) {
|
|
return {
|
|
...next,
|
|
discord: {
|
|
...next.discord,
|
|
enabled: true,
|
|
...(input.useEnv ? {} : input.token ? { token: input.token } : {}),
|
|
},
|
|
};
|
|
}
|
|
return {
|
|
...next,
|
|
discord: {
|
|
...next.discord,
|
|
enabled: true,
|
|
accounts: {
|
|
...next.discord?.accounts,
|
|
[accountId]: {
|
|
...next.discord?.accounts?.[accountId],
|
|
enabled: true,
|
|
...(input.token ? { token: input.token } : {}),
|
|
},
|
|
},
|
|
},
|
|
};
|
|
},
|
|
},
|
|
outbound: {
|
|
deliveryMode: "direct",
|
|
chunker: null,
|
|
textChunkLimit: 2000,
|
|
pollMaxOptions: 10,
|
|
resolveTarget: ({ to }) => {
|
|
const trimmed = to?.trim();
|
|
if (!trimmed) {
|
|
return {
|
|
ok: false,
|
|
error: new Error(
|
|
"Delivering to Discord requires --to <channelId|user:ID|channel:ID>",
|
|
),
|
|
};
|
|
}
|
|
return { ok: true, to: trimmed };
|
|
},
|
|
sendText: async ({ to, text, accountId, deps, replyToId }) => {
|
|
const send = deps?.sendDiscord ?? sendMessageDiscord;
|
|
const result = await send(to, text, {
|
|
verbose: false,
|
|
replyTo: replyToId ?? undefined,
|
|
accountId: accountId ?? undefined,
|
|
});
|
|
return { provider: "discord", ...result };
|
|
},
|
|
sendMedia: async ({ to, text, mediaUrl, accountId, deps, replyToId }) => {
|
|
const send = deps?.sendDiscord ?? sendMessageDiscord;
|
|
const result = await send(to, text, {
|
|
verbose: false,
|
|
mediaUrl,
|
|
replyTo: replyToId ?? undefined,
|
|
accountId: accountId ?? undefined,
|
|
});
|
|
return { provider: "discord", ...result };
|
|
},
|
|
sendPoll: async ({ to, poll, accountId }) =>
|
|
await sendPollDiscord(to, poll, {
|
|
accountId: accountId ?? undefined,
|
|
}),
|
|
},
|
|
status: {
|
|
defaultRuntime: {
|
|
accountId: DEFAULT_ACCOUNT_ID,
|
|
running: false,
|
|
lastStartAt: null,
|
|
lastStopAt: null,
|
|
lastError: null,
|
|
},
|
|
collectStatusIssues: collectDiscordStatusIssues,
|
|
buildProviderSummary: ({ snapshot }) => ({
|
|
configured: snapshot.configured ?? false,
|
|
tokenSource: snapshot.tokenSource ?? "none",
|
|
running: snapshot.running ?? false,
|
|
lastStartAt: snapshot.lastStartAt ?? null,
|
|
lastStopAt: snapshot.lastStopAt ?? null,
|
|
lastError: snapshot.lastError ?? null,
|
|
probe: snapshot.probe,
|
|
lastProbeAt: snapshot.lastProbeAt ?? null,
|
|
}),
|
|
probeAccount: async ({ account, timeoutMs }) =>
|
|
probeDiscord(account.token, timeoutMs, { includeApplication: true }),
|
|
auditAccount: async ({ account, timeoutMs, cfg }) => {
|
|
const { channelIds, unresolvedChannels } = collectDiscordAuditChannelIds({
|
|
cfg,
|
|
accountId: account.accountId,
|
|
});
|
|
if (!channelIds.length && unresolvedChannels === 0) return undefined;
|
|
const botToken = account.token?.trim();
|
|
if (!botToken) {
|
|
return {
|
|
ok: unresolvedChannels === 0,
|
|
checkedChannels: 0,
|
|
unresolvedChannels,
|
|
channels: [],
|
|
elapsedMs: 0,
|
|
};
|
|
}
|
|
const audit = await auditDiscordChannelPermissions({
|
|
token: botToken,
|
|
accountId: account.accountId,
|
|
channelIds,
|
|
timeoutMs,
|
|
});
|
|
return { ...audit, unresolvedChannels };
|
|
},
|
|
buildAccountSnapshot: ({ account, runtime, probe, audit }) => {
|
|
const configured = Boolean(account.token?.trim());
|
|
const app =
|
|
runtime?.application ??
|
|
(probe as { application?: unknown })?.application;
|
|
const bot = runtime?.bot ?? (probe as { bot?: unknown })?.bot;
|
|
return {
|
|
accountId: account.accountId,
|
|
name: account.name,
|
|
enabled: account.enabled,
|
|
configured,
|
|
tokenSource: account.tokenSource,
|
|
running: runtime?.running ?? false,
|
|
lastStartAt: runtime?.lastStartAt ?? null,
|
|
lastStopAt: runtime?.lastStopAt ?? null,
|
|
lastError: runtime?.lastError ?? null,
|
|
application: app ?? undefined,
|
|
bot: bot ?? undefined,
|
|
probe,
|
|
audit,
|
|
lastInboundAt: runtime?.lastInboundAt ?? null,
|
|
lastOutboundAt: runtime?.lastOutboundAt ?? null,
|
|
};
|
|
},
|
|
},
|
|
gateway: {
|
|
startAccount: async (ctx) => {
|
|
const account = ctx.account;
|
|
const token = account.token.trim();
|
|
let discordBotLabel = "";
|
|
try {
|
|
const probe = await probeDiscord(token, 2500, {
|
|
includeApplication: true,
|
|
});
|
|
const username = probe.ok ? probe.bot?.username?.trim() : null;
|
|
if (username) discordBotLabel = ` (@${username})`;
|
|
ctx.setStatus({
|
|
accountId: account.accountId,
|
|
bot: probe.bot,
|
|
application: probe.application,
|
|
});
|
|
const messageContent = probe.application?.intents?.messageContent;
|
|
if (messageContent === "disabled") {
|
|
ctx.log?.warn(
|
|
`[${account.accountId}] Discord Message Content Intent is disabled; bot may not respond to channel messages. Enable it in Discord Dev Portal (Bot → Privileged Gateway Intents) or require mentions.`,
|
|
);
|
|
} else if (messageContent === "limited") {
|
|
ctx.log?.info(
|
|
`[${account.accountId}] Discord Message Content Intent is limited; bots under 100 servers can use it without verification.`,
|
|
);
|
|
}
|
|
} catch (err) {
|
|
if (shouldLogVerbose()) {
|
|
ctx.log?.debug?.(
|
|
`[${account.accountId}] bot probe failed: ${String(err)}`,
|
|
);
|
|
}
|
|
}
|
|
ctx.log?.info(
|
|
`[${account.accountId}] starting provider${discordBotLabel}`,
|
|
);
|
|
// Lazy import: the monitor pulls the reply pipeline; avoid ESM init cycles.
|
|
const { monitorDiscordProvider } = await import("../../discord/index.js");
|
|
return monitorDiscordProvider({
|
|
token,
|
|
accountId: account.accountId,
|
|
config: ctx.cfg,
|
|
runtime: ctx.runtime,
|
|
abortSignal: ctx.abortSignal,
|
|
mediaMaxMb: account.config.mediaMaxMb,
|
|
historyLimit: account.config.historyLimit,
|
|
});
|
|
},
|
|
},
|
|
};
|