@byok-sdk/client 0.6.1 → 0.7.0

package/README.md

@@ -122,17 +122,19 @@ contents.
 
 ## Agent egress and explicit content reads
 
-`agentEgress` is consumed configuration, not a profile projection. The host
-must select one exact policy revision and an authenticated tenant id. Omitting
-contentful mode keeps runtime activity metadata/status-only; enabling it is an
-explicit product decision and requires the server capability. Reliable events
-are fsynced under the canonical Agent home and retire only after an exact ack.
+`agentEgress` is consumed policy configuration, not a profile or tenant
+projection. The host selects one exact policy revision. The daemon obtains its
+tenant binding only from the authenticated pair response persisted in the
+atomic local `DeviceRecord`; there is no `agentEgress.tenantId` setting and no
+Profile/config, deviceId, or access-token fallback. Omitting contentful mode
+keeps runtime activity metadata/status-only; enabling it is an explicit product
+decision and requires the server capability. Reliable events are fsynced under
+the canonical Agent home and retire only after an exact ack.
 
 ```ts
 createDaemon({
   // ...normal device, transport and agentHome configuration
   agentEgress: {
-    tenantId: 'tenant-authority-from-salesko',
     policy: {
       policyRevision: 'salesko-agent-egress-r1',
       activity: { mode: 'metadata-status', delivery: 'latest-value' },
@@ -163,8 +165,9 @@ its local supplement. The local supplement can only narrow root, text, MIME,
 size and sensitive-name behavior; it cannot enable a wire-disabled surface.
 The SDK derives `agents/<agentId>`, `.byok/egress`, runtime-session evidence and
 the per-Agent content-read audit path. Salesko must not compose those paths.
-Tenant/device identity comes from authenticated host state; a request cannot
-override it. Transcript reads additionally require the exact persisted
+Tenant/device identity comes from the persisted authenticated enrollment; a
+request or editable host configuration cannot override it. Transcript reads
+additionally require the exact persisted
 AgentRef/session/runtime/cwd handoff. Allowed content is uploaded through the
 authenticated blob channel. The content-free receipt is fsynced into the
 Agent-local reliable spool with stable event/cursor identity before send and

package/dist/bin/byok-agent.js

@@ -2,10 +2,10 @@
 import { randomUUID, createHash, randomBytes, timingSafeEqual, createHmac, createPrivateKey, generateKeyPairSync, sign } from 'crypto';
 import { readFileSync, promises, linkSync, fstatSync, lstatSync, unlinkSync, constants, readSync, openSync, writeFileSync, fchmodSync, fsyncSync, closeSync, opendirSync, existsSync, realpathSync, mkdirSync, renameSync, chmodSync, statSync, readdirSync } from 'fs';
 import path, { isAbsolute, join } from 'path';
-import { AGENT_CONTENT_ARTIFACT_READ_CAPABILITY, AGENT_CONTENT_TRANSCRIPT_READ_CAPABILITY, AGENT_CONTENT_WORKSPACE_READ_CAPABILITY, TASK_STATES, AgentEgressPolicySchema, CONFIGURED_TOOLSETS_MAX_ITEMS, ToolsetIdSchema, AgentRefSchema, AgentContentReceiptPayloadSchema, BYOK_PAIR_PATH, BYOK_CHALLENGE_PATH, BYOK_TOKEN_PATH, partitionAgentEvents, TASK_TRANSITIONS, encodeEnvelope, PROTOCOL_VERSION, createEnvelope, AGENT_EGRESS_RELIABLE_ACK_CAPABILITY, byokBlobUrlPath, BYOK_BLOBS_PATH, byokBlobFinalizePath, AGENT_EGRESS_POLICY_CAPABILITY, parseMessage, checkResultDocument, MAX_MESSAGES_PER_BATCH, BYOK_CAPABILITIES_PATH, BYOK_PRESENCE_PATH, RuntimeIdSchema, TERMINAL_INFERENCE_USAGE_MAX_DURATION_MS, TERMINAL_INFERENCE_USAGE_MAX_TOKENS, RESULT_DOCUMENT_MAX_BYTES, decodeEnvelope, BYOK_EVENTS_PATH, BYOK_MESSAGES_PATH, MessagesSendResponseSchema, UnknownMessageTypeError, BYOK_WS_PATH } from '@byok-sdk/protocol';
+import { AGENT_CONTENT_ARTIFACT_READ_CAPABILITY, AGENT_CONTENT_TRANSCRIPT_READ_CAPABILITY, AGENT_CONTENT_WORKSPACE_READ_CAPABILITY, TASK_STATES, AgentEgressPolicySchema, CONFIGURED_TOOLSETS_MAX_ITEMS, ToolsetIdSchema, AgentRefSchema, AgentContentReceiptPayloadSchema, BYOK_PAIR_PATH, PairResponseSchema, BYOK_CHALLENGE_PATH, BYOK_TOKEN_PATH, partitionAgentEvents, TASK_TRANSITIONS, encodeEnvelope, PROTOCOL_VERSION, createEnvelope, AGENT_EGRESS_RELIABLE_ACK_CAPABILITY, byokBlobUrlPath, BYOK_BLOBS_PATH, byokBlobFinalizePath, AGENT_EGRESS_POLICY_CAPABILITY, parseMessage, checkResultDocument, MAX_MESSAGES_PER_BATCH, BYOK_CAPABILITIES_PATH, BYOK_PRESENCE_PATH, RuntimeIdSchema, TERMINAL_INFERENCE_USAGE_MAX_DURATION_MS, TERMINAL_INFERENCE_USAGE_MAX_TOKENS, RESULT_DOCUMENT_MAX_BYTES, decodeEnvelope, BYOK_EVENTS_PATH, BYOK_MESSAGES_PATH, MessagesSendResponseSchema, UnknownMessageTypeError, BYOK_WS_PATH } from '@byok-sdk/protocol';
 import { execFile, spawn } from 'child_process';
 import os from 'os';
-import { DEVICE_ASSERTION_AUDIENCE_MAX_BYTES, DEVICE_ASSERTION_DEFAULT_TTL_MS, DEVICE_ASSERTION_MAX_TTL_MS, nonceSigningBytes, CapabilityDeclarationSchema, hasCapability, DeviceAssertionClaimsSchema, deviceAssertionSigningInput, DEVICE_ASSERTION_SCHEMA_ID } from '@byok-sdk/core';
+import { isTenantId, DEVICE_ASSERTION_AUDIENCE_MAX_BYTES, DEVICE_ASSERTION_DEFAULT_TTL_MS, DEVICE_ASSERTION_MAX_TTL_MS, nonceSigningBytes, CapabilityDeclarationSchema, hasCapability, DeviceAssertionClaimsSchema, deviceAssertionSigningInput, DEVICE_ASSERTION_SCHEMA_ID } from '@byok-sdk/core';
 import { promisify } from 'util';
 import { fileURLToPath } from 'url';
 import 'readline';
@@ -4732,6 +4732,13 @@ var ApprovalRegistry = class {
   }
 };
 var MAX_DEVICE_RECORD_BYTES = 256 * 1024;
+var REPAIR_REQUIRED_MESSAGE = "device enrollment record is missing or has an invalid authenticated tenant binding; re-pair required";
+var DeviceRecordRePairRequiredError = class extends Error {
+  constructor() {
+    super(REPAIR_REQUIRED_MESSAGE);
+    this.name = "DeviceRecordRePairRequiredError";
+  }
+};
 function sameInode(left, right) {
   return left.dev === right.dev && left.ino === right.ino;
 }
@@ -4741,18 +4748,32 @@ function sameFileState(left, right) {
 function sameContentState(left, right) {
   return sameInode(left, right) && left.size === right.size && left.mtimeNs === right.mtimeNs;
 }
+function assertDeviceRecord(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    throw new DeviceRecordRePairRequiredError();
+  }
+  const parsed = value;
+  if (typeof parsed.deviceId === "string" && isTenantId(parsed.tenantId) && typeof parsed.accessToken === "string" && typeof parsed.expiresAt === "string" && typeof parsed.devicePrivateKeyPem === "string" && typeof parsed.devicePublicKey === "string") {
+    return;
+  }
+  throw new DeviceRecordRePairRequiredError();
+}
 function parseDeviceRecord(raw) {
-  const parsed = JSON.parse(raw);
-  if (typeof parsed.deviceId === "string" && typeof parsed.accessToken === "string" && typeof parsed.expiresAt === "string" && typeof parsed.devicePrivateKeyPem === "string" && typeof parsed.devicePublicKey === "string") {
-    return {
-      deviceId: parsed.deviceId,
-      accessToken: parsed.accessToken,
-      expiresAt: parsed.expiresAt,
-      devicePrivateKeyPem: parsed.devicePrivateKeyPem,
-      devicePublicKey: parsed.devicePublicKey
-    };
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch {
+    throw new DeviceRecordRePairRequiredError();
   }
-  return void 0;
+  assertDeviceRecord(parsed);
+  return {
+    deviceId: parsed.deviceId,
+    tenantId: parsed.tenantId,
+    accessToken: parsed.accessToken,
+    expiresAt: parsed.expiresAt,
+    devicePrivateKeyPem: parsed.devicePrivateKeyPem,
+    devicePublicKey: parsed.devicePublicKey
+  };
 }
 var DeviceStore = class _DeviceStore {
   /**
@@ -4822,6 +4843,7 @@ var DeviceStore = class _DeviceStore {
     }
   }
   async save(record) {
+    assertDeviceRecord(record);
     const storeDir = path.dirname(this.filePath);
     await ensureSecureDir(storeDir, this.secureDirOptions);
     await atomicWriteFile(this.filePath, JSON.stringify(record, null, 2), { mode: 384 });
@@ -4992,7 +5014,14 @@ var AuthManager = class {
     this.proactiveTimer = void 0;
     try {
       return await this.runCredentialMutation(async () => {
-        const existing = this.record ?? await this.opts.store.load();
+        let existing = this.record;
+        if (!existing) {
+          try {
+            existing = await this.opts.store.load();
+          } catch (error) {
+            if (!(error instanceof DeviceRecordRePairRequiredError)) throw error;
+          }
+        }
         const keyPair = existing ? { privateKey: importPrivateKeyPem(existing.devicePrivateKeyPem), publicKeyBase64Url: existing.devicePublicKey } : generateDeviceKeyPair();
         const url = new URL(BYOK_PAIR_PATH, toHttpBase(this.opts.serverUrl));
         const res = await fetch(url, {
@@ -5007,9 +5036,10 @@ var AuthManager = class {
         if (!res.ok) {
           throw new Error(`pairing failed: HTTP ${res.status} ${await safeErrorText(res)}`.trimEnd());
         }
-        const body = await res.json();
+        const body = PairResponseSchema.parse(await res.json());
         const record = {
           deviceId: body.deviceId,
+          tenantId: body.tenantId,
           accessToken: body.accessToken,
           expiresAt: resolvePairExpiry(body.refreshHint),
           devicePrivateKeyPem: exportPrivateKeyPem(keyPair.privateKey),
@@ -12428,9 +12458,14 @@ var AgentEgressController = class {
   latestStatus = emptyLane();
   reliableStatus = emptyLane();
   drops = [];
+  active = true;
   get policy() {
     return this.options.policy;
   }
+  /** Permanently fail closed after its authenticated enrollment is replaced. */
+  deactivate() {
+    this.active = false;
+  }
   status() {
     const reliable = this.reliableRecords();
     return Object.freeze({
@@ -12448,6 +12483,10 @@ var AgentEgressController = class {
   /** Project before TaskRunner builds a `task.progress` envelope. */
   projectLatestValue(input) {
     if (input.agentRef === void 0) return Object.freeze([...input.events]);
+    if (!this.active) {
+      this.noteDrop("latest-value", "policy_denied", input.agentRef);
+      return [];
+    }
     if (this.options.policy.activity.mode === "contentful-trajectory" && !input.serverCapabilities.includes("agent-egress-policy")) {
       this.noteDrop("latest-value", "capability_missing", input.agentRef);
       return [];
@@ -12470,7 +12509,7 @@ var AgentEgressController = class {
     return latest === void 0 ? [] : Object.freeze([latest]);
   }
   async appendReliable(input) {
-    if (this.options.tenantId === void 0) {
+    if (!this.active || this.options.tenantId === void 0) {
       this.noteDrop("reliable", "policy_denied", input.agentRef);
       return { ok: false, reason: "policy_denied" };
     }
@@ -12508,7 +12547,7 @@ var AgentEgressController = class {
    * with `wireType: agent.content.receipt` before any transport attempt.
    */
   async appendContentReceipt(input) {
-    if (this.options.tenantId === void 0) {
+    if (!this.active || this.options.tenantId === void 0) {
       this.noteDrop("reliable", "policy_denied", input.agentRef);
       return { ok: false, reason: "policy_denied" };
     }
@@ -12532,7 +12571,7 @@ var AgentEgressController = class {
   /** Retires only the record whose full Agent/tenant/revision/id/cursor tuple matches. */
   async acknowledge(ack) {
     const spool = this.spools.get(agentKey(ack.agentRef));
-    if (!spool || this.options.tenantId === void 0 || ack.tenantId !== this.options.tenantId) {
+    if (!this.active || !spool || this.options.tenantId === void 0 || ack.tenantId !== this.options.tenantId) {
       this.noteDrop("reliable", "ack_mismatch", ack.agentRef);
       return false;
     }
@@ -12543,6 +12582,7 @@ var AgentEgressController = class {
   /** Re-open every existing Agent-local spool before retrying stable records after restart. */
   async recover(agentsRoot) {
     if (!path.isAbsolute(agentsRoot)) throw new Error("Agent egress recovery root must be absolute");
+    if (!this.active) throw new Error("Agent egress recovery requires an active authenticated enrollment");
     if (this.options.tenantId === void 0) {
       throw new Error("Agent egress recovery requires one authenticated tenant authority");
     }
@@ -13650,9 +13690,6 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
   if (config.agentEgress !== void 0 && config.agentHome === void 0) {
     throw new Error("DaemonConfig.agentEgress requires DaemonConfig.agentHome for the per-Agent local spool");
   }
-  if (config.agentEgress !== void 0 && (config.agentEgress.tenantId.length === 0 || config.agentEgress.tenantId.trim() !== config.agentEgress.tenantId)) {
-    throw new Error("DaemonConfig.agentEgress.tenantId must be a non-empty canonical authenticated tenant id");
-  }
   const egressPolicy = resolveAgentEgressPolicy(config.agentEgress?.policy);
   const egressBatcherOptions = egressPolicy.activity.mode === "contentful-trajectory" ? {
     ...config.progressBatch,
@@ -13683,9 +13720,8 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
   });
   const agentSessionHandoffs = config.agentHome === void 0 ? void 0 : new AgentSessionHandoffStore();
   const agentContentReadPolicies = resolveContentReadPolicies(egressPolicy, config.agentEgress?.contentRead);
-  const agentEgress = new AgentEgressController({
+  let agentEgress = new AgentEgressController({
     policy: egressPolicy,
-    ...config.agentEgress?.tenantId === void 0 ? {} : { tenantId: config.agentEgress.tenantId },
     ...config.agentEgress?.sanitizer === void 0 ? {} : { sanitizer: config.agentEgress.sanitizer }
   });
   const gitWorkspaceManager = config.gitWorkspace ? overrides.gitWorkspace?.manager ?? new GitWorkspaceManager(config.workspaceRoot, { ownerId: stableGitWorkspaceOwnerId(storeDir, config.productId) }) : void 0;
@@ -13695,11 +13731,6 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
     if (config.hostedJournal.mode !== "sqlite") {
       throw new Error(`DaemonConfig.hostedJournal.mode must be "sqlite" \u2014 got ${JSON.stringify(config.hostedJournal.mode)}`);
     }
-    if (typeof config.hostedJournal.tenantId !== "string" || config.hostedJournal.tenantId.trim() === "") {
-      throw new Error(
-        "DaemonConfig.hostedJournal.tenantId must be a non-empty tenant id \u2014 a hosted journal row with no tenant is durable evidence nobody can act on"
-      );
-    }
     if (config.hostedJournal.storagePolicy) {
       resolvedStoragePolicy = resolveLocalStoragePolicy(config.hostedJournal.storagePolicy);
     }
@@ -13768,6 +13799,8 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
   const approvalRegistry = new ApprovalRegistry();
   let connection;
   let connectionState = "closed";
+  let daemonStarted = false;
+  let tenantRebinding = false;
   let runner;
   let controlServerHandle;
   let daemonOwnerLease;
@@ -13820,11 +13853,31 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
     return runLifecycleMutation(() => pairUnderLease(pairingCode));
   }
   async function pairUnderLease(pairingCode) {
+    const wasRunning = daemonStarted;
+    if (wasRunning) tenantRebinding = true;
     const acquiredHere = daemonOwnerLease === void 0;
     if (acquiredHere) daemonOwnerLease = await acquireDaemonOwner(storeDir, "daemon");
+    let replacementPersisted = false;
     try {
-      const previous = await store.load();
+      let previous;
+      try {
+        previous = await store.load();
+      } catch (error) {
+        if (!(error instanceof DeviceRecordRePairRequiredError)) throw error;
+      }
       const record = await auth.pair(pairingCode);
+      replacementPersisted = true;
+      if (config.agentEgress !== void 0) {
+        agentEgress.deactivate();
+        agentEgress = new AgentEgressController({
+          policy: egressPolicy,
+          ...config.agentEgress.sanitizer === void 0 ? {} : { sanitizer: config.agentEgress.sanitizer }
+        });
+      }
+      if (wasRunning) {
+        await runShutdownSequence("re-pairing enrollment binding");
+        tenantRebinding = false;
+      }
       if (previous && previous.deviceId !== record.deviceId) {
         await cursorStore.clear(config.serverUrl, previous.deviceId);
       }
@@ -13835,6 +13888,7 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
       }
       return record;
     } catch (err) {
+      if (wasRunning && !replacementPersisted) tenantRebinding = false;
       if (acquiredHere) {
         await daemonOwnerLease?.release();
         daemonOwnerLease = void 0;
@@ -13857,6 +13911,13 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
       if (!record) {
         throw new Error("device is not paired yet; call pair(pairingCode) first");
       }
+      if (config.agentEgress !== void 0) {
+        agentEgress = new AgentEgressController({
+          policy: egressPolicy,
+          tenantId: record.tenantId,
+          ...config.agentEgress.sanitizer === void 0 ? {} : { sanitizer: config.agentEgress.sanitizer }
+        });
+      }
       await agentHomeManager?.preflight();
       if (config.agentEgress !== void 0 && config.agentHome !== void 0) {
         await agentEgress.recover(path.join(config.agentHome.hostStorageRoot, "agents"));
@@ -13917,7 +13978,7 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
         config.agentEgress !== void 0,
         agentContentReadPolicies
       );
-      const journalIdentity = config.hostedJournal ? { tenantId: config.hostedJournal.tenantId, productId: config.productId, deviceId: record.deviceId } : void 0;
+      const journalIdentity = config.hostedJournal ? { tenantId: record.tenantId, productId: config.productId, deviceId: record.deviceId } : void 0;
       const sendSanitizedEnvelope = activeJournal && journalIdentity ? (envelope) => {
         observer.handleOutboundEnvelope(envelope);
         const terminalKind = terminalKindOf(envelope.type);
@@ -14055,9 +14116,8 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
       runner = new TaskRunner(deps);
       const handleAgentEgressEnvelope = async (envelope) => {
         if (envelope.type !== "agent.egress.ack") return false;
-        const tenantId = config.agentEgress?.tenantId;
-        if (tenantId === void 0) return true;
-        await agentEgress.acknowledge({ ...envelope.payload, tenantId });
+        if (config.agentEgress === void 0) return true;
+        await agentEgress.acknowledge({ ...envelope.payload, tenantId: record.tenantId });
         return true;
       };
       const handleAgentContentReadEnvelope = async (envelope) => {
@@ -14109,7 +14169,7 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
         const result = await policyEngine.read({
           requestId: payload.requestId,
           actor: payload.actor,
-          tenantId: config.agentEgress.tenantId,
+          tenantId: record.tenantId,
           deviceId: record.deviceId,
           agentRef: payload.agentRef,
           surface: payload.surface,
@@ -14205,6 +14265,9 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
         //
         // The no-journal branch is the ORIGINAL closure, unchanged.
         onEnvelope: activeJournal && journalIdentity ? async (envelope) => {
+          if (tenantRebinding) {
+            throw new Error("tenant enrollment is being re-paired; inbound work is blocked until restart");
+          }
           observer.handleInboundEnvelope(envelope);
           if (await handleAgentEgressEnvelope(envelope)) return;
           if (await handleAgentContentReadEnvelope(envelope)) return;
@@ -14212,6 +14275,9 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
           await activeJournal.appendEnvelope(toJournalEnvelopeRecord(envelope, journalIdentity));
           return runner?.handleEnvelope(envelope) ?? Promise.resolve();
         } : (envelope) => {
+          if (tenantRebinding) {
+            return Promise.reject(new Error("tenant enrollment is being re-paired; inbound work is blocked until restart"));
+          }
           observer.handleInboundEnvelope(envelope);
           if (envelope.type === "agent.egress.ack") return handleAgentEgressEnvelope(envelope).then(() => void 0);
           if (envelope.type === "agent.content.read") return handleAgentContentReadEnvelope(envelope).then(() => void 0);
@@ -14243,6 +14309,7 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
         dispatchReliableRecord(record2);
       }
       startPresenceProducer();
+      daemonStarted = true;
     } catch (err) {
       try {
         await runShutdownSequence("startup failed", { drainTimeoutMs: 0 });
@@ -14387,6 +14454,7 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
     if (!mutationBarrierComplete) {
       throw new Error("daemon shutdown mutation barrier is incomplete; ownership lease retained");
     }
+    daemonStarted = false;
   }
   async function stop(opts = {}) {
     shuttingDown = true;
@@ -14708,6 +14776,9 @@ function buildDaemonWithAdapters(config, adapters, overrides = {}, assertionProb
     if (config.agentEgress === void 0 || agentHomeManager === void 0) {
       throw new Error("Agent reliable egress is not configured");
     }
+    if (tenantRebinding) {
+      throw new Error("tenant enrollment is being re-paired; reliable egress is blocked until restart");
+    }
     const binding = await agentHomeManager.acquire(input.agentRef);
     try {
       await agentHomeManager.initialize(binding);
@@ -15381,7 +15452,7 @@ function createServiceLifecycle(def, opts = {}) {
 
 // src/bin/official-release.ts
 var OFFICIAL_LOCAL_AGENT_RELEASE = resolveLocalAgentReleaseIdentity({
-  version: "0.6.1"
+  version: "0.7.0"
 });
 
 // src/bin/config.ts