@kmmao/happy-wire 0.19.2 → 0.21.0

package/dist/index.cjs

@@ -148,6 +148,24 @@ const sessionTaskEndEventSchema = z__namespace.object({
     durationMs: z__namespace.number()
   }).optional()
 });
+const sessionTaskUpdatedEventSchema = z__namespace.object({
+  t: z__namespace.literal("task-updated"),
+  taskId: z__namespace.string(),
+  patch: z__namespace.object({
+    status: z__namespace.enum(["pending", "running", "completed", "failed", "killed", "paused"]).optional(),
+    description: z__namespace.string().optional(),
+    endTime: z__namespace.number().optional(),
+    error: z__namespace.string().optional(),
+    isBackgrounded: z__namespace.boolean().optional()
+  })
+});
+const sessionRateLimitEventSchema = z__namespace.object({
+  t: z__namespace.literal("rate-limit"),
+  status: z__namespace.enum(["allowed", "allowed_warning", "rejected"]),
+  resetsAt: z__namespace.number().optional(),
+  rateLimitType: z__namespace.string().optional(),
+  utilization: z__namespace.number().optional()
+});
 const sessionToolProgressEventSchema = z__namespace.object({
   t: z__namespace.literal("tool-progress"),
   toolUseId: z__namespace.string(),
@@ -215,6 +233,8 @@ const sessionEventSchema = z__namespace.discriminatedUnion("t", [
   sessionTaskStartEventSchema,
   sessionTaskProgressEventSchema,
   sessionTaskEndEventSchema,
+  sessionTaskUpdatedEventSchema,
+  sessionRateLimitEventSchema,
   sessionToolProgressEventSchema,
   sessionPromptSuggestionEventSchema,
   sessionNeedsContinueEventSchema,
@@ -239,7 +259,7 @@ const sessionEnvelopeSchema = z__namespace.object({
       path: ["role"]
     });
   }
-  if ((envelope.ev.t === "start" || envelope.ev.t === "stop" || envelope.ev.t === "usage-update" || envelope.ev.t === "task-start" || envelope.ev.t === "task-progress" || envelope.ev.t === "task-end" || envelope.ev.t === "tool-progress" || envelope.ev.t === "prompt-suggestion" || envelope.ev.t === "needs-continue" || envelope.ev.t === "session-state-changed" || envelope.ev.t === "context-usage" || envelope.ev.t === "task-log") && envelope.role !== "agent") {
+  if ((envelope.ev.t === "start" || envelope.ev.t === "stop" || envelope.ev.t === "usage-update" || envelope.ev.t === "task-start" || envelope.ev.t === "task-progress" || envelope.ev.t === "task-end" || envelope.ev.t === "task-updated" || envelope.ev.t === "rate-limit" || envelope.ev.t === "tool-progress" || envelope.ev.t === "prompt-suggestion" || envelope.ev.t === "needs-continue" || envelope.ev.t === "session-state-changed" || envelope.ev.t === "context-usage" || envelope.ev.t === "task-log") && envelope.role !== "agent") {
     ctx.addIssue({
       code: z__namespace.ZodIssueCode.custom,
       message: `${envelope.ev.t} events must use role "agent"`,
@@ -2224,6 +2244,142 @@ const GetMcpServersResponseSchema = z.z.object({
     })).optional()
   }))
 }).strict();
+const SetMcpServersRequestSchema = z.z.object({
+  /** Full MCP server config map — keys are server names, values are McpServerConfig. */
+  servers: z.z.record(z.z.string(), z.z.object({
+    /** Transport type: 'stdio', 'sse', 'streamable-http', or 'url'. */
+    type: z.z.string().optional(),
+    /** Command for stdio transport. */
+    command: z.z.string().optional(),
+    /** Args for stdio transport. */
+    args: z.z.array(z.z.string()).optional(),
+    /** Environment variables for stdio transport. */
+    env: z.z.record(z.z.string(), z.z.string()).optional(),
+    /** URL for sse / streamable-http / url transports. */
+    url: z.z.string().optional()
+  }))
+}).strict();
+const SetMcpServersResponseSchema = z.z.object({
+  /** Server names that were newly connected. */
+  added: z.z.array(z.z.string()),
+  /** Server names that were disconnected. */
+  removed: z.z.array(z.z.string()),
+  /** Per-server errors keyed by server name. Empty when all succeeded. */
+  errors: z.z.record(z.z.string(), z.z.string())
+}).strict();
+const ReconnectMcpServerRequestSchema = z.z.object({
+  serverName: z.z.string().min(1).max(256)
+}).strict();
+const ReconnectMcpServerResponseSchema = z.z.object({
+  success: z.z.literal(true)
+}).strict();
+const ToggleMcpServerRequestSchema = z.z.object({
+  serverName: z.z.string().min(1).max(256),
+  enabled: z.z.boolean()
+}).strict();
+const ToggleMcpServerResponseSchema = z.z.object({
+  success: z.z.literal(true)
+}).strict();
+const AddMcpServerRequestSchema = z.z.object({
+  /** Unique server name. Must not collide with protected names. */
+  name: z.z.string().min(1).max(256),
+  /** Server transport config — same shape as SetMcpServersRequest entries. */
+  config: z.z.object({
+    type: z.z.string().optional(),
+    command: z.z.string().optional(),
+    args: z.z.array(z.z.string()).optional(),
+    env: z.z.record(z.z.string(), z.z.string()).optional(),
+    url: z.z.string().optional()
+  })
+}).strict();
+const AddMcpServerResponseSchema = z.z.object({
+  /** Whether the server was successfully added and connected. */
+  success: z.z.boolean(),
+  /** Server names that were newly connected (typically just the one added). */
+  added: z.z.array(z.z.string()),
+  /** Per-server errors. Empty on success. */
+  errors: z.z.record(z.z.string(), z.z.string()),
+  /** Error message when success is false. */
+  errorMessage: z.z.string().optional()
+}).strict();
+const RemoveMcpServerRequestSchema = z.z.object({
+  /** Server name to remove. */
+  name: z.z.string().min(1).max(256)
+}).strict();
+const RemoveMcpServerResponseSchema = z.z.object({
+  success: z.z.boolean(),
+  /** Server names that were disconnected (typically just the one removed). */
+  removed: z.z.array(z.z.string()),
+  /** Error message when success is false. */
+  errorMessage: z.z.string().optional()
+}).strict();
+const ApplySettingsRequestSchema = z.z.object({
+  /** Partial Settings object — only supplied keys are merged. */
+  settings: z.z.record(z.z.string(), z.z.unknown())
+}).strict();
+const ApplySettingsResponseSchema = z.z.object({
+  success: z.z.literal(true)
+}).strict();
+const SdkSessionInfoSchema = z.z.object({
+  sessionId: z.z.string(),
+  summary: z.z.string(),
+  lastModified: z.z.number(),
+  fileSize: z.z.number().optional(),
+  customTitle: z.z.string().optional(),
+  firstPrompt: z.z.string().optional(),
+  gitBranch: z.z.string().optional(),
+  cwd: z.z.string().optional(),
+  tag: z.z.string().optional(),
+  createdAt: z.z.number().optional()
+});
+const ListSessionsRequestSchema = z.z.object({
+  /** Directory to scope results to. Omit for all projects. */
+  dir: z.z.string().optional(),
+  limit: z.z.number().int().positive().max(500).optional(),
+  offset: z.z.number().int().nonnegative().optional()
+}).strict();
+const ListSessionsResponseSchema = z.z.object({
+  sessions: z.z.array(SdkSessionInfoSchema)
+}).strict();
+const GetSessionInfoRequestSchema = z.z.object({
+  /** Session ID to look up. */
+  targetSessionId: z.z.string().min(1),
+  dir: z.z.string().optional()
+}).strict();
+const GetSessionInfoResponseSchema = z.z.object({
+  session: SdkSessionInfoSchema.nullable()
+}).strict();
+const DeleteSessionRequestSchema = z.z.object({
+  targetSessionId: z.z.string().min(1),
+  dir: z.z.string().optional()
+}).strict();
+const DeleteSessionResponseSchema = z.z.object({
+  success: z.z.literal(true)
+}).strict();
+const RenameSessionRequestSchema = z.z.object({
+  targetSessionId: z.z.string().min(1),
+  title: z.z.string().min(1).max(500),
+  dir: z.z.string().optional()
+}).strict();
+const RenameSessionResponseSchema = z.z.object({
+  success: z.z.literal(true)
+}).strict();
+const GetSessionMessagesRequestSchema = z.z.object({
+  targetSessionId: z.z.string().min(1),
+  dir: z.z.string().optional(),
+  limit: z.z.number().int().positive().max(1e3).optional(),
+  offset: z.z.number().int().nonnegative().optional(),
+  includeSystemMessages: z.z.boolean().optional()
+}).strict();
+const GetSessionMessagesResponseSchema = z.z.object({
+  messages: z.z.array(z.z.object({
+    type: z.z.enum(["user", "assistant", "system"]),
+    uuid: z.z.string(),
+    sessionId: z.z.string(),
+    content: z.z.unknown()
+  })),
+  totalCount: z.z.number().int().nonnegative()
+}).strict();
 const CLAUDE_CONTROL_METHODS = [
   "get_session_cost",
   "get_binary_version",
@@ -2232,10 +2388,118 @@ const CLAUDE_CONTROL_METHODS = [
   "mcp_call",
   "get_context_usage",
   "get_mcp_servers",
-  "get_context_detail"
+  "get_context_detail",
+  "set_mcp_servers",
+  "reconnect_mcp_server",
+  "toggle_mcp_server",
+  "add_mcp_server",
+  "remove_mcp_server",
+  "apply_settings",
+  "list_sessions",
+  "get_session_info",
+  "delete_session",
+  "rename_session",
+  "get_session_messages"
 ];
 
+const McpStdioConfigSchema = z.z.object({
+  type: z.z.literal("stdio"),
+  command: z.z.string().min(1),
+  args: z.z.array(z.z.string()).optional(),
+  env: z.z.record(z.z.string(), z.z.string()).optional()
+});
+const McpSseConfigSchema = z.z.object({
+  type: z.z.literal("sse"),
+  url: z.z.string().url()
+});
+const McpUrlConfigSchema = z.z.object({
+  type: z.z.literal("url"),
+  url: z.z.string().url()
+});
+const McpStreamableHttpConfigSchema = z.z.object({
+  type: z.z.literal("streamable-http"),
+  url: z.z.string().url()
+});
+const McpTransportConfigSchema = z.z.discriminatedUnion("type", [
+  McpStdioConfigSchema,
+  McpSseConfigSchema,
+  McpUrlConfigSchema,
+  McpStreamableHttpConfigSchema
+]);
+const McpRegistryEntrySchema = z.z.object({
+  /** Server display name — must be unique within the registry. */
+  name: z.z.string().min(1).max(128),
+  /** Transport configuration (determines how the SDK connects). */
+  transport: McpTransportConfigSchema,
+  /** Whether this server is enabled. Disabled servers are persisted but not loaded. */
+  enabled: z.z.boolean().default(true),
+  /**
+   * Machine ID scope. When set, this entry is only loaded by the specified
+   * machine. When null/undefined, the entry is account-wide (all machines).
+   * Typically set for stdio servers whose binaries exist on a specific host.
+   */
+  machineId: z.z.string().optional(),
+  /** Optional human-readable description. */
+  description: z.z.string().max(512).optional(),
+  /** ISO timestamp of when this entry was created. */
+  createdAt: z.z.string().optional(),
+  /** ISO timestamp of the last update. */
+  updatedAt: z.z.string().optional(),
+  // ── Extended metadata (optional, backward-compatible) ──
+  /** Categorize the server for UI grouping (e.g. 'search', 'data', 'code', 'infra'). */
+  category: z.z.string().max(64).optional(),
+  /** Icon identifier for display (e.g. emoji or icon-pack name). */
+  icon: z.z.string().max(64).optional(),
+  /** Free-form tags for filtering and search. */
+  tags: z.z.array(z.z.string().max(64)).max(20).optional(),
+  /** npm package name or GitHub repo (e.g. '@anthropic-ai/mcp-server-filesystem'). */
+  packageName: z.z.string().max(256).optional(),
+  /** Semantic version of the server binary/package. */
+  version: z.z.string().max(64).optional(),
+  /** Author or organization name. */
+  author: z.z.string().max(128).optional(),
+  /** Homepage / documentation URL. */
+  homepage: z.z.string().max(512).optional(),
+  /** Cached tool names discovered from the server on last connect. */
+  toolInventory: z.z.array(z.z.string().max(128)).max(200).optional(),
+  /** ISO timestamp of the last successful connection. */
+  lastConnectedAt: z.z.string().optional(),
+  /** Total number of times this server has been connected. */
+  connectionCount: z.z.number().int().nonnegative().optional()
+});
+const McpRegistrySchema = z.z.object({
+  /** Schema version for forward compatibility. */
+  version: z.z.literal(1),
+  /** Server entries keyed by name. */
+  servers: z.z.record(z.z.string(), McpRegistryEntrySchema)
+});
+const MCP_REGISTRY_KV_KEY = "mcp:servers";
+function createEmptyMcpRegistry() {
+  return { version: 1, servers: {} };
+}
+function registryToSdkConfig(registry, machineId) {
+  const result = {};
+  for (const [name, entry] of Object.entries(registry.servers)) {
+    if (!entry.enabled) continue;
+    if (entry.machineId && entry.machineId !== machineId) continue;
+    const { type, ...transportConfig } = entry.transport;
+    result[name] = { type, ...transportConfig };
+  }
+  return result;
+}
+function parseMcpRegistry(raw) {
+  if (!raw) return createEmptyMcpRegistry();
+  try {
+    const parsed = JSON.parse(raw);
+    return McpRegistrySchema.parse(parsed);
+  } catch {
+    return createEmptyMcpRegistry();
+  }
+}
+
 exports.AIBackendProfileSchema = AIBackendProfileSchema;
+exports.AddMcpServerRequestSchema = AddMcpServerRequestSchema;
+exports.AddMcpServerResponseSchema = AddMcpServerResponseSchema;
 exports.AgentLoopSummarySchema = AgentLoopSummarySchema;
 exports.AgentMessageSchema = AgentMessageSchema;
 exports.AnthropicConfigSchema = AnthropicConfigSchema;
@@ -2243,6 +2507,8 @@ exports.ApiMessageSchema = ApiMessageSchema;
 exports.ApiUpdateMachineStateSchema = ApiUpdateMachineStateSchema;
 exports.ApiUpdateNewMessageSchema = ApiUpdateNewMessageSchema;
 exports.ApiUpdateSessionStateSchema = ApiUpdateSessionStateSchema;
+exports.ApplySettingsRequestSchema = ApplySettingsRequestSchema;
+exports.ApplySettingsResponseSchema = ApplySettingsResponseSchema;
 exports.AutoDreamProfileSummarySchema = AutoDreamProfileSummarySchema;
 exports.AutomationAuditEventSummarySchema = AutomationAuditEventSummarySchema;
 exports.AutomationAuditStatsSchema = AutomationAuditStatsSchema;
@@ -2290,6 +2556,8 @@ exports.CrossProjectSearchResultSchema = CrossProjectSearchResultSchema;
 exports.CustomModelSchema = CustomModelSchema;
 exports.DaemonStateSchema = DaemonStateSchema;
 exports.DefaultPermissionModeSchema = DefaultPermissionModeSchema;
+exports.DeleteSessionRequestSchema = DeleteSessionRequestSchema;
+exports.DeleteSessionResponseSchema = DeleteSessionResponseSchema;
 exports.EnvironmentVariableSchema = EnvironmentVariableSchema;
 exports.GetBinaryVersionRequestSchema = GetBinaryVersionRequestSchema;
 exports.GetBinaryVersionResponseSchema = GetBinaryVersionResponseSchema;
@@ -2301,6 +2569,10 @@ exports.GetMcpServersRequestSchema = GetMcpServersRequestSchema;
 exports.GetMcpServersResponseSchema = GetMcpServersResponseSchema;
 exports.GetSessionCostRequestSchema = GetSessionCostRequestSchema;
 exports.GetSessionCostResponseSchema = GetSessionCostResponseSchema;
+exports.GetSessionInfoRequestSchema = GetSessionInfoRequestSchema;
+exports.GetSessionInfoResponseSchema = GetSessionInfoResponseSchema;
+exports.GetSessionMessagesRequestSchema = GetSessionMessagesRequestSchema;
+exports.GetSessionMessagesResponseSchema = GetSessionMessagesResponseSchema;
 exports.HAPPY_MCP_AUTO_APPROVE_TOOL_NAMES = HAPPY_MCP_AUTO_APPROVE_TOOL_NAMES;
 exports.HAPPY_MCP_SILENT_SUCCESS_TOOL_NAMES = HAPPY_MCP_SILENT_SUCCESS_TOOL_NAMES;
 exports.HAPPY_MCP_TOOL_NAMES = HAPPY_MCP_TOOL_NAMES;
@@ -2323,10 +2595,20 @@ exports.KnowledgeInjectionRequestSchema = KnowledgeInjectionRequestSchema;
 exports.KnowledgeInjectionResponseSchema = KnowledgeInjectionResponseSchema;
 exports.KnowledgeStatusSchema = KnowledgeStatusSchema;
 exports.LegacyMessageContentSchema = LegacyMessageContentSchema;
+exports.ListSessionsRequestSchema = ListSessionsRequestSchema;
+exports.ListSessionsResponseSchema = ListSessionsResponseSchema;
 exports.LiveKitTokenResponseSchema = LiveKitTokenResponseSchema;
+exports.MCP_REGISTRY_KV_KEY = MCP_REGISTRY_KV_KEY;
 exports.MachineMetadataSchema = MachineMetadataSchema;
 exports.McpCallRequestSchema = McpCallRequestSchema;
 exports.McpCallResponseSchema = McpCallResponseSchema;
+exports.McpRegistryEntrySchema = McpRegistryEntrySchema;
+exports.McpRegistrySchema = McpRegistrySchema;
+exports.McpSseConfigSchema = McpSseConfigSchema;
+exports.McpStdioConfigSchema = McpStdioConfigSchema;
+exports.McpStreamableHttpConfigSchema = McpStreamableHttpConfigSchema;
+exports.McpTransportConfigSchema = McpTransportConfigSchema;
+exports.McpUrlConfigSchema = McpUrlConfigSchema;
 exports.MessageContentSchema = MessageContentSchema;
 exports.MessageMetaSchema = MessageMetaSchema;
 exports.OpenAIConfigSchema = OpenAIConfigSchema;
@@ -2336,9 +2618,16 @@ exports.QueryKnowledgeParamsSchema = QueryKnowledgeParamsSchema;
 exports.RESOLVED_RUNTIME_PROFILE_SCHEMA_VERSION = RESOLVED_RUNTIME_PROFILE_SCHEMA_VERSION;
 exports.ReadFileRequestSchema = ReadFileRequestSchema;
 exports.ReadFileResponseSchema = ReadFileResponseSchema;
+exports.ReconnectMcpServerRequestSchema = ReconnectMcpServerRequestSchema;
+exports.ReconnectMcpServerResponseSchema = ReconnectMcpServerResponseSchema;
+exports.RemoveMcpServerRequestSchema = RemoveMcpServerRequestSchema;
+exports.RemoveMcpServerResponseSchema = RemoveMcpServerResponseSchema;
+exports.RenameSessionRequestSchema = RenameSessionRequestSchema;
+exports.RenameSessionResponseSchema = RenameSessionResponseSchema;
 exports.ResolvedRuntimeProfileSchema = ResolvedRuntimeProfileSchema;
 exports.RuntimeProfileSourceSchema = RuntimeProfileSourceSchema;
 exports.RuntimeProfileTrustSchema = RuntimeProfileTrustSchema;
+exports.SdkSessionInfoSchema = SdkSessionInfoSchema;
 exports.SessionEventCreatedSchema = SessionEventCreatedSchema;
 exports.SessionEventReportSchema = SessionEventReportSchema;
 exports.SessionEventSummarySchema = SessionEventSummarySchema;
@@ -2348,6 +2637,8 @@ exports.SessionMessageSchema = SessionMessageSchema;
 exports.SessionProtocolMessageSchema = SessionProtocolMessageSchema;
 exports.SetColorRequestSchema = SetColorRequestSchema;
 exports.SetColorResponseSchema = SetColorResponseSchema;
+exports.SetMcpServersRequestSchema = SetMcpServersRequestSchema;
+exports.SetMcpServersResponseSchema = SetMcpServersResponseSchema;
 exports.SkillContentSchema = SkillContentSchema;
 exports.SkillSummarySchema = SkillSummarySchema;
 exports.TailscaleInfoSchema = TailscaleInfoSchema;
@@ -2362,6 +2653,8 @@ exports.TaskTriggerDataSchema = TaskTriggerDataSchema;
 exports.TaskTriggerTypeSchema = TaskTriggerTypeSchema;
 exports.TmuxConfigSchema = TmuxConfigSchema;
 exports.TogetherAIConfigSchema = TogetherAIConfigSchema;
+exports.ToggleMcpServerRequestSchema = ToggleMcpServerRequestSchema;
+exports.ToggleMcpServerResponseSchema = ToggleMcpServerResponseSchema;
 exports.TunnelEntrySchema = TunnelEntrySchema;
 exports.TunnelProviderInfoSchema = TunnelProviderInfoSchema;
 exports.TunnelStateSchema = TunnelStateSchema;
@@ -2380,6 +2673,7 @@ exports.VersionedNullableEncryptedValueSchema = VersionedNullableEncryptedValueS
 exports.VoiceTokenAllowedSchema = VoiceTokenAllowedSchema;
 exports.VoiceTokenDeniedSchema = VoiceTokenDeniedSchema;
 exports.VoiceTokenResponseSchema = VoiceTokenResponseSchema;
+exports.createEmptyMcpRegistry = createEmptyMcpRegistry;
 exports.createEnvelope = createEnvelope;
 exports.createResolvedRuntimeProfile = createResolvedRuntimeProfile;
 exports.getBuiltInAIBackendProfile = getBuiltInAIBackendProfile;
@@ -2394,6 +2688,8 @@ exports.isHappyMcpToolName = isHappyMcpToolName;
 exports.isTrustedRuntimeProfile = isTrustedRuntimeProfile;
 exports.normalizeHappyMcpToolName = normalizeHappyMcpToolName;
 exports.normalizeResolvedRuntimeProfile = normalizeResolvedRuntimeProfile;
+exports.parseMcpRegistry = parseMcpRegistry;
+exports.registryToSdkConfig = registryToSdkConfig;
 exports.resolveCodexResolvedBackend = resolveCodexResolvedBackend;
 exports.resolveCodexResumableThreadId = resolveCodexResumableThreadId;
 exports.resolveRequestedCodexBackend = resolveRequestedCodexBackend;
@@ -2409,6 +2705,7 @@ exports.sessionProgressStateSchema = sessionProgressStateSchema;
 exports.sessionProgressTodoSchema = sessionProgressTodoSchema;
 exports.sessionProgressTodoStatusSchema = sessionProgressTodoStatusSchema;
 exports.sessionPromptSuggestionEventSchema = sessionPromptSuggestionEventSchema;
+exports.sessionRateLimitEventSchema = sessionRateLimitEventSchema;
 exports.sessionRoleSchema = sessionRoleSchema;
 exports.sessionServiceMessageEventSchema = sessionServiceMessageEventSchema;
 exports.sessionStartEventSchema = sessionStartEventSchema;
@@ -2422,6 +2719,7 @@ exports.sessionTaskEndEventSchema = sessionTaskEndEventSchema;
 exports.sessionTaskLogEventSchema = sessionTaskLogEventSchema;
 exports.sessionTaskProgressEventSchema = sessionTaskProgressEventSchema;
 exports.sessionTaskStartEventSchema = sessionTaskStartEventSchema;
+exports.sessionTaskUpdatedEventSchema = sessionTaskUpdatedEventSchema;
 exports.sessionTextDeltaEventSchema = sessionTextDeltaEventSchema;
 exports.sessionTextEventSchema = sessionTextEventSchema;
 exports.sessionToolCallEndEventSchema = sessionToolCallEndEventSchema;