feat: add allowedChannelIds for Discord channel filtering (#29)

* feat: add allowedChannelIds config for Discord channel filtering

- Add allowedChannelIds to DiscordImConfig (config.jsonl)
- Propagate through runtime env as ALLOWED_CHANNEL_IDS
- Filter messages and interactions in Discord index.ts
- Thread messages check parentId against allowlist
- Empty list = no filtering (backward compatible)
- 5 new tests covering allow/deny/thread/empty scenarios

* chore: bump patch version (1.1.1 / 1.2.1)

Author: Doctor-wu

apps/agent-inbox/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@doctorwu/agent-inbox",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "type": "module",
   "main": "dist/index.mjs",
   "types": "dist/index.d.mts",

apps/agent-inbox/src/config.ts

@@ -13,6 +13,7 @@ export type DiscordImConfig = {
   token?: string;
   clientId?: string;
   guildIds?: string[];
+  allowedChannelIds?: string[];
 };
 
 export type FeishuImConfig = {
@@ -57,7 +58,7 @@ export type AvailableIm =
   | {
     id: 'discord';
     note?: string;
-    config: Required<Pick<DiscordImConfig, 'token' | 'clientId'>> & Pick<DiscordImConfig, 'guildIds'>;
+    config: Required<Pick<DiscordImConfig, 'token' | 'clientId'>> & Pick<DiscordImConfig, 'guildIds' | 'allowedChannelIds'>;
   }
   | {
     id: 'feishu';
@@ -151,6 +152,7 @@ function normalizeDiscordImRecord(value: Record<string, unknown>): DiscordImReco
       token: asString(config.token),
       clientId: asString(config.clientId),
       guildIds: asStringList(config.guildIds),
+      allowedChannelIds: asStringList(config.allowedChannelIds),
     },
   };
 }
@@ -292,6 +294,7 @@ export function resolveAvailableIms(records: AppConfigRecord[]): AvailableIm[] {
           token: record.config.token,
           clientId: record.config.clientId,
           guildIds: record.config.guildIds,
+          allowedChannelIds: record.config.allowedChannelIds,
         },
       }];
     }

apps/agent-inbox/src/runtime.ts

@@ -35,6 +35,7 @@ export function applyRuntimeEnvironment(
   delete process.env['DISCORD_TOKEN'];
   delete process.env['DISCORD_CLIENT_ID'];
   delete process.env['GUILD_IDS'];
+  delete process.env['ALLOWED_CHANNEL_IDS'];
   delete process.env['FEISHU_APP_ID'];
   delete process.env['FEISHU_APP_SECRET'];
   delete process.env['FEISHU_VERIFICATION_TOKEN'];
@@ -46,6 +47,7 @@ export function applyRuntimeEnvironment(
     process.env['DISCORD_TOKEN'] = selectedIm.config.token;
     process.env['DISCORD_CLIENT_ID'] = selectedIm.config.clientId;
     setOptionalEnv('GUILD_IDS', selectedIm.config.guildIds?.join(','));
+    setOptionalEnv('ALLOWED_CHANNEL_IDS', selectedIm.config.allowedChannelIds?.join(','));
     return;
   }
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-im-relay-workspace",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "private": true,
   "description": "Inbox-first IM launcher for local AI agents",
   "type": "module",

packages/core/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-im-relay/core",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "type": "module",
   "main": "dist/index.mjs",
   "types": "dist/index.d.mts",

packages/discord/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-im-relay/discord",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "type": "module",
   "main": "dist/index.mjs",
   "scripts": {

packages/discord/src/__tests__/index.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 const { clientMock, restMock } = vi.hoisted(() => ({
   clientMock: {
@@ -127,6 +127,7 @@ vi.mock('../commands/thread-setup.js', () => ({
 
 import { handleDiscordMessageCreate } from '../index.js';
 import { handleSkillAutocomplete } from '../commands/skill.js';
+import { config as discordConfig } from '../config.js';
 import {
   applyMessageControlDirectives,
   getAvailableBackendCapabilities,
@@ -469,3 +470,119 @@ describe('handleDiscordMessageCreate', () => {
     });
   });
 });
+
+describe('allowedChannelIds filter', () => {
+  afterEach(() => {
+    discordConfig.allowedChannelIds = [];
+  });
+
+  it('drops messages from channels not in the allowlist', async () => {
+    discordConfig.allowedChannelIds = ['allowed-channel'];
+
+    const message = createBaseMessage();
+    message.channel.id = 'other-channel';
+
+    await handleDiscordMessageCreate(message, {
+      botUser: { id: 'relay-bot' },
+      hasOpenStickyThreadSession: () => false,
+      runThreadConversation: vi.fn(),
+      ensureMentionThread: vi.fn(),
+      promptThreadSetup: vi.fn(),
+      applySetupResult: vi.fn(),
+    });
+
+    expect(message.react).not.toHaveBeenCalled();
+  });
+
+  it('accepts messages from allowed channels', async () => {
+    discordConfig.allowedChannelIds = ['allowed-channel'];
+
+    const message = createBaseMessage();
+    message.channel.id = 'allowed-channel';
+
+    const ensureMentionThread = vi.fn(async () => ({
+      id: 'thread-1',
+      send: vi.fn(async () => undefined),
+    }));
+
+    await handleDiscordMessageCreate(message, {
+      botUser: { id: 'relay-bot' },
+      hasOpenStickyThreadSession: () => false,
+      runThreadConversation: vi.fn(async () => true),
+      ensureMentionThread,
+      promptThreadSetup: vi.fn(async () => ({ kind: 'skip' })),
+      applySetupResult: vi.fn(),
+    });
+
+    expect(message.react).toHaveBeenCalled();
+  });
+
+  it('accepts thread messages whose parent is in the allowlist', async () => {
+    discordConfig.allowedChannelIds = ['allowed-channel'];
+
+    const message = createBaseMessage();
+    message.channel = {
+      id: 'thread-in-allowed',
+      parentId: 'allowed-channel',
+      isThread: () => true,
+      send: vi.fn(async () => undefined),
+    };
+
+    await handleDiscordMessageCreate(message, {
+      botUser: { id: 'relay-bot' },
+      hasOpenStickyThreadSession: () => true,
+      runThreadConversation: vi.fn(async () => true),
+      ensureMentionThread: vi.fn(),
+      promptThreadSetup: vi.fn(),
+      applySetupResult: vi.fn(),
+    });
+
+    expect(message.react).toHaveBeenCalled();
+  });
+
+  it('drops thread messages whose parent is not in the allowlist', async () => {
+    discordConfig.allowedChannelIds = ['allowed-channel'];
+
+    const message = createBaseMessage();
+    message.channel = {
+      id: 'thread-in-other',
+      parentId: 'other-channel',
+      isThread: () => true,
+      send: vi.fn(async () => undefined),
+    };
+
+    await handleDiscordMessageCreate(message, {
+      botUser: { id: 'relay-bot' },
+      hasOpenStickyThreadSession: () => true,
+      runThreadConversation: vi.fn(),
+      ensureMentionThread: vi.fn(),
+      promptThreadSetup: vi.fn(),
+      applySetupResult: vi.fn(),
+    });
+
+    expect(message.react).not.toHaveBeenCalled();
+  });
+
+  it('allows all channels when allowedChannelIds is empty', async () => {
+    discordConfig.allowedChannelIds = [];
+
+    const message = createBaseMessage();
+    message.channel.id = 'any-channel';
+
+    const ensureMentionThread = vi.fn(async () => ({
+      id: 'thread-1',
+      send: vi.fn(async () => undefined),
+    }));
+
+    await handleDiscordMessageCreate(message, {
+      botUser: { id: 'relay-bot' },
+      hasOpenStickyThreadSession: () => false,
+      runThreadConversation: vi.fn(async () => true),
+      ensureMentionThread,
+      promptThreadSetup: vi.fn(async () => ({ kind: 'skip' })),
+      applySetupResult: vi.fn(),
+    });
+
+    expect(message.react).toHaveBeenCalled();
+  });
+});

packages/discord/src/config.ts

@@ -29,6 +29,9 @@ export const config = {
   guildIds: process.env['GUILD_IDS']
     ? process.env['GUILD_IDS'].split(',').map((id) => id.trim()).filter(Boolean)
     : [],
+  allowedChannelIds: process.env['ALLOWED_CHANNEL_IDS']
+    ? process.env['ALLOWED_CHANNEL_IDS'].split(',').map((id) => id.trim()).filter(Boolean)
+    : [],
   streamUpdateIntervalMs: numberEnv('STREAM_UPDATE_INTERVAL_MS', 1000),
   discordMessageCharLimit: numberEnv('DISCORD_MESSAGE_CHAR_LIMIT', 1900),
   maxAttachmentSizeBytes: coreConfig.artifactMaxSizeBytes,

packages/discord/src/index.ts

@@ -45,6 +45,12 @@ import {
 } from './commands/skill.js';
 import { promptThreadSetup, applySetupResult } from './commands/thread-setup.js';
 
+function isChannelAllowed(channelId: string, parentId: string | null): boolean {
+  if (config.allowedChannelIds.length === 0) return true;
+  return config.allowedChannelIds.includes(channelId)
+    || (parentId !== null && config.allowedChannelIds.includes(parentId));
+}
+
 type CommandHandler = (interaction: ChatInputCommandInteraction) => Promise<void>;
 type AutocompleteHandler = (interaction: AutocompleteInteraction) => Promise<void>;
 
@@ -164,6 +170,11 @@ export async function handleDiscordMessageCreate(
   const botUser = dependencies.botUser ?? client.user;
   if (!botUser) return;
 
+  // Channel allowlist filter
+  const channelId = message.channel.id;
+  const parentId = message.channel.isThread() ? message.channel.parentId : null;
+  if (!isChannelAllowed(channelId, parentId)) return;
+
   const isActiveThread = message.channel.isThread()
     && (dependencies.hasOpenStickyThreadSession ?? hasOpenStickyThreadSession)(message.channel.id);
   const routedMessage = resolveInboundDiscordMessage({
@@ -302,6 +313,13 @@ client.on(Events.Error, (error) => {
 
 client.on(Events.InteractionCreate, async (interaction) => {
   try {
+    // Channel allowlist filter
+    if (interaction.channel) {
+      const channelId = interaction.channel.id;
+      const parentId = interaction.channel.isThread() ? interaction.channel.parentId : null;
+      if (!isChannelAllowed(channelId, parentId)) return;
+    }
+
     if (interaction.isChatInputCommand()) {
       const handler = commandHandlers.get(interaction.commandName);
       if (!handler) return;

packages/feishu/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-im-relay/feishu",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "type": "module",
   "main": "dist/index.mjs",
   "types": "dist/index.d.mts",