@byok-sdk/cloud-dataplane 0.14.0 → 0.16.0

package/dist/index.js

@@ -804,10 +804,11 @@ function toRecord(row) {
     proofKeyEpoch: row.proof_key_epoch,
     revoked: row.revoked,
     ...row.machine_id == null ? {} : { machineId: row.machine_id },
+    ...row.harnesses == null ? {} : { harnesses: row.harnesses },
     ...row.capabilities == null ? {} : { capabilities: Object.freeze([...row.capabilities]) }
   };
 }
-var SELECT_COLUMNS = "tenant_id, device_id, product_id, device_name, device_public_key, proof_key_id, proof_key_epoch, revoked, machine_id, capabilities";
+var SELECT_COLUMNS = "tenant_id, device_id, product_id, device_name, device_public_key, proof_key_id, proof_key_epoch, revoked, machine_id, capabilities, harnesses";
 var DEVICE_STATE_DELETES = [
   "DELETE FROM device_presence WHERE tenant_id = $1 AND device_id = ANY($2::text[])",
   "DELETE FROM auth_nonce WHERE tenant_id = $1 AND device_id = ANY($2::text[])",
@@ -854,7 +855,7 @@ async function registerDeviceOnClient(client, tenant, input) {
            proof_key_epoch = EXCLUDED.proof_key_epoch,
            revoked = false,
            machine_id = EXCLUDED.machine_id,
-           capabilities = NULL
+           capabilities = NULL, harnesses = NULL
      RETURNING ${SELECT_COLUMNS}`,
     [
       tenant,
@@ -939,10 +940,10 @@ var PostgresDeviceDirectory = class {
   async recordCapabilities(tenant, input) {
     const result = await this.#pool.query(
       `UPDATE device
-          SET capabilities = $3::jsonb
+          SET capabilities = $3::jsonb, harnesses = $4::jsonb
         WHERE tenant_id = $1 AND device_id = $2 AND revoked = false
       RETURNING ${SELECT_COLUMNS}`,
-      [tenant, input.deviceId, JSON.stringify([...input.capabilities])]
+      [tenant, input.deviceId, JSON.stringify([...input.capabilities]), JSON.stringify(input.harnesses ?? [])]
     );
     const row = result.rows[0];
     return row === void 0 ? void 0 : toRecord(row);
@@ -1356,7 +1357,7 @@ var PostgresProofRequestReceiptStore = class {
     return row === void 0 ? void 0 : toReceipt2(row);
   }
 };
-var TASK_SELECT_COLUMNS = "tenant_id, task_id, device_id, agent_id, agent_profile_revision, owner_device_id, status, terminal_cause, cancel_requested_at, cancel_reason, cancel_message_id, claimed_runtime, claimed_runtime_capabilities, updated_at";
+var TASK_SELECT_COLUMNS = "tenant_id, task_id, device_id, agent_id, agent_profile_revision, owner_device_id, status, terminal_cause, cancel_requested_at, cancel_reason, cancel_message_id, claimed_runtime, claimed_harness_id, claimed_runtime_capabilities, updated_at";
 var AGENT_MESSAGE_ADMISSION_SELECT_COLUMNS = "message_id, payload_body, terminal_body";
 function admissionRowToRecord(row) {
   return {
@@ -1384,6 +1385,7 @@ function taskRowToAttempt(row) {
     status: row.status,
     // `claimed_runtime` is written only by the guarded claim UPDATE below, so a
     // non-null value here is always the winning claim's own self-report.
+    ...row.claimed_harness_id == null ? {} : { claimedHarnessId: row.claimed_harness_id },
     ...row.claimed_runtime === null ? {} : { claimedRuntime: row.claimed_runtime },
     ...row.claimed_runtime_capabilities === null ? {} : { claimedRuntimeCapabilities: row.claimed_runtime_capabilities },
     ...row.terminal_cause == null ? {} : { terminalCause: row.terminal_cause },
@@ -1559,6 +1561,7 @@ var PostgresTaskAttemptStore = class {
           SET owner_device_id = $3,
               status = 'claimed',
               claimed_runtime = $5,
+              claimed_harness_id = $7,
               claimed_runtime_capabilities = $6::jsonb,
               updated_at = $4
         WHERE tenant_id = $1 AND task_id = $2 AND device_id = $3 AND owner_device_id IS NULL
@@ -1570,7 +1573,8 @@ var PostgresTaskAttemptStore = class {
         input.deviceId,
         this.#now(),
         input.runtime ?? null,
-        input.capabilities === void 0 ? null : JSON.stringify(input.capabilities)
+        input.capabilities === void 0 ? null : JSON.stringify(input.capabilities),
+        input.harnessId ?? null
       ]
     );
     const won = claimed.rows[0];