@ahmadposten/talos-wire 0.1.0

package/dist/index.mjs

@@ -0,0 +1,304 @@
+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(),
+  // The tool's own output. Optional for back-compat: older CLIs send only
+  // `call`, and clients must keep treating a missing result as "no output".
+  // Without this the protocol drops every result — which silently broke the
+  // todo panel, since TaskCreate reports the task id only in its result.
+  result: z.unknown().optional(),
+  isError: z.boolean().optional()
+});
+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(),
+  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 } : {},
+    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(),
+  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 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 encryptionContexts = {
+  content: "Happy EnCoder",
+  analytics: "Happy Coder",
+  blobs: "Happy Blobs",
+  serverTokens: "happy-server-tokens"
+};
+const legacyServerBanner = "Welcome to Happy Server!";
+const authenticationContexts = {
+  persistent: "handy",
+  github: "github-happy"
+};
+const rpcMethods = {
+  spawnSession: "spawn-happy-session",
+  resumeSession: "resume-happy-session"
+};
+const legacyInstallation = {
+  accountLinkPrefix: "happy:///account?",
+  homeDirectory: ".happy",
+  homeEnvironment: "HAPPY_HOME_DIR",
+  serverEnvironment: "HAPPY_SERVER_URL",
+  webappEnvironment: "HAPPY_WEBAPP_URL",
+  defaultServerUrl: "https://api.happy.ahposten.com",
+  defaultWebappUrl: "https://app.happy.engineering"
+};
+const metadataAliases = {
+  happyCliVersion: "talosCliVersion",
+  happyHomeDir: "talosHomeDir",
+  happyLibDir: "talosLibDir",
+  happyToolsDir: "talosToolsDir",
+  requiresHappyAgentAuth: "requiresTalosAgentAuth",
+  happyAgentAuthenticated: "talosAgentAuthenticated"
+};
+function isRecord(value) {
+  return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+function normalizeMetadata(value) {
+  if (!isRecord(value)) return value;
+  const result = { ...value };
+  for (const [oldKey, key] of Object.entries(metadataAliases)) {
+    if (result[key] === void 0 && result[oldKey] !== void 0) result[key] = result[oldKey];
+    delete result[oldKey];
+  }
+  if (isRecord(result.resumeSupport)) result.resumeSupport = normalizeMetadata(result.resumeSupport);
+  if (result.shutdownSource === "happy-app") result.shutdownSource = "talos-app";
+  if (result.shutdownSource === "happy-cli") result.shutdownSource = "talos-cli";
+  return result;
+}
+function toWireMetadata(value) {
+  if (!isRecord(value)) return value;
+  const result = { ...value };
+  for (const [oldKey, key] of Object.entries(metadataAliases)) {
+    if (result[key] !== void 0) result[oldKey] = result[key];
+  }
+  if (isRecord(result.resumeSupport)) result.resumeSupport = toWireMetadata(result.resumeSupport);
+  if (result.shutdownSource === "talos-app") result.shutdownSource = "happy-app";
+  if (result.shutdownSource === "talos-cli") result.shutdownSource = "happy-cli";
+  return result;
+}
+
+export { AgentMessageSchema, ApiMessageSchema, ApiUpdateMachineStateSchema, ApiUpdateNewMessageSchema, ApiUpdateSessionStateSchema, CoreUpdateBodySchema, CoreUpdateContainerSchema, LegacyMessageContentSchema, MessageContentSchema, MessageMetaSchema, SessionMessageContentSchema, SessionMessageSchema, SessionProtocolMessageSchema, UpdateBodySchema, UpdateMachineBodySchema, UpdateNewMessageBodySchema, UpdateSchema, UpdateSessionBodySchema, UserMessageSchema, VersionedEncryptedValueSchema, VersionedMachineEncryptedValueSchema, VersionedNullableEncryptedValueSchema, VoiceConversationDeniedSchema, VoiceConversationGrantedSchema, VoiceConversationResponseSchema, VoiceUsageResponseSchema, authenticationContexts, createEnvelope, encryptionContexts, legacyInstallation, legacyServerBanner, normalizeMetadata, rpcMethods, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema, toWireMetadata };

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "@ahmadposten/talos-wire",
+  "version": "0.1.0",
+  "description": "Shared message wire types and Zod schemas for Talos clients and services",
+  "author": "Kirill Dubovitskiy",
+  "license": "MIT",
+  "type": "module",
+  "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": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org"
+  },
+  "homepage": "https://talosapp.ai",
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "shx rm -rf dist && tsc --noEmit && pkgroll",
+    "test": "pnpm run build && vitest run",
+    "release": "node ../../scripts/release.cjs wire"
+  }
+}