mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-18 14:25:51 +00:00
fix: 修正 channel permission relay 路由与能力判定
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
@@ -0,0 +1,38 @@
|
||||
import { describe, expect, test } from 'bun:test'
|
||||
import { getLatestChannelContextHint } from '../interactiveHandler.js'
|
||||
|
||||
describe('getLatestChannelContextHint', () => {
|
||||
test('extracts source server and chat id from latest channel user message', () => {
|
||||
expect(
|
||||
getLatestChannelContextHint([
|
||||
{
|
||||
type: 'user',
|
||||
origin: { kind: 'channel', server: 'plugin:weixin:weixin' },
|
||||
message: {
|
||||
content: [
|
||||
{
|
||||
type: 'text',
|
||||
text: '<channel source="plugin:weixin:weixin" chat_id="user-1" sender_id="user-1">\nhello\n</channel>',
|
||||
},
|
||||
],
|
||||
},
|
||||
},
|
||||
]),
|
||||
).toEqual({
|
||||
sourceServer: 'plugin:weixin:weixin',
|
||||
chatId: 'user-1',
|
||||
})
|
||||
})
|
||||
|
||||
test('returns null when there is no channel-origin user message', () => {
|
||||
expect(
|
||||
getLatestChannelContextHint([
|
||||
{
|
||||
type: 'user',
|
||||
origin: { kind: 'manual' },
|
||||
message: { content: [{ type: 'text', text: 'hello' }] },
|
||||
},
|
||||
]),
|
||||
).toBeNull()
|
||||
})
|
||||
})
|
||||
@@ -1,6 +1,7 @@
|
||||
import { feature } from 'bun:bundle'
|
||||
import type { ContentBlockParam } from '@anthropic-ai/sdk/resources/messages.mjs'
|
||||
import { randomUUID } from 'crypto'
|
||||
import { CHANNEL_TAG } from 'src/constants/xml.js'
|
||||
import { logForDebugging } from 'src/utils/debug.js'
|
||||
import { getAllowedChannels } from '../../../bootstrap/state.js'
|
||||
import type { BridgePermissionCallbacks } from '../../../bridge/bridgePermissionCallbacks.js'
|
||||
@@ -46,6 +47,76 @@ type InteractivePermissionParams = {
|
||||
channelCallbacks?: ChannelPermissionCallbacks
|
||||
}
|
||||
|
||||
type ChannelContextHint = {
|
||||
sourceServer?: string
|
||||
chatId?: string
|
||||
}
|
||||
|
||||
function getTextBlocksText(content: unknown): string {
|
||||
if (typeof content === 'string') {
|
||||
return content
|
||||
}
|
||||
if (!Array.isArray(content)) {
|
||||
return ''
|
||||
}
|
||||
return content
|
||||
.filter(
|
||||
(block): block is { type: 'text'; text: string } =>
|
||||
typeof block === 'object' &&
|
||||
block !== null &&
|
||||
(block as { type?: unknown }).type === 'text' &&
|
||||
typeof (block as { text?: unknown }).text === 'string',
|
||||
)
|
||||
.map(block => block.text)
|
||||
.join('\n')
|
||||
}
|
||||
|
||||
function parseChannelContextHintFromText(text: string): ChannelContextHint | null {
|
||||
const tagMatch = text.match(new RegExp(`<${CHANNEL_TAG}\\b([^>]*)>`))
|
||||
if (!tagMatch?.[1]) {
|
||||
return null
|
||||
}
|
||||
|
||||
const attrs = tagMatch[1]
|
||||
const sourceServer = attrs.match(/\bsource="([^"]+)"/)?.[1]
|
||||
const chatId = attrs.match(/\bchat_id="([^"]+)"/)?.[1]
|
||||
|
||||
if (!sourceServer && !chatId) {
|
||||
return null
|
||||
}
|
||||
|
||||
return { sourceServer, chatId }
|
||||
}
|
||||
|
||||
export function getLatestChannelContextHint(messages: readonly unknown[]): ChannelContextHint | null {
|
||||
for (let index = messages.length - 1; index >= 0; index--) {
|
||||
const message = messages[index] as {
|
||||
type?: unknown
|
||||
origin?: { kind?: unknown; server?: unknown }
|
||||
message?: { content?: unknown }
|
||||
}
|
||||
|
||||
if (message?.type !== 'user' || message?.origin?.kind !== 'channel') {
|
||||
continue
|
||||
}
|
||||
|
||||
const text = getTextBlocksText(message.message?.content)
|
||||
const parsed = parseChannelContextHintFromText(text)
|
||||
if (parsed) {
|
||||
return {
|
||||
sourceServer:
|
||||
parsed.sourceServer ||
|
||||
(typeof message.origin.server === 'string'
|
||||
? message.origin.server
|
||||
: undefined),
|
||||
chatId: parsed.chatId,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return null
|
||||
}
|
||||
|
||||
/**
|
||||
* Handles the interactive (main-agent) permission flow.
|
||||
*
|
||||
@@ -420,6 +491,17 @@ function handleInteractivePermission(
|
||||
description,
|
||||
input_preview: truncateForPreview(displayInput),
|
||||
}
|
||||
const channelContext = getLatestChannelContextHint(
|
||||
ctx.toolUseContext.messages,
|
||||
)
|
||||
if (channelContext?.sourceServer || channelContext?.chatId) {
|
||||
params.channel_context = {
|
||||
...(channelContext.sourceServer && {
|
||||
source_server: channelContext.sourceServer,
|
||||
}),
|
||||
...(channelContext.chatId && { chat_id: channelContext.chatId }),
|
||||
}
|
||||
}
|
||||
|
||||
for (const client of channelClients) {
|
||||
if (client.type !== 'connected') continue // refine for TS
|
||||
|
||||
Reference in New Issue
Block a user