@kmmao/happy-wire 0.1.0

package/dist/index.mjs

@@ -0,0 +1,212 @@
+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 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(),
+  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,
+  model: z.string().optional(),
+  usage: z.object({
+    input_tokens: z.number(),
+    output_tokens: z.number(),
+    cache_creation_input_tokens: z.number().optional(),
+    cache_read_input_tokens: z.number().optional()
+  }).optional(),
+  durationMs: z.number().optional()
+});
+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(),
+  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 } : {},
+    ev
+  });
+}
+
+const MessageMetaSchema = z.object({
+  sentFrom: z.string().optional(),
+  permissionMode: z.enum(["default", "acceptEdits", "bypassPermissions", "plan", "read-only", "safe-yolo", "yolo"]).optional(),
+  model: z.string().nullable().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(),
+  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;
+
+export { AgentMessageSchema, ApiMessageSchema, ApiUpdateMachineStateSchema, ApiUpdateNewMessageSchema, ApiUpdateSessionStateSchema, CoreUpdateBodySchema, CoreUpdateContainerSchema, LegacyMessageContentSchema, MessageContentSchema, MessageMetaSchema, SessionMessageContentSchema, SessionMessageSchema, SessionProtocolMessageSchema, UpdateBodySchema, UpdateMachineBodySchema, UpdateNewMessageBodySchema, UpdateSchema, UpdateSessionBodySchema, UserMessageSchema, VersionedEncryptedValueSchema, VersionedMachineEncryptedValueSchema, VersionedNullableEncryptedValueSchema, createEnvelope, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema };

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@kmmao/happy-wire",
+  "version": "0.1.0",
+  "description": "Shared message wire types and Zod schemas for Happy clients and services",
+  "author": "kmmao",
+  "license": "MIT",
+  "type": "module",
+  "homepage": "https://github.com/kmmao/happy/tree/main/packages/happy-wire",
+  "bugs": "https://github.com/kmmao/happy/issues",
+  "repository": "kmmao/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"
+  ],
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "shx rm -rf dist && npx tsc --noEmit && pkgroll",
+    "test": "$npm_execpath run build && vitest run",
+    "prepublishOnly": "$npm_execpath run build && $npm_execpath run test",
+    "release": "npx --no-install release-it"
+  },
+  "dependencies": {
+    "@paralleldrive/cuid2": "^2.2.2",
+    "zod": "3.25.76"
+  },
+  "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"
+  },
+  "packageManager": "yarn@1.22.22"
+}