| |
| import { |
| normalizeLowercaseStringOrEmpty, |
| normalizeOptionalString, |
| } from "@openclaw/normalization-core/string-coerce"; |
| import { buildSubagentRunReadIndex } from "../../../agents/subagents/registry/subagent-registry-read.js"; |
| import type { SubagentRunRecord } from "../../../agents/subagents/registry/subagent-registry.types.js"; |
| import { buildSubagentRunView } from "../../../agents/subagents/registry/subagent-run-view.js"; |
| import { |
| resolveInternalSessionKey, |
| resolveMainSessionAlias, |
| } from "../../../agents/tools/sessions-helpers.js"; |
| import { isNativeCommandTurn, resolveCommandTurnContext } from "../../command-turn-context.js"; |
| import { commandReply } from "../command-gates.js"; |
| import { extractSubagentMessageText, type ChatMessage } from "../commands-subagents-text.js"; |
| import type { CommandHandler, CommandHandlerResult } from "../commands-types.js"; |
| import { formatRunLabel } from "../subagents-utils.js"; |
|
|
| export type { ChatMessage } from "../commands-subagents-text.js"; |
|
|
| export const RECENT_WINDOW_MINUTES = 30; |
|
|
| type SubagentsCommandParams = Parameters<CommandHandler>[0]; |
|
|
| export type SubagentsCommandContext = { |
| params: SubagentsCommandParams; |
| requesterKey: string; |
| runs: SubagentRunRecord[]; |
| restTokens: string[]; |
| }; |
|
|
| export function resolveSubagentEntryForToken( |
| runs: SubagentRunRecord[], |
| token: string | undefined, |
| ): { entry: SubagentRunRecord } | { reply: CommandHandlerResult } { |
| const fail = (message: string) => ({ reply: commandReply(`⚠️ ${message}`) }); |
| const trimmed = normalizeOptionalString(token); |
| if (!trimmed) { |
| return fail("Missing subagent id."); |
| } |
| const readIndex = buildSubagentRunReadIndex(); |
| const { latest, active, recent } = buildSubagentRunView({ |
| runs, |
| recentMinutes: RECENT_WINDOW_MINUTES, |
| countPendingDescendantRuns: (sessionKey) => readIndex.countPendingDescendantRuns(sessionKey), |
| }); |
| if (trimmed === "last") { |
| const entry = latest[0]; |
| return entry ? { entry } : fail("Unknown subagent."); |
| } |
| const numericOrder = [...active, ...recent]; |
| if (/^\d+$/.test(trimmed)) { |
| const entry = numericOrder[Number.parseInt(trimmed, 10) - 1]; |
| return entry ? { entry } : fail(`Invalid subagent index: ${trimmed}`); |
| } |
| if (trimmed.includes(":")) { |
| const entry = latest.find((run) => run.childSessionKey === trimmed); |
| return entry ? { entry } : fail(`Unknown subagent session: ${trimmed}`); |
| } |
| const lowered = normalizeLowercaseStringOrEmpty(trimmed); |
| const match = (entries: SubagentRunRecord[], ambiguity: string) => { |
| if (entries.length > 1) { |
| return fail(`${ambiguity}: ${trimmed}`); |
| } |
| const entry = entries[0]; |
| return entry ? { entry } : undefined; |
| }; |
| return ( |
| match( |
| numericOrder.filter((entry) => normalizeLowercaseStringOrEmpty(entry.taskName) === lowered), |
| "Ambiguous subagent label", |
| ) ?? |
| match( |
| latest.filter((entry) => normalizeLowercaseStringOrEmpty(formatRunLabel(entry)) === lowered), |
| "Ambiguous subagent label", |
| ) ?? |
| match( |
| numericOrder.filter((entry) => |
| normalizeLowercaseStringOrEmpty(entry.taskName).startsWith(lowered), |
| ), |
| "Ambiguous subagent label prefix", |
| ) ?? |
| match( |
| latest.filter((entry) => |
| normalizeLowercaseStringOrEmpty(formatRunLabel(entry)).startsWith(lowered), |
| ), |
| "Ambiguous subagent label prefix", |
| ) ?? |
| match( |
| latest.filter((entry) => entry.runId.startsWith(trimmed)), |
| "Ambiguous run id prefix", |
| ) ?? |
| fail(`Unknown subagent id: ${trimmed}`) |
| ); |
| } |
|
|
| export function resolveRequesterSessionKey( |
| params: SubagentsCommandParams, |
| opts?: { preferCommandTarget?: boolean }, |
| ): string | undefined { |
| const commandTarget = normalizeOptionalString(params.ctx.CommandTargetSessionKey); |
| const commandSession = normalizeOptionalString(params.sessionKey); |
| const shouldPreferCommandTarget = |
| opts?.preferCommandTarget ?? isNativeCommandTurn(resolveCommandTurnContext(params.ctx)); |
| const raw = shouldPreferCommandTarget |
| ? commandTarget || commandSession |
| : commandSession || commandTarget; |
| if (!raw) { |
| return undefined; |
| } |
| const { mainKey, alias } = resolveMainSessionAlias(params.cfg); |
| return resolveInternalSessionKey({ key: raw, alias, mainKey }); |
| } |
|
|
| export function buildSubagentsHelp() { |
| return [ |
| "Subagents", |
| "Usage:", |
| "- /subagents list", |
| "- /subagents log <id|#> [limit] [tools]", |
| "- /subagents info <id|#>", |
| "- /session unbind", |
| "- /agents", |
| "- /session idle <duration|off>", |
| "- /session max-age <duration|off>", |
| "", |
| "Ids: use the list index (#), runId/session prefix, label, or full session key.", |
| ].join("\n"); |
| } |
|
|
| export function formatLogLines(messages: ChatMessage[]) { |
| const lines: string[] = []; |
| for (const msg of messages) { |
| const extracted = extractSubagentMessageText(msg); |
| if (!extracted) { |
| continue; |
| } |
| const label = extracted.role === "assistant" ? "Assistant" : "User"; |
| lines.push(`${label}: ${extracted.text}`); |
| } |
| return lines; |
| } |
|
|
