mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-18 14:25:51 +00:00
feat: 远程群控 (#243)
* feat: restore pipe IPC, LAN pipes, monitor tool, and PR-package features Core IPC system (UDS_INBOX): - PipeServer/PipeClient with UDS + TCP dual transport, NDJSON protocol - PipeRegistry: machineId-based role assignment, file locking - Master/slave attach, prompt relay, permission forwarding - Heartbeat lifecycle with parallel isPipeAlive probes - Commands: /pipes, /attach, /detach, /send, /claim-main, /pipe-status LAN Pipes (LAN_PIPES): - UDP multicast beacon (224.0.71.67:7101) for zero-config LAN discovery - PipeServer TCP listener, PipeClient TCP connect mode - Heartbeat auto-attaches LAN peers via TCP - Cross-machine attach allowed regardless of role - /pipes shows [LAN] peers with role + hostname/IP - SendMessageTool supports tcp: scheme with user consent Architecture — extracted hooks from REPL.tsx (~830 lines → ~20 lines): - usePipeIpc: lifecycle (bootstrap, handlers, heartbeat, cleanup) - usePipeRelay: slave→master message relay via module singleton - usePipePermissionForward: permission request/cancel forwarding - usePipeRouter: selected pipe input routing with role+IP labels - Shared ndjsonFramer.ts replaces 3 duplicate NDJSON parsers Key fixes applied during development: - Multicast binds to correct LAN interface (not WSL/Docker) - Beacon ref stored as module singleton (not Zustand state mutation) - Heartbeat preserves LAN peers in discoveredPipes and selectedPipes - Disconnect handler calls removeSlaveClient (fixes listener leak) - cleanupStaleEntries probes without lock, writes briefly under lock - getMachineId uses async execFile (not blocking execSync) - globalThis.__pipeSendToMaster replaced with setPipeRelay singleton - M key only toggles route mode when selector panel is expanded - User prompt displayed in message list on pipe broadcast - Broadcast notifications show [role] + hostname/IP for LAN peers Other restored features: - Monitor tool: /monitor command, MonitorTool, MonitorMcpTask lifecycle - Daemon supervisor and remoteControlServer command - Tools: SnipTool, SleepTool, ListPeersTool, SendUserFileTool, WebBrowserTool, WorkflowTool, and 10+ stub→implementation rewrites - Feature flags: UDS_INBOX, LAN_PIPES, MONITOR_TOOL, FORK_SUBAGENT, KAIROS, COORDINATOR_MODE, WORKFLOW_SCRIPTS, HISTORY_SNIP Tests: 2190 pass / 0 fail (15 new: lanBeacon 7, peerAddress 8) * fix: resolve merge conflicts and fix all tsc/test errors after main merge - Export ToolResultBlockParam from Tool.ts (14 tool files fixed) - Migrate ink imports from ../../ink.js to @anthropic/ink (7 files) - Fix toolUseID → toolUseId typo in monitor.ts and MonitorTool.tsx - Add fallback values for string|undefined type errors (8 locations) - Fix AppState type in assistant.ts, add NewInstallWizard stubs - Fix ParsedRepository.repo → .name in subscribe-pr.ts - Fix AgentId/string type mismatch in BackgroundTasksDialog.tsx - Fix PipeRelayFn return type in pipePermissionRelay.ts - Use PipeMessage type in usePipeRelay.ts - Fix lanBeacon.test.ts mock type assertions - Create missing MouseActionEvent class for ink package - Use ansi: color format instead of bare "green"/"red" - Resolve theme.permission access via getTheme() Result: 0 tsc errors, 2496 tests pass, 0 fail Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix: 恢复 /poor 的说明 --------- Co-authored-by: unraid <local@unraid.local> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
327
src/hooks/useMasterMonitor.ts
Normal file
327
src/hooks/useMasterMonitor.ts
Normal file
@@ -0,0 +1,327 @@
|
||||
/**
|
||||
* useMasterMonitor — master-side slave registry helpers plus an optional hook
|
||||
*
|
||||
* The module-level registry helpers are the live integration point used by
|
||||
* attach/send/status flows. The hook remains available for history syncing if
|
||||
* a caller wants AppState to mirror slave session events.
|
||||
*
|
||||
* The master CLI itself remains fully functional — this hook only collects
|
||||
* data from slaves for review via /history and /status commands.
|
||||
*/
|
||||
|
||||
import { useEffect, useSyncExternalStore } from 'react'
|
||||
import { useAppState, useSetAppState } from '../state/AppState.js'
|
||||
import {
|
||||
getPipeIpc,
|
||||
type PipeClient,
|
||||
type PipeMessage,
|
||||
type PipeIpcSlaveState,
|
||||
} from '../utils/pipeTransport.js'
|
||||
import { logForDebugging } from '../utils/debug.js'
|
||||
|
||||
/** Session history entry for pipe IPC monitoring. */
|
||||
export type SessionEntry = {
|
||||
type: string
|
||||
content: string
|
||||
from: string
|
||||
timestamp: string
|
||||
meta?: Record<string, unknown>
|
||||
}
|
||||
|
||||
function summarizePipeEntry(entry: SessionEntry): string | undefined {
|
||||
const content = entry.content.trim()
|
||||
switch (entry.type) {
|
||||
case 'prompt':
|
||||
return content ? `Queued: ${content}` : 'Queued prompt'
|
||||
case 'prompt_ack':
|
||||
return content || 'Accepted'
|
||||
case 'stream':
|
||||
return content || undefined
|
||||
case 'tool_start':
|
||||
return content ? `Tool: ${content}` : 'Tool started'
|
||||
case 'tool_result':
|
||||
return content ? `Tool result: ${content}` : 'Tool completed'
|
||||
case 'done':
|
||||
return content || 'Completed'
|
||||
case 'error':
|
||||
return content || 'Error'
|
||||
default:
|
||||
return content || undefined
|
||||
}
|
||||
}
|
||||
|
||||
function statusForPipeEntry(
|
||||
currentStatus: PipeIpcSlaveState['status'],
|
||||
entryType: SessionEntry['type'],
|
||||
): PipeIpcSlaveState['status'] {
|
||||
switch (entryType) {
|
||||
case 'prompt':
|
||||
case 'prompt_ack':
|
||||
case 'stream':
|
||||
case 'tool_start':
|
||||
case 'tool_result':
|
||||
return 'busy'
|
||||
case 'done':
|
||||
return 'idle'
|
||||
case 'error':
|
||||
return 'error'
|
||||
default:
|
||||
return currentStatus
|
||||
}
|
||||
}
|
||||
|
||||
export function applyPipeEntryToSlaveState(
|
||||
slave: PipeIpcSlaveState,
|
||||
entry: SessionEntry,
|
||||
): PipeIpcSlaveState {
|
||||
return {
|
||||
...slave,
|
||||
status: statusForPipeEntry(slave.status, entry.type),
|
||||
lastActivityAt: entry.timestamp,
|
||||
lastSummary: summarizePipeEntry(entry),
|
||||
lastEventType: entry.type as PipeIpcSlaveState['lastEventType'],
|
||||
unreadCount: (slave.unreadCount ?? 0) + 1,
|
||||
history: [...slave.history, entry],
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Module-level registry of connected slave PipeClients.
|
||||
* Keyed by slave pipe name. Managed by /attach and /detach commands.
|
||||
*/
|
||||
const _slaveClients = new Map<string, PipeClient>()
|
||||
const _slaveClientRegistryListeners = new Set<() => void>()
|
||||
const _pipeEntryListeners = new Set<
|
||||
(slaveName: string, entry: SessionEntry) => void
|
||||
>()
|
||||
const _pipeEntryHandlers = new Map<string, (msg: PipeMessage) => void>()
|
||||
let _slaveClientRegistryVersion = 0
|
||||
|
||||
const MONITORED_PIPE_ENTRY_TYPES = [
|
||||
'prompt_ack',
|
||||
'stream',
|
||||
'tool_start',
|
||||
'tool_result',
|
||||
'done',
|
||||
'error',
|
||||
'prompt',
|
||||
'permission_request',
|
||||
'permission_cancel',
|
||||
]
|
||||
|
||||
function isMonitoredPipeEntryType(type: string): boolean {
|
||||
return MONITORED_PIPE_ENTRY_TYPES.includes(type)
|
||||
}
|
||||
|
||||
function pipeMessageToSessionEntry(
|
||||
slaveName: string,
|
||||
msg: PipeMessage,
|
||||
): SessionEntry {
|
||||
return {
|
||||
type: msg.type as SessionEntry['type'],
|
||||
content: msg.data ?? '',
|
||||
from: msg.from ?? slaveName,
|
||||
timestamp: msg.ts ?? new Date().toISOString(),
|
||||
meta: msg.meta,
|
||||
}
|
||||
}
|
||||
|
||||
function emitPipeEntry(slaveName: string, entry: SessionEntry): void {
|
||||
for (const listener of _pipeEntryListeners) {
|
||||
listener(slaveName, entry)
|
||||
}
|
||||
}
|
||||
|
||||
export function subscribePipeEntries(
|
||||
listener: (slaveName: string, entry: SessionEntry) => void,
|
||||
): () => void {
|
||||
_pipeEntryListeners.add(listener)
|
||||
return () => {
|
||||
_pipeEntryListeners.delete(listener)
|
||||
}
|
||||
}
|
||||
|
||||
function detachPipeEntryEmitter(name: string, client?: PipeClient): void {
|
||||
const handler = _pipeEntryHandlers.get(name)
|
||||
if (!handler) return
|
||||
client?.removeListener?.('message', handler)
|
||||
_pipeEntryHandlers.delete(name)
|
||||
}
|
||||
|
||||
function attachPipeEntryEmitter(name: string, client: PipeClient): void {
|
||||
detachPipeEntryEmitter(name, _slaveClients.get(name))
|
||||
if (typeof client.on !== 'function') return
|
||||
const handler = (msg: PipeMessage) => {
|
||||
if (!isMonitoredPipeEntryType(msg.type)) return
|
||||
emitPipeEntry(name, pipeMessageToSessionEntry(name, msg))
|
||||
}
|
||||
_pipeEntryHandlers.set(name, handler)
|
||||
client.on('message', handler)
|
||||
}
|
||||
|
||||
function emitSlaveClientRegistryChanged(): void {
|
||||
_slaveClientRegistryVersion += 1
|
||||
for (const listener of _slaveClientRegistryListeners) {
|
||||
listener()
|
||||
}
|
||||
}
|
||||
|
||||
function subscribeToSlaveClientRegistry(listener: () => void): () => void {
|
||||
_slaveClientRegistryListeners.add(listener)
|
||||
return () => {
|
||||
_slaveClientRegistryListeners.delete(listener)
|
||||
}
|
||||
}
|
||||
|
||||
function getSlaveClientRegistryVersion(): number {
|
||||
return _slaveClientRegistryVersion
|
||||
}
|
||||
|
||||
export function addSlaveClient(name: string, client: PipeClient): void {
|
||||
attachPipeEntryEmitter(name, client)
|
||||
_slaveClients.set(name, client)
|
||||
emitSlaveClientRegistryChanged()
|
||||
}
|
||||
|
||||
export function removeSlaveClient(name: string): PipeClient | undefined {
|
||||
const client = _slaveClients.get(name)
|
||||
detachPipeEntryEmitter(name, client)
|
||||
_slaveClients.delete(name)
|
||||
emitSlaveClientRegistryChanged()
|
||||
return client
|
||||
}
|
||||
|
||||
export function getSlaveClient(name: string): PipeClient | undefined {
|
||||
return _slaveClients.get(name)
|
||||
}
|
||||
|
||||
export function getAllSlaveClients(): Map<string, PipeClient> {
|
||||
return _slaveClients
|
||||
}
|
||||
|
||||
export type ConnectedSlaveTarget = {
|
||||
name: string
|
||||
client: PipeClient
|
||||
}
|
||||
|
||||
/**
|
||||
* Resolve a selection list to currently connected slave clients.
|
||||
*
|
||||
* The pipe selector can include discovered-but-not-attached names. Routing
|
||||
* should only treat attached, connected clients as broadcast targets.
|
||||
*/
|
||||
export function getConnectedSlaveTargets(
|
||||
selectedNames: string[],
|
||||
): ConnectedSlaveTarget[] {
|
||||
const targets: ConnectedSlaveTarget[] = []
|
||||
for (const name of selectedNames) {
|
||||
const client = _slaveClients.get(name)
|
||||
if (client?.connected) {
|
||||
targets.push({ name, client })
|
||||
}
|
||||
}
|
||||
return targets
|
||||
}
|
||||
|
||||
export function resetSlaveClientsForTesting(): void {
|
||||
for (const [name, client] of _slaveClients.entries()) {
|
||||
detachPipeEntryEmitter(name, client)
|
||||
}
|
||||
_slaveClients.clear()
|
||||
emitSlaveClientRegistryChanged()
|
||||
}
|
||||
|
||||
export function useMasterMonitor(): void {
|
||||
const role = useAppState(s => getPipeIpc(s).role)
|
||||
const setAppState = useSetAppState()
|
||||
const registryVersion = useSyncExternalStore(
|
||||
subscribeToSlaveClientRegistry,
|
||||
getSlaveClientRegistryVersion,
|
||||
getSlaveClientRegistryVersion,
|
||||
)
|
||||
|
||||
useEffect(() => {
|
||||
if (role !== 'master' && _slaveClients.size === 0) return
|
||||
|
||||
// Set up listeners for each connected slave client
|
||||
const cleanups: (() => void)[] = []
|
||||
|
||||
for (const [slaveName, client] of _slaveClients.entries()) {
|
||||
const handler = (msg: PipeMessage) => {
|
||||
const entry = pipeMessageToSessionEntry(slaveName, msg)
|
||||
|
||||
// Only record relevant message types
|
||||
if (!isMonitoredPipeEntryType(msg.type)) {
|
||||
return
|
||||
}
|
||||
|
||||
setAppState(prev => {
|
||||
const slave = getPipeIpc(prev).slaves[slaveName]
|
||||
if (!slave) return prev
|
||||
|
||||
const newStatus =
|
||||
msg.type === 'done' || msg.type === 'error'
|
||||
? 'idle'
|
||||
: msg.type === 'prompt'
|
||||
? 'busy'
|
||||
: slave.status
|
||||
|
||||
return {
|
||||
...prev,
|
||||
pipeIpc: {
|
||||
...getPipeIpc(prev),
|
||||
slaves: {
|
||||
...getPipeIpc(prev).slaves,
|
||||
[slaveName]: applyPipeEntryToSlaveState(
|
||||
{
|
||||
...slave,
|
||||
status: newStatus,
|
||||
},
|
||||
entry,
|
||||
),
|
||||
},
|
||||
},
|
||||
}
|
||||
})
|
||||
|
||||
if (msg.type === 'done') {
|
||||
logForDebugging(`[MasterMonitor] Slave "${slaveName}" turn complete`)
|
||||
}
|
||||
}
|
||||
|
||||
client.on('message', handler)
|
||||
|
||||
// Handle slave disconnect
|
||||
const onDisconnect = () => {
|
||||
logForDebugging(`[MasterMonitor] Slave "${slaveName}" disconnected`)
|
||||
removeSlaveClient(slaveName)
|
||||
setAppState(prev => {
|
||||
const { [slaveName]: _removed, ...remainingSlaves } =
|
||||
getPipeIpc(prev).slaves
|
||||
const hasSlaves = Object.keys(remainingSlaves).length > 0
|
||||
return {
|
||||
...prev,
|
||||
pipeIpc: {
|
||||
...getPipeIpc(prev),
|
||||
role: hasSlaves ? 'master' : 'main',
|
||||
displayRole: hasSlaves ? 'master' : 'main',
|
||||
slaves: remainingSlaves,
|
||||
},
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
client.on('disconnect', onDisconnect)
|
||||
cleanups.push(() => {
|
||||
client.removeListener('message', handler)
|
||||
client.removeListener('disconnect', onDisconnect)
|
||||
})
|
||||
}
|
||||
|
||||
return () => {
|
||||
for (const cleanup of cleanups) {
|
||||
cleanup()
|
||||
}
|
||||
}
|
||||
}, [registryVersion, role, setAppState])
|
||||
}
|
||||
Reference in New Issue
Block a user