@sokdak/happy-wire 0.1.0-main.5

package/dist/index.mjs

@@ -0,0 +1,404 @@
+import * as z from 'zod';
+import { isCuid, createId } from '@paralleldrive/cuid2';
+
+const sessionRoleSchema = z.enum(["user", "agent"]);
+const sessionTextEventSchema = z.object({
+  t: z.literal("text"),
+  text: z.string(),
+  thinking: z.boolean().optional()
+});
+const sessionUsageSchema = z.object({
+  input_tokens: z.number().int().nonnegative(),
+  cache_creation_input_tokens: z.number().int().nonnegative().optional(),
+  cache_read_input_tokens: z.number().int().nonnegative().optional(),
+  output_tokens: z.number().int().nonnegative(),
+  context_window: z.number().int().positive().optional(),
+  service_tier: z.string().optional()
+}).passthrough();
+const sessionServiceMessageEventSchema = z.object({
+  t: z.literal("service"),
+  text: z.string()
+});
+const sessionToolCallStartEventSchema = z.object({
+  t: z.literal("tool-call-start"),
+  call: z.string(),
+  name: z.string(),
+  title: z.string(),
+  description: z.string(),
+  args: z.record(z.string(), z.unknown())
+});
+const sessionToolCallEndEventSchema = z.object({
+  t: z.literal("tool-call-end"),
+  call: z.string()
+});
+const sessionFileEventSchema = z.object({
+  t: z.literal("file"),
+  ref: z.string(),
+  name: z.string(),
+  size: z.number(),
+  mimeType: z.string().optional(),
+  image: z.object({
+    width: z.number(),
+    height: z.number(),
+    thumbhash: z.string()
+  }).optional()
+});
+const sessionTurnStartEventSchema = z.object({
+  t: z.literal("turn-start")
+});
+const sessionStartEventSchema = z.object({
+  t: z.literal("start"),
+  title: z.string().optional()
+});
+const sessionTurnEndStatusSchema = z.enum(["completed", "failed", "cancelled"]);
+const sessionTurnEndEventSchema = z.object({
+  t: z.literal("turn-end"),
+  status: sessionTurnEndStatusSchema
+});
+const sessionStopEventSchema = z.object({
+  t: z.literal("stop")
+});
+const sessionEventSchema = z.discriminatedUnion("t", [
+  sessionTextEventSchema,
+  sessionServiceMessageEventSchema,
+  sessionToolCallStartEventSchema,
+  sessionToolCallEndEventSchema,
+  sessionFileEventSchema,
+  sessionTurnStartEventSchema,
+  sessionStartEventSchema,
+  sessionTurnEndEventSchema,
+  sessionStopEventSchema
+]);
+const sessionEnvelopeSchema = z.object({
+  id: z.string(),
+  time: z.number(),
+  role: sessionRoleSchema,
+  turn: z.string().optional(),
+  subagent: z.string().refine((value) => isCuid(value), {
+    message: "subagent must be a cuid2 value"
+  }).optional(),
+  // Underlying agent-protocol message id (e.g. Claude's `uuid` in the
+  // session JSONL). Set on text-bearing envelopes so the app can let
+  // users pick a precise rewind point for session fork / duplicate.
+  claudeUuid: z.string().min(1).optional(),
+  // Codex app-server item id for this envelope. Used as the precise
+  // rollback point for Codex thread duplicate/fork-from-message.
+  codexItemId: z.string().min(1).optional(),
+  // Optional model usage carried by the source agent message. Consumers use
+  // this to update session context meters without rendering a separate row.
+  usage: sessionUsageSchema.optional(),
+  ev: sessionEventSchema
+}).superRefine((envelope, ctx) => {
+  if (envelope.ev.t === "service" && envelope.role !== "agent") {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      message: 'service events must use role "agent"',
+      path: ["role"]
+    });
+  }
+  if ((envelope.ev.t === "start" || envelope.ev.t === "stop") && envelope.role !== "agent") {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      message: `${envelope.ev.t} events must use role "agent"`,
+      path: ["role"]
+    });
+  }
+});
+function createEnvelope(role, ev, opts = {}) {
+  return sessionEnvelopeSchema.parse({
+    id: opts.id ?? createId(),
+    time: opts.time ?? Date.now(),
+    role,
+    ...opts.turn ? { turn: opts.turn } : {},
+    ...opts.subagent ? { subagent: opts.subagent } : {},
+    ...opts.claudeUuid ? { claudeUuid: opts.claudeUuid } : {},
+    ...opts.codexItemId ? { codexItemId: opts.codexItemId } : {},
+    ...opts.usage ? { usage: opts.usage } : {},
+    ev
+  });
+}
+
+const MessageMetaSchema = z.object({
+  sentFrom: z.string().optional(),
+  // Native clients may publish their own mode codes (for example Rig's
+  // auto/workspace_write/read_only/full_access), so this stays open-ended.
+  permissionMode: z.string().optional(),
+  model: z.string().nullable().optional(),
+  modelProviderId: z.string().optional(),
+  fallbackModel: z.string().nullable().optional(),
+  customSystemPrompt: z.string().nullable().optional(),
+  appendSystemPrompt: z.string().nullable().optional(),
+  allowedTools: z.array(z.string()).nullable().optional(),
+  disallowedTools: z.array(z.string()).nullable().optional(),
+  effort: z.string().nullable().optional(),
+  displayText: z.string().optional()
+});
+
+const UserMessageSchema = z.object({
+  role: z.literal("user"),
+  content: z.object({
+    type: z.literal("text"),
+    text: z.string()
+  }),
+  localKey: z.string().optional(),
+  meta: MessageMetaSchema.optional()
+});
+const AgentMessageSchema = z.object({
+  role: z.literal("agent"),
+  content: z.object({
+    type: z.string()
+  }).passthrough(),
+  meta: MessageMetaSchema.optional()
+});
+const LegacyMessageContentSchema = z.discriminatedUnion("role", [UserMessageSchema, AgentMessageSchema]);
+
+const SessionMessageContentSchema = z.object({
+  c: z.string(),
+  t: z.literal("encrypted")
+});
+const SessionMessageSchema = z.object({
+  id: z.string(),
+  seq: z.number(),
+  localId: z.string().nullish(),
+  content: SessionMessageContentSchema,
+  createdAt: z.number(),
+  updatedAt: z.number()
+});
+const SessionProtocolMessageSchema = z.object({
+  role: z.literal("session"),
+  content: sessionEnvelopeSchema,
+  meta: MessageMetaSchema.optional()
+});
+const MessageContentSchema = z.discriminatedUnion("role", [
+  UserMessageSchema,
+  AgentMessageSchema,
+  SessionProtocolMessageSchema
+]);
+const VersionedEncryptedValueSchema = z.object({
+  version: z.number(),
+  value: z.string()
+});
+const VersionedNullableEncryptedValueSchema = z.object({
+  version: z.number(),
+  value: z.string().nullable()
+});
+const UpdateNewMessageBodySchema = z.object({
+  t: z.literal("new-message"),
+  sid: z.string(),
+  message: SessionMessageSchema
+});
+const UpdateSessionBodySchema = z.object({
+  t: z.literal("update-session"),
+  id: z.string(),
+  metadata: VersionedEncryptedValueSchema.nullish(),
+  agentState: VersionedNullableEncryptedValueSchema.nullish()
+});
+const VersionedMachineEncryptedValueSchema = z.object({
+  version: z.number(),
+  value: z.string()
+});
+const UpdateMachineBodySchema = z.object({
+  t: z.literal("update-machine"),
+  machineId: z.string(),
+  metadata: VersionedMachineEncryptedValueSchema.nullish(),
+  daemonState: VersionedMachineEncryptedValueSchema.nullish(),
+  active: z.boolean().optional(),
+  activeAt: z.number().optional()
+});
+const CoreUpdateBodySchema = z.discriminatedUnion("t", [
+  UpdateNewMessageBodySchema,
+  UpdateSessionBodySchema,
+  UpdateMachineBodySchema
+]);
+const CoreUpdateContainerSchema = z.object({
+  id: z.string(),
+  seq: z.number(),
+  body: CoreUpdateBodySchema,
+  createdAt: z.number()
+});
+const ApiMessageSchema = SessionMessageSchema;
+const ApiUpdateNewMessageSchema = UpdateNewMessageBodySchema;
+const ApiUpdateSessionStateSchema = UpdateSessionBodySchema;
+const ApiUpdateMachineStateSchema = UpdateMachineBodySchema;
+const UpdateBodySchema = UpdateNewMessageBodySchema;
+const UpdateSchema = CoreUpdateContainerSchema;
+
+const TASK_NOTIFICATION_OPEN = "<task-notification>";
+const TASK_NOTIFICATION_CLOSE = "</task-notification>";
+function leadingTaskNotificationEnd(text) {
+  if (!text.startsWith(TASK_NOTIFICATION_OPEN)) {
+    return null;
+  }
+  let depth = 1;
+  let cursor = TASK_NOTIFICATION_OPEN.length;
+  while (depth > 0) {
+    const nextOpen = text.indexOf(TASK_NOTIFICATION_OPEN, cursor);
+    const nextClose = text.indexOf(TASK_NOTIFICATION_CLOSE, cursor);
+    if (nextClose === -1) {
+      return null;
+    }
+    if (nextOpen !== -1 && nextOpen < nextClose) {
+      depth += 1;
+      cursor = nextOpen + TASK_NOTIFICATION_OPEN.length;
+      continue;
+    }
+    depth -= 1;
+    cursor = nextClose + TASK_NOTIFICATION_CLOSE.length;
+  }
+  return cursor;
+}
+function stripLeadingTaskNotificationWrappers(text) {
+  let remainder = text.trimStart();
+  let stripped = false;
+  while (remainder.startsWith(TASK_NOTIFICATION_OPEN)) {
+    const wrapperEnd = leadingTaskNotificationEnd(remainder);
+    if (wrapperEnd === null) {
+      break;
+    }
+    stripped = true;
+    remainder = remainder.slice(wrapperEnd).trimStart();
+  }
+  return stripped ? remainder : text;
+}
+
+const VoiceConversationGrantedSchema = z.object({
+  allowed: z.literal(true),
+  conversationToken: z.string(),
+  conversationId: z.string(),
+  agentId: z.string(),
+  elevenUserId: z.string(),
+  usedSeconds: z.number(),
+  limitSeconds: z.number()
+});
+const VoiceConversationDeniedSchema = z.object({
+  allowed: z.literal(false),
+  reason: z.enum(["voice_hard_limit_reached", "subscription_required", "voice_conversation_limit_reached"]),
+  usedSeconds: z.number(),
+  limitSeconds: z.number(),
+  agentId: z.string()
+});
+const VoiceConversationResponseSchema = z.discriminatedUnion("allowed", [
+  VoiceConversationGrantedSchema,
+  VoiceConversationDeniedSchema
+]);
+const VoiceUsageResponseSchema = z.object({
+  usedSeconds: z.number(),
+  limitSeconds: z.number(),
+  conversationCount: z.number(),
+  conversationLimit: z.number(),
+  elevenUserId: z.string()
+});
+
+const RigProviderSchema = z.object({
+  id: z.string(),
+  kind: z.string(),
+  name: z.string()
+}).passthrough();
+const RigModelSchema = z.object({
+  id: z.string(),
+  code: z.string(),
+  name: z.string(),
+  value: z.string(),
+  providerId: z.string(),
+  providerKind: z.string(),
+  providerName: z.string(),
+  provider: RigProviderSchema,
+  contextWindow: z.number().optional(),
+  serviceTiers: z.array(z.string()),
+  thinkingLevels: z.array(z.string()),
+  defaultThinkingLevel: z.string()
+}).passthrough();
+const RigOperatingModeKindSchema = z.string();
+const RigOperatingModeSchema = z.object({
+  code: z.string(),
+  value: z.string(),
+  description: z.string(),
+  kind: RigOperatingModeKindSchema
+}).passthrough();
+const RigCapabilitiesSchema = z.object({
+  abort: z.boolean(),
+  attachments: z.object({
+    enabled: z.boolean(),
+    maxBytes: z.number(),
+    mediaTypes: z.array(z.string())
+  }).passthrough(),
+  files: z.object({
+    browse: z.boolean(),
+    read: z.boolean(),
+    search: z.boolean(),
+    write: z.boolean()
+  }).passthrough(),
+  modelSelection: z.boolean(),
+  reasoningSelection: z.boolean(),
+  permissionModeSelection: z.boolean(),
+  resume: z.boolean(),
+  rpcMethods: z.array(z.string()),
+  shell: z.boolean(),
+  steering: z.boolean()
+}).passthrough();
+const RigActivitySchema = z.object({
+  subagents: z.object({
+    running: z.number(),
+    queued: z.number(),
+    total: z.number()
+  }).passthrough(),
+  workflows: z.object({
+    running: z.number(),
+    total: z.number()
+  }).passthrough(),
+  processes: z.object({
+    running: z.number()
+  }).passthrough(),
+  tasks: z.object({
+    pending: z.number(),
+    inProgress: z.number(),
+    completed: z.number(),
+    total: z.number()
+  }).passthrough()
+}).passthrough();
+const RigMetadataV1Schema = z.object({
+  // Parse later additive revisions with the v1-compatible fields we know.
+  rigMetadataVersion: z.number().int().min(1),
+  client: z.object({
+    id: z.literal("rig"),
+    name: z.string(),
+    version: z.string()
+  }).passthrough(),
+  provider: RigProviderSchema,
+  providers: z.array(RigProviderSchema),
+  model: z.object({
+    providerId: z.string(),
+    id: z.string()
+  }).passthrough(),
+  models: z.array(RigModelSchema),
+  currentModelProviderId: z.string(),
+  currentModelCode: z.string(),
+  permissionMode: z.string(),
+  currentOperatingModeCode: z.string(),
+  operatingModes: z.array(RigOperatingModeSchema),
+  reasoning: z.object({
+    current: z.string().nullable(),
+    levels: z.array(z.string())
+  }).passthrough(),
+  currentThoughtLevelCode: z.string().optional(),
+  thoughtLevels: z.array(z.object({
+    code: z.string(),
+    value: z.string()
+  }).passthrough()),
+  session: z.object({
+    status: z.string(),
+    permissionMode: z.string(),
+    modelLocked: z.boolean(),
+    serviceTier: z.string().optional()
+  }).passthrough(),
+  capabilities: RigCapabilitiesSchema,
+  activity: RigActivitySchema,
+  mcpServers: z.array(z.object({
+    name: z.string(),
+    status: z.string()
+  }).passthrough()),
+  tools: z.array(z.string()),
+  skills: z.array(z.string())
+}).passthrough();
+
+export { AgentMessageSchema, ApiMessageSchema, ApiUpdateMachineStateSchema, ApiUpdateNewMessageSchema, ApiUpdateSessionStateSchema, CoreUpdateBodySchema, CoreUpdateContainerSchema, LegacyMessageContentSchema, MessageContentSchema, MessageMetaSchema, RigActivitySchema, RigCapabilitiesSchema, RigMetadataV1Schema, RigModelSchema, RigOperatingModeKindSchema, RigOperatingModeSchema, RigProviderSchema, SessionMessageContentSchema, SessionMessageSchema, SessionProtocolMessageSchema, UpdateBodySchema, UpdateMachineBodySchema, UpdateNewMessageBodySchema, UpdateSchema, UpdateSessionBodySchema, UserMessageSchema, VersionedEncryptedValueSchema, VersionedMachineEncryptedValueSchema, VersionedNullableEncryptedValueSchema, VoiceConversationDeniedSchema, VoiceConversationGrantedSchema, VoiceConversationResponseSchema, VoiceUsageResponseSchema, createEnvelope, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema, sessionUsageSchema, stripLeadingTaskNotificationWrappers };

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "@sokdak/happy-wire",
+  "version": "0.1.0-main.5",
+  "description": "Shared message wire types and Zod schemas for Happy clients and services",
+  "author": "Kirill Dubovitskiy",
+  "license": "MIT",
+  "type": "module",
+  "homepage": "https://github.com/slopus/happy/tree/main/packages/happy-wire",
+  "bugs": "https://github.com/slopus/happy/issues",
+  "repository": "slopus/happy",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.cts",
+  "exports": {
+    ".": {
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      },
+      "import": {
+        "types": "./dist/index.d.mts",
+        "default": "./dist/index.mjs"
+      }
+    }
+  },
+  "files": [
+    "dist",
+    "package.json",
+    "README.md"
+  ],
+  "dependencies": {
+    "@paralleldrive/cuid2": "^2.2.2",
+    "zod": "^4.0.0"
+  },
+  "devDependencies": {
+    "@types/node": ">=20",
+    "pkgroll": "^2.14.2",
+    "release-it": "^19.0.6",
+    "shx": "^0.3.3",
+    "typescript": "5.9.3",
+    "vitest": "^3.2.4"
+  },
+  "publishConfig": {
+    "registry": "https://registry.npmjs.org",
+    "access": "public"
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "shx rm -rf dist && tsc --noEmit && pkgroll",
+    "test": "$npm_execpath run build && vitest run",
+    "release": "npx --no-install release-it"
+  }
+}