agl 22.0.4 → 23.0.0

package/ts_web/classes.controllersocketclient.ts

@@ -3,6 +3,7 @@ import * as interfaces from '../ts_interfaces/index.js';
 
 const controllerEventTypes: ReadonlySet<string> = new Set(interfaces.controllerEventTypes);
 const controllerRequestTimeoutMs = 35_000;
+const textEncoder = new TextEncoder();
 
 const isQualifiedRuntimeId = (
   valueArg: unknown,
@@ -16,7 +17,7 @@ const isQualifiedRuntimeId = (
     && allowedHarnessIdsArg.has(candidate.harnessId as interfaces.TControllerHarnessId)
     && typeof candidate.nativeId === 'string'
     && candidate.nativeId.length > 0
-    && new TextEncoder().encode(candidate.nativeId).byteLength <= 512
+    && textEncoder.encode(candidate.nativeId).byteLength <= 512
   );
 };
 
@@ -36,6 +37,14 @@ const textUpdateKeys = new Set([
   'sessionId', 'messageId', 'partId', 'text', 'status', 'order',
   'sourceUpdatedAt', 'revision', 'streamEpoch',
 ]);
+const messageDeltaKeys = new Set([
+  'sessionId', 'messageId', 'partId', 'delta', 'baseTextUtf8Bytes', 'textUtf8Bytes',
+  'order', 'sourceUpdatedAt', 'revision', 'streamEpoch',
+]);
+const messageDeltaRequiredKeys = [
+  'sessionId', 'messageId', 'partId', 'delta', 'baseTextUtf8Bytes', 'textUtf8Bytes',
+  'sourceUpdatedAt', 'revision', 'streamEpoch',
+] as const;
 
 const isOptionalFiniteNumber = (valueArg: unknown): boolean => (
   valueArg === undefined || (typeof valueArg === 'number' && Number.isFinite(valueArg))
@@ -53,12 +62,26 @@ const isTranscriptOrder = (valueArg: unknown): boolean => {
 
 const serializedBytesWithin = (valueArg: unknown, byteLimitArg: number): boolean => {
   try {
-    return new TextEncoder().encode(JSON.stringify(valueArg)).byteLength <= byteLimitArg;
+    return textEncoder.encode(JSON.stringify(valueArg)).byteLength <= byteLimitArg;
   } catch {
     return false;
   }
 };
 
+const hasExactDeltaUtf8Coordinates = (
+  deltaArg: string,
+  baseTextUtf8BytesArg: number,
+  textUtf8BytesArg: number,
+): boolean => {
+  const encodedDeltaBytes = textEncoder.encode(deltaArg).byteLength;
+  const increment = textUtf8BytesArg - baseTextUtf8BytesArg;
+  if (increment === encodedDeltaBytes) return true;
+  const firstCodeUnit = deltaArg.charCodeAt(0);
+  return firstCodeUnit >= 0xDC00
+    && firstCodeUnit <= 0xDFFF
+    && increment === encodedDeltaBytes - 2;
+};
+
 const isDraftAttachment = (valueArg: unknown): valueArg is interfaces.IControllerDraftAttachment => {
   if (!valueArg || typeof valueArg !== 'object' || Array.isArray(valueArg)) return false;
   const candidate = valueArg as Partial<interfaces.IControllerDraftAttachment>;
@@ -68,13 +91,13 @@ const isDraftAttachment = (valueArg: unknown): valueArg is interfaces.IControlle
     ))
     || typeof candidate.id !== 'string'
     || candidate.id.length === 0
-    || new TextEncoder().encode(candidate.id).byteLength > 512
+    || textEncoder.encode(candidate.id).byteLength > 512
     || typeof candidate.name !== 'string'
     || candidate.name.length === 0
-    || new TextEncoder().encode(candidate.name).byteLength > 255
+    || textEncoder.encode(candidate.name).byteLength > 255
     || typeof candidate.mediaType !== 'string'
     || candidate.mediaType.length === 0
-    || new TextEncoder().encode(candidate.mediaType).byteLength > 129
+    || textEncoder.encode(candidate.mediaType).byteLength > 129
     || !Number.isSafeInteger(candidate.size)
     || (candidate.size ?? -1) < 0
     || (candidate.size ?? 0) > interfaces.controllerMaxDraftAttachmentBytes
@@ -101,7 +124,7 @@ const isSessionDraftUpdate = (
     || (candidate.text === undefined && candidate.attachments === undefined)
     || (candidate.text !== undefined && (
       typeof candidate.text !== 'string'
-      || new TextEncoder().encode(candidate.text).byteLength > interfaces.controllerMaxDraftTextBytes
+      || textEncoder.encode(candidate.text).byteLength > interfaces.controllerMaxDraftTextBytes
     ))
     || (candidate.attachments !== undefined && (
       !Array.isArray(candidate.attachments)
@@ -130,7 +153,7 @@ const isControllerToolExecution = (
     && isQualifiedRuntimeId(candidate.callId, executionHarnessIds)
     && typeof candidate.toolName === 'string'
     && candidate.toolName.length > 0
-    && new TextEncoder().encode(candidate.toolName).byteLength <= 2048
+    && textEncoder.encode(candidate.toolName).byteLength <= 2048
     && ['pending', 'running', 'completed', 'error'].includes(candidate.status ?? '')
     && (candidate.exitCode === undefined
       || (Number.isSafeInteger(candidate.exitCode) && candidate.exitCode >= 0))
@@ -166,7 +189,7 @@ const isControllerReasoningUpdate = (
     && isQualifiedRuntimeId(candidate.messageId, harnessIds)
     && isQualifiedRuntimeId(candidate.partId, harnessIds)
     && typeof candidate.text === 'string'
-    && new TextEncoder().encode(candidate.text).byteLength
+    && textEncoder.encode(candidate.text).byteLength
       <= interfaces.controllerMaxLiveMessageEventBytes
     && ['running', 'completed', 'cancelled'].includes(candidate.status ?? '')
     && (candidate.order === undefined || isTranscriptOrder(candidate.order))
@@ -193,7 +216,7 @@ const isControllerTextUpdate = (
     && isQualifiedRuntimeId(candidate.messageId, harnessIds)
     && isQualifiedRuntimeId(candidate.partId, harnessIds)
     && typeof candidate.text === 'string'
-    && new TextEncoder().encode(candidate.text).byteLength
+    && textEncoder.encode(candidate.text).byteLength
       <= interfaces.controllerMaxLiveMessageEventBytes
     && ['running', 'completed'].includes(candidate.status ?? '')
     && (candidate.order === undefined || isTranscriptOrder(candidate.order))
@@ -207,6 +230,48 @@ const isControllerTextUpdate = (
     && (candidate.streamEpoch ?? -1) >= 0;
 };
 
+const isControllerMessageDelta = (
+  valueArg: unknown,
+): valueArg is interfaces.IControllerReasoningDelta | interfaces.IControllerTextDelta => {
+  if (!valueArg || typeof valueArg !== 'object' || Array.isArray(valueArg)) return false;
+  const candidate = valueArg as Partial<interfaces.IControllerReasoningDelta>;
+  if (!isQualifiedRuntimeId(candidate.sessionId, sessionHarnessIds)) return false;
+  const harnessIds: ReadonlySet<interfaces.TControllerHarnessId> = new Set([
+    candidate.sessionId.harnessId,
+  ]);
+  const keys = Object.keys(candidate);
+  const hasOrder = Object.prototype.hasOwnProperty.call(candidate, 'order');
+  return keys.length === messageDeltaRequiredKeys.length + (hasOrder ? 1 : 0)
+    && keys.every((key) => messageDeltaKeys.has(key))
+    && messageDeltaRequiredKeys.every((key) => (
+      Object.prototype.hasOwnProperty.call(candidate, key)
+    ))
+    && isQualifiedRuntimeId(candidate.messageId, harnessIds)
+    && isQualifiedRuntimeId(candidate.partId, harnessIds)
+    && typeof candidate.delta === 'string'
+    && candidate.delta.length > 0
+    && textEncoder.encode(candidate.delta).byteLength
+      <= interfaces.controllerMaxLiveMessageEventBytes
+    && Number.isSafeInteger(candidate.baseTextUtf8Bytes)
+    && (candidate.baseTextUtf8Bytes ?? -1) >= 0
+    && Number.isSafeInteger(candidate.textUtf8Bytes)
+    && (candidate.textUtf8Bytes ?? -1) > (candidate.baseTextUtf8Bytes ?? -1)
+    && hasExactDeltaUtf8Coordinates(
+      candidate.delta,
+      candidate.baseTextUtf8Bytes!,
+      candidate.textUtf8Bytes!,
+    )
+    && (!hasOrder || isTranscriptOrder(candidate.order))
+    && (candidate.sessionId.harnessId !== 'flex' || hasOrder)
+    && (candidate.sessionId.harnessId !== 'opencode' || !hasOrder)
+    && typeof candidate.sourceUpdatedAt === 'number'
+    && Number.isFinite(candidate.sourceUpdatedAt)
+    && Number.isSafeInteger(candidate.revision)
+    && (candidate.revision ?? -1) >= 0
+    && Number.isSafeInteger(candidate.streamEpoch)
+    && (candidate.streamEpoch ?? -1) >= 0;
+};
+
 export interface IControllerSocketClientOptions {
   onStatus: (status: plugins.typedsocket.TConnectionStatus) => void;
   onControllerEvent: (event: interfaces.IControllerEvent) => void | Promise<void>;
@@ -240,9 +305,18 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
   const hasSessionMarker = candidate.sessionStatus !== undefined || candidate.sessionError !== undefined;
   const hasToolExecution = candidate.toolExecution !== undefined;
   const hasReasoningUpdate = candidate.reasoningUpdate !== undefined;
+  const hasReasoningDelta = candidate.reasoningDelta !== undefined;
   const hasTextUpdate = candidate.textUpdate !== undefined;
+  const hasTextDelta = candidate.textDelta !== undefined;
   const hasDraftUpdate = candidate.sessionDraftUpdate !== undefined;
   const hasUpgrade = candidate.upgrade !== undefined;
+  const livePayloadCount = [
+    hasToolExecution,
+    hasReasoningUpdate,
+    hasReasoningDelta,
+    hasTextUpdate,
+    hasTextDelta,
+  ].filter(Boolean).length;
   return (
     typeof candidate.type === 'string'
     && controllerEventTypes.has(candidate.type as interfaces.TControllerEventType)
@@ -269,14 +343,19 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
     && (candidate.toolExecution === undefined || isControllerToolExecution(candidate.toolExecution))
     && (candidate.reasoningUpdate === undefined
       || isControllerReasoningUpdate(candidate.reasoningUpdate))
+    && (candidate.reasoningDelta === undefined
+      || isControllerMessageDelta(candidate.reasoningDelta))
     && (candidate.textUpdate === undefined || isControllerTextUpdate(candidate.textUpdate))
+    && (candidate.textDelta === undefined || isControllerMessageDelta(candidate.textDelta))
     && (candidate.sessionDraftUpdate === undefined
       || isSessionDraftUpdate(candidate.sessionDraftUpdate))
+    && livePayloadCount <= 1
     && (!hasToolExecution || serializedBytesWithin(
       candidate,
       interfaces.controllerMaxToolEventBytes,
     ))
-    && (!(hasReasoningUpdate || hasTextUpdate) || serializedBytesWithin(
+    && (!(hasReasoningUpdate || hasReasoningDelta || hasTextUpdate || hasTextDelta)
+      || serializedBytesWithin(
       candidate,
       interfaces.controllerMaxLiveMessageEventBytes,
     ))
@@ -312,6 +391,16 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
         === interfaces.controllerRuntimeIdKey(candidate.reasoningUpdate!.sessionId)
     ))
     && (hasReasoningUpdate || candidate.type !== 'session.reasoning.updated')
+    && (!hasReasoningDelta || (
+      candidate.type === 'session.reasoning.delta'
+      && (candidate.harnessId === 'opencode' || candidate.harnessId === 'flex')
+      && typeof candidate.projectId === 'string'
+      && isQualifiedRuntimeId(candidate.sessionId, sessionHarnessIds)
+      && candidate.harnessId === candidate.reasoningDelta!.sessionId.harnessId
+      && interfaces.controllerRuntimeIdKey(candidate.sessionId)
+        === interfaces.controllerRuntimeIdKey(candidate.reasoningDelta!.sessionId)
+    ))
+    && (hasReasoningDelta || candidate.type !== 'session.reasoning.delta')
     && (!hasTextUpdate || (
       candidate.type === 'session.text.updated'
       && (candidate.harnessId === 'opencode' || candidate.harnessId === 'flex')
@@ -322,6 +411,16 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
         === interfaces.controllerRuntimeIdKey(candidate.textUpdate!.sessionId)
     ))
     && (hasTextUpdate || candidate.type !== 'session.text.updated')
+    && (!hasTextDelta || (
+      candidate.type === 'session.text.delta'
+      && (candidate.harnessId === 'opencode' || candidate.harnessId === 'flex')
+      && typeof candidate.projectId === 'string'
+      && isQualifiedRuntimeId(candidate.sessionId, sessionHarnessIds)
+      && candidate.harnessId === candidate.textDelta!.sessionId.harnessId
+      && interfaces.controllerRuntimeIdKey(candidate.sessionId)
+        === interfaces.controllerRuntimeIdKey(candidate.textDelta!.sessionId)
+    ))
+    && (hasTextDelta || candidate.type !== 'session.text.delta')
     && (!hasDraftUpdate || (
       candidate.type === 'session.draft.changed'
       && typeof candidate.projectId === 'string'
@@ -337,7 +436,9 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
       candidate.type === 'harness.changed'
       && (candidate.harnessId === 'opencode' || candidate.harnessId === 'flex')
       && !hasReasoningUpdate
+      && !hasReasoningDelta
       && !hasTextUpdate
+      && !hasTextDelta
     ))
     && (candidate.upgrade === undefined || (
       typeof candidate.upgrade === 'object'
@@ -345,10 +446,10 @@ export const isControllerEvent = (valueArg: unknown): valueArg is interfaces.ICo
       && Object.keys(candidate.upgrade).length === 3
       && typeof candidate.upgrade.fromVersion === 'string'
       && candidate.upgrade.fromVersion.length > 0
-      && new TextEncoder().encode(candidate.upgrade.fromVersion).byteLength <= 128
+      && textEncoder.encode(candidate.upgrade.fromVersion).byteLength <= 128
       && typeof candidate.upgrade.toVersion === 'string'
       && candidate.upgrade.toVersion.length > 0
-      && new TextEncoder().encode(candidate.upgrade.toVersion).byteLength <= 128
+      && textEncoder.encode(candidate.upgrade.toVersion).byteLength <= 128
       && ['preparing', 'pausing', 'installing', 'restarting', 'continuing', 'completed', 'failed']
         .includes(candidate.upgrade.phase)
     ))